From 9f55bab45b3709a1b675411461e6adca59ac2606 Mon Sep 17 00:00:00 2001 From: Daniel Svitan Date: Thu, 20 Mar 2025 17:16:36 +0100 Subject: [PATCH] :tada: Initial server commit --- server/go.mod | 21 ++++++ server/go.sum | 35 +++++++++ server/main.go | 195 +++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 251 insertions(+) create mode 100644 server/go.mod create mode 100644 server/go.sum create mode 100644 server/main.go diff --git a/server/go.mod b/server/go.mod new file mode 100644 index 0000000..2b032e9 --- /dev/null +++ b/server/go.mod @@ -0,0 +1,21 @@ +module svitan.dev/keys/server + +go 1.24.1 + +require ( + github.com/labstack/echo/v4 v4.13.3 + github.com/labstack/gommon v0.4.2 +) + +require ( + github.com/joho/godotenv v1.5.1 // indirect + github.com/mattn/go-colorable v0.1.13 // indirect + github.com/mattn/go-isatty v0.0.20 // indirect + github.com/valyala/bytebufferpool v1.0.0 // indirect + github.com/valyala/fasttemplate v1.2.2 // indirect + golang.org/x/crypto v0.31.0 // indirect + golang.org/x/net v0.33.0 // indirect + golang.org/x/sys v0.28.0 // indirect + golang.org/x/text v0.21.0 // indirect + golang.org/x/time v0.8.0 // indirect +) diff --git a/server/go.sum b/server/go.sum new file mode 100644 index 0000000..f75a767 --- /dev/null +++ b/server/go.sum @@ -0,0 +1,35 @@ +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0= +github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4= +github.com/labstack/echo/v4 v4.13.3 h1:pwhpCPrTl5qry5HRdM5FwdXnhXSLSY+WE+YQSeCaafY= +github.com/labstack/echo/v4 v4.13.3/go.mod h1:o90YNEeQWjDozo584l7AwhJMHN0bOC4tAfg+Xox9q5g= +github.com/labstack/gommon v0.4.2 h1:F8qTUNXgG1+6WQmqoUWnz8WiEU60mXVVw0P4ht1WRA0= +github.com/labstack/gommon v0.4.2/go.mod h1:QlUFxVM+SNXhDL/Z7YhocGIBYOiwB0mXm1+1bAPHPyU= +github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA= +github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg= +github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM= +github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= +github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA= +github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= +github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw= +github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc= +github.com/valyala/fasttemplate v1.2.2 h1:lxLXG0uE3Qnshl9QyaK6XJxMXlQZELvChBOCmQD0Loo= +github.com/valyala/fasttemplate v1.2.2/go.mod h1:KHLXt3tVN2HBp8eijSv/kGJopbvo7S+qRAEEKiv+SiQ= +golang.org/x/crypto v0.31.0 h1:ihbySMvVjLAeSH1IbfcRTkD/iNscyz8rGzjF/E5hV6U= +golang.org/x/crypto v0.31.0/go.mod h1:kDsLvtWBEx7MV9tJOj9bnXsPbxwJQ6csT/x4KIN4Ssk= +golang.org/x/net v0.33.0 h1:74SYHlV8BIgHIFC/LrYkOGIwL19eTYXQ5wc6TBuO36I= +golang.org/x/net v0.33.0/go.mod h1:HXLR5J+9DxmrqMwG9qjGCxZ+zKXxBru04zlTvWlWuN4= +golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.28.0 h1:Fksou7UEQUWlKvIdsqzJmUmCX3cZuD2+P3XyyzwMhlA= +golang.org/x/sys v0.28.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo= +golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ= +golang.org/x/time v0.8.0 h1:9i3RxcPv3PZnitoVGMPDKZSq1xW1gK1Xy3ArNOGZfEg= +golang.org/x/time v0.8.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/server/main.go b/server/main.go new file mode 100644 index 0000000..c795479 --- /dev/null +++ b/server/main.go @@ -0,0 +1,195 @@ +package main + +import ( + "errors" + "fmt" + "net/http" + "os" + "syscall" + "time" + + "github.com/joho/godotenv" + "github.com/labstack/echo/v4" + "github.com/labstack/echo/v4/middleware" + "github.com/labstack/gommon/log" + "golang.org/x/net/websocket" +) + +const TimeFormat = "2006-01-02 15:04:05" + +type Client struct { + ID int + conn *websocket.Conn + Exit bool +} + +var id = 0 +var clients []*Client +var keyDir = "." +var token = "" + +type ReqData struct { + ID int `json:"id"` +} + +func main() { + err := godotenv.Load() + if err != nil { + log.Fatal(err) + } + keyDir = os.Getenv("KEY_DIR") + if keyDir == "" { + log.Fatal("KEY_DIR is not defined") + } + token = os.Getenv("TOKEN") + if token == "" { + log.Fatal("TOKEN is not defined") + } + + app := echo.New() + app.Logger.SetLevel(log.INFO) + + app.Use(middleware.RecoverWithConfig(middleware.RecoverConfig{ + StackSize: 1 << 10, + LogLevel: log.ERROR, + })) + app.Use(middleware.Secure()) + + app.Use(middleware.LoggerWithConfig(middleware.LoggerConfig{ + Format: "${time_custom} ${method} ${uri} ---> ${status} in ${latency_human} (${bytes_out} bytes)", + CustomTimeFormat: TimeFormat, + })) + app.Use(middleware.RemoveTrailingSlash()) + + app.OnAddRouteHandler = func(host string, route echo.Route, handler echo.HandlerFunc, middleware []echo.MiddlewareFunc) { + now := time.Now() + fmt.Printf("%s registered %-6s %s\n", now.Format(TimeFormat), route.Method, route.Path) + } + app.HTTPErrorHandler = func(err error, c echo.Context) { + if c.Response().Committed { + log.Error(err) + return + } + + code := http.StatusInternalServerError + var response any = err.Error() + + var httpErr *echo.HTTPError + if errors.As(err, &httpErr) { + code = httpErr.Code + response = httpErr + } + + if code >= 500 { + log.Error(err) + } + + err2 := c.JSON(code, response) + if err2 != nil { + log.Error(err2) + } + } + + app.GET("/", func(c echo.Context) error { + return c.JSON(http.StatusOK, "Hello World!") + }) + app.GET("/key", key) + + app.GET("/clients", func(c echo.Context) error { + if c.Request().Header.Get("Authorization") != token { + return c.NoContent(http.StatusUnauthorized) + } + + var a []echo.Map + for _, client := range clients { + a = append(a, echo.Map{ + "id": client.ID, + "exit": client.Exit, + }) + } + + return c.JSON(http.StatusOK, a) + }) + app.POST("/exit", func(c echo.Context) error { + if c.Request().Header.Get("Authorization") != token { + return c.NoContent(http.StatusUnauthorized) + } + + var data ReqData + err := c.Bind(&data) + if err != nil { + return c.JSON(http.StatusBadRequest, echo.Map{"message": "missing field 'id'"}) + } + + for _, client := range clients { + if client.ID == data.ID { + client.Exit = true + return c.JSON(http.StatusOK, echo.Map{"message": "ok"}) + } + } + return c.JSON(http.StatusNotFound, echo.Map{"message": "not found"}) + }) + + log.Fatal(app.Start(":8080")) +} + +func key(c echo.Context) error { + websocket.Handler(func(ws *websocket.Conn) { + client := Client{ + ID: id, + conn: ws, + Exit: false, + } + clients = append(clients, &client) + id += 1 + fmt.Printf("%s client %-3d connected", time.Now().Format(TimeFormat), client.ID) + + defer ws.Close() + err := websocket.Message.Send(ws, "welcome") + if err != nil { + log.Error(err) + } + + f, err := os.OpenFile(fmt.Sprintf("%s/%s_%d.txt", keyDir, time.Now().Format("2006-01-02_15-04"), client.ID), syscall.O_CREAT|syscall.O_APPEND|syscall.O_WRONLY, 0644) + if err != nil { + log.Error(err) + fmt.Printf("%s client %-3d crashed (couldn't open file)", time.Now().Format(TimeFormat), client.ID) + client.Exit = true + return + } + + for { + if client.Exit { + _ = websocket.Message.Send(ws, "exit") + _ = ws.Close() + break + } + + msg := "" + err = websocket.Message.Receive(ws, &msg) + if err != nil { + log.Error(err) + _ = websocket.Message.Send(ws, "exit") + _ = ws.Close() + fmt.Printf("%s client %-3d crashed (websocket error)", time.Now().Format(TimeFormat), client.ID) + client.Exit = true + break + } + + _, err = f.WriteString(msg) + if err != nil { + log.Error(err) + _ = websocket.Message.Send(ws, "exit") + _ = ws.Close() + fmt.Printf("%s client %-3d crashed (file write error)", time.Now().Format(TimeFormat), client.ID) + client.Exit = true + break + } + } + + fmt.Printf("%s client %-3d disconnected", time.Now().Format(TimeFormat), client.ID) + client.Exit = true + }).ServeHTTP(c.Response(), c.Request()) + + return nil +}