package net

import (
	"github.com/gorilla/websocket"
)

type wsConn struct {
	conn *websocket.Conn
}

func newWSConn(conn *websocket.Conn) *wsConn {
	return &wsConn{conn: conn}
}

func (w *wsConn) ReadMessage() (string, error) {
	_, msg, err := w.conn.ReadMessage()
	return string(msg), err
}

func (w *wsConn) Write(b []byte) (int, error) {
	err := w.conn.WriteMessage(websocket.TextMessage, b)
	return len(b), err
}

func (w *wsConn) Close() error {
	return w.conn.Close()
}

func (w *wsConn) SetEcho(on bool) error {
	if on {
		return w.conn.WriteMessage(websocket.TextMessage, []byte("\x1b]10;0\x07"))
	}
	return w.conn.WriteMessage(websocket.TextMessage, []byte("\x1b]10;1\x07"))
}