1 files changed, 34 insertions, 0 deletions
diff --git a/internal/net/wsconn.go b/internal/net/wsconn.go
new file mode 100644
index 0000000..d6f45dc
--- /dev/null
+++ b/internal/net/wsconn.go
@@ -0,0 +1,34 @@
+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"))
+}