main.go

// Package api_show /main.go
package main

import (
	"embed"
	"fmt"
	"os"
	"runtime"
	"runtime/debug"
	"time"

	"github.com/0xYeah/api_show/api"
	"github.com/0xYeah/api_show/as_config"
	"github.com/0xYeah/api_show/auth"
	"github.com/0xYeah/api_show/common"
	"github.com/0xYeah/api_show/custom_cmd"
	"github.com/george012/gtbox"
	"github.com/george012/gtbox/gtbox_cmd"
	"github.com/george012/gtbox/gtbox_log"
)

//go:embed webroot
var webrootFS embed.FS

var (
	mRunMode       = ""
	mGitCommitHash = ""
	mGitCommitTime = ""
	mPackageOS     = ""
	mPackageTime   = ""
	mGoVersion     = ""
)

func setupApp() {
	runMode := gtbox.RunModeDebug
	switch mRunMode {
	case "debug":
		runMode = gtbox.RunModeDebug
	case "test":
		runMode = gtbox.RunModeTest
	case "release":
		runMode = gtbox.RunModeRelease
	default:
		runMode = gtbox.RunModeDebug
	}

	as_config.CurrentApp = as_config.NewApp(
		as_config.ProjectName,
		as_config.ProjectBundleID,
		fmt.Sprintf("%s service",
			as_config.ProjectName,
		),
		runMode,
	)

	if as_config.CurrentApp.CurrentRunMode == gtbox.RunModeDebug {
		cmdMap := map[string]string{
			"git_commit_hash": "git show -s --format=%H",
			"git_commit_time": "git show -s --format=\"%ci\" | cut -d ' ' -f 1,2 | sed 's/ /_/'",
			"build_os":        "go env GOOS",
			"go_version":      "go version | awk '{print $3}'",
		}
		cmdRes := gtbox_cmd.RunWith(cmdMap)

		if cmdRes != nil {
			mGitCommitHash = cmdRes["git_commit_hash"]
			mGitCommitTime = cmdRes["git_commit_time"]
			mPackageOS = cmdRes["build_os"]
			mGoVersion = cmdRes["go_version"]
			mPackageTime = time.Now().UTC().Format("2006-01-02_15:04:05")
		}
	}

	as_config.CurrentApp.GitCommitHash = mGitCommitHash
	as_config.CurrentApp.GitCommitTime = mGitCommitTime
	as_config.CurrentApp.GoVersion = mGoVersion
	as_config.CurrentApp.PackageOS = mPackageOS
	as_config.CurrentApp.PackageTime = mPackageTime

	custom_cmd.HandleCustomCmds(os.Args, as_config.CurrentApp)

	gtbox.SetupGTBox(as_config.CurrentApp.AppName,
		as_config.CurrentApp.CurrentRunMode,
		as_config.CurrentApp.AppLogPath,
		30,
		gtbox_log.GTLogSaveHours,
		as_config.CurrentApp.HTTPRequestTimeOut,
	)
}

func main() {
	runtime.LockOSThread()

	setupApp()

	defer func() {
		if r := recover(); r != nil {
			fmt.Println("Recovered from panic:", r)
			gtbox_log.LogErrorf("系统崩溃:%v", r)
			gtbox_log.LogErrorf("系统崩溃:\n%s", debug.Stack())
		}
	}()

	if as_config.CurrentApp.CurrentRunMode == gtbox.RunModeDebug {
		as_config.CurrentApp.AppConfigFilePath = "./example_files/config_local.toml"
	} else {
		as_config.CurrentApp.AppConfigFilePath = "./conf/config.toml"
	}

	if err := as_config.SyncConfigFile(as_config.CurrentApp.AppConfigFilePath); err != nil {
		gtbox_log.LogErrorf("[sync config error] %s: %v", as_config.CurrentApp.AppConfigFilePath, err)
		os.Exit(1)
	}

	cfg := as_config.GlobalConfig

	if err := auth.LoadUsersFromConfig(cfg.Users); err != nil {
		gtbox_log.LogErrorf("[api_show] load users: %v", err)
		os.Exit(1)
	}

	if cfg.Secret == "" || len(cfg.Secret) < 16 {
		gtbox_log.LogErrorf("[api_show] secret missing or <16 bytes — refusing to start")
		os.Exit(1)
	}
	secret := []byte(cfg.Secret)

	dataDir := cfg.DataDir
	if dataDir == "" {
		dataDir = as_config.DefaultDataDir
	}
	listenPort := cfg.Port
	if listenPort == 0 {
		listenPort = as_config.ListenPort
	}

	api.StartAPIServices(&api.Options{
		Deps: auth.HandlerDeps{
			Secret: secret,
			TOTP:   auth.NewTOTPStore(dataDir, secret),
			Pwd:    auth.NewPwdStore(dataDir, secret),
		},
		DataDir:     dataDir,
		ListenPort:  listenPort,
		UploadToken: cfg.UploadToken,
		OnlyLocal:   cfg.OnlyLocal,
		WebrootFS:   webrootFS,
	})

	common.LoadSigHandle(nil, nil)
}