mirror of
https://github.com/chaitin/MonkeyCode.git
synced 2026-02-02 06:43:23 +08:00
376 lines
14 KiB
Go
376 lines
14 KiB
Go
package domain
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/GoYoko/web"
|
|
|
|
"github.com/chaitin/MonkeyCode/backend/consts"
|
|
"github.com/chaitin/MonkeyCode/backend/db"
|
|
"github.com/chaitin/MonkeyCode/backend/ent/types"
|
|
"github.com/chaitin/MonkeyCode/backend/pkg/cvt"
|
|
)
|
|
|
|
type UserUsecase interface {
|
|
Login(ctx context.Context, req *LoginReq) (*LoginResp, error)
|
|
Update(ctx context.Context, req *UpdateUserReq) (*User, error)
|
|
Delete(ctx context.Context, id string) error
|
|
InitAdmin(ctx context.Context) error
|
|
AdminLogin(ctx context.Context, req *LoginReq) (*AdminUser, error)
|
|
DeleteAdmin(ctx context.Context, id string) error
|
|
CreateAdmin(ctx context.Context, req *CreateAdminReq) (*AdminUser, error)
|
|
VSCodeAuthInit(ctx context.Context, req *VSCodeAuthInitReq) (*VSCodeAuthInitResp, error)
|
|
List(ctx context.Context, req ListReq) (*ListUserResp, error)
|
|
AdminList(ctx context.Context, page *web.Pagination) (*ListAdminUserResp, error)
|
|
LoginHistory(ctx context.Context, page *web.Pagination) (*ListLoginHistoryResp, error)
|
|
AdminLoginHistory(ctx context.Context, page *web.Pagination) (*ListAdminLoginHistoryResp, error)
|
|
Invite(ctx context.Context, userID string) (*InviteResp, error)
|
|
Register(ctx context.Context, req *RegisterReq) (*User, error)
|
|
GetSetting(ctx context.Context) (*Setting, error)
|
|
UpdateSetting(ctx context.Context, req *UpdateSettingReq) (*Setting, error)
|
|
OAuthSignUpOrIn(ctx context.Context, req *OAuthSignUpOrInReq) (*OAuthURLResp, error)
|
|
OAuthCallback(ctx context.Context, req *OAuthCallbackReq) (string, error)
|
|
}
|
|
|
|
type UserRepo interface {
|
|
List(ctx context.Context, page *web.Pagination) ([]*db.User, *db.PageInfo, error)
|
|
Update(ctx context.Context, id string, fn func(*db.UserUpdateOne) error) (*db.User, error)
|
|
Delete(ctx context.Context, id string) error
|
|
InitAdmin(ctx context.Context, username, password string) error
|
|
CreateUser(ctx context.Context, user *db.User) (*db.User, error)
|
|
CreateAdmin(ctx context.Context, admin *db.Admin) (*db.Admin, error)
|
|
DeleteAdmin(ctx context.Context, id string) error
|
|
AdminByName(ctx context.Context, username string) (*db.Admin, error)
|
|
GetByName(ctx context.Context, username string) (*db.User, error)
|
|
GetOrCreateApiKey(ctx context.Context, userID string) (*db.ApiKey, error)
|
|
AdminList(ctx context.Context, page *web.Pagination) ([]*db.Admin, *db.PageInfo, error)
|
|
CreateInviteCode(ctx context.Context, userID string, code string) (*db.InviteCode, error)
|
|
ValidateInviteCode(ctx context.Context, code string) (*db.InviteCode, error)
|
|
UserLoginHistory(ctx context.Context, page *web.Pagination) ([]*db.UserLoginHistory, *db.PageInfo, error)
|
|
AdminLoginHistory(ctx context.Context, page *web.Pagination) ([]*db.AdminLoginHistory, *db.PageInfo, error)
|
|
GetSetting(ctx context.Context) (*db.Setting, error)
|
|
UpdateSetting(ctx context.Context, fn func(*db.Setting, *db.SettingUpdateOne)) (*db.Setting, error)
|
|
OAuthRegister(ctx context.Context, platform consts.UserPlatform, inviteCode string, req *OAuthUserInfo) (*db.User, error)
|
|
OAuthLogin(ctx context.Context, platform consts.UserPlatform, req *OAuthUserInfo) (*db.User, error)
|
|
SignUpOrIn(ctx context.Context, platform consts.UserPlatform, req *OAuthUserInfo) (*db.User, error)
|
|
SaveAdminLoginHistory(ctx context.Context, adminID, ip string) error
|
|
SaveUserLoginHistory(ctx context.Context, userID, ip string, session *VSCodeSession) error
|
|
}
|
|
|
|
type UpdateUserReq struct {
|
|
ID string `json:"id" validate:"required"` // 用户ID
|
|
Status *consts.UserStatus `json:"status"` // 用户状态 active: 正常 locked: 锁定 inactive: 禁用
|
|
Password *string `json:"password"` // 重置密码
|
|
}
|
|
|
|
type CreateAdminReq struct {
|
|
Username string `json:"username"` // 用户名
|
|
Password string `json:"password"` // 密码
|
|
}
|
|
|
|
type VSCodeAuthInitReq struct {
|
|
ClientID string `json:"client_id" validate:"required"` // 客户端ID
|
|
RedirectURI string `json:"redirect_uri" validate:"required"` // 重定向URI
|
|
State string `json:"state" validate:"required"` // 状态
|
|
Version string `json:"version"` // 版本
|
|
OSType consts.OSType `json:"os_type"` // 操作系统类型
|
|
OSRelease consts.OSRelease `json:"os_release"` // 操作系统版本
|
|
Hostname string `json:"hostname"` // 主机名
|
|
}
|
|
|
|
type VSCodeAuthInitResp struct {
|
|
AuthURL string `json:"auth_url"` // 授权URL
|
|
}
|
|
|
|
type LoginReq struct {
|
|
SessionID string `json:"session_id"` // 会话Id
|
|
Username string `json:"username"` // 用户名
|
|
Password string `json:"password"` // 密码
|
|
IP string `json:"-"` // IP地址
|
|
}
|
|
|
|
type AdminLoginReq struct {
|
|
Account string `json:"account"` // 用户名
|
|
Password string `json:"password"` // 密码
|
|
}
|
|
|
|
type LoginResp struct {
|
|
RedirectURL string `json:"redirect_url"` // 重定向URL
|
|
}
|
|
|
|
type ListReq struct {
|
|
web.Pagination
|
|
|
|
Search string `json:"search" query:"search"` // 搜索
|
|
}
|
|
|
|
type RegisterReq struct {
|
|
Username string `json:"username" validate:"required"` // 用户名
|
|
Email string `json:"email" validate:"required"` // 邮箱
|
|
Password string `json:"password" validate:"required"` // 密码
|
|
Code string `json:"code" validate:"required"` // 邀请码
|
|
}
|
|
|
|
type ListLoginHistoryResp struct {
|
|
*db.PageInfo
|
|
|
|
LoginHistories []*UserLoginHistory `json:"login_histories"`
|
|
}
|
|
|
|
type ListAdminLoginHistoryResp struct {
|
|
*db.PageInfo
|
|
|
|
LoginHistories []*AdminLoginHistory `json:"login_histories"`
|
|
}
|
|
|
|
type InviteResp struct {
|
|
Code string `json:"code"` // 邀请码
|
|
}
|
|
|
|
type IPInfo struct {
|
|
IP string `json:"ip"` // IP地址
|
|
Country string `json:"country"` // 国家
|
|
Province string `json:"province"` // 省份
|
|
City string `json:"city"` // 城市
|
|
ISP string `json:"isp"` // 运营商
|
|
ASN string `json:"asn"` // ASN
|
|
}
|
|
|
|
type UserLoginHistory struct {
|
|
User *User `json:"user"` // 用户信息
|
|
IPInfo *IPInfo `json:"ip_info"` // IP信息
|
|
ClientVersion string `json:"client_version"` // 客户端版本
|
|
Device string `json:"device"` // 设备信息
|
|
CreatedAt int64 `json:"created_at"` // 登录时间
|
|
}
|
|
|
|
func (l *UserLoginHistory) From(e *db.UserLoginHistory) *UserLoginHistory {
|
|
if e == nil {
|
|
return l
|
|
}
|
|
|
|
l.User = cvt.From(e.Edges.Owner, &User{})
|
|
l.IPInfo = &IPInfo{
|
|
IP: e.IP,
|
|
Country: e.Country,
|
|
Province: e.Province,
|
|
City: e.City,
|
|
ISP: e.Isp,
|
|
ASN: e.Asn,
|
|
}
|
|
l.ClientVersion = e.ClientVersion
|
|
l.Device = e.OsType.Name()
|
|
if e.Hostname != "" {
|
|
l.Device += fmt.Sprintf(" (%s)", e.Hostname)
|
|
}
|
|
l.CreatedAt = e.CreatedAt.Unix()
|
|
|
|
return l
|
|
}
|
|
|
|
type AdminLoginHistory struct {
|
|
User *AdminUser `json:"user"` // 用户信息
|
|
IPInfo *IPInfo `json:"ip_info"` // IP信息
|
|
ClientVersion string `json:"client_version"` // 客户端版本
|
|
Device string `json:"device"` // 设备信息
|
|
CreatedAt int64 `json:"created_at"` // 登录时间
|
|
}
|
|
|
|
func (l *AdminLoginHistory) From(e *db.AdminLoginHistory) *AdminLoginHistory {
|
|
if e == nil {
|
|
return l
|
|
}
|
|
|
|
l.User = cvt.From(e.Edges.Owner, &AdminUser{})
|
|
l.IPInfo = &IPInfo{
|
|
IP: e.IP,
|
|
Country: e.Country,
|
|
Province: e.Province,
|
|
City: e.City,
|
|
ISP: e.Isp,
|
|
ASN: e.Asn,
|
|
}
|
|
l.ClientVersion = e.ClientVersion
|
|
l.Device = e.Device
|
|
l.CreatedAt = e.CreatedAt.Unix()
|
|
|
|
return l
|
|
}
|
|
|
|
type ListUserResp struct {
|
|
*db.PageInfo
|
|
|
|
Users []*User `json:"users"`
|
|
}
|
|
|
|
type ListAdminUserResp struct {
|
|
*db.PageInfo
|
|
|
|
Users []*AdminUser `json:"users"`
|
|
}
|
|
|
|
type User struct {
|
|
ID string `json:"id"` // 用户ID
|
|
Username string `json:"username"` // 用户名
|
|
Email string `json:"email"` // 邮箱
|
|
TwoStepAuth bool `json:"two_step_auth"` // 是否开启两步验证
|
|
Status consts.UserStatus `json:"status"` // 用户状态 active: 正常 locked: 锁定 inactive: 禁用
|
|
AvatarURL string `json:"avatar_url"` // 头像URL
|
|
CreatedAt int64 `json:"created_at"` // 创建时间
|
|
LastActiveAt int64 `json:"last_active_at"` // 最后活跃时间
|
|
}
|
|
|
|
func (u *User) From(e *db.User) *User {
|
|
if e == nil {
|
|
return u
|
|
}
|
|
|
|
u.ID = e.ID.String()
|
|
u.Username = e.Username
|
|
u.Email = e.Email
|
|
u.Status = e.Status
|
|
u.AvatarURL = e.AvatarURL
|
|
u.CreatedAt = e.CreatedAt.Unix()
|
|
|
|
return u
|
|
}
|
|
|
|
type AdminUser struct {
|
|
ID string `json:"id"` // 用户ID
|
|
Username string `json:"username"` // 用户名
|
|
LastActiveAt int64 `json:"last_active_at"` // 最后活跃时间
|
|
Status consts.AdminStatus `json:"status"` // 用户状态 active: 正常 inactive: 禁用
|
|
CreatedAt int64 `json:"created_at"` // 创建时间
|
|
}
|
|
|
|
func (a *AdminUser) From(e *db.Admin) *AdminUser {
|
|
if e == nil {
|
|
return a
|
|
}
|
|
|
|
a.ID = e.ID.String()
|
|
a.Username = e.Username
|
|
a.Status = e.Status
|
|
a.CreatedAt = e.CreatedAt.Unix()
|
|
|
|
return a
|
|
}
|
|
|
|
type VSCodeSession struct {
|
|
ID string `json:"id"` // 会话ID
|
|
State string `json:"state"` // 状态
|
|
RedirectURI string `json:"redirect_uri"` // 重定向URI
|
|
Version string `json:"version"` // 版本
|
|
OSType consts.OSType `json:"os_type"` // 操作系统类型
|
|
OSRelease consts.OSRelease `json:"os_release"` // 操作系统版本
|
|
Hostname string `json:"hostname"` // 主机名
|
|
}
|
|
|
|
type UpdateSettingReq struct {
|
|
EnableSSO *bool `json:"enable_sso"` // 是否开启SSO
|
|
ForceTwoFactorAuth *bool `json:"force_two_factor_auth"` // 是否强制两步验证
|
|
DisablePasswordLogin *bool `json:"disable_password_login"` // 是否禁用密码登录
|
|
EnableAutoLogin *bool `json:"enable_auto_login"` // 是否开启自动登录
|
|
DingtalkOAuth *DingtalkOAuthReq `json:"dingtalk_oauth"` // 钉钉OAuth配置
|
|
CustomOAuth *CustomOAuthReq `json:"custom_oauth"` // 自定义OAuth配置
|
|
}
|
|
|
|
type DingtalkOAuthReq struct {
|
|
Enable *bool `json:"enable"` // 钉钉OAuth开关
|
|
ClientID *string `json:"client_id"` // 钉钉客户端ID
|
|
ClientSecret *string `json:"client_secret"` // 钉钉客户端密钥
|
|
}
|
|
|
|
type DingtalkOAuth struct {
|
|
Enable bool `json:"enable"` // 钉钉OAuth开关
|
|
ClientID string `json:"client_id"` // 钉钉客户端ID
|
|
ClientSecret string `json:"client_secret"` // 钉钉客户端密钥
|
|
}
|
|
|
|
func (d *DingtalkOAuth) From(e *types.DingtalkOAuth) *DingtalkOAuth {
|
|
if e == nil {
|
|
d.Enable = false
|
|
return d
|
|
}
|
|
|
|
d.Enable = e.Enable
|
|
d.ClientID = e.ClientID
|
|
return d
|
|
}
|
|
|
|
type CustomOAuthReq struct {
|
|
Enable *bool `json:"enable"` // 自定义OAuth开关
|
|
ClientID *string `json:"client_id"` // 自定义客户端ID
|
|
ClientSecret *string `json:"client_secret"` // 自定义客户端密钥
|
|
AuthorizeURL *string `json:"authorize_url"` // 自定义OAuth授权URL
|
|
AccessTokenURL *string `json:"access_token_url"` // 自定义OAuth访问令牌URL
|
|
UserInfoURL *string `json:"userinfo_url"` // 自定义OAuth用户信息URL
|
|
Scopes []string `json:"scopes"` // 自定义OAuth Scope列表
|
|
IDField *string `json:"id_field"` // 用户信息回包中的ID字段名
|
|
NameField *string `json:"name_field"` // 用户信息回包中的用户名字段名
|
|
AvatarField *string `json:"avatar_field"` // 用户信息回包中的头像URL字段名
|
|
EmailField *string `json:"email_field"` // 用户信息回包中的邮箱字段名
|
|
}
|
|
|
|
type CustomOAuth struct {
|
|
Enable bool `json:"enable"` // 自定义OAuth开关
|
|
ClientID string `json:"client_id"` // 自定义客户端ID
|
|
ClientSecret string `json:"client_secret"` // 自定义客户端密钥
|
|
AuthorizeURL string `json:"authorize_url"` // 自定义OAuth授权URL
|
|
AccessTokenURL string `json:"access_token_url"` // 自定义OAuth访问令牌URL
|
|
UserInfoURL string `json:"userinfo_url"` // 自定义OAuth用户信息URL
|
|
Scopes []string `json:"scopes"` // 自定义OAuth Scope列表
|
|
IDField string `json:"id_field"` // 用户信息回包中的ID字段名
|
|
NameField string `json:"name_field"` // 用户信息回包中的用户名字段名
|
|
AvatarField string `json:"avatar_field"` // 用户信息回包中的头像URL字段名
|
|
EmailField string `json:"email_field"` // 用户信息回包中的邮箱字段名
|
|
}
|
|
|
|
func (c *CustomOAuth) From(e *types.CustomOAuth) *CustomOAuth {
|
|
if e == nil {
|
|
c.Enable = false
|
|
return c
|
|
}
|
|
|
|
c.Enable = e.Enable
|
|
c.ClientID = e.ClientID
|
|
c.AuthorizeURL = e.AuthorizeURL
|
|
c.AccessTokenURL = e.AccessTokenURL
|
|
c.UserInfoURL = e.UserInfoURL
|
|
c.Scopes = e.Scopes
|
|
c.IDField = e.IDField
|
|
c.NameField = e.NameField
|
|
c.AvatarField = e.AvatarField
|
|
c.EmailField = e.EmailField
|
|
return c
|
|
}
|
|
|
|
type Setting struct {
|
|
EnableSSO bool `json:"enable_sso"` // 是否开启SSO
|
|
ForceTwoFactorAuth bool `json:"force_two_factor_auth"` // 是否强制两步验证
|
|
DisablePasswordLogin bool `json:"disable_password_login"` // 是否禁用密码登录
|
|
EnableAutoLogin bool `json:"enable_auto_login"` // 是否开启自动登录
|
|
DingtalkOAuth DingtalkOAuth `json:"dingtalk_oauth"` // 钉钉OAuth接入
|
|
CustomOAuth CustomOAuth `json:"custom_oauth"` // 自定义OAuth接入
|
|
CreatedAt int64 `json:"created_at"` // 创建时间
|
|
UpdatedAt int64 `json:"updated_at"` // 更新时间
|
|
}
|
|
|
|
func (s *Setting) From(e *db.Setting) *Setting {
|
|
if e == nil {
|
|
return s
|
|
}
|
|
|
|
s.EnableSSO = e.EnableSSO
|
|
s.ForceTwoFactorAuth = e.ForceTwoFactorAuth
|
|
s.DisablePasswordLogin = e.DisablePasswordLogin
|
|
s.EnableAutoLogin = e.EnableAutoLogin
|
|
s.DingtalkOAuth = *cvt.From(e.DingtalkOauth, &DingtalkOAuth{})
|
|
s.CustomOAuth = *cvt.From(e.CustomOauth, &CustomOAuth{})
|
|
s.CreatedAt = e.CreatedAt.Unix()
|
|
s.UpdatedAt = e.UpdatedAt.Unix()
|
|
|
|
return s
|
|
}
|