mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
aa87442fdd
Source files keep their SPDX-License-Identifier and Hanzo copyright line; the Gitea, Gogs and LFS Test Server copyrights they used to repeat now live once in NOTICE, alongside LICENSE. The header lint accepts any copyright line above the SPDX line, so it no longer demands an upstream one. Co-authored-by: Hanzo Dev <[email protected]>
117 lines
3.0 KiB
Go
117 lines
3.0 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package user
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/hanzoai/git/models/db"
|
|
"github.com/hanzoai/git/modules/util"
|
|
|
|
"github.com/hanzoai/builder"
|
|
)
|
|
|
|
// UserOpenID is the list of all OpenID identities of a user.
|
|
// Since this is a middle table, name it OpenID is not suitable, so we ignore the lint here
|
|
type UserOpenID struct { //revive:disable-line:exported
|
|
ID int64 `xorm:"pk autoincr"`
|
|
UID int64 `xorm:"INDEX NOT NULL"`
|
|
URI string `xorm:"UNIQUE NOT NULL"`
|
|
Show bool `xorm:"DEFAULT false"`
|
|
}
|
|
|
|
func init() {
|
|
db.RegisterModel(new(UserOpenID))
|
|
}
|
|
|
|
// GetUserOpenIDs returns all openid addresses that belongs to given user.
|
|
func GetUserOpenIDs(ctx context.Context, uid int64) ([]*UserOpenID, error) {
|
|
openids := make([]*UserOpenID, 0, 5)
|
|
if err := db.GetEngine(ctx).
|
|
Where("uid=?", uid).
|
|
Asc("id").
|
|
Find(&openids); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return openids, nil
|
|
}
|
|
|
|
// isOpenIDUsed returns true if the openid has been used.
|
|
func isOpenIDUsed(ctx context.Context, uri string) (bool, error) {
|
|
if len(uri) == 0 {
|
|
return true, nil
|
|
}
|
|
|
|
return db.Exist[UserOpenID](ctx, builder.Eq{"uri": uri})
|
|
}
|
|
|
|
// ErrOpenIDAlreadyUsed represents a "OpenIDAlreadyUsed" kind of error.
|
|
type ErrOpenIDAlreadyUsed struct {
|
|
OpenID string
|
|
}
|
|
|
|
// IsErrOpenIDAlreadyUsed checks if an error is a ErrOpenIDAlreadyUsed.
|
|
func IsErrOpenIDAlreadyUsed(err error) bool {
|
|
_, ok := err.(ErrOpenIDAlreadyUsed)
|
|
return ok
|
|
}
|
|
|
|
func (err ErrOpenIDAlreadyUsed) Error() string {
|
|
return fmt.Sprintf("OpenID already in use [oid: %s]", err.OpenID)
|
|
}
|
|
|
|
func (err ErrOpenIDAlreadyUsed) Unwrap() error {
|
|
return util.ErrAlreadyExist
|
|
}
|
|
|
|
// AddUserOpenID adds an pre-verified/normalized OpenID URI to given user.
|
|
// NOTE: make sure openid.URI is normalized already
|
|
func AddUserOpenID(ctx context.Context, openid *UserOpenID) error {
|
|
used, err := isOpenIDUsed(ctx, openid.URI)
|
|
if err != nil {
|
|
return err
|
|
} else if used {
|
|
return ErrOpenIDAlreadyUsed{openid.URI}
|
|
}
|
|
|
|
return db.Insert(ctx, openid)
|
|
}
|
|
|
|
// DeleteUserOpenID deletes an openid address of given user.
|
|
func DeleteUserOpenID(ctx context.Context, openid *UserOpenID) (err error) {
|
|
var deleted int64
|
|
// ask to check UID
|
|
address := UserOpenID{
|
|
UID: openid.UID,
|
|
}
|
|
if openid.ID > 0 {
|
|
deleted, err = db.GetEngine(ctx).ID(openid.ID).Delete(&address)
|
|
} else {
|
|
deleted, err = db.GetEngine(ctx).
|
|
Where("openid=?", openid.URI).
|
|
Delete(&address)
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
} else if deleted != 1 {
|
|
return util.NewNotExistErrorf("OpenID is unknown")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// ToggleUserOpenIDVisibility toggles visibility of an openid address of given user.
|
|
func ToggleUserOpenIDVisibility(ctx context.Context, id int64, user *User) error {
|
|
affected, err := db.GetEngine(ctx).Exec("update `user_open_id` set `show` = not `show` where `id` = ? AND uid = ?", id, user.ID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if n, _ := affected.RowsAffected(); n != 1 {
|
|
return util.NewNotExistErrorf("OpenID is unknown")
|
|
}
|
|
return nil
|
|
}
|