certimate/main.go
2025-01-05 00:08:12 +08:00

65 lines
1.4 KiB
Go

package main
import (
"flag"
"log"
"os"
"strings"
_ "time/tzdata"
"github.com/labstack/echo/v5"
"github.com/labstack/echo/v5/middleware"
"github.com/pocketbase/pocketbase/core"
"github.com/pocketbase/pocketbase/plugins/migratecmd"
"github.com/usual2970/certimate/internal/app"
"github.com/usual2970/certimate/internal/routes"
"github.com/usual2970/certimate/internal/scheduler"
"github.com/usual2970/certimate/internal/workflow"
"github.com/usual2970/certimate/ui"
_ "github.com/usual2970/certimate/migrations"
)
func main() {
app := app.GetApp()
isGoRun := strings.HasPrefix(os.Args[0], os.TempDir())
// 获取启动命令中的http参数
var httpFlag string
flag.StringVar(&httpFlag, "http", "127.0.0.1:8090", "HTTP server address")
// "serve"影响解析
_ = flag.CommandLine.Parse(os.Args[2:])
migratecmd.MustRegister(app, app.RootCmd, migratecmd.Config{
// enable auto creation of migration files when making collection changes in the Admin UI
// (the isGoRun check is to enable it only during development)
Automigrate: isGoRun,
})
workflow.RegisterEvents()
app.OnBeforeServe().Add(func(e *core.ServeEvent) error {
routes.Register(e.Router)
scheduler.Register()
e.Router.GET(
"/*",
echo.StaticDirectoryHandler(ui.DistDirFS, false),
middleware.Gzip(),
)
return nil
})
defer log.Println("Exit!")
log.Printf("Visit the website: http://%s", httpFlag)
if err := app.Start(); err != nil {
log.Fatal(err)
}
}