feat: add control client

internal/client/trok.go

@@ -0,0 +1,71 @@
+/*
+Copyright © 2024 tux <0xtux@pm.me>
+*/
+
+package client
+
+import (
+	"net"
+
+	"github.com/0xtux/trok/internal/lib"
+	"github.com/rs/zerolog/log"
+)
+
+type Trok struct {
+	controlClient TCPClient
+}
+
+func (t *Trok) Init(port uint16) error {
+	err := t.controlClient.Init(port, "Controller")
+	return err
+}
+
+func (t *Trok) Start() {
+	go t.controlClient.Start(t.ControlConnHandler)
+	log.Info().Msgf("started Trok client on port %d", t.controlClient.Port())
+}
+
+func (t *Trok) Stop() {
+	t.controlClient.Stop()
+	log.Info().Msgf("stopped Trok client on port %d", t.controlClient.Port())
+}
+
+func (t *Trok) ControlConnHandler(conn net.Conn) {
+	p := lib.InitProtocolHandler(conn)
+
+	p.WriteMessage(&lib.Message{CMD: "HELO", ARG: "Trok"})
+
+	for {
+		m, err := p.ReadMessage()
+		if err != nil {
+			if netErr, ok := err.(net.Error); ok && netErr.Timeout() {
+				log.Warn().Msgf("connection timed out: %s", conn.RemoteAddr())
+			} else {
+				log.Logger.Info().Msgf("connection closed: %s", err)
+			}
+			return
+		}
+
+		switch m.CMD {
+
+		case "EHLO":
+			t.hanldeCMDEHLO(p, m)
+
+		case "CNCT":
+			t.handleCMDCNCT(p, m)
+
+		default:
+			log.Info().Msgf("invalid command")
+		}
+	}
+}
+
+func (t *Trok) hanldeCMDEHLO(p *lib.ProtocolHandler, m *lib.Message) {
+	log.Info().Msgf("[CMD] %s [ARG] %s", m.CMD, m.ARG)
+	p.WriteMessage(m)
+}
+
+func (t *Trok) handleCMDCNCT(p *lib.ProtocolHandler, m *lib.Message) {
+	log.Info().Msgf("[CMD] %s [ARG] %s", m.CMD, m.ARG)
+	p.WriteMessage(m)
+}