69 lines
1.7 KiB
Go
69 lines
1.7 KiB
Go
package main
|
||
|
||
import (
|
||
"embed"
|
||
"flag"
|
||
|
||
"github.com/cloudreve/Cloudreve/v3/bootstrap"
|
||
"github.com/cloudreve/Cloudreve/v3/pkg/conf"
|
||
"github.com/cloudreve/Cloudreve/v3/pkg/util"
|
||
"github.com/cloudreve/Cloudreve/v3/routers"
|
||
)
|
||
|
||
var (
|
||
isEject bool
|
||
confPath string
|
||
scriptName string
|
||
)
|
||
|
||
//go:embed assets/build
|
||
var staticEmbed embed.FS
|
||
|
||
func init() {
|
||
flag.StringVar(&confPath, "c", util.RelativePath("conf.ini"), "配置文件路径")
|
||
flag.BoolVar(&isEject, "eject", false, "导出内置静态资源")
|
||
flag.StringVar(&scriptName, "database-script", "", "运行内置数据库助手脚本")
|
||
flag.Parse()
|
||
bootstrap.Init(confPath, staticEmbed)
|
||
}
|
||
|
||
func main() {
|
||
if isEject {
|
||
// 开始导出内置静态资源文件
|
||
bootstrap.Eject(staticEmbed)
|
||
return
|
||
}
|
||
|
||
if scriptName != "" {
|
||
// 开始运行助手数据库脚本
|
||
bootstrap.RunScript(scriptName)
|
||
return
|
||
}
|
||
|
||
api := routers.InitRouter()
|
||
|
||
// 如果启用了SSL
|
||
if conf.SSLConfig.CertPath != "" {
|
||
go func() {
|
||
util.Log().Info("开始监听 %s", conf.SSLConfig.Listen)
|
||
if err := api.RunTLS(conf.SSLConfig.Listen,
|
||
conf.SSLConfig.CertPath, conf.SSLConfig.KeyPath); err != nil {
|
||
util.Log().Error("无法监听[%s]，%s", conf.SSLConfig.Listen, err)
|
||
}
|
||
}()
|
||
}
|
||
|
||
// 如果启用了Unix
|
||
if conf.UnixConfig.Listen != "" {
|
||
util.Log().Info("开始监听 %s", conf.UnixConfig.Listen)
|
||
if err := api.RunUnix(conf.UnixConfig.Listen); err != nil {
|
||
util.Log().Error("无法监听[%s]，%s", conf.UnixConfig.Listen, err)
|
||
}
|
||
return
|
||
}
|
||
|
||
util.Log().Info("开始监听 %s", conf.SystemConfig.Listen)
|
||
if err := api.Run(conf.SystemConfig.Listen); err != nil {
|
||
util.Log().Error("无法监听[%s]，%s", conf.SystemConfig.Listen, err)
|
||
}
|
||
}
|