59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
"github.com/charmbracelet/log"
|
|
_ "github.com/lib/pq"
|
|
|
|
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/models"
|
|
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/tasks"
|
|
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/tasks/runner"
|
|
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/users"
|
|
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
|
|
)
|
|
|
|
const (
|
|
host = "localhost"
|
|
port = 5432
|
|
user = "postgres"
|
|
password = "verysafepassword"
|
|
dbname = "aistuff"
|
|
)
|
|
|
|
func main() {
|
|
psqlInfo := fmt.Sprintf("host=%s port=%d user=%s "+
|
|
"password=%s dbname=%s sslmode=disable",
|
|
host, port, user, password, dbname)
|
|
|
|
db, err := sql.Open("postgres", psqlInfo)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer db.Close()
|
|
log.Info("Starting server on :5002!")
|
|
|
|
config := LoadConfig()
|
|
log.Info("Config loaded!", "config", config)
|
|
config.GenerateToken(db)
|
|
|
|
db.SetMaxOpenConns(config.DbInfo.MaxConnections)
|
|
|
|
StartRunners(db, config)
|
|
|
|
//TODO check if file structure exists to save data
|
|
handle := NewHandler(db, config)
|
|
|
|
config.Cleanup(db)
|
|
|
|
// TODO Handle this in other way
|
|
handle.ReadTypesFilesApi("/savedData/", ".", []string{".png", ".jpeg"}, []string{"image/png", "image/jpeg"})
|
|
|
|
UsersEndpints(db, handle)
|
|
HandleModels(handle)
|
|
HandleTasks(handle)
|
|
|
|
handle.Startup()
|
|
}
|