main.go

package main

import (
	"flag"
	"fmt"
	"log"
	"os"
	"path/filepath"
	"runtime/debug"
	"smart-git/config"
	"smart-git/database"
	"smart-git/gitc"
	"strings"

	"github.com/WJQSERVER-STUDIO/logger"
)

var (
	cfgfile string = "./config/config"
	cfg     *config.Config
)

var (
	logw       = logger.Logw
	logDump    = logger.LogDump
	logDebug   = logger.LogDebug
	logInfo    = logger.LogInfo
	logWarning = logger.LogWarning
	logError   = logger.LogError
)

func ReadFlag() {
	cfgfilePtr := flag.String("c", "./config/config", "config file path")
	cfgfileCompatPtr := flag.String("cfg", "", "config file path (compat alias)")
	flag.Parse()
	if *cfgfileCompatPtr != "" {
		cfgfile = *cfgfileCompatPtr
		return
	}
	cfgfile = *cfgfilePtr
}

func loadConfig() {
	var err error
	// 初始化配置
	cfg, err = config.LoadConfig(cfgfile)
	if err != nil {
		log.Fatalf("Failed to load config: %v", err)
	}
	fmt.Printf("Loaded config: %v\n", cfg)
}

func setMemLimit(cfg *config.Config) {
	if cfg.Server.MemLimit > 0 {
		debug.SetMemoryLimit((cfg.Server.MemLimit) * 1024 * 1024)
		logInfo("Set Memory Limit to %d MB", cfg.Server.MemLimit)
	}
}

func bootstrap() error {
	for _, arg := range os.Args[1:] {
		if strings.HasPrefix(arg, "-test.") || arg == "-test.v" {
			return nil
		}
	}

	ReadFlag()
	loadConfig()
	setMemLimit(cfg)

	// 创建根目录 os
	err := os.MkdirAll(cfg.Server.BaseDir, 0755)
	if err != nil {
		return fmt.Errorf("fail to create dir: %w", err)
	}

	if cfg.Log.LogFilePath != "" {
		err = os.MkdirAll(filepath.Dir(cfg.Log.LogFilePath), 0755)
		if err != nil {
			return fmt.Errorf("fail to create log dir: %w", err)
		}
	}

	if cfg.Database.Path != "" {
		err = os.MkdirAll(filepath.Dir(cfg.Database.Path), 0755)
		if err != nil {
			return fmt.Errorf("fail to create db dir: %w", err)
		}
	}

	err = logger.Init(cfg.Log.LogFilePath, cfg.Log.MaxLogSize)
	if err != nil {
		return fmt.Errorf("fail to init logger: %w", err)
	}
	if cfg.Log.Level != "" {
		logger.SetLogLevel(cfg.Log.Level)
	}

	database.SetDBInfo(cfg)
	if err := gitc.RecoverPendingRepos(cfg); err != nil {
		return fmt.Errorf("fail to recover pending repos: %w", err)
	}

	return nil
}

func main() {
	if err := bootstrap(); err != nil {
		log.Fatalf("startup failed: %v", err)
	}

	if database.DB != nil {
		defer database.DB.Close()
	}

	addr := fmt.Sprintf("%s:%d", cfg.Server.Host, cfg.Server.Port)

	// 运行HTTP Git Server
	err := RunHTTP(addr, cfg.Server.BaseDir)
	if err != nil {
		fmt.Printf("Fail to run http: %v\n", err)
		return
	}

}