From ab8aaf619b91b0f76545214f184b72d5fd7dcd01 Mon Sep 17 00:00:00 2001
From: Denis Arh <denis.arh@kendu.si>
Date: Sat, 28 Jul 2018 17:50:02 +0200
Subject: [PATCH] =?UTF-8?q?Add=20simplistic=20cli=20chat=C2=A0helper?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cmd/chat/main.go | 126 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 126 insertions(+)
 create mode 100644 cmd/chat/main.go

diff --git a/cmd/chat/main.go b/cmd/chat/main.go
new file mode 100644
index 000000000..f3039c5a1
--- /dev/null
+++ b/cmd/chat/main.go
@@ -0,0 +1,126 @@
+package main
+
+import (
+	"bufio"
+	"encoding/json"
+	"github.com/SentimensRG/sigctx"
+	"github.com/crusttech/crust/sam/websocket/incoming"
+	"github.com/gorilla/websocket"
+	"log"
+	"os"
+	"strings"
+)
+
+func main() {
+	var ctx = sigctx.New()
+
+	var wsd = websocket.Dialer{}
+	conn, _, err := wsd.Dial("ws://localhost:3000/websocket/?jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE1MzM5ODg4MzksInN1YiI6IjIwNDA4NTk3MDc2MzUxMzk2NSJ9.aJQE-_hH_bxbri0SDmBok_UzvYnJrvDYc16E_6meJ6w", map[string][]string{})
+	must(err)
+
+	defer conn.Close()
+	go func() {
+
+		log := log.New(os.Stderr, "<<<<", log.LstdFlags)
+		// Reader
+		for {
+			mt, p, err := conn.ReadMessage()
+			must(err)
+			log.Printf("%d %s", mt, string(p))
+		}
+	}()
+
+	go func() {
+		reader := bufio.NewReader(os.Stdin)
+		var chanId = ""
+
+		// Writer
+		for {
+			func() {
+				if chanId == "" {
+					print("NO CHANNEL")
+				} else {
+					print(chanId)
+				}
+				print("> ")
+				text, _ := reader.ReadString('\n')
+				text = strings.TrimSpace(text)
+				if len(text) == 0 {
+					return
+				}
+
+				if text[:1] == "/" {
+					cmdSplit := strings.Split(text, " ")
+					switch cmdSplit[0] {
+					case "/join":
+						if len(cmdSplit) < 2 {
+							println("/join <channelid>")
+						} else {
+							joinCh(conn, cmdSplit[1])
+							chanId = cmdSplit[1]
+							openCh(conn, chanId)
+						}
+					case "/part":
+						partCh(conn, chanId)
+						chanId = ""
+
+					case "/list":
+						listCh(conn)
+
+					default:
+						println("Unknown command, try:")
+						println("   /join <channel-id>")
+						println("   /part")
+						println("   /list")
+					}
+				} else {
+					sendMsg(conn, text, chanId)
+				}
+			}()
+		}
+	}()
+
+	<-ctx.Done()
+}
+
+func must(err error) {
+	if err != nil {
+		log.Fatal(err)
+	}
+}
+
+func should(err error) {
+	if err != nil {
+		log.Printf("Warning: %+v", err)
+	}
+}
+
+func sendMsg(conn *websocket.Conn, text, chanId string) {
+	pb, err := json.Marshal(incoming.Payload{MessageCreate: &incoming.MessageCreate{Message: text, ChannelID: chanId}})
+	should(err)
+	should(conn.WriteMessage(websocket.TextMessage, pb))
+}
+
+func listCh(conn *websocket.Conn) {
+	pb, err := json.Marshal(incoming.Payload{ChannelList: &incoming.ChannelList{}})
+	should(err)
+	should(conn.WriteMessage(websocket.TextMessage, pb))
+}
+
+func joinCh(conn *websocket.Conn, channelId string) {
+	pb, err := json.Marshal(incoming.Payload{ChannelJoin: &incoming.ChannelJoin{ChannelID: channelId}})
+	should(err)
+	should(conn.WriteMessage(websocket.TextMessage, pb))
+}
+
+func partCh(conn *websocket.Conn, channelId string) {
+	pb, err := json.Marshal(incoming.Payload{ChannelPart: &incoming.ChannelPart{ChannelID: &channelId}})
+	should(err)
+	should(conn.WriteMessage(websocket.TextMessage, pb))
+}
+
+func openCh(conn *websocket.Conn, channelId string) {
+	pb, err := json.Marshal(incoming.Payload{ChannelOpen: &incoming.ChannelOpen{ChannelID: channelId}})
+	should(err)
+	should(conn.WriteMessage(websocket.TextMessage, pb))
+}