fusenapi/model/gmodel/fs_user_logic.go

package gmodel

import (
	"context"
	"encoding/json"
	"fmt"

	"fusenapi/utils/auth"
	"log"
	"time"

	"gorm.io/gorm"
)

// TODO: 使用model的属性做你想做的

type UserBasicInfoForSave struct {
	ID                   uint   `gorm:"primary_key" json:"id"`
	FirstName            string `gorm:"-" json:"first_name"`
	LastName             string `gorm:"-" json:"last_name"`
	Mobile               string `gorm:"-" json:"mobile"`
	Company              string `gorm:"-" json:"company"`
	IsOrderStatusEmail   int64  `gorm:"-" json:"is_order_status_email"`
	IsEmailAdvertisement int64  `gorm:"-" json:"is_email_advertisement"`
	IsOrderStatusPhone   int64  `gorm:"-" json:"is_order_status_phone"`
	IsPhoneAdvertisement int64  `gorm:"-" json:"is_phone_advertisement"`
	Type                 int64  `gorm:"-" json:"type"`
	IsOpenRender         int64  `gorm:"-" json:"is_open_render"`
	IsLowRendering       int64  `gorm:"-" json:"is_low_rendering"`
	IsRemoveBg           int64  `gorm:"-" json:"is_remove_bg"`
}

func (u *FsUserModel) FindUserByEmail(ctx context.Context, emailname string) (resp FsUser, err error) {
	err = u.db.WithContext(ctx).Model(&FsUser{}).Where("`email` = ?", emailname).Take(&resp).Error
	return resp, err
}

func (u *FsUserModel) FindUserById(ctx context.Context, Id int64) (resp FsUser, err error) {
	err = u.db.WithContext(ctx).Model(&FsUser{}).Where("`id` = ? and is_del = ?", Id, 0).Take(&resp).Error
	return resp, err
}

func (u *FsUserModel) FindUserByGoogleId(ctx context.Context, Id int64) (resp *FsUser, err error) {
	err = u.db.WithContext(ctx).Model(resp).Where("`google_id` = ? and is_del = ?", Id, 0).Take(resp).Error
	return resp, err
}

func (u *FsUserModel) Transaction(ctx context.Context, fc func(tx *gorm.DB) error) (err error) {
	return u.db.WithContext(ctx).Transaction(fc)
}

// 继承guest_id的资源表
func InheritGuestIdResource(tx *gorm.DB, userId, guestId int64, afterDo func(txResouce *gorm.DB, txUserMaterial *gorm.DB, txUserInfo *gorm.DB) error) error {
	var err error
	if guestId != 0 {
		// 继承guest_id的资源表
		txRes := tx.Model(&FsResource{})
		err = txRes.
			Where("guest_id = ?", guestId).
			UpdateColumn("user_id", userId).Error

		if err != nil {
			return err
		}
		txUserMaterial := tx.Model(&FsUserMaterial{})
		err = txUserMaterial.
			Where("guest_id = ?", guestId).
			UpdateColumn("user_id", userId).Error

		if err != nil {
			return err
		}

		txUserInfo := tx.Model(&FsUserInfo{})
		err = txUserInfo.
			Where("guest_id = ?", guestId).
			UpdateColumn("user_id", userId).Error
		if err != nil {
			return err
		}

		if afterDo != nil {
			return afterDo(txRes, txUserMaterial, txUserInfo)
		}
	}
	return fmt.Errorf("guest_id must not be 0")
}

// 谷歌平台的注册流程
func (u *FsUserModel) RegisterByGoogleOAuth(ctx context.Context, token *auth.RegisterToken) (*FsUser, error) {
	user := &FsUser{}

	err := u.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
		googleId := token.Extend["google_id"].(int64)

		err := tx.Model(user).Where("email = ?", token.Email).Take(user).Error
		if err != nil {
			// 没有找到在数据库就创建注册
			if err == gorm.ErrRecordNotFound {

				createAt := time.Now().UTC().Unix()
				user.Email = &token.Email
				user.CreatedAt = &createAt
				user.GoogleId = &googleId
				user.PasswordHash = &token.Password
				err = tx.Model(user).Create(user).Error
				if err != nil {
					return err
				}

				if token.GuestId != 0 {
					// 继承guest_id的资源表
					return InheritGuestIdResource(tx, user.Id, token.GuestId, nil)
				}
				return err
			}

			return err
		}

		// 如果已经存在,把谷歌id 加入到用户信息里
		user.GoogleId = &googleId
		return tx.Model(user).Update("google_id", user).Error
	})

	if err != nil {
		return nil, err
	}

	return user, nil
}

type UserProfile struct {
	FirstName   string `json:"first_name"`
	LastName    string `json:"last_name"`
	Resetaurant string `json:"resetaurant"`
}

// 自平台的注册流程
func (u *FsUserModel) RegisterByFusen(ctx context.Context, token *auth.RegisterToken) (*FsUser, error) {
	user := &FsUser{}

	err := u.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {

		UserTx := tx.Model(user)
		err := UserTx.Where("email = ?", token.Email).Take(user).Error
		if err != nil {
			// 没有找到在数据库就创建注册
			if err == gorm.ErrRecordNotFound {
				FirstName := token.Extend["first_name"].(string)
				LastName := token.Extend["last_name"].(string)
				Resetaurant := token.Extend["resetaurant"].(string)

				createAt := time.Now().UTC().Unix()
				user.Email = &token.Email
				user.CreatedAt = &createAt
				user.PasswordHash = &token.Password
				user.FirstName = &FirstName
				user.LastName = &LastName

				err = UserTx.Create(user).Error
				log.Println("create")
				if err != nil {
					return err
				}

				if token.GuestId != 0 {

					// 继承guest_id的资源表
					return InheritGuestIdResource(tx, user.Id, token.GuestId, func(txResouce, txUserMaterial, txUserInfo *gorm.DB) error {
						userProfile := &UserProfile{
							FirstName:   FirstName,
							LastName:    LastName,
							Resetaurant: Resetaurant,
						}
						metadata, err := json.Marshal(userProfile)
						if err != nil {
							return err
						}
						now := time.Now()
						uinfo := &FsUserInfo{
							Module:   FsString("profile"),
							UserId:   &user.Id,
							GuestId:  &token.GuestId,
							Metadata: &metadata,
							Ctime:    &now,
							Utime:    &now,
						}
						return txUserInfo.Create(uinfo).Error
					})
				}
				return err
			}
			return err
		}

		if err == nil {
			return fmt.Errorf("the email had registered")
		}
		return err
	})

	if err != nil {
		return nil, err
	}

	return user, nil
}

func (u *FsUserModel) UpdateUserBasicInfoById(ctx context.Context, Id int64, user *UserBasicInfoForSave) (err error) {

	err = u.db.WithContext(ctx).Model(&FsUser{}).
		Where("`id` = ? and is_del = ? and status = ?", Id, 0, 1).
		Updates(map[string]interface{}{
			"first_name":             user.FirstName,
			"last_name":              user.LastName,
			"mobile":                 user.Mobile,
			"company":                user.Company,
			"is_order_status_email":  user.IsOrderStatusEmail,
			"is_email_advertisement": user.IsEmailAdvertisement,
			"is_order_status_phone":  user.IsOrderStatusPhone,
			"is_phone_advertisement": user.IsPhoneAdvertisement,
			"type":                   user.Type,
			"is_open_render":         user.IsOpenRender,
			"is_low_rendering":       user.IsLowRendering,
			"is_remove_bg":           user.IsRemoveBg,
		}).Error
	return err
}