feat: read config from config.yaml (#10)

Makefile

@@ -10,6 +10,7 @@ test:
 ### Formatting, linting, and vetting
 fmt:
 	gofmt -s -w .
+	go mod tidy
 
 check:
 	golangci-lint run \

cmd/main.go

@@ -1,11 +1,13 @@
 package main
 
 import (
+	"os"
+
 	"github.com/zurvan-lab/TimeTraceDB/src"
 )
 
 func main() {
-	database := src.CreateDataBase()
+	database := src.CreateDataBase(os.Args[1])
 	database.InitSocket()
 	database.InitUsers()
 }

config/config.yaml

@@ -2,17 +2,18 @@ name: TimeTraceConfig
 authorization: false
 
 proc:
- - cores: 2
- - threads: 4 
+  cores: 2
+  threads: 4
 
 server:
-  - listen: localhost
-  - port: 7070
+  listen: localhost
+  port: 7070
 
 log:
-  - path: ./log
+  path: ./log
 
 user:
-  - name: root
-  - token: super_secret_token
-  - cmd: []
+  name: root
+  token: super_secret_token
+  cmd:
+    - all

src/config.go

@@ -8,20 +8,32 @@ import (
 )
 
 type Config struct {
-	Authorization bool
-	Proc          struct {
-		Cores, Threads int
-	}
-	Listen struct {
-		IP, Port string
-	}
-	Log struct {
-		Path string
-	}
-	User struct {
-		Name  string
-		Token string
-	}
+	Name          string    `yaml:"name"`
+	Authorization bool      `yaml:"authorization"`
+	Proc          Proc      `yaml:"proc"`
+	Listen        Listen    `yaml:"server"`
+	Log           Log       `yaml:"log"`
+	FirstUser     FirstUser `yaml:"user"`
+}
+
+type Proc struct {
+	Cores   int `yaml:"cores"`
+	Threads int `yaml:"threads"`
+}
+
+type Listen struct {
+	IP   string `yaml:"listen"`
+	Port string `yaml:"port"`
+}
+
+type Log struct {
+	Path string `yaml:"path"`
+}
+
+type FirstUser struct {
+	Name  string   `yaml:"name"`
+	Token string   `yaml:"token"`
+	Cmd   []string `yaml:"cmd"`
 }
 
 func createConfig() *Config {

src/database.go

@@ -12,10 +12,10 @@ type Database struct {
 	Users  Users
 }
 
-func CreateDataBase() *Database {
+func CreateDataBase(path string) *Database {
 	return &Database{
 		Sets:   *NewSets(),
-		Config: *ReadConfigFile(""),
+		Config: *ReadConfigFile(path),
 	}
 }
 
@@ -31,5 +31,5 @@ func (db *Database) InitUsers() {
 	users := CreateUsers()
 	db.Users = *users
 	cmds := []string{"all"}
-	users.NewUser(db.Config.User.Name, db.Config.User.Token, cmds)
+	users.NewUser(db.Config.FirstUser.Name, db.Config.FirstUser.Token, cmds)
 }