CasaOS/main.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()

}