238 lines
7 KiB
Go
238 lines
7 KiB
Go
//go:generate bash -c "mkdir -p codegen && go run github.com/deepmap/oapi-codegen/cmd/oapi-codegen@v1.12.4 -generate types,server,spec -package codegen api/casaos/openapi.yaml > codegen/casaos_api.go"
|
|
//go:generate bash -c "mkdir -p codegen/message_bus && go run github.com/deepmap/oapi-codegen/cmd/oapi-codegen@v1.12.4 -generate types,client -package message_bus https://raw.githubusercontent.com/IceWhaleTech/CasaOS-MessageBus/main/api/message_bus/openapi.yaml > codegen/message_bus/api.go"
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
_ "embed"
|
|
"flag"
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/IceWhaleTech/CasaOS-Common/model"
|
|
"github.com/IceWhaleTech/CasaOS-Common/utils/constants"
|
|
"github.com/IceWhaleTech/CasaOS-Common/utils/logger"
|
|
|
|
util_http "github.com/IceWhaleTech/CasaOS-Common/utils/http"
|
|
|
|
"github.com/IceWhaleTech/CasaOS/codegen/message_bus"
|
|
"github.com/IceWhaleTech/CasaOS/common"
|
|
"github.com/IceWhaleTech/CasaOS/pkg/cache"
|
|
"github.com/IceWhaleTech/CasaOS/pkg/config"
|
|
"github.com/IceWhaleTech/CasaOS/pkg/sqlite"
|
|
"github.com/IceWhaleTech/CasaOS/pkg/utils/command"
|
|
"github.com/IceWhaleTech/CasaOS/pkg/utils/file"
|
|
"github.com/IceWhaleTech/CasaOS/route"
|
|
"github.com/IceWhaleTech/CasaOS/service"
|
|
"github.com/coreos/go-systemd/daemon"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/robfig/cron"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
const LOCALHOST = "127.0.0.1"
|
|
|
|
var sqliteDB *gorm.DB
|
|
|
|
var (
|
|
commit = "private build"
|
|
date = "private build"
|
|
|
|
//go:embed api/index.html
|
|
_docHTML string
|
|
|
|
//go:embed api/casaos/openapi.yaml
|
|
_docYAML string
|
|
|
|
configFlag = flag.String("c", "", "config address")
|
|
dbFlag = flag.String("db", "", "db path")
|
|
versionFlag = flag.Bool("v", false, "version")
|
|
)
|
|
|
|
func init() {
|
|
flag.Parse()
|
|
if *versionFlag {
|
|
fmt.Println("v" + common.VERSION)
|
|
return
|
|
}
|
|
|
|
println("git commit:", commit)
|
|
println("build date:", date)
|
|
|
|
config.InitSetup(*configFlag)
|
|
|
|
logger.LogInit(config.AppInfo.LogPath, config.AppInfo.LogSaveName, config.AppInfo.LogFileExt)
|
|
if len(*dbFlag) == 0 {
|
|
*dbFlag = config.AppInfo.DBPath + "/db"
|
|
}
|
|
|
|
sqliteDB = sqlite.GetDb(*dbFlag)
|
|
// gredis.GetRedisConn(config.RedisInfo),
|
|
|
|
service.MyService = service.NewService(sqliteDB, config.CommonInfo.RuntimePath)
|
|
|
|
service.Cache = cache.Init()
|
|
|
|
service.GetCPUThermalZone()
|
|
|
|
route.InitFunction()
|
|
|
|
///
|
|
// service.MountLists = make(map[string]*mountlib.MountPoint)
|
|
// configfile.Install()
|
|
service.MyService.Storage().CheckAndMountAll()
|
|
}
|
|
|
|
// @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 *versionFlag {
|
|
return
|
|
}
|
|
|
|
v1Router := route.InitV1Router()
|
|
|
|
v2Router := route.InitV2Router()
|
|
v2DocRouter := route.InitV2DocRouter(_docHTML, _docYAML)
|
|
v3file := route.InitFile()
|
|
v4dir := route.InitDir()
|
|
mux := &util_http.HandlerMultiplexer{
|
|
HandlerMap: map[string]http.Handler{
|
|
"v1": v1Router,
|
|
"v2": v2Router,
|
|
"doc": v2DocRouter,
|
|
"v3": v3file,
|
|
"v4": v4dir,
|
|
},
|
|
}
|
|
|
|
cron2 := cron.New()
|
|
// every day execution
|
|
|
|
err := cron2.AddFunc("0/5 * * * * *", func() {
|
|
route.SendAllHardwareStatusBySocket()
|
|
})
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
cron2.Start()
|
|
|
|
defer cron2.Stop()
|
|
|
|
listener, err := net.Listen("tcp", net.JoinHostPort(LOCALHOST, "0"))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
routers := []string{
|
|
"/v1/sys",
|
|
"/v1/port",
|
|
"/v1/file",
|
|
"/v1/folder",
|
|
"/v1/batch",
|
|
"/v1/image",
|
|
"/v1/samba",
|
|
"/v1/notify",
|
|
"/v1/driver",
|
|
"/v1/cloud",
|
|
"/v1/recover",
|
|
"/v1/other",
|
|
route.V2APIPath,
|
|
route.V2DocPath,
|
|
route.V3FilePath,
|
|
}
|
|
for _, apiPath := range routers {
|
|
err = service.MyService.Gateway().CreateRoute(&model.Route{
|
|
Path: apiPath,
|
|
Target: "http://" + listener.Addr().String(),
|
|
})
|
|
|
|
if err != nil {
|
|
fmt.Println("err", err)
|
|
panic(err)
|
|
}
|
|
}
|
|
var events []message_bus.EventType
|
|
events = append(events, message_bus.EventType{Name: "casaos:system:utilization", SourceID: common.SERVICENAME, PropertyTypeList: []message_bus.PropertyType{}})
|
|
events = append(events, message_bus.EventType{Name: "casaos:file:recover", SourceID: common.SERVICENAME, PropertyTypeList: []message_bus.PropertyType{}})
|
|
events = append(events, message_bus.EventType{Name: "casaos:file:operate", SourceID: common.SERVICENAME, PropertyTypeList: []message_bus.PropertyType{}})
|
|
// register at message bus
|
|
for i := 0; i < 10; i++ {
|
|
response, err := service.MyService.MessageBus().RegisterEventTypesWithResponse(context.Background(), events)
|
|
if err != nil {
|
|
logger.Error("error when trying to register one or more event types - some event type will not be discoverable", zap.Error(err))
|
|
}
|
|
if response != nil && response.StatusCode() != http.StatusOK {
|
|
logger.Error("error when trying to register one or more event types - some event type will not be discoverable", zap.String("status", response.Status()), zap.String("body", string(response.Body)))
|
|
}
|
|
if response.StatusCode() == http.StatusOK {
|
|
break
|
|
}
|
|
time.Sleep(time.Second)
|
|
}
|
|
|
|
go func() {
|
|
time.Sleep(time.Second * 2)
|
|
// v0.3.6
|
|
if config.ServerInfo.HttpPort != "" {
|
|
changePort := model.ChangePortRequest{}
|
|
changePort.Port = config.ServerInfo.HttpPort
|
|
err := service.MyService.Gateway().ChangePort(&changePort)
|
|
if err == nil {
|
|
config.Cfg.Section("server").Key("HttpPort").SetValue("")
|
|
config.Cfg.SaveTo(config.SystemConfigInfo.ConfigPath)
|
|
}
|
|
}
|
|
}()
|
|
|
|
urlFilePath := filepath.Join(config.CommonInfo.RuntimePath, "casaos.url")
|
|
if err := file.CreateFileAndWriteContent(urlFilePath, "http://"+listener.Addr().String()); err != nil {
|
|
logger.Error("error when creating address file", zap.Error(err),
|
|
zap.Any("address", listener.Addr().String()),
|
|
zap.Any("filepath", urlFilePath),
|
|
)
|
|
}
|
|
|
|
// run any script that needs to be executed
|
|
scriptDirectory := filepath.Join(constants.DefaultConfigPath, "start.d")
|
|
command.ExecuteScripts(scriptDirectory)
|
|
|
|
if supported, err := daemon.SdNotify(false, daemon.SdNotifyReady); err != nil {
|
|
logger.Error("Failed to notify systemd that casaos main service is ready", zap.Any("error", err))
|
|
} else if supported {
|
|
logger.Info("Notified systemd that casaos main service is ready")
|
|
} else {
|
|
logger.Info("This process is not running as a systemd service.")
|
|
}
|
|
// http.HandleFunc("/v1/file/test", func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
// //http.ServeFile(w, r, r.URL.Path[1:])
|
|
// http.ServeFile(w, r, "/DATA/test.img")
|
|
// })
|
|
// go http.ListenAndServe(":8081", nil)
|
|
|
|
s := &http.Server{
|
|
Handler: mux,
|
|
ReadHeaderTimeout: 5 * time.Second, // fix G112: Potential slowloris attack (see https://github.com/securego/gosec)
|
|
}
|
|
|
|
logger.Info("CasaOS main service is listening...", zap.Any("address", listener.Addr().String()))
|
|
//defer service.MyService.Storage().UnmountAllStorage()
|
|
err = s.Serve(listener) // not using http.serve() to fix G114: Use of net/http serve function that has no support for setting timeouts (see https://github.com/securego/gosec)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|