Update system log

common/init.go

@@ -11,19 +11,16 @@ import (
 var (
 	Port         = flag.Int("port", 3000, "the listening port")
 	PrintVersion = flag.Bool("version", false, "print version and exit")
+	PrintHelp    = flag.Bool("help", false, "print help and exit")
 	LogDir       = flag.String("log-dir", "", "specify the log directory")
-	//Host         = flag.String("host", "localhost", "the server's ip address or domain")
-	//Path         = flag.String("path", "", "specify a local path to public")
-	//VideoPath    = flag.String("video", "", "specify a video folder to public")
-	//NoBrowser    = flag.Bool("no-browser", false, "open browser or not")
 )
 
-// UploadPath Maybe override by ENV_VAR
-var UploadPath = "upload"
-
-//var ExplorerRootPath = UploadPath
-//var ImageUploadPath = "upload/images"
-//var VideoServePath = "upload"
+func printHelp() {
+	fmt.Println("One API " + Version + " - All in one API service for OpenAI API.")
+	fmt.Println("Copyright (C) 2023 JustSong. All rights reserved.")
+	fmt.Println("GitHub: https://github.com/songquanpeng/one-api")
+	fmt.Println("Usage: one-api [--port <port>] [--log-dir <log directory>] [--version] [--help]")
+}
 
 func init() {
 	flag.Parse()
@@ -33,18 +30,17 @@ func init() {
 		os.Exit(0)
 	}
 
+	if *PrintHelp {
+		printHelp()
+		os.Exit(0)
+	}
+
 	if os.Getenv("SESSION_SECRET") != "" {
 		SessionSecret = os.Getenv("SESSION_SECRET")
 	}
 	if os.Getenv("SQLITE_PATH") != "" {
 		SQLitePath = os.Getenv("SQLITE_PATH")
 	}
-	if os.Getenv("UPLOAD_PATH") != "" {
-		UploadPath = os.Getenv("UPLOAD_PATH")
-		//ExplorerRootPath = UploadPath
-		//ImageUploadPath = path.Join(UploadPath, "images")
-		//VideoServePath = UploadPath
-	}
 	if *LogDir != "" {
 		var err error
 		*LogDir, err = filepath.Abs(*LogDir)
@@ -58,21 +54,4 @@ func init() {
 			}
 		}
 	}
-	//if *Path != "" {
-	//	ExplorerRootPath = *Path
-	//}
-	//if *VideoPath != "" {
-	//	VideoServePath = *VideoPath
-	//}
-	//
-	//ExplorerRootPath, _ = filepath.Abs(ExplorerRootPath)
-	//VideoServePath, _ = filepath.Abs(VideoServePath)
-	//ImageUploadPath, _ = filepath.Abs(ImageUploadPath)
-	//
-	if _, err := os.Stat(UploadPath); os.IsNotExist(err) {
-		_ = os.Mkdir(UploadPath, 0777)
-	}
-	//if _, err := os.Stat(ImageUploadPath); os.IsNotExist(err) {
-	//	_ = os.Mkdir(ImageUploadPath, 0777)
-	//}
 }

main.go

@@ -24,7 +24,7 @@ var indexPage []byte
 
 func main() {
 	common.SetupGinLog()
-	common.SysLog("system started")
+	common.SysLog("One API " + common.Version + " started")
 	if os.Getenv("GIN_MODE") != "debug" {
 		gin.SetMode(gin.ReleaseMode)
 	}