|
|
package main
|
|
|
|
|
|
import (
|
|
|
_ "embed"
|
|
|
"flag"
|
|
|
"io"
|
|
|
"io/fs"
|
|
|
"strings"
|
|
|
|
|
|
"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"
|
|
|
|
|
|
"github.com/mholt/archiver/v4"
|
|
|
)
|
|
|
|
|
|
var (
|
|
|
isEject bool
|
|
|
confPath string
|
|
|
scriptName string
|
|
|
)
|
|
|
|
|
|
//go:embed assets.zip
|
|
|
var staticZip string
|
|
|
|
|
|
var staticFS fs.FS
|
|
|
|
|
|
func init() {
|
|
|
flag.StringVar(&confPath, "c", util.RelativePath("conf.ini"), "配置文件路径")
|
|
|
flag.BoolVar(&isEject, "eject", false, "导出内置静态资源")
|
|
|
flag.StringVar(&scriptName, "database-script", "", "运行内置数据库助手脚本")
|
|
|
flag.Parse()
|
|
|
|
|
|
staticFS = archiver.ArchiveFS{
|
|
|
Stream: io.NewSectionReader(strings.NewReader(staticZip), 0, int64(len(staticZip))),
|
|
|
Format: archiver.Zip{},
|
|
|
}
|
|
|
bootstrap.Init(confPath, staticFS)
|
|
|
}
|
|
|
|
|
|
func main() {
|
|
|
if isEject {
|
|
|
// 开始导出内置静态资源文件
|
|
|
bootstrap.Eject(staticFS)
|
|
|
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)
|
|
|
}
|
|
|
}
|
