Streamer272/keys
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

🚧 Adds database connection

Browse Source
This commit is contained in:
Daniel Svitan
2025-03-26 21:12:35 +01:00
parent 28883c620d
commit ebe87d1eb3
4 changed files with 119 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
server/main.go
View File

@@ -26,7 +26,7 @@ type Client struct {
var id = 0
var clients = []*Client{}
var keyDir = "."
var dataDir = "."
var token = ""
var upgrader = websocket.Upgrader{}
@@ -36,8 +36,8 @@ type ReqData struct {
func main() {
_ = godotenv.Load()
keyDir = os.Getenv("KEY_DIR")
if keyDir == "" {
dataDir = os.Getenv("DATA_DIR")
if dataDir == "" {
log.Fatal("KEY_DIR is not defined")
}
token = os.Getenv("TOKEN")
@@ -45,9 +45,11 @@ func main() {
log.Fatal("TOKEN is not defined")
}
f, err := os.OpenFile(path.Join(keyDir, "startup"), syscall.O_CREAT|syscall.O_APPEND|syscall.O_WRONLY, 0644)
ConnectDbFromEnv()
f, err := os.OpenFile(path.Join(dataDir, "startup"), syscall.O_CREAT|syscall.O_APPEND|syscall.O_WRONLY, 0644)
if err != nil {
log.Fatalf("KEY_DIR %s is not writable (%v)\n", keyDir, err)
log.Fatalf("KEY_DIR %s is not writable (%v)\n", dataDir, err)
}
_, err = f.Write([]byte(fmt.Sprintf("%s\n", time.Now().Format(TimeFormat))))
if err != nil {
@@ -166,7 +168,7 @@ func keys(c echo.Context) error {
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(fmt.Sprintf("%s/%s_%d.txt", dataDir, 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)\n", time.Now().Format(TimeFormat), client.ID)