feat: init repo

templates/project/pkg/service/http/http.go.tpl

@@ -0,0 +1,65 @@
+package http
+
+import (
+	"{{.ModuleName}}/pkg/service"
+	"{{.ModuleName}}/providers/app"
+	"{{.ModuleName}}/providers/hashids"
+	"{{.ModuleName}}/providers/http"
+	"{{.ModuleName}}/providers/jwt"
+	"{{.ModuleName}}/providers/postgres"
+
+	"git.ipao.vip/rogeecn/atom"
+	"git.ipao.vip/rogeecn/atom/container"
+	"git.ipao.vip/rogeecn/atom/contracts"
+	"github.com/gofiber/fiber/v3/middleware/favicon"
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+	"go.uber.org/dig"
+)
+
+func defaultProviders() container.Providers {
+	return service.Default(container.Providers{
+		http.DefaultProvider(),
+		postgres.DefaultProvider(),
+		jwt.DefaultProvider(),
+		hashids.DefaultProvider(),
+	}...)
+}
+
+func Command() atom.Option {
+	return atom.Command(
+		atom.Name("serve"),
+		atom.Short("run http server"),
+		atom.RunE(Serve),
+		atom.Providers(defaultProviders()),
+	)
+}
+
+type Http struct {
+	dig.In
+
+	App      *app.Config
+	Service  *http.Service
+	Initials []contracts.Initial   `group:"initials"`
+	Routes   []contracts.HttpRoute `group:"routes"`
+}
+
+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)
+		}
+
+		http.Service.Engine.Use(favicon.New(favicon.Config{
+			Data: []byte{},
+		}))
+
+		group := http.Service.Engine.Group("/v1")
+
+		for _, route := range http.Routes {
+			route.Register(group)
+		}
+
+		return http.Service.Serve()
+	})
+}

templates/project/pkg/service/migrate/migrate.go.tpl

@@ -0,0 +1,52 @@
+package migrate
+
+import (
+	"context"
+	"database/sql"
+
+	"{{.ModuleName}}/database"
+	"{{.ModuleName}}/providers/postgres"
+
+	"git.ipao.vip/rogeecn/atom"
+	"git.ipao.vip/rogeecn/atom/container"
+	"github.com/pressly/goose/v3"
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+	"go.uber.org/dig"
+)
+
+func Default(providers ...container.ProviderContainer) container.Providers {
+	return append(container.Providers{
+		postgres.DefaultProvider(),
+	}, providers...)
+}
+
+func Command() atom.Option {
+	return atom.Command(
+		atom.Name("migrate"),
+		atom.Short("run migrations"),
+		atom.RunE(Serve),
+		atom.Providers(Default()),
+	)
+}
+
+type Migrate struct {
+	dig.In
+	DB *sql.DB
+}
+
+func Serve(cmd *cobra.Command, args []string) error {
+	return container.Container.Invoke(func(migrate Migrate) error {
+		if len(args) == 0 {
+			args = append(args, "up")
+		}
+
+		action, args := args[0], args[1:]
+		log.Infof("migration action: %s args: %+v", action, args)
+
+		goose.SetBaseFS(database.MigrationFS)
+		goose.SetTableName("migrations")
+
+		return goose.RunContext(context.Background(), action, migrate.DB, "migrations", args...)
+	})
+}

templates/project/pkg/service/model/gen.go.tpl

@@ -0,0 +1,129 @@
+package model
+
+import (
+	"database/sql"
+	"fmt"
+	"strings"
+
+	db "{{.ModuleName}}/providers/postgres"
+
+	"git.ipao.vip/rogeecn/atom"
+	"git.ipao.vip/rogeecn/atom/container"
+	"github.com/go-jet/jet/v2/generator/metadata"
+	"github.com/go-jet/jet/v2/generator/postgres"
+	"github.com/go-jet/jet/v2/generator/template"
+	pg "github.com/go-jet/jet/v2/postgres"
+	"github.com/gofiber/fiber/v3/log"
+	_ "github.com/lib/pq"
+	"github.com/samber/lo"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+	"go.uber.org/dig"
+)
+
+func Default(providers ...container.ProviderContainer) container.Providers {
+	return append(container.Providers{
+		db.DefaultProvider(),
+	}, providers...)
+}
+
+func Options() []atom.Option {
+	return []atom.Option{
+		atom.Name("model"),
+		atom.Short("run model generator"),
+		atom.RunE(Serve),
+		atom.Providers(Default()),
+		atom.Arguments(func(cmd *cobra.Command) {
+			cmd.Flags().String("path", "./database/models", "generate to path")
+			cmd.Flags().String("transform", "./database/.transform.yaml", "transform config")
+		}),
+	}
+}
+
+func Command() atom.Option {
+	return atom.Command(Options()...)
+}
+
+type Migrate struct {
+	dig.In
+	DB     *sql.DB
+	Config *db.Config
+}
+
+type Transformer struct {
+	Ignores []string                     `mapstructure:"ignores"`
+	Types   map[string]map[string]string `mapstructure:"types"`
+}
+
+func Serve(cmd *cobra.Command, args []string) error {
+	v := viper.New()
+	v.SetConfigType("yaml")
+	v.SetConfigFile(cmd.Flag("transform").Value.String())
+
+	if err := v.ReadInConfig(); err != nil {
+		return err
+	}
+
+	var conf Transformer
+	if err := v.Unmarshal(&conf); err != nil {
+		return err
+	}
+
+	return container.Container.Invoke(func(migrate Migrate) error {
+		return postgres.GenerateDSN(
+			migrate.Config.DSN(),
+			migrate.Config.Schema,
+			cmd.Flag("path").Value.String(),
+			template.Default(pg.Dialect).
+				UseSchema(func(schema metadata.Schema) template.Schema {
+					return template.
+						DefaultSchema(schema).
+						UseModel(
+							template.
+								DefaultModel().
+								UseTable(func(table metadata.Table) template.TableModel {
+									if lo.Contains(conf.Ignores, table.Name) {
+										table := template.DefaultTableModel(table)
+										table.Skip = true
+										return table
+									}
+
+									return template.DefaultTableModel(table).UseField(func(column metadata.Column) template.TableModelField {
+										defaultTableModelField := template.DefaultTableModelField(column)
+										defaultTableModelField = defaultTableModelField.UseTags(fmt.Sprintf(`json:"%s"`, column.Name))
+
+										if schema.Name != migrate.Config.Schema {
+											return defaultTableModelField
+										}
+
+										fields, ok := conf.Types[table.Name]
+										if !ok {
+											return defaultTableModelField
+										}
+
+										toType, ok := fields[column.Name]
+										if !ok {
+											return defaultTableModelField
+										}
+
+										splits := strings.Split(toType, ".")
+										typeName := splits[len(splits)-1]
+
+										pkgSplits := strings.Split(splits[0], "/")
+										typePkg := pkgSplits[len(pkgSplits)-1]
+
+										defaultTableModelField = defaultTableModelField.
+											UseType(template.Type{
+												Name:       fmt.Sprintf("%s.%s", typePkg, typeName),
+												ImportPath: splits[0],
+											})
+
+										log.Infof("Convert table %s field %s type to : %s", table.Name, column.Name, toType)
+										return defaultTableModelField
+									})
+								}),
+						)
+				}),
+		)
+	})
+}

templates/project/pkg/service/service.go.tpl

@@ -0,0 +1,13 @@
+package service
+
+import (
+	"{{.ModuleName}}/providers/app"
+
+	"git.ipao.vip/rogeecn/atom/container"
+)
+
+func Default(providers ...container.ProviderContainer) container.Providers {
+	return append(container.Providers{
+		app.DefaultProvider(),
+	}, providers...)
+}

templates/project/pkg/service/testx/testing.go.tpl

@@ -0,0 +1,29 @@
+package testx
+
+import (
+	"os"
+	"testing"
+
+	"git.ipao.vip/rogeecn/atom"
+	"git.ipao.vip/rogeecn/atom/container"
+	"github.com/rogeecn/fabfile"
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+func Default(providers ...container.ProviderContainer) container.Providers {
+	return append(container.Providers{}, providers...)
+}
+
+func Serve(providers container.Providers, t *testing.T, invoke any) {
+	Convey("tests boot up", t, func() {
+		file := fabfile.MustFind("config.toml")
+
+		localEnv := os.Getenv("ENV_LOCAL")
+		if localEnv != "" {
+			file = fabfile.MustFind("config." + localEnv + ".toml")
+		}
+
+		So(atom.LoadProviders(file, providers), ShouldBeNil)
+		So(container.Container.Invoke(invoke), ShouldBeNil)
+	})
+}