KeyDB/main.go at main · creatorkostas/KeyDB · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
package main

import (
	"flag"
	"os"
	"strconv"

	"github.com/creatorkostas/KeyDB/database/database_core/conf"
	internal "github.com/creatorkostas/KeyDB/database/database_core/conf"
	"github.com/creatorkostas/KeyDB/database/database_core/users"
	cmd_interface "github.com/creatorkostas/KeyDB/database/database_interfaces/cmd"
	web_interface "github.com/creatorkostas/KeyDB/database/database_interfaces/web"
	// cmd_api "github.com/creatorkostas/KeyDB/database/database_interfaces/cmd_interface"
	// "github.com/creatorkostas/KeyDB/internal/database/database_test"
)

// var DB_filename string = "db.gob"

// var router = gin.New()

// func initialize() {
// 	// gin.SetMode(gin.ReleaseMode)
// 	gin.SetMode(gin.DebugMode)

// 	db_utils.LoadDB(internal.DB_filename)
// 	users.LoadAccounts(internal.Accounts_filename)

// 	// c := make(chan os.Signal, 2)
// 	// signal.Notify(c, os.Interrupt, syscall.SIGTERM)
// 	// go func() {
// 	// 	<-c
// 	// 	cleanup()
// 	// 	os.Exit(1)
// 	// }()

// 	persistance.Start_writers(1)

// 	web_api.Setup_router(router)
// 	web_api.Add_endpoints(router)

// }

func main() {

	cmd := false
	unix := false
	devMode := false
	conf_path := "config.yaml"
	flag.BoolVar(&devMode, "dev", devMode, "enable dev mode")
	flag.BoolVar(&cmd, "cmd", cmd, "enable cmd mode")
	flag.BoolVar(&unix, "unix", unix, "enable communication through unix port (/tmp/keydb_sock.sock)")
	flag.StringVar(&conf_path, "conf", conf_path, "Set the config file")
	flag.Parse()

	internal.Load_configs(conf_path)
	// initialize()
	var port = os.Getenv("PORT")

	conf.StartUnix = unix
	var local_acc users.Account = users.MakeDefaultUser()
	local_acc.MakeAdmin()
	local_acc.Username = "Local cmd admin account"

	if port == "" {
		cmd_interface.StartKeyDB(&local_acc, devMode, conf.StartWeb, strconv.Itoa(8080), conf.StartUnix, web_interface.Setup_router, web_interface.Add_endpoints)
	} else {
		cmd_interface.StartKeyDB(&local_acc, devMode, conf.StartWeb, port, conf.StartUnix, web_interface.Setup_router, web_interface.Add_endpoints)
	}

	if cmd {
		conf.Number_of_writers = 0
		cmd_interface.StartKeyDB(&local_acc, devMode, false, port, false, web_interface.Setup_router, web_interface.Add_endpoints)
		cmd_interface.Cmd_start(&local_acc)
	}
	// router.Run(":8080")
}