|
@@ -21,7 +21,7 @@ import (
|
|
)
|
|
)
|
|
|
|
|
|
func runWebServer() {
|
|
func runWebServer() {
|
|
- log.Printf("%v %v", config.GetName(), config.GetVersion())
|
|
|
|
|
|
+ log.Printf("Starting %v %v", config.GetName(), config.GetVersion())
|
|
|
|
|
|
switch config.GetLogLevel() {
|
|
switch config.GetLogLevel() {
|
|
case config.Debug:
|
|
case config.Debug:
|
|
@@ -35,31 +35,29 @@ func runWebServer() {
|
|
case config.Error:
|
|
case config.Error:
|
|
logger.InitLogger(logging.ERROR)
|
|
logger.InitLogger(logging.ERROR)
|
|
default:
|
|
default:
|
|
- log.Fatal("unknown log level:", config.GetLogLevel())
|
|
|
|
|
|
+ log.Fatalf("Unknown log level: %v", config.GetLogLevel())
|
|
}
|
|
}
|
|
|
|
|
|
err := database.InitDB(config.GetDBPath())
|
|
err := database.InitDB(config.GetDBPath())
|
|
if err != nil {
|
|
if err != nil {
|
|
- log.Fatal(err)
|
|
|
|
|
|
+ log.Fatalf("Error initializing database: %v", err)
|
|
}
|
|
}
|
|
|
|
|
|
var server *web.Server
|
|
var server *web.Server
|
|
-
|
|
|
|
server = web.NewServer()
|
|
server = web.NewServer()
|
|
global.SetWebServer(server)
|
|
global.SetWebServer(server)
|
|
err = server.Start()
|
|
err = server.Start()
|
|
if err != nil {
|
|
if err != nil {
|
|
- log.Println(err)
|
|
|
|
|
|
+ log.Fatalf("Error starting web server: %v", err)
|
|
return
|
|
return
|
|
}
|
|
}
|
|
|
|
|
|
var subServer *sub.Server
|
|
var subServer *sub.Server
|
|
subServer = sub.NewServer()
|
|
subServer = sub.NewServer()
|
|
global.SetSubServer(subServer)
|
|
global.SetSubServer(subServer)
|
|
-
|
|
|
|
err = subServer.Start()
|
|
err = subServer.Start()
|
|
if err != nil {
|
|
if err != nil {
|
|
- log.Println(err)
|
|
|
|
|
|
+ log.Fatalf("Error starting sub server: %v", err)
|
|
return
|
|
return
|
|
}
|
|
}
|
|
|
|
|
|
@@ -71,34 +69,39 @@ func runWebServer() {
|
|
|
|
|
|
switch sig {
|
|
switch sig {
|
|
case syscall.SIGHUP:
|
|
case syscall.SIGHUP:
|
|
|
|
+ logger.Info("Received SIGHUP signal. Restarting servers...")
|
|
|
|
+
|
|
err := server.Stop()
|
|
err := server.Stop()
|
|
if err != nil {
|
|
if err != nil {
|
|
- logger.Warning("stop server err:", err)
|
|
|
|
|
|
+ logger.Warning("Error stopping web server:", err)
|
|
}
|
|
}
|
|
err = subServer.Stop()
|
|
err = subServer.Stop()
|
|
if err != nil {
|
|
if err != nil {
|
|
- logger.Warning("stop server err:", err)
|
|
|
|
|
|
+ logger.Warning("Error stopping sub server:", err)
|
|
}
|
|
}
|
|
|
|
|
|
server = web.NewServer()
|
|
server = web.NewServer()
|
|
global.SetWebServer(server)
|
|
global.SetWebServer(server)
|
|
err = server.Start()
|
|
err = server.Start()
|
|
if err != nil {
|
|
if err != nil {
|
|
- log.Println(err)
|
|
|
|
|
|
+ log.Fatalf("Error restarting web server: %v", err)
|
|
return
|
|
return
|
|
}
|
|
}
|
|
|
|
+ log.Println("Web server restarted successfully.")
|
|
|
|
|
|
subServer = sub.NewServer()
|
|
subServer = sub.NewServer()
|
|
global.SetSubServer(subServer)
|
|
global.SetSubServer(subServer)
|
|
-
|
|
|
|
err = subServer.Start()
|
|
err = subServer.Start()
|
|
if err != nil {
|
|
if err != nil {
|
|
- log.Println(err)
|
|
|
|
|
|
+ log.Fatalf("Error restarting sub server: %v", err)
|
|
return
|
|
return
|
|
}
|
|
}
|
|
|
|
+ log.Println("Sub server restarted successfully.")
|
|
|
|
+
|
|
default:
|
|
default:
|
|
server.Stop()
|
|
server.Stop()
|
|
subServer.Stop()
|
|
subServer.Stop()
|
|
|
|
+ log.Println("Shutting down servers.")
|
|
return
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|
|
@@ -107,16 +110,16 @@ func runWebServer() {
|
|
func resetSetting() {
|
|
func resetSetting() {
|
|
err := database.InitDB(config.GetDBPath())
|
|
err := database.InitDB(config.GetDBPath())
|
|
if err != nil {
|
|
if err != nil {
|
|
- fmt.Println(err)
|
|
|
|
|
|
+ fmt.Println("Failed to initialize database:", err)
|
|
return
|
|
return
|
|
}
|
|
}
|
|
|
|
|
|
settingService := service.SettingService{}
|
|
settingService := service.SettingService{}
|
|
err = settingService.ResetSettings()
|
|
err = settingService.ResetSettings()
|
|
if err != nil {
|
|
if err != nil {
|
|
- fmt.Println("reset setting failed:", err)
|
|
|
|
|
|
+ fmt.Println("Failed to reset settings:", err)
|
|
} else {
|
|
} else {
|
|
- fmt.Println("reset setting success")
|
|
|
|
|
|
+ fmt.Println("Settings successfully reset.")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
@@ -179,7 +182,7 @@ func updateTgbotEnableSts(status bool) {
|
|
func updateTgbotSetting(tgBotToken string, tgBotChatid string, tgBotRuntime string) {
|
|
func updateTgbotSetting(tgBotToken string, tgBotChatid string, tgBotRuntime string) {
|
|
err := database.InitDB(config.GetDBPath())
|
|
err := database.InitDB(config.GetDBPath())
|
|
if err != nil {
|
|
if err != nil {
|
|
- fmt.Println(err)
|
|
|
|
|
|
+ fmt.Println("Error initializing database:", err)
|
|
return
|
|
return
|
|
}
|
|
}
|
|
|
|
|
|
@@ -188,68 +191,65 @@ func updateTgbotSetting(tgBotToken string, tgBotChatid string, tgBotRuntime stri
|
|
if tgBotToken != "" {
|
|
if tgBotToken != "" {
|
|
err := settingService.SetTgBotToken(tgBotToken)
|
|
err := settingService.SetTgBotToken(tgBotToken)
|
|
if err != nil {
|
|
if err != nil {
|
|
- fmt.Println(err)
|
|
|
|
|
|
+ fmt.Printf("Error setting Telegram bot token: %v\n", err)
|
|
return
|
|
return
|
|
- } else {
|
|
|
|
- logger.Info("updateTgbotSetting tgBotToken success")
|
|
|
|
}
|
|
}
|
|
|
|
+ logger.Info("Successfully updated Telegram bot token.")
|
|
}
|
|
}
|
|
|
|
|
|
if tgBotRuntime != "" {
|
|
if tgBotRuntime != "" {
|
|
err := settingService.SetTgbotRuntime(tgBotRuntime)
|
|
err := settingService.SetTgbotRuntime(tgBotRuntime)
|
|
if err != nil {
|
|
if err != nil {
|
|
- fmt.Println(err)
|
|
|
|
|
|
+ fmt.Printf("Error setting Telegram bot runtime: %v\n", err)
|
|
return
|
|
return
|
|
- } else {
|
|
|
|
- logger.Infof("updateTgbotSetting tgBotRuntime[%s] success", tgBotRuntime)
|
|
|
|
}
|
|
}
|
|
|
|
+ logger.Infof("Successfully updated Telegram bot runtime to [%s].", tgBotRuntime)
|
|
}
|
|
}
|
|
|
|
|
|
if tgBotChatid != "" {
|
|
if tgBotChatid != "" {
|
|
err := settingService.SetTgBotChatId(tgBotChatid)
|
|
err := settingService.SetTgBotChatId(tgBotChatid)
|
|
if err != nil {
|
|
if err != nil {
|
|
- fmt.Println(err)
|
|
|
|
|
|
+ fmt.Printf("Error setting Telegram bot chat ID: %v\n", err)
|
|
return
|
|
return
|
|
- } else {
|
|
|
|
- logger.Info("updateTgbotSetting tgBotChatid success")
|
|
|
|
}
|
|
}
|
|
|
|
+ logger.Info("Successfully updated Telegram bot chat ID.")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
func updateSetting(port int, username string, password string, webBasePath string) {
|
|
func updateSetting(port int, username string, password string, webBasePath string) {
|
|
err := database.InitDB(config.GetDBPath())
|
|
err := database.InitDB(config.GetDBPath())
|
|
if err != nil {
|
|
if err != nil {
|
|
- fmt.Println(err)
|
|
|
|
|
|
+ fmt.Println("Database initialization failed:", err)
|
|
return
|
|
return
|
|
}
|
|
}
|
|
|
|
|
|
settingService := service.SettingService{}
|
|
settingService := service.SettingService{}
|
|
|
|
+ userService := service.UserService{}
|
|
|
|
|
|
if port > 0 {
|
|
if port > 0 {
|
|
err := settingService.SetPort(port)
|
|
err := settingService.SetPort(port)
|
|
if err != nil {
|
|
if err != nil {
|
|
- fmt.Println("set port failed:", err)
|
|
|
|
|
|
+ fmt.Println("Failed to set port:", err)
|
|
} else {
|
|
} else {
|
|
- fmt.Printf("set port %v success", port)
|
|
|
|
|
|
+ fmt.Printf("Port set successfully: %v\n", port)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
if username != "" || password != "" {
|
|
if username != "" || password != "" {
|
|
- userService := service.UserService{}
|
|
|
|
err := userService.UpdateFirstUser(username, password)
|
|
err := userService.UpdateFirstUser(username, password)
|
|
if err != nil {
|
|
if err != nil {
|
|
- fmt.Println("set username and password failed:", err)
|
|
|
|
|
|
+ fmt.Println("Failed to update username and password:", err)
|
|
} else {
|
|
} else {
|
|
- fmt.Println("set username and password success")
|
|
|
|
|
|
+ fmt.Println("Username and password updated successfully")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
if webBasePath != "" {
|
|
if webBasePath != "" {
|
|
err := settingService.SetBasePath(webBasePath)
|
|
err := settingService.SetBasePath(webBasePath)
|
|
if err != nil {
|
|
if err != nil {
|
|
- fmt.Println("set base URI path failed:", err)
|
|
|
|
|
|
+ fmt.Println("Failed to set base URI path:", err)
|
|
} else {
|
|
} else {
|
|
- fmt.Println("set base URI path success")
|
|
|
|
|
|
+ fmt.Println("Base URI path set successfully")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
@@ -348,19 +348,19 @@ func main() {
|
|
var reset bool
|
|
var reset bool
|
|
var show bool
|
|
var show bool
|
|
var remove_secret bool
|
|
var remove_secret bool
|
|
- settingCmd.BoolVar(&reset, "reset", false, "reset all settings")
|
|
|
|
- settingCmd.BoolVar(&show, "show", false, "show current settings")
|
|
|
|
- settingCmd.BoolVar(&remove_secret, "remove_secret", false, "remove secret")
|
|
|
|
- settingCmd.IntVar(&port, "port", 0, "set panel port")
|
|
|
|
- settingCmd.StringVar(&username, "username", "", "set login username")
|
|
|
|
- settingCmd.StringVar(&password, "password", "", "set login password")
|
|
|
|
- settingCmd.StringVar(&webBasePath, "webBasePath", "", "set web base path")
|
|
|
|
- settingCmd.StringVar(&webCertFile, "webCert", "", "set web public key path")
|
|
|
|
- settingCmd.StringVar(&webKeyFile, "webCertKey", "", "set web private key path")
|
|
|
|
- settingCmd.StringVar(&tgbottoken, "tgbottoken", "", "set telegram bot token")
|
|
|
|
- settingCmd.StringVar(&tgbotRuntime, "tgbotRuntime", "", "set telegram bot cron time")
|
|
|
|
- settingCmd.StringVar(&tgbotchatid, "tgbotchatid", "", "set telegram bot chat id")
|
|
|
|
- settingCmd.BoolVar(&enabletgbot, "enabletgbot", false, "enable telegram bot notify")
|
|
|
|
|
|
+ settingCmd.BoolVar(&reset, "reset", false, "Reset all settings")
|
|
|
|
+ settingCmd.BoolVar(&show, "show", false, "Display current settings")
|
|
|
|
+ settingCmd.BoolVar(&remove_secret, "remove_secret", false, "Remove secret key")
|
|
|
|
+ settingCmd.IntVar(&port, "port", 0, "Set panel port number")
|
|
|
|
+ settingCmd.StringVar(&username, "username", "", "Set login username")
|
|
|
|
+ settingCmd.StringVar(&password, "password", "", "Set login password")
|
|
|
|
+ settingCmd.StringVar(&webBasePath, "webBasePath", "", "Set base path for Panel")
|
|
|
|
+ settingCmd.StringVar(&webCertFile, "webCert", "", "Set path to public key file for panel")
|
|
|
|
+ settingCmd.StringVar(&webKeyFile, "webCertKey", "", "Set path to private key file for panel")
|
|
|
|
+ settingCmd.StringVar(&tgbottoken, "tgbottoken", "", "Set token for Telegram bot")
|
|
|
|
+ settingCmd.StringVar(&tgbotRuntime, "tgbotRuntime", "", "Set cron time for Telegram bot notifications")
|
|
|
|
+ settingCmd.StringVar(&tgbotchatid, "tgbotchatid", "", "Set chat ID for Telegram bot notifications")
|
|
|
|
+ settingCmd.BoolVar(&enabletgbot, "enabletgbot", false, "Enable notifications via Telegram bot")
|
|
|
|
|
|
oldUsage := flag.Usage
|
|
oldUsage := flag.Usage
|
|
flag.Usage = func() {
|
|
flag.Usage = func() {
|