package http import ( "backend/modules/medias" "backend/modules/middlewares" "backend/modules/users" "backend/providers/app" "backend/providers/http" "backend/providers/jwt" "backend/providers/postgres" "backend/providers/storage" "backend/providers/wechat" "git.ipao.vip/rogeecn/atom" "git.ipao.vip/rogeecn/atom/container" "git.ipao.vip/rogeecn/atom/contracts" log "github.com/sirupsen/logrus" "github.com/spf13/cobra" "go.uber.org/dig" ) func defaultProviders(providers ...container.ProviderContainer) container.Providers { return append(container.Providers{ app.DefaultProvider(), http.DefaultProvider(), postgres.DefaultProvider(), jwt.DefaultProvider(), }, providers...) } func Command() atom.Option { return atom.Command( atom.Name("serve"), atom.Short("run http server"), atom.RunE(Serve), atom.Providers(defaultProviders( storage.DefaultProvider(), wechat.DefaultProvider(), ).With( middlewares.Provide, users.Provide, medias.Provide, )), ) } type Http struct { dig.In App *app.Config Service *http.Service Initials []contracts.Initial `group:"initials"` Routes []contracts.HttpRoute `group:"routes"` Middlewares *middlewares.Middlewares } func Serve(cmd *cobra.Command, args []string) error { return container.Container.Invoke(func(http Http) error { if http.App.Mode == app.AppModeDevelopment { log.SetLevel(log.DebugLevel) } mid := http.Middlewares http.Service.Engine.Use(mid.DebugMode) http.Service.Engine.Use(mid.WeChatVerify) http.Service.Engine.Use(mid.WeChatAuthUserInfo) http.Service.Engine.Use(mid.WeChatSilentAuth) http.Service.Engine.Use(mid.ParseJWT) group := http.Service.Engine.Group("/v1") for _, route := range http.Routes { route.Register(group) } return http.Service.Serve() }) }