mirror of
https://github.com/linux-do/new-api.git
synced 2025-09-18 00:16:37 +08:00
65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
package common
|
||
|
||
import (
|
||
"flag"
|
||
"fmt"
|
||
"log"
|
||
"os"
|
||
"path/filepath"
|
||
)
|
||
|
||
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", "./logs", "specify the log directory")
|
||
)
|
||
|
||
func printHelp() {
|
||
fmt.Println("New 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()
|
||
|
||
if *PrintVersion {
|
||
fmt.Println(Version)
|
||
os.Exit(0)
|
||
}
|
||
|
||
if *PrintHelp {
|
||
printHelp()
|
||
os.Exit(0)
|
||
}
|
||
|
||
if os.Getenv("SESSION_SECRET") != "" {
|
||
ss := os.Getenv("SESSION_SECRET")
|
||
if ss == "random_string" {
|
||
log.Println("WARNING: SESSION_SECRET is set to the default value 'random_string', please change it to a random string.")
|
||
log.Println("警告：SESSION_SECRET被设置为默认值'random_string'，请修改为随机字符串。")
|
||
log.Fatal("Please set SESSION_SECRET to a random string.")
|
||
} else {
|
||
SessionSecret = ss
|
||
}
|
||
}
|
||
if os.Getenv("SQLITE_PATH") != "" {
|
||
SQLitePath = os.Getenv("SQLITE_PATH")
|
||
}
|
||
if *LogDir != "" {
|
||
var err error
|
||
*LogDir, err = filepath.Abs(*LogDir)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
if _, err := os.Stat(*LogDir); os.IsNotExist(err) {
|
||
err = os.Mkdir(*LogDir, 0777)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
}
|
||
}
|
||
}
|