mirror of https://github.com/veypi/OneAuth.git
feat: simplify user login
veypi
6 months ago
parent
545f49c0f3
commit
7e7e6ed506
@ -0,0 +1,49 @@
|
||||
package token
|
||||
|
||||
import (
|
||||
"oa/cfg"
|
||||
"oa/models"
|
||||
"time"
|
||||
|
||||
"github.com/veypi/OneBD/rest"
|
||||
)
|
||||
|
||||
var _ = Router.Patch("/:token_id", tokenPatch)
|
||||
|
||||
type patchOpts struct {
|
||||
ID string `json:"id" parse:"path@token_id"`
|
||||
ExpiredAt *time.Time `json:"expired_at" parse:"json"`
|
||||
OverPerm *string `json:"over_perm" parse:"json"`
|
||||
}
|
||||
|
||||
func tokenPatch(x *rest.X) (any, error) {
|
||||
opts := &patchOpts{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := &models.Token{}
|
||||
|
||||
err = cfg.DB().Where("id = ?", opts.ID).First(data).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
optsMap := make(map[string]interface{})
|
||||
if opts.ExpiredAt != nil {
|
||||
optsMap["expired_at"] = opts.ExpiredAt
|
||||
}
|
||||
if opts.OverPerm != nil {
|
||||
optsMap["over_perm"] = opts.OverPerm
|
||||
}
|
||||
err = cfg.DB().Model(data).Updates(optsMap).Error
|
||||
|
||||
return data, err
|
||||
}
|
||||
|
||||
var _ = Router.Delete("/:token_id", tokenDelete)
|
||||
|
||||
func tokenDelete(x *rest.X) (any, error) {
|
||||
data := &models.Token{}
|
||||
err := cfg.DB().Where("id = ?", x.Params.Get("token_id")).Delete(data).Error
|
||||
return data, err
|
||||
}
|
||||
@ -0,0 +1,55 @@
|
||||
//
|
||||
// get.go
|
||||
// Copyright (C) 2025 veypi <i@veypi.com>
|
||||
// 2025-05-09 17:24
|
||||
// Distributed under terms of the MIT license.
|
||||
//
|
||||
|
||||
package token
|
||||
|
||||
import (
|
||||
"oa/cfg"
|
||||
"oa/models"
|
||||
|
||||
"github.com/veypi/OneBD/rest"
|
||||
)
|
||||
|
||||
type getOpts struct {
|
||||
ID string `json:"id" parse:"path@token_id"`
|
||||
}
|
||||
|
||||
var _ = Router.Get("/:token_id", tokenGet)
|
||||
|
||||
func tokenGet(x *rest.X) (any, error) {
|
||||
opts := &getOpts{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := &models.Token{}
|
||||
err = cfg.DB().Where("id = ?", opts.ID).First(data).Error
|
||||
return data, err
|
||||
}
|
||||
|
||||
type listOpts struct {
|
||||
Limit int `json:"limit"`
|
||||
UserID string `json:"user_id" gorm:"index;type:varchar(32)" parse:"query"`
|
||||
AppID string `json:"app_id" gorm:"index;type:varchar(32)" parse:"query"`
|
||||
}
|
||||
|
||||
var _ = Router.Get("/", tokenList)
|
||||
|
||||
func tokenList(x *rest.X) (any, error) {
|
||||
opts := &listOpts{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := make([]*models.Token, 0, 10)
|
||||
|
||||
query := cfg.DB()
|
||||
query = query.Where("user_id = ?", opts.UserID)
|
||||
query = query.Where("app_id = ?", opts.AppID)
|
||||
err = query.Limit(opts.Limit).Find(&data).Error
|
||||
return data, err
|
||||
}
|
||||
@ -1,258 +0,0 @@
|
||||
package token
|
||||
|
||||
import (
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"oa/cfg"
|
||||
"oa/errs"
|
||||
"oa/libs/auth"
|
||||
M "oa/models"
|
||||
"time"
|
||||
|
||||
"github.com/golang-jwt/jwt/v5"
|
||||
"github.com/veypi/OneBD/rest"
|
||||
"github.com/veypi/utils"
|
||||
"github.com/veypi/utils/logv"
|
||||
)
|
||||
|
||||
func useToken(r rest.Router) {
|
||||
r.Post("/salt", tokenSalt)
|
||||
r.Post("/", tokenPost)
|
||||
r.Get("/:token_id", tokenGet)
|
||||
r.Patch("/:token_id", tokenPatch)
|
||||
r.Delete("/:token_id", tokenDelete)
|
||||
r.Get("/", tokenList)
|
||||
}
|
||||
func tokenSalt(x *rest.X) (any, error) {
|
||||
opts := &M.TokenSalt{}
|
||||
err := x.Parse(opts)
|
||||
logv.Warn().Msg(opts.Username)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := &M.User{}
|
||||
query := "username = ?"
|
||||
if opts.Typ == nil {
|
||||
} else if *opts.Typ == "email" {
|
||||
query = "email = ?"
|
||||
} else if *opts.Typ == "phone" {
|
||||
query = "phone = ?"
|
||||
}
|
||||
|
||||
err = cfg.DB().Where(query, opts.Username).First(data).Error
|
||||
return map[string]string{"salt": data.Salt, "id": data.ID}, err
|
||||
}
|
||||
|
||||
// for user login app
|
||||
func tokenPost(x *rest.X) (any, error) {
|
||||
opts := &M.TokenPost{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
aid := cfg.Config.ID
|
||||
if opts.AppID != nil && *opts.AppID != "" {
|
||||
aid = *opts.AppID
|
||||
}
|
||||
data := &M.Token{}
|
||||
claim := &auth.Claims{}
|
||||
claim.IssuedAt = jwt.NewNumericDate(time.Now())
|
||||
claim.Issuer = cfg.Config.ID
|
||||
if opts.Refresh != nil {
|
||||
typ := "app"
|
||||
if opts.Typ != nil {
|
||||
typ = *opts.Typ
|
||||
}
|
||||
// for other app redirect
|
||||
refresh, err := auth.ParseJwt(*opts.Refresh)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if refresh.ID == "" {
|
||||
return nil, errs.AuthInvalid
|
||||
}
|
||||
err = cfg.DB().Where("id = ?", refresh.ID).First(data).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
claim.AID = aid
|
||||
claim.UID = refresh.UID
|
||||
claim.Name = refresh.Name
|
||||
claim.Icon = refresh.Icon
|
||||
claim.ExpiresAt = jwt.NewNumericDate(time.Now().Add(time.Minute * 10))
|
||||
if typ == "app" {
|
||||
if refresh.AID == aid {
|
||||
// refresh token
|
||||
acList := make(auth.Access, 0, 10)
|
||||
logv.AssertError(cfg.DB().Table("accesses a").
|
||||
Select("a.name, a.t_id, a.level").
|
||||
Joins("INNER JOIN user_roles ur ON ur.role_id = a.role_id AND ur.user_id = ? AND a.app_id = ?", refresh.UID, aid).
|
||||
Scan(&acList).Error)
|
||||
claim.Access = acList
|
||||
if aid == cfg.Config.ID {
|
||||
return auth.GenJwt(claim)
|
||||
}
|
||||
app := &M.App{}
|
||||
err = cfg.DB().Where("id = ?", aid).First(app).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return auth.GenJwtWithKey(claim, app.Key)
|
||||
} else if aid != cfg.Config.ID {
|
||||
// 只能生成其他应用的refresh token
|
||||
newToken := &M.Token{}
|
||||
newToken.UserID = refresh.UID
|
||||
newToken.AppID = aid
|
||||
newToken.ExpiredAt = time.Now().Add(time.Hour * 24)
|
||||
if opts.OverPerm != nil {
|
||||
newToken.OverPerm = *opts.OverPerm
|
||||
}
|
||||
if opts.Device != nil {
|
||||
newToken.Device = *opts.Device
|
||||
}
|
||||
newToken.Ip = x.GetRemoteIp()
|
||||
logv.AssertError(cfg.DB().Create(newToken).Error)
|
||||
// gen other app token
|
||||
claim.ID = newToken.ID
|
||||
claim.ExpiresAt = jwt.NewNumericDate(newToken.ExpiredAt)
|
||||
return auth.GenJwt(claim)
|
||||
} else {
|
||||
// 其他应用获取访问oa的token
|
||||
claim.Access = make(auth.Access, 0, 10)
|
||||
if data.OverPerm != "" {
|
||||
err = json.Unmarshal([]byte(data.OverPerm), &claim.Access)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
return auth.GenJwt(claim)
|
||||
}
|
||||
} else if typ == "ufs" {
|
||||
claim.AID = refresh.AID
|
||||
claim.UID = refresh.UID
|
||||
claim.Name = refresh.Name
|
||||
claim.Icon = refresh.Icon
|
||||
claim.ExpiresAt = jwt.NewNumericDate(time.Now().Add(time.Minute * 10))
|
||||
claim.Access = auth.Access{
|
||||
{Name: "fs", TID: "/", Level: auth.Do},
|
||||
}
|
||||
token := logv.AssertFuncErr(auth.GenJwt(claim))
|
||||
|
||||
cookie := &http.Cookie{
|
||||
Name: "fstoken", // Cookie 的名称
|
||||
Value: token, // Cookie 的值
|
||||
Path: "/fs/u/", // Cookie 的路径，通常是根路径
|
||||
MaxAge: 600, // Cookie 的最大年龄，单位是秒
|
||||
HttpOnly: true, // 是否仅限 HTTP(S) 访问
|
||||
Secure: false, // 是否通过安全连接传输 Cookie
|
||||
}
|
||||
http.SetCookie(x, cookie)
|
||||
|
||||
return token, nil
|
||||
} else {
|
||||
return nil, errs.ArgsInvalid
|
||||
}
|
||||
} else if opts.Code != nil && aid == cfg.Config.ID && opts.Salt != nil && opts.UserID != nil {
|
||||
// for oa login
|
||||
user := &M.User{}
|
||||
err = cfg.DB().Where("id = ?", opts.UserID).Find(user).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
logv.Info().Str("user", user.ID).Msg("login")
|
||||
code := *opts.Code
|
||||
salt := logv.AssertFuncErr(hex.DecodeString(*opts.Salt))
|
||||
key := logv.AssertFuncErr(hex.DecodeString(user.Code))
|
||||
de, err := utils.AesDecrypt([]byte(code), key, salt)
|
||||
if err != nil || de != user.ID {
|
||||
return nil, errs.AuthFailed
|
||||
}
|
||||
data.UserID = *opts.UserID
|
||||
data.AppID = aid
|
||||
data.ExpiredAt = time.Now().Add(time.Hour * 72)
|
||||
if opts.OverPerm != nil {
|
||||
data.OverPerm = *opts.OverPerm
|
||||
}
|
||||
if opts.Device != nil {
|
||||
data.Device = *opts.Device
|
||||
}
|
||||
data.Ip = x.GetRemoteIp()
|
||||
logv.AssertError(cfg.DB().Create(data).Error)
|
||||
claim.ID = data.ID
|
||||
claim.AID = aid
|
||||
claim.UID = user.ID
|
||||
claim.Name = user.Username
|
||||
claim.Icon = user.Icon
|
||||
claim.ExpiresAt = jwt.NewNumericDate(data.ExpiredAt)
|
||||
if user.Nickname != "" {
|
||||
claim.Name = user.Nickname
|
||||
}
|
||||
return auth.GenJwt(claim)
|
||||
} else {
|
||||
return nil, errs.ArgsInvalid
|
||||
}
|
||||
}
|
||||
|
||||
func tokenGet(x *rest.X) (any, error) {
|
||||
opts := &M.TokenGet{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := &M.Token{}
|
||||
|
||||
err = cfg.DB().Where("id = ?", opts.ID).First(data).Error
|
||||
|
||||
return data, err
|
||||
}
|
||||
func tokenPatch(x *rest.X) (any, error) {
|
||||
opts := &M.TokenPatch{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := &M.Token{}
|
||||
|
||||
err = cfg.DB().Where("id = ?", opts.ID).First(data).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
optsMap := make(map[string]interface{})
|
||||
if opts.ExpiredAt != nil {
|
||||
optsMap["expired_at"] = opts.ExpiredAt
|
||||
}
|
||||
if opts.OverPerm != nil {
|
||||
optsMap["over_perm"] = opts.OverPerm
|
||||
}
|
||||
err = cfg.DB().Model(data).Updates(optsMap).Error
|
||||
|
||||
return data, err
|
||||
}
|
||||
func tokenDelete(x *rest.X) (any, error) {
|
||||
opts := &M.TokenDelete{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := &M.Token{}
|
||||
|
||||
err = cfg.DB().Where("id = ?", opts.ID).Delete(data).Error
|
||||
|
||||
return data, err
|
||||
}
|
||||
func tokenList(x *rest.X) (any, error) {
|
||||
opts := &M.TokenList{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := make([]*M.Token, 0, 10)
|
||||
|
||||
query := cfg.DB()
|
||||
query = query.Where("user_id = ?", opts.UserID)
|
||||
query = query.Where("app_id = ?", opts.AppID)
|
||||
err = query.Limit(opts.Limit).Find(&data).Error
|
||||
|
||||
return data, err
|
||||
}
|
||||
@ -0,0 +1,108 @@
|
||||
//
|
||||
// create.go
|
||||
// Copyright (C) 2025 veypi <i@veypi.com>
|
||||
// 2025-05-06 15:05
|
||||
// Distributed under terms of the MIT license.
|
||||
//
|
||||
|
||||
package user
|
||||
|
||||
import (
|
||||
"encoding/base64"
|
||||
"fmt"
|
||||
"oa/cfg"
|
||||
M "oa/models"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"math/rand"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/veypi/OneBD/rest"
|
||||
"github.com/veypi/utils"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
var _ = Router.Post("/", userPost)
|
||||
|
||||
type postOpts struct {
|
||||
Username string `json:"username" gorm:"varchar(100);unique;default:not null" parse:"json"`
|
||||
Code string `json:"code" gorm:"varchar(128)" parse:"json"`
|
||||
Nickname *string `json:"nickname" parse:"json"`
|
||||
Icon *string `json:"icon" parse:"json"`
|
||||
Email *string `json:"email" gorm:"varchar(20);unique;default:null" parse:"json"`
|
||||
Phone *string `json:"phone" gorm:"varchar(50);unique;default:null" parse:"json"`
|
||||
}
|
||||
|
||||
func userPost(x *rest.X) (any, error) {
|
||||
opts := &postOpts{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := &M.User{}
|
||||
|
||||
data.ID = strings.ReplaceAll(uuid.New().String(), "-", "")
|
||||
data.Username = opts.Username
|
||||
data.Code = opts.Code
|
||||
data.Salt = utils.RandSeq(16)
|
||||
if len(data.Username) < 2 {
|
||||
return nil, rest.ErrArgInvalid.WithArgs("username length")
|
||||
}
|
||||
code, err := base64.URLEncoding.DecodeString(opts.Code)
|
||||
if err != nil || len(code) < 8 {
|
||||
return nil, rest.ErrArgInvalid.WithArgs("code")
|
||||
}
|
||||
code = utils.PKCS7Padding(code, 32)
|
||||
data.Code, err = utils.AesEncrypt([]byte(data.ID), code, []byte(data.Salt))
|
||||
if err != nil {
|
||||
return nil, rest.ErrArgInvalid.WithArgs("code")
|
||||
}
|
||||
ncode, err := utils.AesDecrypt([]byte(data.Code), code, []byte(data.Salt))
|
||||
if err != nil || ncode != data.ID {
|
||||
return nil, rest.ErrInternalServer.AppendString("code decrypt failed")
|
||||
}
|
||||
if opts.Nickname != nil {
|
||||
data.Nickname = *opts.Nickname
|
||||
}
|
||||
if opts.Icon != nil {
|
||||
data.Icon = *opts.Icon
|
||||
} else {
|
||||
data.Icon = fmt.Sprintf("https://public.veypi.com/img/avatar/%04d.jpg", rand.New(rand.NewSource(time.Now().UnixNano())).Intn(220))
|
||||
}
|
||||
if opts.Email != nil {
|
||||
data.Email = *opts.Email
|
||||
}
|
||||
if opts.Phone != nil {
|
||||
data.Phone = *opts.Phone
|
||||
}
|
||||
data.Status = 1
|
||||
err = cfg.DB().Transaction(func(tx *gorm.DB) error {
|
||||
err := tx.Create(data).Error
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
app := &M.App{}
|
||||
err = tx.Where("id = ?", cfg.Config.ID).First(app).Error
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
status := "ok"
|
||||
switch app.Typ {
|
||||
case "private":
|
||||
return rest.ErrNotPermitted.WithArgs("not enable register")
|
||||
case "apply":
|
||||
status = "applying"
|
||||
case "public":
|
||||
}
|
||||
if app.Typ != "public" {
|
||||
}
|
||||
|
||||
return tx.Create(&M.AppUser{
|
||||
UserID: data.ID,
|
||||
AppID: cfg.Config.ID,
|
||||
Status: status,
|
||||
}).Error
|
||||
})
|
||||
return data, err
|
||||
}
|
||||
@ -0,0 +1,101 @@
|
||||
//
|
||||
// login.go
|
||||
// Copyright (C) 2025 veypi <i@veypi.com>
|
||||
// 2025-05-12 17:35
|
||||
// Distributed under terms of the MIT license.
|
||||
//
|
||||
|
||||
package user
|
||||
|
||||
import (
|
||||
"encoding/base64"
|
||||
"oa/cfg"
|
||||
"oa/libs/auth"
|
||||
"oa/models"
|
||||
"time"
|
||||
|
||||
"github.com/golang-jwt/jwt/v5"
|
||||
"github.com/veypi/OneBD/rest"
|
||||
"github.com/veypi/utils"
|
||||
"github.com/veypi/utils/logv"
|
||||
)
|
||||
|
||||
var _ = Router.Post("/login", userLogin)
|
||||
|
||||
type loginOpts struct {
|
||||
UserName string `json:"username" parse:"json"`
|
||||
Code string `json:"code" parse:"json"`
|
||||
Phone *string `json:"phone" parse:"json"`
|
||||
Email *string `json:"email" parse:"json"`
|
||||
Type *string `json:"type" parse:"json"`
|
||||
AppID *string `json:"app_id" parse:"json"`
|
||||
Device *string `json:"device" parse:"json"`
|
||||
}
|
||||
|
||||
func userLogin(x *rest.X) (any, error) {
|
||||
// Implement login logic here
|
||||
// For example, validate user credentials and return a token
|
||||
opts := &loginOpts{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
user := &models.User{}
|
||||
query := cfg.DB()
|
||||
typ := ""
|
||||
if opts.Type != nil {
|
||||
typ = *opts.Type
|
||||
}
|
||||
switch typ {
|
||||
case "phone":
|
||||
query = query.Where("phone = ?", opts.Phone)
|
||||
case "email":
|
||||
query = query.Where("email = ?", opts.Email)
|
||||
default:
|
||||
query = query.Where("username = ?", opts.UserName)
|
||||
}
|
||||
err = query.First(user).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
logv.Info().Str("user", user.ID).Msg("login")
|
||||
code, err := base64.URLEncoding.DecodeString(opts.Code)
|
||||
if err != nil {
|
||||
return nil, rest.ErrArgInvalid.WithArgs("code")
|
||||
}
|
||||
logv.Warn().Msgf("code: %s", code)
|
||||
|
||||
ncode, err := utils.AesDecrypt([]byte(user.Code), utils.PKCS7Padding(code, 32), []byte(user.Salt))
|
||||
logv.Warn().Msgf("id: %s\n%s", ncode, user.ID)
|
||||
if err != nil || string(ncode) != user.ID {
|
||||
return nil, rest.ErrNotAuthorized
|
||||
}
|
||||
aid := cfg.Config.ID
|
||||
if opts.AppID != nil && *opts.AppID != "" {
|
||||
aid = *opts.AppID
|
||||
}
|
||||
|
||||
data := &models.Token{}
|
||||
|
||||
data.UserID = user.ID
|
||||
data.AppID = aid
|
||||
data.ExpiredAt = time.Now().Add(time.Hour * 72)
|
||||
if opts.Device != nil {
|
||||
data.Device = *opts.Device
|
||||
}
|
||||
data.Ip = x.GetRemoteIp()
|
||||
logv.AssertError(cfg.DB().Create(data).Error)
|
||||
claim := &auth.Claims{}
|
||||
claim.IssuedAt = jwt.NewNumericDate(time.Now())
|
||||
claim.Issuer = cfg.Config.ID
|
||||
claim.ID = data.ID
|
||||
claim.AID = aid
|
||||
claim.UID = user.ID
|
||||
claim.Name = user.Username
|
||||
claim.Icon = user.Icon
|
||||
claim.ExpiresAt = jwt.NewNumericDate(data.ExpiredAt)
|
||||
if user.Nickname != "" {
|
||||
claim.Name = user.Nickname
|
||||
}
|
||||
return auth.GenJwt(claim)
|
||||
}
|
||||
@ -0,0 +1,40 @@
|
||||
//
|
||||
// crud.go
|
||||
// Copyright (C) 2025 veypi <i@veypi.com>
|
||||
// 2025-05-06 15:12
|
||||
// Distributed under terms of the MIT license.
|
||||
//
|
||||
|
||||
package role
|
||||
|
||||
import (
|
||||
"github.com/veypi/OneBD/rest"
|
||||
"oa/cfg"
|
||||
M "oa/models"
|
||||
)
|
||||
|
||||
var _ = Router.Post("/", userRolePost)
|
||||
|
||||
type postOpts struct {
|
||||
UserID string `json:"user_id" parse:"path"`
|
||||
RoleID string `json:"role_id" parse:"json"`
|
||||
AppID string `json:"app_id" parse:"json"`
|
||||
Status string `json:"status" parse:"json"`
|
||||
}
|
||||
|
||||
func userRolePost(x *rest.X) (any, error) {
|
||||
opts := &postOpts{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := &M.UserRole{}
|
||||
|
||||
data.UserID = opts.UserID
|
||||
data.RoleID = opts.RoleID
|
||||
data.AppID = opts.AppID
|
||||
data.Status = opts.Status
|
||||
err = cfg.DB().Create(data).Error
|
||||
|
||||
return data, err
|
||||
}
|
||||
@ -0,0 +1,27 @@
|
||||
//
|
||||
// del.go
|
||||
// Copyright (C) 2025 veypi <i@veypi.com>
|
||||
// 2025-05-06 15:24
|
||||
// Distributed under terms of the MIT license.
|
||||
//
|
||||
|
||||
package role
|
||||
|
||||
import (
|
||||
"oa/cfg"
|
||||
"oa/models"
|
||||
|
||||
"github.com/veypi/OneBD/rest"
|
||||
)
|
||||
|
||||
var _ = Router.Delete("/:id", userRoleDelete)
|
||||
|
||||
func userRoleDelete(x *rest.X) (any, error) {
|
||||
id := x.Params.Get("id")
|
||||
if id == "" {
|
||||
return nil, rest.ErrArgInvalid.WithArgs("id")
|
||||
}
|
||||
data := &models.UserRole{}
|
||||
err := cfg.DB().Where("id = ?", id).Delete(data).Error
|
||||
return data, err
|
||||
}
|
||||
@ -0,0 +1,71 @@
|
||||
//
|
||||
// get.go
|
||||
// Copyright (C) 2025 veypi <i@veypi.com>
|
||||
// 2025-05-06 15:21
|
||||
// Distributed under terms of the MIT license.
|
||||
//
|
||||
|
||||
package role
|
||||
|
||||
import (
|
||||
"oa/cfg"
|
||||
M "oa/models"
|
||||
|
||||
"github.com/veypi/OneBD/rest"
|
||||
)
|
||||
|
||||
var _ = Router.Get("/:id", `
|
||||
get user role
|
||||
`, userRoleGet)
|
||||
|
||||
func userRoleGet(x *rest.X) (any, error) {
|
||||
data := &M.UserRole{}
|
||||
err := cfg.DB().Where("id = ?", x.Params.Get("id")).First(data).Error
|
||||
return data, err
|
||||
}
|
||||
|
||||
var _ = Router.Get("/", `
|
||||
list user roles
|
||||
`, userRoleList)
|
||||
|
||||
type listOpts struct {
|
||||
UserID *string `json:"user_id" parse:"path"`
|
||||
RoleID *string `json:"role_id" parse:"query"`
|
||||
AppID *string `json:"app_id" parse:"query"`
|
||||
Status *string `json:"status" parse:"query"`
|
||||
}
|
||||
|
||||
func userRoleList(x *rest.X) (any, error) {
|
||||
opts := &listOpts{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := make([]*struct {
|
||||
M.UserRole
|
||||
Username string `json:"username"`
|
||||
Nickname string `json:"nickname"`
|
||||
Icon string `json:"icon"`
|
||||
RoleName string `json:"role_name"`
|
||||
}, 0, 10)
|
||||
// data := make([]*M.UserRole, 0, 10)
|
||||
|
||||
query := cfg.DB().Debug().Table("user_roles").Select("user_roles.*,users.username,users.nickname,users.icon,roles.name as role_name").
|
||||
Joins("JOIN users ON users.id = user_roles.user_id").
|
||||
Joins("JOIN roles ON roles.id = user_roles.role_id")
|
||||
if opts.UserID != nil && *opts.UserID != "-" {
|
||||
query = query.Where("user_id LIKE ?", opts.UserID)
|
||||
}
|
||||
if opts.RoleID != nil {
|
||||
query = query.Where("role_id LIKE ?", opts.RoleID)
|
||||
}
|
||||
if opts.AppID != nil {
|
||||
query = query.Where("app_id LIKE ?", opts.AppID)
|
||||
}
|
||||
if opts.Status != nil {
|
||||
query = query.Where("status LIKE ?", opts.Status)
|
||||
}
|
||||
err = query.Scan(&data).Error
|
||||
|
||||
return data, err
|
||||
}
|
||||
@ -0,0 +1,12 @@
|
||||
//
|
||||
// init.go
|
||||
// Copyright (C) 2025 veypi <i@veypi.com>
|
||||
// 2025-05-06 15:12
|
||||
// Distributed under terms of the MIT license.
|
||||
//
|
||||
|
||||
package role
|
||||
|
||||
import "github.com/veypi/OneBD/rest"
|
||||
|
||||
var Router = rest.NewRouter()
|
||||
@ -0,0 +1,42 @@
|
||||
//
|
||||
// crud.go
|
||||
// Copyright (C) 2025 veypi <i@veypi.com>
|
||||
// 2025-05-06 15:12
|
||||
// Distributed under terms of the MIT license.
|
||||
//
|
||||
|
||||
package role
|
||||
|
||||
import (
|
||||
"github.com/veypi/OneBD/rest"
|
||||
"oa/cfg"
|
||||
M "oa/models"
|
||||
)
|
||||
|
||||
type patchOpts struct {
|
||||
ID string `json:"id" parse:"path"`
|
||||
Status *string `json:"status" parse:"json"`
|
||||
}
|
||||
|
||||
var _ = Router.Patch("/:id", userRolePatch)
|
||||
|
||||
func userRolePatch(x *rest.X) (any, error) {
|
||||
opts := &patchOpts{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := &M.UserRole{}
|
||||
|
||||
err = cfg.DB().Where("id = ?", opts.ID).First(data).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
optsMap := make(map[string]interface{})
|
||||
if opts.Status != nil {
|
||||
optsMap["status"] = opts.Status
|
||||
}
|
||||
err = cfg.DB().Model(data).Updates(optsMap).Error
|
||||
|
||||
return data, err
|
||||
}
|
||||
@ -1,111 +0,0 @@
|
||||
package user
|
||||
|
||||
import (
|
||||
"github.com/veypi/OneBD/rest"
|
||||
"oa/cfg"
|
||||
M "oa/models"
|
||||
)
|
||||
|
||||
func init() {
|
||||
r := Router.SubRouter(":user_id/user_role")
|
||||
r.Delete("/:user_role_id", userRoleDelete)
|
||||
r.Get("/:user_role_id", userRoleGet)
|
||||
r.Get("/", userRoleList)
|
||||
r.Patch("/:user_role_id", userRolePatch)
|
||||
r.Post("/", userRolePost)
|
||||
}
|
||||
|
||||
func userRoleList(x *rest.X) (any, error) {
|
||||
opts := &M.UserRoleList{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := make([]*struct {
|
||||
M.UserRole
|
||||
Username string `json:"username"`
|
||||
Nickname string `json:"nickname"`
|
||||
Icon string `json:"icon"`
|
||||
RoleName string `json:"role_name"`
|
||||
}, 0, 10)
|
||||
// data := make([]*M.UserRole, 0, 10)
|
||||
|
||||
query := cfg.DB().Debug().Table("user_roles").Select("user_roles.*,users.username,users.nickname,users.icon,roles.name as role_name").
|
||||
Joins("JOIN users ON users.id = user_roles.user_id").
|
||||
Joins("JOIN roles ON roles.id = user_roles.role_id")
|
||||
if opts.UserID != nil && *opts.UserID != "-" {
|
||||
query = query.Where("user_id LIKE ?", opts.UserID)
|
||||
}
|
||||
if opts.RoleID != nil {
|
||||
query = query.Where("role_id LIKE ?", opts.RoleID)
|
||||
}
|
||||
if opts.AppID != nil {
|
||||
query = query.Where("app_id LIKE ?", opts.AppID)
|
||||
}
|
||||
if opts.Status != nil {
|
||||
query = query.Where("status LIKE ?", opts.Status)
|
||||
}
|
||||
err = query.Scan(&data).Error
|
||||
|
||||
return data, err
|
||||
}
|
||||
func userRolePost(x *rest.X) (any, error) {
|
||||
opts := &M.UserRolePost{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := &M.UserRole{}
|
||||
|
||||
data.UserID = opts.UserID
|
||||
data.RoleID = opts.RoleID
|
||||
data.AppID = opts.AppID
|
||||
data.Status = opts.Status
|
||||
err = cfg.DB().Create(data).Error
|
||||
|
||||
return data, err
|
||||
}
|
||||
func userRoleGet(x *rest.X) (any, error) {
|
||||
opts := &M.UserRoleGet{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := &M.UserRole{}
|
||||
|
||||
err = cfg.DB().Where("id = ?", opts.ID).First(data).Error
|
||||
|
||||
return data, err
|
||||
}
|
||||
func userRolePatch(x *rest.X) (any, error) {
|
||||
opts := &M.UserRolePatch{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := &M.UserRole{}
|
||||
|
||||
err = cfg.DB().Where("id = ?", opts.ID).First(data).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
optsMap := make(map[string]interface{})
|
||||
if opts.Status != nil {
|
||||
optsMap["status"] = opts.Status
|
||||
}
|
||||
err = cfg.DB().Model(data).Updates(optsMap).Error
|
||||
|
||||
return data, err
|
||||
}
|
||||
func userRoleDelete(x *rest.X) (any, error) {
|
||||
opts := &M.UserRoleDelete{}
|
||||
err := x.Parse(opts)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data := &M.UserRole{}
|
||||
|
||||
err = cfg.DB().Where("id = ?", opts.ID).Delete(data).Error
|
||||
|
||||
return data, err
|
||||
}
|
||||
@ -1,66 +0,0 @@
|
||||
package models
|
||||
|
||||
import ()
|
||||
|
||||
type UserGet struct {
|
||||
ID string `json:"id" gorm:"primaryKey;type:varchar(32)" parse:"path@user_id"`
|
||||
}
|
||||
|
||||
type UserPatch struct {
|
||||
ID string `json:"id" gorm:"primaryKey;type:varchar(32)" parse:"path@user_id"`
|
||||
Username *string `json:"username" gorm:"varchar(100);unique;default:not null" parse:"json"`
|
||||
Nickname *string `json:"nickname" parse:"json"`
|
||||
Icon *string `json:"icon" parse:"json"`
|
||||
Email *string `json:"email" gorm:"varchar(20);unique;default:null" parse:"json"`
|
||||
Phone *string `json:"phone" gorm:"varchar(50);unique;default:null" parse:"json"`
|
||||
Status *uint `json:"status" parse:"json"`
|
||||
}
|
||||
|
||||
type UserDelete struct {
|
||||
ID string `json:"id" gorm:"primaryKey;type:varchar(32)" parse:"path@user_id"`
|
||||
}
|
||||
|
||||
type UserPost struct {
|
||||
Username string `json:"username" gorm:"varchar(100);unique;default:not null" parse:"json"`
|
||||
Nickname *string `json:"nickname" parse:"json"`
|
||||
Icon *string `json:"icon" parse:"json"`
|
||||
Email *string `json:"email" gorm:"varchar(20);unique;default:null" parse:"json"`
|
||||
Phone *string `json:"phone" gorm:"varchar(50);unique;default:null" parse:"json"`
|
||||
Salt string `json:"salt" gorm:"varchar(32)" parse:"json"`
|
||||
Code string `json:"code" gorm:"varchar(128)" parse:"json"`
|
||||
}
|
||||
|
||||
type UserList struct {
|
||||
Username *string `json:"username" gorm:"type:varchar(100);unique;default:not null" parse:"query"`
|
||||
Nickname *string `json:"nickname" gorm:"type:varchar(100)" parse:"query"`
|
||||
Email *string `json:"email" gorm:"unique;type:varchar(50);default:null" parse:"query"`
|
||||
Phone *string `json:"phone" gorm:"type:varchar(30);unique;default:null" parse:"query"`
|
||||
Status *uint `json:"status" parse:"query"`
|
||||
}
|
||||
|
||||
type UserRoleGet struct {
|
||||
ID string `json:"id" gorm:"primaryKey;type:varchar(32)" parse:"path@user_role_id"`
|
||||
}
|
||||
|
||||
type UserRolePatch struct {
|
||||
ID string `json:"id" gorm:"primaryKey;type:varchar(32)" parse:"path@user_role_id"`
|
||||
Status *string `json:"status" parse:"json"`
|
||||
}
|
||||
|
||||
type UserRoleDelete struct {
|
||||
ID string `json:"id" gorm:"primaryKey;type:varchar(32)" parse:"path@user_role_id"`
|
||||
}
|
||||
|
||||
type UserRoleList struct {
|
||||
UserID *string `json:"user_id" parse:"path"`
|
||||
RoleID *string `json:"role_id" parse:"query"`
|
||||
AppID *string `json:"app_id" parse:"query"`
|
||||
Status *string `json:"status" parse:"query"`
|
||||
}
|
||||
|
||||
type UserRolePost struct {
|
||||
UserID string `json:"user_id" parse:"path"`
|
||||
RoleID string `json:"role_id" parse:"json"`
|
||||
AppID string `json:"app_id" parse:"json"`
|
||||
Status string `json:"status" parse:"json"`
|
||||
}
|
||||
Loading…
Reference in New Issue