chore: 简化入口函数

cmd/app/builder.go

@@ -0,0 +1,40 @@
+package main
+
+import (
+	"github.com/TBXark/sphere/internal/biz/task"
+	"github.com/TBXark/sphere/internal/server/api"
+	"github.com/TBXark/sphere/internal/server/bot"
+	"github.com/TBXark/sphere/internal/server/dash"
+	"github.com/TBXark/sphere/internal/server/docs"
+	"github.com/TBXark/sphere/pkg/utils/boot"
+)
+
+func newApplication(dash *dash.Web, api *api.Web, docs *docs.Web, initialize *task.DashInitialize, cleaner *task.ConnectCleaner) *boot.Application {
+	return boot.NewApplication(
+		dash,
+		api,
+		docs,
+		initialize,
+		cleaner,
+	)
+}
+
+func newAPIApplication(api *api.Web, initialize *task.DashInitialize, cleaner *task.ConnectCleaner) *boot.Application {
+	return boot.NewApplication(
+		api,
+		initialize,
+		cleaner,
+	)
+}
+
+func newDashApplication(dash *dash.Web, initialize *task.DashInitialize, cleaner *task.ConnectCleaner) *boot.Application {
+	return boot.NewApplication(
+		dash,
+		initialize,
+		cleaner,
+	)
+}
+
+func newBotApplication(app *bot.Bot) *boot.Application {
+	return boot.NewApplication(app)
+}

cmd/app/main.go

@@ -1,24 +1,24 @@
 package main
 
 import (
-	"github.com/TBXark/sphere/internal/biz/task"
+	"flag"
+	"github.com/TBXark/sphere/internal/config"
 	"github.com/TBXark/sphere/internal/pkg/app"
-	"github.com/TBXark/sphere/internal/server/api"
-	"github.com/TBXark/sphere/internal/server/dash"
-	"github.com/TBXark/sphere/internal/server/docs"
 	"github.com/TBXark/sphere/pkg/utils/boot"
 )
 
 func main() {
-	app.Execute(NewApplication)
-}
-
-func newApplication(dash *dash.Web, api *api.Web, docs *docs.Web, initialize *task.DashInitialize, cleaner *task.ConnectCleaner) *boot.Application {
-	return boot.NewApplication(
-		dash,
-		api,
-		docs,
-		initialize,
-		cleaner,
-	)
+	mode := flag.String("mode", "app", "run mode: app, api, dash, bot")
+	app.Execute(func(config *config.Config) (*boot.Application, error) {
+		switch *mode {
+		case "api":
+			return NewAPIApplication(config)
+		case "dash":
+			return NewDashApplication(config)
+		case "bot":
+			return NewBotApplication(config)
+		default:
+			return NewApplication(config)
+		}
+	})
 }

cmd/app/wire.go

@@ -14,3 +14,18 @@ func NewApplication(conf *config.Config) (*boot.Application, error) {
 	wire.Build(internal.ProviderSet, wire.NewSet(newApplication))
 	return &boot.Application{}, nil
 }
+
+func NewAPIApplication(conf *config.Config) (*boot.Application, error) {
+	wire.Build(internal.ProviderSet, wire.NewSet(newAPIApplication))
+	return &boot.Application{}, nil
+}
+
+func NewDashApplication(conf *config.Config) (*boot.Application, error) {
+	wire.Build(internal.ProviderSet, wire.NewSet(newDashApplication))
+	return &boot.Application{}, nil
+}
+
+func NewBotApplication(conf *config.Config) (*boot.Application, error) {
+	wire.Build(internal.ProviderSet, wire.NewSet(newBotApplication))
+	return &boot.Application{}, nil
+}