diff --git a/common/init.go b/common/init.go index 66fcd7e1..0f22c69b 100644 --- a/common/init.go +++ b/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 ] [--log-dir ] [--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) - //} } diff --git a/main.go b/main.go index 3fd87c1c..21db969b 100644 --- a/main.go +++ b/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) }