├── README.md
├── go.mod
├── go.sum
├── index.html
└── main.go


/README.md:
--------------------------------------------------------------------------------
1 | Deploy: https://chat-1nz.vercel.app


--------------------------------------------------------------------------------
/go.mod:
--------------------------------------------------------------------------------
 1 | module v2
 2 | 
 3 | go 1.14
 4 | 
 5 | require (
 6 | 	github.com/dgrijalva/jwt-go v3.2.0+incompatible // indirect
 7 | 	github.com/go-playground/validator/v10 v10.3.0
 8 | 	github.com/gorilla/securecookie v1.1.1
 9 | 	github.com/gorilla/websocket v1.4.1
10 | 	github.com/labstack/echo v3.3.10+incompatible
11 | 	github.com/labstack/gommon v0.3.0 // indirect
12 | 	github.com/mattn/go-colorable v0.1.7 // indirect
13 | 	github.com/novalagung/gubrak/v2 v2.0.1
14 | 	github.com/sirupsen/logrus v1.6.0 // indirect
15 | 	github.com/valyala/fasttemplate v1.2.1 // indirect
16 | 	golang.org/x/crypto v0.0.0-20200820211705-5c72a883971a // indirect
17 | 	golang.org/x/net v0.0.0-20200904194848-62affa334b73 // indirect
18 | 	golang.org/x/sys v0.0.0-20200909081042-eff7692f9009 // indirect
19 | 	golang.org/x/text v0.3.3 // indirect
20 | )
21 | 


--------------------------------------------------------------------------------
/go.sum:
--------------------------------------------------------------------------------
 1 | github.com/DefinitelyMod/gocsv v0.0.0-20181205141819-acfa5f112b45/go.mod h1:+nlrAh0au59iC1KN5RA1h1NdiOQYlNOBrbtE1Plqht4=
 2 | github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 3 | github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 4 | github.com/dgrijalva/jwt-go v1.0.2 h1:KPldsxuKGsS2FPWsNeg9ZO18aCrGKujPoWXn2yo+KQM=
 5 | github.com/dgrijalva/jwt-go v3.2.0+incompatible h1:7qlOGliEKZXTDg6OTjfoBKDXWrumCAMpl/TFQ4/5kLM=
 6 | github.com/dgrijalva/jwt-go v3.2.0+incompatible/go.mod h1:E3ru+11k8xSBh+hMPgOLZmtrrCbhqsmaPHjLKYnJCaQ=
 7 | github.com/go-playground/assert/v2 v2.0.1/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
 8 | github.com/go-playground/locales v0.13.0 h1:HyWk6mgj5qFqCT5fjGBuRArbVDfE4hi8+e8ceBS/t7Q=
 9 | github.com/go-playground/locales v0.13.0/go.mod h1:taPMhCMXrRLJO55olJkUXHZBHCxTMfnGwq/HNwmWNS8=
10 | github.com/go-playground/universal-translator v0.17.0 h1:icxd5fm+REJzpZx7ZfpaD876Lmtgy7VtROAbHHXk8no=
11 | github.com/go-playground/universal-translator v0.17.0/go.mod h1:UkSxE5sNxxRwHyU+Scu5vgOQjsIJAF8j9muTVoKLVtA=
12 | github.com/go-playground/validator v9.31.0+incompatible h1:UA72EPEogEnq76ehGdEDp4Mit+3FDh548oRqwVgNsHA=
13 | github.com/go-playground/validator/v10 v10.3.0 h1:nZU+7q+yJoFmwvNgv/LnPUkwPal62+b2xXj0AU1Es7o=
14 | github.com/go-playground/validator/v10 v10.3.0/go.mod h1:uOYAAleCW8F/7oMFd6aG0GOhaH6EGOAJShg8Id5JGkI=
15 | github.com/gorilla/securecookie v1.1.1 h1:miw7JPhV+b/lAHSXz4qd/nN9jRiAFV5FwjeKyCS8BvQ=
16 | github.com/gorilla/securecookie v1.1.1/go.mod h1:ra0sb63/xPlUeL+yeDciTfxMRAA+MP+HVt/4epWDjd4=
17 | github.com/gorilla/websocket v1.4.1 h1:q7AeDBpnBk8AogcD4DSag/Ukw/KV+YhzLj2bP5HvKCM=
18 | github.com/gorilla/websocket v1.4.1/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
19 | github.com/konsorten/go-windows-terminal-sequences v1.0.3 h1:CE8S1cTafDpPvMhIxNJKvHsGVBgn1xWYf1NbHQhywc8=
20 | github.com/konsorten/go-windows-terminal-sequences v1.0.3/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
21 | github.com/labstack/echo v1.4.4 h1:1bEiBNeGSUKxcPDGfZ/7IgdhJJZx8wV/pICJh4W2NJI=
22 | github.com/labstack/echo v3.3.10+incompatible h1:pGRcYk231ExFAyoAjAfD85kQzRJCRI8bbnE7CX5OEgg=
23 | github.com/labstack/echo v3.3.10+incompatible/go.mod h1:0INS7j/VjnFxD4E2wkz67b8cVwCLbBmJyDaka6Cmk1s=
24 | github.com/labstack/gommon v0.3.0 h1:JEeO0bvc78PKdyHxloTKiF8BD5iGrH8T6MSeGvSgob0=
25 | github.com/labstack/gommon v0.3.0/go.mod h1:MULnywXg0yavhxWKc+lOruYdAhDwPK9wf0OL7NoOu+k=
26 | github.com/leodido/go-urn v1.2.0 h1:hpXL4XnriNwQ/ABnpepYM/1vCLWNDfUNts8dX3xTG6Y=
27 | github.com/leodido/go-urn v1.2.0/go.mod h1:+8+nEpDfqqsY+g338gtMEUOtuK+4dEMhiQEgxpxOKII=
28 | github.com/mattn/go-colorable v0.1.2 h1:/bC9yWikZXAL9uJdulbSfyVNIR3n3trXl+v8+1sx8mU=
29 | github.com/mattn/go-colorable v0.1.2/go.mod h1:U0ppj6V5qS13XJ6of8GYAs25YV2eR4EVcfRqFIhoBtE=
30 | github.com/mattn/go-colorable v0.1.7 h1:bQGKb3vps/j0E9GfJQ03JyhRuxsvdAanXlT9BTw3mdw=
31 | github.com/mattn/go-colorable v0.1.7/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope9wVRipJSqc=
32 | github.com/mattn/go-isatty v0.0.8/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s=
33 | github.com/mattn/go-isatty v0.0.9 h1:d5US/mDsogSGW37IV293h//ZFaeajb69h+EHFsv2xGg=
34 | github.com/mattn/go-isatty v0.0.9/go.mod h1:YNRxwqDuOph6SZLI9vUUz6OYw3QyUt7WiY2yME+cCiQ=
35 | github.com/mattn/go-isatty v0.0.12 h1:wuysRhFDzyxgEmMf5xjvJ2M9dZoWAXNNr5LSBS7uHXY=
36 | github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU=
37 | github.com/novalagung/gubrak v1.0.0 h1:+iDvzUcSHUoa3bwP/ig40K2h9X+5cX2w5qcBb3izAwo=
38 | github.com/novalagung/gubrak/v2 v2.0.1 h1:OmabeSZsnf8Yd4FSZfm3F6VG2XoEsQBpktwtkjiCLdg=
39 | github.com/novalagung/gubrak/v2 v2.0.1/go.mod h1:zilliNLzP2RSdZ67Sz7NdqI4bg3j5zUTFR34tuXopA0=
40 | github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
41 | github.com/sirupsen/logrus v1.6.0 h1:UBcNElsrwanuuMsnGSlYmtmgbb23qDR5dG+6X6Oo89I=
42 | github.com/sirupsen/logrus v1.6.0/go.mod h1:7uNnSEd1DgxDLC74fIahvMZmmYsHGZGEOFrfsX/uA88=
43 | github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
44 | github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
45 | github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
46 | github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
47 | github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
48 | github.com/valyala/fasttemplate v1.0.1 h1:tY9CJiPnMXf1ERmG2EyK7gNUd+c6RKGD0IfU8WdUSz8=
49 | github.com/valyala/fasttemplate v1.0.1/go.mod h1:UQGH1tvbgY+Nz5t2n7tXsz52dQxojPUpymEIMZ47gx8=
50 | github.com/valyala/fasttemplate v1.2.1 h1:TVEnxayobAdVkhQfrfes2IzOB6o+z4roRkPF52WA1u4=
51 | github.com/valyala/fasttemplate v1.2.1/go.mod h1:KHLXt3tVN2HBp8eijSv/kGJopbvo7S+qRAEEKiv+SiQ=
52 | golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
53 | golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
54 | golang.org/x/crypto v0.0.0-20200820211705-5c72a883971a h1:vclmkQCjlDX5OydZ9wv8rBCcS0QyQY66Mpf/7BZbInM=
55 | golang.org/x/crypto v0.0.0-20200820211705-5c72a883971a/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
56 | golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3 h1:0GoQqolDA55aaLxZyTzK/Y2ePZzZTUrRacwib7cNsYQ=
57 | golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
58 | golang.org/x/net v0.0.0-20200904194848-62affa334b73 h1:MXfv8rhZWmFeqX3GNZRsd6vOLoaCHjYEX3qkRo3YBUA=
59 | golang.org/x/net v0.0.0-20200904194848-62affa334b73/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=
60 | golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
61 | golang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
62 | golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
63 | golang.org/x/sys v0.0.0-20190422165155-953cdadca894/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
64 | golang.org/x/sys v0.0.0-20190813064441-fde4db37ae7a h1:aYOabOQFp6Vj6W1F80affTUvO9UxmJRx8K0gsfABByQ=
65 | golang.org/x/sys v0.0.0-20190813064441-fde4db37ae7a/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
66 | golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
67 | golang.org/x/sys v0.0.0-20200223170610-d5e6a3e2c0ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
68 | golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
69 | golang.org/x/sys v0.0.0-20200909081042-eff7692f9009 h1:W0lCpv29Hv0UaM1LXb9QlBHLNP8UFfcKjblhVCWftOM=
70 | golang.org/x/sys v0.0.0-20200909081042-eff7692f9009/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
71 | golang.org/x/text v0.3.0 h1:g61tztE5qeGQ89tm6NTjjM9VPIm088od1l6aSorWRWg=
72 | golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
73 | golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=
74 | golang.org/x/text v0.3.3 h1:cokOdA+Jmi5PJGXLlLllQSgYigAEfHXJAERHVMaCc2k=
75 | golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
76 | golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
77 | gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
78 | gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
79 | 


--------------------------------------------------------------------------------
/index.html:
--------------------------------------------------------------------------------
  1 | <!DOCTYPE html>
  2 | <html>
  3 | <head>
  4 |     <title>WebSocket</title>
  5 | 
  6 |     <style type="text/css">
  7 |         .form {
  8 |             position: fixed;
  9 |             left: 0;
 10 |             bottom: 0;
 11 |             right: 0;
 12 |             background-color: #f9f9f9;
 13 |             border-top: 1px solid #78b8ef;
 14 |             padding: 5px 10px;
 15 |         }
 16 |         .form .placeholder,
 17 |         .form .input-message,
 18 |         .form button {
 19 |             display: block;
 20 |             margin-bottom: 5px;
 21 |         }
 22 |         .form .input-message {
 23 |             padding: 7px;
 24 |             border: 1px solid #ecebeb;
 25 |             border-radius: 4px;
 26 |             width: -webkit-fill-available;
 27 |         }
 28 |         .form button {
 29 |             width: 100%;
 30 |             color: white;
 31 |             padding: 7px 10px;
 32 |             border-radius: 4px;
 33 |             background-color: #78b8ef;
 34 |             border: 1px solid #5a9ed8;
 35 |         }
 36 |         .container {
 37 |             margin-bottom: 50px;
 38 |         }
 39 |         .container p {
 40 |             display: block;
 41 |         }
 42 |     </style>
 43 | </head>
 44 | <body>
 45 |     <div class="container"></div>
 46 | 
 47 |     <div class="form">
 48 |         <form onsubmit="app.doSendMessage(); return false;">
 49 |             <div class="placeholder">
 50 |                 <label>Hello <b class="username"></b>. Say something:</label>
 51 |             </div>
 52 |             <input class="input-message" type="text" placeholder="Enter message">
 53 |             <button type="submit">Send</button>
 54 |         </form>
 55 |     </div>
 56 | 
 57 |     <script type="text/javascript">
 58 |         var app = {}
 59 |         app.ws = undefined
 60 |         app.container = undefined
 61 | 
 62 |         app.print = function (message) {
 63 |             var el = document.createElement("p")
 64 |             el.innerHTML = message
 65 |             app.container.append(el)
 66 |         }
 67 | 
 68 |         app.doSendMessage = function () {
 69 |             var messageRaw = document.querySelector('.input-message').value
 70 |             app.ws.send(JSON.stringify({
 71 |                 Message: messageRaw
 72 |             }));
 73 | 
 74 |             var message = '<b>me</b>: ' + messageRaw
 75 |             app.print(message)
 76 | 
 77 |             document.querySelector('.input-message').value = ''
 78 |         }
 79 | 
 80 |         app.init = function () {
 81 |             if (!(window.WebSocket)) {
 82 |                 alert('Your browser does not support WebSocket')
 83 |                 return
 84 |             }
 85 | 
 86 |             var name = prompt('Enter your name please:') || "No name"
 87 |             document.querySelector('.username').innerText = name
 88 | 
 89 |             app.container = document.querySelector('.container')
 90 | 
 91 |             app.ws = new WebSocket("ws://localhost:8080/ws?username=" + name)
 92 | 
 93 |             app.ws.onopen = function() {
 94 |                 var message = '<b>me</b>: connected'
 95 |                 app.print(message)
 96 |             }
 97 | 
 98 |             app.ws.onmessage = function (event) {
 99 |                 var res = JSON.parse(event.data)
100 | 
101 |                 var messsage = ''
102 |                 if (res.Type === 'New User') {
103 |                     message = 'User <b>' + res.From + '</b>: connected'
104 |                 } else if (res.Type === 'Leave') {
105 |                     message = 'User <b>' + res.From + '</b>: disconnected'
106 |                 } else {
107 |                     message = '<b>' + res.From + '</b>: ' + res.Message 
108 |                 }
109 | 
110 |                 app.print(message)
111 |             }
112 | 
113 |             app.ws.onclose = function () {
114 |                 var message = '<b>me</b>: disconnected'
115 |                 app.print(message)
116 |             }
117 |         }
118 | 
119 |         window.onload = app.init
120 |     </script>
121 | </body>
122 | </html>


--------------------------------------------------------------------------------
/main.go:
--------------------------------------------------------------------------------
  1 | package main
  2 | 
  3 | import (
  4 | 	"fmt"
  5 | 	"io/ioutil"
  6 | 	"log"
  7 | 	"net/http"
  8 | 	"strings"
  9 | 
 10 | 	"github.com/gorilla/websocket"
 11 | 	gubrak "github.com/novalagung/gubrak/v2"
 12 | )
 13 | 
 14 | type M map[string]interface{}
 15 | 
 16 | const MESSAGE_NEW_USER = "New User"
 17 | const MESSAGE_CHAT = "Chat"
 18 | const MESSAGE_LEAVE = "Leave"
 19 | 
 20 | var connections = make([]*WebSocketConnection, 0)
 21 | 
 22 | type SocketPayload struct {
 23 | 	Message string
 24 | }
 25 | 
 26 | type SocketResponse struct {
 27 | 	From    string
 28 | 	Type    string
 29 | 	Message string
 30 | }
 31 | 
 32 | type WebSocketConnection struct {
 33 | 	*websocket.Conn
 34 | 	Username string
 35 | }
 36 | 
 37 | func main() {
 38 | 	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 39 | 		content, err := ioutil.ReadFile("index.html")
 40 | 		if err != nil {
 41 | 			http.Error(w, "Could not open requested file", http.StatusInternalServerError)
 42 | 			return
 43 | 		}
 44 | 
 45 | 		fmt.Fprintf(w, "%s", content)
 46 | 	})
 47 | 
 48 | 	http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
 49 | 		currentGorillaConn, err := websocket.Upgrade(w, r, w.Header(), 1024, 1024)
 50 | 		if err != nil {
 51 | 			http.Error(w, "Could not open websocket connection", http.StatusBadRequest)
 52 | 		}
 53 | 
 54 | 		username := r.URL.Query().Get("username")
 55 | 		currentConn := WebSocketConnection{Conn: currentGorillaConn, Username: username}
 56 | 		connections = append(connections, &currentConn)
 57 | 
 58 | 		go handleIO(&currentConn, connections)
 59 | 	})
 60 | 
 61 | 	fmt.Println("Server starting at :8080")
 62 | 	http.ListenAndServe(":8080", nil)
 63 | }
 64 | 
 65 | func handleIO(currentConn *WebSocketConnection, connections []*WebSocketConnection) {
 66 | 	defer func() {
 67 | 		if r := recover(); r != nil {
 68 | 			log.Println("ERROR", fmt.Sprintf("%v", r))
 69 | 		}
 70 | 	}()
 71 | 
 72 | 	broadcastMessage(currentConn, MESSAGE_NEW_USER, "")
 73 | 
 74 | 	for {
 75 | 		payload := SocketPayload{}
 76 | 		err := currentConn.ReadJSON(&payload)
 77 | 		if err != nil {
 78 | 			if strings.Contains(err.Error(), "websocket: close") {
 79 | 				broadcastMessage(currentConn, MESSAGE_LEAVE, "")
 80 | 				ejectConnection(currentConn)
 81 | 				return
 82 | 			}
 83 | 
 84 | 			log.Println("ERROR", err.Error())
 85 | 			continue
 86 | 		}
 87 | 
 88 | 		broadcastMessage(currentConn, MESSAGE_CHAT, payload.Message)
 89 | 	}
 90 | }
 91 | 
 92 | func ejectConnection(currentConn *WebSocketConnection) {
 93 | 	filtered := gubrak.From(connections).Reject(func(each *WebSocketConnection) bool {
 94 | 		return each == currentConn
 95 | 	}).Result()
 96 | 	connections = filtered.([]*WebSocketConnection)
 97 | }
 98 | 
 99 | func broadcastMessage(currentConn *WebSocketConnection, kind, message string) {
100 | 	for _, eachConn := range connections {
101 | 		if eachConn == currentConn {
102 | 			continue
103 | 		}
104 | 
105 | 		eachConn.WriteJSON(SocketResponse{
106 | 			From:    currentConn.Username,
107 | 			Type:    kind,
108 | 			Message: message,
109 | 		})
110 | 	}
111 | }


--------------------------------------------------------------------------------