fix: user wechat auth

backend/app/events/publishers/user_register.go

@@ -0,0 +1,23 @@
+package publishers
+
+import (
+	"encoding/json"
+
+	"backend/app/events"
+
+	"git.ipao.vip/rogeecn/atom/contracts"
+)
+
+var _ contracts.EventPublisher = (*UserRegister)(nil)
+
+type UserRegister struct {
+	ID int64 `json:"id"`
+}
+
+func (e *UserRegister) Marshal() ([]byte, error) {
+	return json.Marshal(e)
+}
+
+func (e *UserRegister) Topic() string {
+	return events.TopicUserRegister.String()
+}

backend/app/events/subscribers/provider.gen.go

@@ -1,6 +1,7 @@
-package events
+package subscribers
 
 import (
+	"backend/app/http/users"
 	"backend/providers/event"
 
 	"git.ipao.vip/rogeecn/atom"
@@ -12,8 +13,11 @@ import (
 func Provide(opts ...opt.Option) error {
 	if err := container.Container.Provide(func(
 		__event *event.PubSub,
+		userSvc *users.Service,
 	) (contracts.Initial, error) {
-		obj := &UserRegister{}
+		obj := &UserRegister{
+			userSvc: userSvc,
+		}
 		if err := obj.Prepare(); err != nil {
 			return nil, err
 		}

backend/app/events/subscribers/user_register.go

@@ -0,0 +1,66 @@
+package subscribers
+
+import (
+	"context"
+	"encoding/json"
+
+	"backend/app/events"
+	"backend/app/events/publishers"
+	"backend/app/http/users"
+	"backend/database/fields"
+
+	"git.ipao.vip/rogeecn/atom/contracts"
+	"github.com/ThreeDotsLabs/watermill/message"
+	"github.com/pkg/errors"
+	"github.com/sirupsen/logrus"
+)
+
+var _ contracts.EventHandler = (*UserRegister)(nil)
+
+// @provider(event)
+type UserRegister struct {
+	log *logrus.Entry `inject:"false"`
+
+	userSvc *users.Service
+}
+
+func (e *UserRegister) Prepare() error {
+	e.log = logrus.WithField("module", "events.subscribers.user_register")
+	return nil
+}
+
+// PublishToTopic implements contracts.EventHandler.
+func (e *UserRegister) PublishToTopic() string {
+	return events.TopicProcessed.String()
+}
+
+// Topic implements contracts.EventHandler.
+func (e *UserRegister) Topic() string {
+	return events.TopicUserRegister.String()
+}
+
+// Handler implements contracts.EventHandler.
+func (e *UserRegister) Handler(msg *message.Message) ([]*message.Message, error) {
+	var payload publishers.UserRegister
+	err := json.Unmarshal(msg.Payload, &payload)
+	if err != nil {
+		return nil, err
+	}
+	e.log.Infof("received event %s", msg.Payload)
+
+	user, err := e.userSvc.GetUserByID(context.Background(), payload.ID)
+	if err != nil {
+		return nil, errors.Wrapf(err, "failed to get user by id: %d", payload.ID)
+	}
+
+	if user.Status != fields.UserStatusPending {
+		return nil, nil
+	}
+
+	err = e.userSvc.SetUserStatusByID(context.Background(), payload.ID, fields.UserStatusVerified)
+	if err != nil {
+		return nil, errors.Wrapf(err, "failed to set user status to %s by id: %d", fields.UserStatusVerified, payload.ID)
+	}
+
+	return nil, nil
+}

backend/app/events/user_register.go

@@ -4,19 +4,12 @@ import (
 	"encoding/json"
 
 	"git.ipao.vip/rogeecn/atom/contracts"
-	"github.com/ThreeDotsLabs/watermill/message"
-	"github.com/sirupsen/logrus"
 )
 
-var (
-	_ contracts.EventHandler   = (*UserRegister)(nil)
-	_ contracts.EventPublisher = (*UserRegister)(nil)
-)
+var _ contracts.EventPublisher = (*UserRegister)(nil)
 
-// @provider(event)
 type UserRegister struct {
-	log *logrus.Entry `inject:"false" json:"-"`
-	ID  int64         `json:"id"`
+	ID int64 `json:"id"`
 }
 
 func (e *UserRegister) Prepare() error {
@@ -28,25 +21,7 @@ func (e *UserRegister) Marshal() ([]byte, error) {
 	return json.Marshal(e)
 }
 
-// PublishToTopic implements contracts.EventHandler.
-func (e *UserRegister) PublishToTopic() string {
-	return TopicProcessed.String()
-}
-
 // Topic implements contracts.EventHandler.
 func (e *UserRegister) Topic() string {
 	return TopicUserRegister.String()
 }
-
-// Handler implements contracts.EventHandler.
-func (e *UserRegister) Handler(msg *message.Message) ([]*message.Message, error) {
-	var payload UserRegister
-	err := json.Unmarshal(msg.Payload, &payload)
-	if err != nil {
-		return nil, err
-	}
-
-	e.log.Infof("received event %+v\n", payload)
-
-	return nil, nil
-}

backend/app/http/auth/controller_wechat.go

@@ -3,11 +3,11 @@ package auth
 import (
 	"fmt"
 	"net/url"
-	"strings"
 	"time"
 
 	"backend/app/consts"
 	"backend/app/http/users"
+	"backend/database/fields"
 	"backend/providers/jwt"
 	"backend/providers/otel"
 	"backend/providers/wechat"
@@ -33,8 +33,7 @@ func (ctl *Controller) Prepare() error {
 	return nil
 }
 
-// @Router			/v1/auth/wechat/jump/:tenant [get]
-// @Bind tenant path
+// @Router			/v1/auth/wechat/jump [get]
 // @Bind redirectUri query
 func (ctl *Controller) JumpToAuth(ctx fiber.Ctx, tenant, redirectUri string) error {
 	_, span := otel.Start(ctx.Context(), "auth.controller.wechat")
@@ -68,21 +67,13 @@ func (ctl *Controller) JumpToAuth(ctx fiber.Ctx, tenant, redirectUri string) err
 	return ctx.Redirect().To(to.String())
 }
 
-// @Router			/v1/auth/login/:tenant [get]
-// @Bind tenant path
+// @Router			/v1/auth/login [get]
 // @Bind code query
 // @Bind state query
 // @Bind redirectUri query
 func (ctl *Controller) Login(ctx fiber.Ctx, code, state, tenant, redirectUri string) error {
 	ctl.log.Debugf("code: %s, state: %s", code, state)
 
-	ctx.Cookie(&fiber.Cookie{
-		Name:     consts.TokenTypeUser.String(),
-		Value:    "",
-		Expires:  time.Now().Add(12 * time.Hour),
-		HTTPOnly: true,
-	})
-
 	// get the openid
 	token, err := ctl.wechat.AuthorizeCode2Token(code)
 	if err != nil {
@@ -90,31 +81,22 @@ func (ctl *Controller) Login(ctx fiber.Ctx, code, state, tenant, redirectUri str
 	}
 	ctl.log.Debugf("tokenInfo %+v", token)
 
-	user, err := ctl.userSvc.GetOrNewFromChannel(ctx.Context(), consts.AuthChannelWeChat, token.OpenID, tenant)
+	userID, err := ctl.userSvc.GetUserIDByOpenID(ctx.Context(), fields.AuthChannelWeChat, token.GetOpenID())
 	if err != nil {
 		return errors.Wrap(err, "failed to get user")
 	}
 
-	claim := c.jwt.CreateClaims(jwt.BaseClaims{
-		OpenID:   user.OpenID,
-		Tenant:   tenantSlug,
-		UserID:   user.ID,
-		TenantID: tenant.ID,
-	})
-	jwtToken, err = c.jwt.CreateToken(claim)
+	jwtToken, err := ctl.jwt.CreateToken(ctl.jwt.CreateClaims(jwt.BaseClaims{UserID: userID}))
 	if err != nil {
 		return errors.Wrap(err, "failed to create token")
 	}
 
 	ctx.Cookie(&fiber.Cookie{
-		Name:     "token",
+		Name:     consts.TokenTypeUser.String(),
 		Value:    jwtToken,
 		Expires:  time.Now().Add(6 * time.Hour),
 		HTTPOnly: true,
 	})
 
-	html := strings.ReplaceAll(string(b), "{{JWT}}", jwtToken)
-	return ctx.SendString(html)
-
-	return ctx.Redirect().To(paramRedirect)
+	return ctx.Redirect().To(redirectUri)
 }

backend/app/http/auth/provider.gen.go

@@ -7,7 +7,9 @@ import (
 	"backend/providers/jwt"
 	"backend/providers/wechat"
 
+	"git.ipao.vip/rogeecn/atom"
 	"git.ipao.vip/rogeecn/atom/container"
+	"git.ipao.vip/rogeecn/atom/contracts"
 	"git.ipao.vip/rogeecn/atom/utils/opt"
 )
 
@@ -32,6 +34,20 @@ func Provide(opts ...opt.Option) error {
 	}); err != nil {
 		return err
 	}
+	if err := container.Container.Provide(func(
+		controller *Controller,
+	) (contracts.HttpRoute, error) {
+		obj := &Routes{
+			controller: controller,
+		}
+		if err := obj.Prepare(); err != nil {
+			return nil, err
+		}
+
+		return obj, nil
+	}, atom.GroupRoutes); err != nil {
+		return err
+	}
 	if err := container.Container.Provide(func(
 		db *sql.DB,
 	) (*Service, error) {

backend/app/http/auth/routes.gen.go

@@ -0,0 +1,45 @@
+// Code generated by the atomctl ; DO NOT EDIT.
+
+package auth
+
+import (
+	. "backend/pkg/f"
+
+	_ "git.ipao.vip/rogeecn/atom"
+	_ "git.ipao.vip/rogeecn/atom/contracts"
+	"github.com/gofiber/fiber/v3"
+	log "github.com/sirupsen/logrus"
+)
+
+// @provider contracts.HttpRoute atom.GroupRoutes
+type Routes struct {
+	log        *log.Entry `inject:"false"`
+	controller *Controller
+}
+
+func (r *Routes) Prepare() error {
+	r.log = log.WithField("module", "routes.auth")
+	return nil
+}
+
+func (r *Routes) Name() string {
+	return "auth"
+}
+
+func (r *Routes) Register(router fiber.Router) {
+	// 注册路由组: Controller
+	router.Get("/v1/auth/wechat/jump/:tenant", Func2(
+		r.controller.JumpToAuth,
+		Path[string]("tenant"),
+		QueryParam[string]("redirectUri"),
+	))
+
+	router.Get("/v1/auth/login/:tenant", Func4(
+		r.controller.Login,
+		Path[string]("tenant"),
+		QueryParam[string]("code"),
+		QueryParam[string]("state"),
+		QueryParam[string]("redirectUri"),
+	))
+
+}

backend/app/http/users/provider.gen.go

@@ -3,6 +3,8 @@ package users
 import (
 	"database/sql"
 
+	"backend/providers/event"
+
 	"git.ipao.vip/rogeecn/atom/container"
 	"git.ipao.vip/rogeecn/atom/utils/opt"
 )
@@ -24,9 +26,11 @@ func Provide(opts ...opt.Option) error {
 	}
 	if err := container.Container.Provide(func(
 		db *sql.DB,
+		event *event.PubSub,
 	) (*Service, error) {
 		obj := &Service{
-			db: db,
+			db:    db,
+			event: event,
 		}
 		if err := obj.Prepare(); err != nil {
 			return nil, err

backend/app/http/users/service.go

@@ -5,7 +5,7 @@ import (
 	"database/sql"
 	"time"
 
-	"backend/app/events"
+	"backend/app/events/publishers"
 	"backend/database/fields"
 	"backend/database/models/qvyun_v2/public/model"
 	"backend/database/models/qvyun_v2/public/table"
@@ -14,6 +14,7 @@ import (
 	"backend/providers/otel"
 
 	. "github.com/go-jet/jet/v2/postgres"
+	"github.com/pkg/errors"
 	"github.com/samber/lo"
 	log "github.com/sirupsen/logrus"
 	"go.opentelemetry.io/otel/attribute"
@@ -35,7 +36,7 @@ func (svc *Service) Prepare() error {
 }
 
 // GetUsersByOpenID Get user by open id
-func (svc *Service) GetUserByOpenID(ctx context.Context, channel fields.AuthChannel, openID string) (*model.Users, error) {
+func (svc *Service) GetUserByOpenIDOfChannel(ctx context.Context, channel fields.AuthChannel, openID string) (*model.Users, error) {
 	_, span := otel.Start(ctx, "users.service.GetUsersByOpenID")
 	defer span.End()
 
@@ -106,7 +107,7 @@ func (svc *Service) CreateUser(ctx context.Context, user *model.Users) (*model.U
 		user.UpdatedAt = time.Now()
 	}
 
-	user.Status = int16(fields.UserStatusPending)
+	user.Status = fields.UserStatusPending
 
 	// use bcrypt to hash password
 	pwd, err := bcrypt.GenerateFromPassword([]byte(user.Password), bcrypt.DefaultCost)
@@ -126,9 +127,11 @@ func (svc *Service) CreateUser(ctx context.Context, user *model.Users) (*model.U
 
 	// if user created successfully, trigger event
 	span.AddEvent("user created")
-	svc.event.Publish(&events.UserRegister{ID: m.ID})
+	if err := svc.event.Publish(&publishers.UserRegister{ID: m.ID}); err != nil {
+		return nil, errors.Wrapf(err, "failed to publish user register event %d", m.ID)
+	}
 
-	return &m, err
+	return &m, nil
 }
 
 // GetUserByID
@@ -199,3 +202,29 @@ func (svc *Service) AttachUserOAuth(ctx context.Context, user *model.Users, chan
 
 	return nil
 }
+
+// SetUserStatusByID
+func (svc *Service) SetUserStatusByID(ctx context.Context, userID int64, status fields.UserStatus) error {
+	_, span := otel.Start(ctx, "users.service.SetUserStatusByID")
+	defer span.End()
+	span.SetAttributes(
+		attribute.Int64("user.id", userID),
+		attribute.String("user.status", status.String()),
+	)
+
+	tbl := table.Users
+	stmt := tbl.
+		UPDATE().
+		SET(
+			tbl.Status.SET(Int16(int16(status))),
+		).
+		WHERE(
+			tbl.ID.EQ(Int64(userID)),
+		)
+	span.SetAttributes(semconv.DBStatementKey.String(stmt.DebugSql()))
+
+	if _, err := stmt.ExecContext(ctx, svc.db); err != nil {
+		return err
+	}
+	return nil
+}

backend/app/service/event/event.go

@@ -3,7 +3,7 @@ package event
 import (
 	"context"
 
-	"backend/app/events"
+	"backend/app/events/subscribers"
 	"backend/app/service"
 	"backend/providers/app"
 	"backend/providers/event"
@@ -31,7 +31,7 @@ func Command() atom.Option {
 		atom.Providers(
 			defaultProviders().
 				With(
-					events.Provide,
+					subscribers.Provide,
 				),
 		),
 	)