wip: 周天,继续

internal/cmd/serve.go

@@ -5,6 +5,7 @@ import (
 	"uauth/internal/opt"
 	"uauth/internal/serve"
 	"uauth/internal/store/cache"
+	"uauth/internal/store/db"
 	"uauth/internal/tool"
 )
 
@@ -14,6 +15,7 @@ func initServe() *cobra.Command {
 		RunE: func(cmd *cobra.Command, args []string) error {
 			tool.TablePrinter(opt.Cfg)
 			tool.Must(cache.Init(opt.Cfg.Svc.Cache))
+			tool.Must(db.Init(cmd.Context(), opt.Cfg.Svc.DB))
 			return serve.Run(cmd.Context())
 		},
 	}
@@ -21,6 +23,7 @@ func initServe() *cobra.Command {
 	svc.Flags().StringVar(&opt.Cfg.Svc.Address, "address", "localhost:8080", "listen address")
 	svc.Flags().StringVar(&opt.Cfg.Svc.Prefix, "prefix", "/api/oauth/v2", "api prefix")
 	svc.Flags().StringVar(&opt.Cfg.Svc.Cache, "cache", "lru::", "cache uri")
+	svc.Flags().StringVar(&opt.Cfg.Svc.DB, "db", "sqlite::data.sqlite", "database uri")
 
 	return svc
 }

internal/opt/config.go

@@ -4,6 +4,7 @@ type svc struct {
 	Address string `json:"address"`
 	Prefix  string `json:"prefix"`
 	Cache   string `json:"cache"`
+	DB      string `json:"db"`
 }
 
 type config struct {

internal/serve/handler/registry.go

@@ -0,0 +1,47 @@
+package handler
+
+import (
+	"errors"
+	"github.com/loveuer/nf"
+	"github.com/loveuer/nf/nft/resp"
+	"gorm.io/gorm"
+	"uauth/internal/store/db"
+	"uauth/internal/tool"
+	"uauth/model"
+)
+
+func ClientRegistry(c *nf.Ctx) error {
+	type Req struct {
+		ClientId string `json:"client_id"`
+		Icon     string `json:"icon"` // url
+		Name     string `json:"name"`
+	}
+
+	var (
+		err error
+		req = new(Req)
+	)
+
+	if err = c.BodyParser(req); err != nil {
+		return resp.Resp400(c, err.Error())
+	}
+
+	Secret := tool.RandomString(32)
+
+	platform := &model.Platform{
+		ClientId:     req.ClientId,
+		Icon:         req.Icon,
+		Name:         req.Name,
+		ClientSecret: Secret,
+	}
+
+	if err = db.Default.Session().Create(platform).Error; err != nil {
+		if errors.Is(err, gorm.ErrDuplicatedKey) {
+			return resp.Resp400(c, err, "当前平台已经存在")
+		}
+
+		return resp.Resp500(c, err)
+	}
+
+	return resp.Resp200(c, platform)
+}

internal/serve/serve.go

@@ -132,6 +132,7 @@ func Run(ctx context.Context) error {
 
 	api := app.Group(opt.Cfg.Svc.Prefix)
 	// 设置路由
+	api.Post("/client/registry", handler.ClientRegistry)
 	api.Get("/login", handler.LoginPage)
 	api.Post("/login", handler.LoginAction)
 	api.Get("/authorize", handler.Authorize)

internal/store/db/client.go

@@ -0,0 +1,45 @@
+package db
+
+import (
+	"context"
+	"uauth/internal/opt"
+	"uauth/internal/tool"
+
+	"gorm.io/gorm"
+)
+
+var (
+	Default *Client
+)
+
+type Client struct {
+	ctx   context.Context
+	cli   *gorm.DB
+	ttype string
+}
+
+func (c *Client) Type() string {
+	return c.ttype
+}
+
+func (c *Client) Session(ctxs ...context.Context) *gorm.DB {
+	var ctx context.Context
+	if len(ctxs) > 0 && ctxs[0] != nil {
+		ctx = ctxs[0]
+	} else {
+		ctx = tool.Timeout(30)
+	}
+
+	session := c.cli.Session(&gorm.Session{Context: ctx})
+
+	if opt.Cfg.Debug {
+		session = session.Debug()
+	}
+
+	return session
+}
+
+func (c *Client) Close() {
+	d, _ := c.cli.DB()
+	d.Close()
+}

internal/store/db/db_test.go

@@ -0,0 +1,9 @@
+package db
+
+import (
+	"testing"
+)
+
+func TestOpen(t *testing.T) {
+
+}

internal/store/db/init.go

@@ -0,0 +1,52 @@
+package db
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	"github.com/glebarez/sqlite"
+	"gorm.io/driver/mysql"
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+)
+
+func New(ctx context.Context, uri string) (*Client, error) {
+	strs := strings.Split(uri, "::")
+
+	if len(strs) != 2 {
+		return nil, fmt.Errorf("db.Init: opt db uri invalid: %s", uri)
+	}
+
+	c := &Client{ttype: strs[0]}
+
+	var (
+		err error
+		dsn = strs[1]
+	)
+
+	switch strs[0] {
+	case "sqlite":
+		c.cli, err = gorm.Open(sqlite.Open(dsn))
+	case "mysql":
+		c.cli, err = gorm.Open(mysql.Open(dsn))
+	case "postgres":
+		c.cli, err = gorm.Open(postgres.Open(dsn))
+	default:
+		return nil, fmt.Errorf("db type only support: [sqlite, mysql, postgres], unsupported db type: %s", strs[0])
+	}
+
+	if err != nil {
+		return nil, fmt.Errorf("db.Init: open %s with dsn:%s, err: %w", strs[0], dsn, err)
+	}
+
+	return c, nil
+}
+
+func Init(ctx context.Context, uri string) (err error) {
+	if Default, err = New(ctx, uri); err != nil {
+		return err
+	}
+
+	return nil
+}