OneAuth/api/user/user.go

package user

import (
	"github.com/veypi/OneAuth/cfg"
	"github.com/veypi/OneAuth/libs/auth"
	"github.com/veypi/OneAuth/models"
	"github.com/vyes/vigo"
	"github.com/vyes/vigo/contrib/crud"
)

var _ = Router.Delete("/:user_id", auth.Check("user", "user_id", auth.DoDelete), userDelete)

func userDelete(x *vigo.X) (any, error) {
	data := &models.User{}
	err := cfg.DB().Where("id = ?", x.Params.Get("user_id")).Delete(data).Error
	return data, err
}

var _ = Router.Get("/:user_id", auth.Check("user", "user_id", auth.DoRead), userGet)

type getOpts struct {
	ID string `json:"id"  parse:"path@user_id"`
}

func userGet(x *vigo.X) (any, error) {
	opts := &getOpts{}
	err := x.Parse(opts)
	if err != nil {
		return nil, err
	}
	data := &models.User{}

	err = cfg.DB().Where("id = ?", opts.ID).First(data).Error

	return data, err
}

var _ = Router.Get("/", "list user", listOpts{}, auth.Check("user", "", auth.DoUpdate), crud.List(cfg.DB, &models.User{}))

type listOpts struct {
	Username *string `json:"username"  parse:"query"`
	Nickname *string `json:"nickname"  parse:"query"`
	Email    *string `json:"email"  parse:"query"`
	Phone    *string `json:"phone"  parse:"query"`
	Status   *uint   `json:"status"  parse:"query"`
}

func userList(x *vigo.X) (any, error) {
	opts := &listOpts{}
	err := x.Parse(opts)
	if err != nil {
		return nil, err
	}
	data := make([]*models.User, 0, 10)

	query := cfg.DB()
	if opts.Username != nil {
		query = query.Where("username LIKE ?", opts.Username)
	}
	if opts.Nickname != nil {
		query = query.Where("nickname LIKE ?", opts.Nickname)
	}
	if opts.Email != nil {
		query = query.Where("email LIKE ?", opts.Email)
	}
	if opts.Phone != nil {
		query = query.Where("phone LIKE ?", opts.Phone)
	}
	if opts.Status != nil {
		query = query.Where("status = ?", opts.Status)
	}
	err = query.Find(&data).Error

	return data, err
}

var _ = Router.Patch("/:user_id", auth.Check("user", "user_id", auth.DoUpdate), userPatch)

type patchOpts struct {
	ID       string  `json:"id"  parse:"path@user_id"`
	Nickname *string `json:"nickname"  parse:"json"`
	Icon     *string `json:"icon"  parse:"json"`

	Username *string `json:"username"  parse:"json"`
	Email    *string `json:"email"  parse:"json"`
	Phone    *string `json:"phone"  parse:"json"`
	Status   *uint   `json:"status"  parse:"json"`
}

func userPatch(x *vigo.X) (any, error) {
	opts := &patchOpts{}
	err := x.Parse(opts)
	if err != nil {
		return nil, err
	}
	data := &models.User{}

	err = cfg.DB().Where("id = ?", opts.ID).First(data).Error
	if err != nil {
		return nil, err
	}
	optsMap := make(map[string]any)
	if opts.Username != nil {
		optsMap["username"] = opts.Username
	}
	if opts.Nickname != nil {
		optsMap["nickname"] = opts.Nickname
	}
	if opts.Icon != nil {
		optsMap["icon"] = opts.Icon
	}
	if opts.Email != nil {
		optsMap["email"] = opts.Email
	}
	if opts.Phone != nil {
		optsMap["phone"] = opts.Phone
	}
	if opts.Status != nil {
		optsMap["status"] = opts.Status
	}
	err = cfg.DB().Model(data).Updates(optsMap).Error

	return data, err
}