100 lines
2.0 KiB
Go
100 lines
2.0 KiB
Go
package cmd
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/sirupsen/logrus"
|
|
"hy2xs-admin/dao"
|
|
"hy2xs-admin/middleware"
|
|
"hy2xs-admin/model/constant"
|
|
"hy2xs-admin/router"
|
|
"hy2xs-admin/service"
|
|
"hy2xs-admin/util"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
func resolveUiBindHost() (string, error) {
|
|
host := os.Getenv("HY2XS_UI_BIND_HOST")
|
|
if host == "" {
|
|
host = "127.0.0.1"
|
|
}
|
|
ip := net.ParseIP(host)
|
|
if ip == nil || ip.To4() == nil {
|
|
return "", errors.New("HY2XS_UI_BIND_HOST must be IPv4")
|
|
}
|
|
return host, nil
|
|
}
|
|
|
|
func runServer(port string) error {
|
|
defer releaseResource()
|
|
|
|
middleware.InitLog()
|
|
if err := initFile(); err != nil {
|
|
return err
|
|
}
|
|
if err := dao.InitSql(port); err != nil {
|
|
return err
|
|
}
|
|
if err := middleware.InitCron(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.InitHysteria2(); err != nil {
|
|
return err
|
|
}
|
|
config, err := dao.GetConfig("key = ?", constant.HUIWebContext)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
r := gin.Default()
|
|
router.Router(r, config.Value)
|
|
|
|
serverPort, crtPath, keyPath, err := service.GetServerPortAndCert()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
bindHost, err := resolveUiBindHost()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
service.InitServer(fmt.Sprintf("%s:%d", bindHost, serverPort), r)
|
|
if err := service.StartServer(crtPath, keyPath); err != nil && err != http.ErrServerClosed {
|
|
logrus.Errorf("start server err: %v", err)
|
|
return errors.New("start server err")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func releaseResource() {
|
|
if err := dao.CloseSqliteDB(); err != nil {
|
|
logrus.Errorf(err.Error())
|
|
}
|
|
if err := service.ReleaseHysteria2(); err != nil {
|
|
logrus.Errorf(err.Error())
|
|
}
|
|
}
|
|
|
|
func initFile() error {
|
|
var dirs = []string{
|
|
constant.LogDir,
|
|
constant.SqliteDBDir,
|
|
constant.BinDir,
|
|
constant.ExportPathDir,
|
|
filepath.Dir(constant.SqliteDBPath),
|
|
}
|
|
for _, item := range dirs {
|
|
if !util.Exists(item) {
|
|
if err := os.MkdirAll(item, 0o750); err != nil {
|
|
logrus.Errorf("%s create err: %v", item, err)
|
|
return errors.New(fmt.Sprintf("%s create err", item))
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|