main.go

package main

import (
	"crypto/md5"
	"fmt"

	"next-terminal/pkg/config"
	"next-terminal/pkg/global"
	"next-terminal/pkg/task"
	"next-terminal/server/api"
	"next-terminal/server/repository"

	"github.com/labstack/gommon/log"
)

const Version = "v0.5.0"

func main() {
	err := Run()
	if err != nil {
		log.Fatal(err)
	}
}

func Run() error {

	fmt.Printf(`
 _______                   __    ___________                  .__              .__   
 \      \   ____ ___  ____/  |_  \__    ___/__________  _____ |__| ____ _____  |  |  
 /   |   \_/ __ \\  \/  /\   __\   |    |_/ __ \_  __ \/     \|  |/    \\__  \ |  |  
/    |    \  ___/ >    <  |  |     |    |\  ___/|  | \/  Y Y  \  |   |  \/ __ \|  |__
\____|__  /\___  >__/\_ \ |__|     |____| \___  >__|  |__|_|  /__|___|  (____  /____/
        \/     \/      \/                     \/            \/        \/     \/      ` + Version + "\n\n")

	// 为了兼容之前调用global包的代码 后期预期会改为调用pgk/config
	global.Config = config.GlobalCfg

	if global.Config.EncryptionKey == "" {
		global.Config.EncryptionKey = "next-terminal"
	}
	md5Sum := fmt.Sprintf("%x", md5.Sum([]byte(global.Config.EncryptionKey)))
	global.Config.EncryptionPassword = []byte(md5Sum)

	global.Cache = api.SetupCache()
	db := api.SetupDB()
	e := api.SetupRoutes(db)

	if global.Config.ResetPassword != "" {
		return api.ResetPassword(global.Config.ResetPassword)
	}
	if global.Config.ResetTotp != "" {
		return api.ResetTotp(global.Config.ResetTotp)
	}

	if global.Config.NewEncryptionKey != "" {
		return api.ChangeEncryptionKey(global.Config.EncryptionKey, global.Config.NewEncryptionKey)
	}

	sessionRepo := repository.NewSessionRepository(db)
	propertyRepo := repository.NewPropertyRepository(db)
	ticker := task.NewTicker(sessionRepo, propertyRepo)
	ticker.SetupTicker()

	if global.Config.Server.Cert != "" && global.Config.Server.Key != "" {
		return e.StartTLS(global.Config.Server.Addr, global.Config.Server.Cert, global.Config.Server.Key)
	} else {
		return e.Start(global.Config.Server.Addr)
	}

}