snapshot of https://github.com/traggo/server.git (master, 2026-09-16; blobs > 131072B omitted)

2026-09-16 05:24:45 +05:00
commit cda7c1a474
272 changed files with 31973 additions and 0 deletions
--- a/main.go
+++ b/main.go
@@ -0,0 +1,100 @@
+package main
+
+import (
+	"os"
+	"time"
+
+	"github.com/gorilla/mux"
+	"github.com/jinzhu/gorm"
+	"github.com/rs/zerolog"
+	"github.com/rs/zerolog/log"
+	"github.com/traggo/server/auth"
+	"github.com/traggo/server/config"
+	"github.com/traggo/server/config/mode"
+	"github.com/traggo/server/database"
+	"github.com/traggo/server/graphql"
+	"github.com/traggo/server/logger"
+	"github.com/traggo/server/model"
+	"github.com/traggo/server/server"
+	"github.com/traggo/server/ui"
+	"github.com/traggo/server/user/password"
+)
+
+var (
+	// BuildMode will be set via ldflags
+	BuildMode = mode.Dev
+	// BuildCommit will be set via ldflags
+	BuildCommit = "-"
+	// BuildVersion will be set via ldflags
+	BuildVersion = "develop"
+	// BuildDate will be set via ldflags
+	BuildDate = "unknown"
+)
+
+func main() {
+	mode.Set(BuildMode)
+
+	version := model.Version{Commit: BuildCommit, BuildDate: BuildDate, Name: BuildVersion}
+
+	conf, errs := config.Get()
+	logger.Init(conf.LogLevel.AsZeroLogLevel())
+	log.Info().Str("commit", BuildCommit).Str("buildDate", BuildDate).Str("buildMode", BuildMode).Str("version", BuildVersion).Msg("Traggo")
+
+	exit := false
+	for _, err := range errs {
+		log.WithLevel(err.Level).Msg(err.Msg)
+		exit = exit || err.Level == zerolog.FatalLevel || err.Level == zerolog.PanicLevel
+	}
+	if exit {
+		os.Exit(1)
+	}
+	log.Debug().Interface("config", conf).Msg("Using")
+
+	db := initDatabase(conf)
+
+	stopCleanUp := initCleanUp(db)
+
+	router := initRouter(db, conf, version)
+
+	log.Info().Int("port", conf.Port).Msg("Start listening")
+	if err := server.Start(router, conf.Port); err != nil {
+		log.Fatal().Err(err).Msg("Server Error")
+	}
+	stopCleanUp <- true
+}
+
+func initRouter(db *gorm.DB, conf config.Config, version model.Version) *mux.Router {
+	gqlHandler := graphql.Handler(
+		"/graphql",
+		graphql.NewResolver(db, conf.PassStrength, version),
+		graphql.NewDirective())
+
+	router := mux.NewRouter()
+	router.Use(auth.Middleware(db))
+	router.HandleFunc("/graphql", gqlHandler)
+	ui.Register(router)
+	return router
+}
+
+func initDatabase(conf config.Config) *gorm.DB {
+	db, err := database.New(conf.DatabaseDialect, conf.DatabaseConnection)
+	if err != nil {
+		log.Fatal().Err(err).Msg("Failed to connect to the database")
+	}
+	c := new(int)
+	if db.Model(new(model.User)).Count(c); *c == 0 {
+		log.Info().Msg("Creating default user.")
+		db.Create(&model.User{
+			Name:  conf.DefaultUserName,
+			Pass:  password.CreatePassword(conf.DefaultUserPass, conf.PassStrength),
+			Admin: true})
+	}
+
+	return db
+}
+
+func initCleanUp(db *gorm.DB) chan<- bool {
+	stopCleanUp := make(chan bool)
+	go auth.CleanUp(db, time.Minute*10, stopCleanUp)
+	return stopCleanUp
+}