mirror of
https://github.com/nitezs/sub2clash.git
synced 2024-12-23 22:04:41 -05:00
79 lines
1.5 KiB
Go
79 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
_ "embed"
|
|
"github.com/gin-gonic/gin"
|
|
"go.uber.org/zap"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"sub2clash/api"
|
|
"sub2clash/config"
|
|
"sub2clash/logger"
|
|
"sub2clash/utils"
|
|
"sub2clash/utils/database"
|
|
)
|
|
|
|
//go:embed templates/template_meta.yaml
|
|
var templateMeta string
|
|
|
|
//go:embed templates/template_clash.yaml
|
|
var templateClash string
|
|
|
|
func writeTemplate(path string, template string) error {
|
|
tPath := filepath.Join(
|
|
"templates", path,
|
|
)
|
|
if _, err := os.Stat(tPath); os.IsNotExist(err) {
|
|
file, err := os.Create(tPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer func(file *os.File) {
|
|
_ = file.Close()
|
|
}(file)
|
|
_, err = file.WriteString(template)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func init() {
|
|
if err := utils.MKDir("subs"); err != nil {
|
|
os.Exit(1)
|
|
}
|
|
if err := utils.MKDir("templates"); err != nil {
|
|
os.Exit(1)
|
|
}
|
|
if err := writeTemplate(config.Default.MetaTemplate, templateMeta); err != nil {
|
|
os.Exit(1)
|
|
}
|
|
if err := writeTemplate(config.Default.ClashTemplate, templateClash); err != nil {
|
|
os.Exit(1)
|
|
}
|
|
err := database.ConnectDB()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
// 设置运行模式
|
|
gin.SetMode(gin.ReleaseMode)
|
|
// 关闭 Gin 的日志输出
|
|
gin.DefaultWriter = io.Discard
|
|
// 创建路由
|
|
r := gin.Default()
|
|
// 设置路由
|
|
api.SetRoute(r)
|
|
logger.Logger.Info("Server is running at http://localhost:" + strconv.Itoa(config.Default.Port))
|
|
err := r.Run(":" + strconv.Itoa(config.Default.Port))
|
|
if err != nil {
|
|
logger.Logger.Error("Server run error", zap.Error(err))
|
|
return
|
|
}
|
|
}
|