🎉 Initial server commit
This commit is contained in:
Daniel Svitan
commit
9f55bab45b
21
server/go.mod
Normal file
21
server/go.mod
Normal file
@ -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
|
||||
)
|
||||
35
server/go.sum
Normal file
35
server/go.sum
Normal file
@ -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=
|
||||
195
server/main.go
Normal file
195
server/main.go
Normal file
@ -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
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user