127 lines
3 KiB
Go
127 lines
3 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"net/http"
|
|
"runtime"
|
|
"time"
|
|
|
|
"github.com/IceWhaleTech/CasaOS/pkg/cache"
|
|
"github.com/IceWhaleTech/CasaOS/pkg/config"
|
|
"github.com/IceWhaleTech/CasaOS/pkg/sqlite"
|
|
loger2 "github.com/IceWhaleTech/CasaOS/pkg/utils/loger"
|
|
"github.com/IceWhaleTech/CasaOS/route"
|
|
"github.com/IceWhaleTech/CasaOS/service"
|
|
|
|
"github.com/robfig/cron"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var sqliteDB *gorm.DB
|
|
|
|
var configFlag = flag.String("c", "", "config address")
|
|
|
|
var showUserInfo = flag.Bool("show-user-info", false, "show user info")
|
|
|
|
func init() {
|
|
flag.Parse()
|
|
config.InitSetup(*configFlag)
|
|
config.UpdateSetup()
|
|
loger2.LogSetup()
|
|
sysType := runtime.GOOS
|
|
if sysType == "windows" {
|
|
config.AppInfo.ProjectPath = "C:\\CasaOS\\service"
|
|
config.Cfg.Section("app").Key("ProjectPath").SetValue("C:\\CasaOS\\service")
|
|
|
|
config.AppInfo.RootPath = "C:\\CasaOS"
|
|
config.Cfg.Section("app").Key("RootPath").SetValue("C:\\CasaOS")
|
|
config.Cfg.SaveTo(config.SystemConfigInfo.ConfigPath)
|
|
}
|
|
if sysType == "darwin" {
|
|
config.AppInfo.ProjectPath = "./CasaOS/service"
|
|
config.Cfg.Section("app").Key("ProjectPath").SetValue("./CasaOS/service")
|
|
|
|
config.AppInfo.RootPath = "./CasaOS"
|
|
config.Cfg.Section("app").Key("RootPath").SetValue("./CasaOS")
|
|
config.Cfg.SaveTo(config.SystemConfigInfo.ConfigPath)
|
|
}
|
|
|
|
sqliteDB = sqlite.GetDb(config.AppInfo.ProjectPath)
|
|
//gredis.GetRedisConn(config.RedisInfo),
|
|
service.MyService = service.NewService(sqliteDB, loger2.NewOLoger())
|
|
service.Cache = cache.Init()
|
|
|
|
go service.UDPService()
|
|
|
|
fmt.Println("token", service.GetToken())
|
|
service.UDPAddressMap = make(map[string]string)
|
|
//go service.SocketConnect()
|
|
service.CancelList = make(map[string]string)
|
|
route.InitFunction()
|
|
|
|
go service.SendIPToServer()
|
|
go service.LoopFriend()
|
|
|
|
}
|
|
|
|
// @title casaOS API
|
|
// @version 1.0.0
|
|
// @contact.name lauren.pan
|
|
// @contact.url https://www.zimaboard.com
|
|
// @contact.email lauren.pan@icewhale.org
|
|
// @description casaOS v1版本api
|
|
// @host 192.168.2.217:8089
|
|
// @securityDefinitions.apikey ApiKeyAuth
|
|
// @in header
|
|
// @name Authorization
|
|
// @BasePath /v1
|
|
func main() {
|
|
if *showUserInfo {
|
|
fmt.Println("CasaOS User Info")
|
|
fmt.Println("UserName:" + config.UserInfo.UserName)
|
|
fmt.Println("Password:" + config.UserInfo.PWD)
|
|
return
|
|
}
|
|
//model.Setup()
|
|
//gredis.Setup()
|
|
r := route.InitRouter()
|
|
//service.SyncTask(sqliteDB)
|
|
cron2 := cron.New()
|
|
//every day execution
|
|
err := cron2.AddFunc("0 0/5 * * * *", func() {
|
|
//service.PushIpInfo(*&config.ServerInfo.Token)
|
|
//service.UpdataDDNSList(mysqldb)
|
|
//service.SyncTask(sqliteDB)
|
|
|
|
service.SendIPToServer()
|
|
|
|
service.LoopFriend()
|
|
|
|
})
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
// err = cron2.AddFunc("0/1 * * * * *", func() {
|
|
|
|
// //service.SendIPToServer()
|
|
// //service.LoopNet()
|
|
|
|
// })
|
|
// if err != nil {
|
|
// fmt.Println(err)
|
|
// }
|
|
cron2.Start()
|
|
defer cron2.Stop()
|
|
s := &http.Server{
|
|
Addr: fmt.Sprintf(":%v", config.ServerInfo.HttpPort),
|
|
Handler: r,
|
|
ReadTimeout: 60 * time.Second,
|
|
WriteTimeout: 60 * time.Second,
|
|
MaxHeaderBytes: 1 << 20,
|
|
}
|
|
|
|
s.ListenAndServe()
|
|
|
|
}
|