wahandler peer 파일 분리

wshandler/wshandler_peer.go

@@ -0,0 +1,159 @@
+package wshandler
+
+import (
+	"encoding/base64"
+	"encoding/hex"
+	"io"
+	"net/http"
+	"strings"
+
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"repositories.action2quare.com/ayo/gocommon"
+	"repositories.action2quare.com/ayo/gocommon/logger"
+	"repositories.action2quare.com/ayo/gocommon/session"
+
+	"github.com/gorilla/websocket"
+)
+
+type WebsocketPeerHandler struct {
+	WebsocketPeerApiBroker
+	sessionConsumer session.Consumer
+}
+
+func NewWebsocketPeerHandler(consumer session.Consumer) WebsocketPeerHandler {
+	return WebsocketPeerHandler{
+		sessionConsumer: consumer,
+	}
+}
+
+func (ws *WebsocketPeerHandler) RegisterHandlers(serveMux *http.ServeMux, prefix string) error {
+	url := gocommon.MakeHttpHandlerPattern(prefix, "ws")
+	if *noAuthFlag {
+		serveMux.HandleFunc(url, ws.upgrade_nosession)
+	} else {
+		serveMux.HandleFunc(url, ws.upgrade)
+	}
+
+	return nil
+}
+
+func (ws *WebsocketPeerHandler) upgrade_core(conn *websocket.Conn, accid primitive.ObjectID, alias string) {
+	go func(c *websocket.Conn, accid primitive.ObjectID) {
+		peer := ws.CreatePeer(accid)
+		ws.ClientConnected(peer, c)
+
+		var closeReason string
+		for {
+			messageType, r, err := c.NextReader()
+			if err != nil {
+				if ce, ok := err.(*websocket.CloseError); ok {
+					closeReason = ce.Text
+				}
+				c.Close()
+				break
+			}
+
+			if messageType == websocket.CloseMessage {
+				closeMsg, _ := io.ReadAll(r)
+				closeReason = string(closeMsg)
+				break
+			}
+
+			if messageType == websocket.BinaryMessage {
+				var size [1]byte
+				r.Read(size[:])
+				cmd := make([]byte, size[0])
+				r.Read(cmd)
+				ws.Call(peer, string(cmd), r)
+			}
+		}
+		ws.ClientDisconnected(peer, closeReason)
+	}(conn, accid)
+}
+
+func (ws *WebsocketPeerHandler) upgrade_nosession(w http.ResponseWriter, r *http.Request) {
+	// 클라이언트 접속
+	defer func() {
+		s := recover()
+		if s != nil {
+			logger.Error(s)
+		}
+		io.Copy(io.Discard, r.Body)
+		r.Body.Close()
+	}()
+
+	auth := strings.Split(r.Header.Get("Authorization"), " ")
+	if len(auth) != 2 {
+		w.WriteHeader(http.StatusBadRequest)
+		return
+	}
+
+	temp, err := hex.DecodeString(auth[1])
+	if err != nil {
+		w.WriteHeader(http.StatusBadRequest)
+		return
+	}
+
+	if len(temp) != len(primitive.NilObjectID) {
+		w.WriteHeader(http.StatusBadRequest)
+		return
+	}
+
+	raw := (*[12]byte)(temp)
+	accid := primitive.ObjectID(*raw)
+
+	var upgrader = websocket.Upgrader{} // use default options
+	conn, err := upgrader.Upgrade(w, r, nil)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+
+	var alias string
+	if v := r.Header.Get("AS-X-ALIAS"); len(v) > 0 {
+		vt, _ := base64.StdEncoding.DecodeString(v)
+		alias = string(vt)
+	} else {
+		alias = accid.Hex()
+	}
+
+	ws.upgrade_core(conn, accid, alias)
+}
+
+func (ws *WebsocketPeerHandler) upgrade(w http.ResponseWriter, r *http.Request) {
+	// 클라이언트 접속
+	defer func() {
+		s := recover()
+		if s != nil {
+			logger.Error(s)
+		}
+		io.Copy(io.Discard, r.Body)
+		r.Body.Close()
+	}()
+
+	sk := r.Header.Get("AS-X-SESSION")
+	logger.Println("WebsocketHandler.upgrade sk :", sk)
+	authinfo, err := ws.sessionConsumer.Query(sk)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		logger.Error("authorize query failed :", err)
+		return
+	}
+
+	var upgrader = websocket.Upgrader{} // use default options
+	conn, err := upgrader.Upgrade(w, r, nil)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+
+	var alias string
+	if v := r.Header.Get("AS-X-ALIAS"); len(v) > 0 {
+		vt, _ := base64.StdEncoding.DecodeString(v)
+		alias = string(vt)
+	} else {
+		alias = authinfo.Account.Hex()
+	}
+
+	ws.upgrade_core(conn, authinfo.Account, alias)
+}