automatically generate a service user at startup closes #83

This commit is contained in:
Andre Henriques 2024-04-15 13:01:16 +01:00
parent 00e862e016
commit 2318bad5d8
3 changed files with 65 additions and 5 deletions

View File

@ -6,6 +6,9 @@ NUMBER_OF_WORKERS=20
SUPRESS_CUDA=1
[ServiceUser]
USER="service"
[Worker]
PULLING_TIME="500ms"
NUMBER_OF_WORKERS=1

View File

@ -1,7 +1,11 @@
package utils
import (
"database/sql"
"os"
"strings"
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/db_types"
toml "github.com/BurntSushi/toml"
"github.com/charmbracelet/log"
@ -12,6 +16,11 @@ type WorkerConfig struct {
Pulling string `toml:"pulling_time"`
}
type ServiceUser struct {
User string `toml:"user"`
UserId string `toml:"__user__id__"`
}
type Config struct {
Hostname string
Port int
@ -19,6 +28,8 @@ type Config struct {
SupressCuda int `toml:"supress_cuda"`
GpuWorker WorkerConfig `toml:"Worker"`
ServiceUser ServiceUser `toml:"ServiceUser"`
}
func LoadConfig() Config {
@ -37,6 +48,10 @@ func LoadConfig() Config {
NumberOfWorkers: 1,
Pulling: "500ms",
},
ServiceUser: ServiceUser{
User: "Service",
UserId: "",
},
}
}
@ -51,3 +66,44 @@ func LoadConfig() Config {
return conf
}
func (c *Config) GenerateToken(db *sql.DB) {
if c.ServiceUser.User == "" {
log.Fatal("A user needs to be set in a configuration file")
}
var user struct {
Password string `db:"password"`
UserId string `db:"id"`
}
err := GetDBOnce(db, &user, "users where username=$1;", c.ServiceUser.User)
if err == NotFoundError {
var newUser = struct {
Username string
Email string
Salt string
Password string
UserType UserType `db:"user_type"`
}{
c.ServiceUser.User,
c.ServiceUser.User,
"",
"",
User_Admin,
}
id, err := InsertReturnId(db, &newUser, "users", "id")
if err != nil {
log.Fatal("Failed to create user", "err", err)
}
c.ServiceUser.UserId = id
} else if err != nil {
log.Fatal("To get user name", "err", err)
return
} else {
if len(strings.ReplaceAll(user.Password, " ", "")) != 0 {
log.Fatal("User already exists and is not the service user", "user", user)
}
c.ServiceUser.UserId = user.UserId
}
}

View File

@ -37,6 +37,7 @@ func main() {
config := LoadConfig()
log.Info("Config loaded!", "config", config)
config.GenerateToken(db)
StartRunners(db, config)