package websocket

import "io"

func Write(w io.Writer, msg []byte) (n int, err error) {
	header := []byte{0x81} // FIN + data frame
	if len(msg) < 126 {
		header = append(header, byte(len(msg)))
	} else {
		header = append(header, 126, byte(len(msg)>>8), byte(len(msg)))
	}
	return w.Write(append(header, msg...))
}