Buriburizaemon/nezha_domains
1
0
Fork 0
mirror of https://github.com/Buriburizaem0n/nezha_domains.git synced 2026-02-05 21:20:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

update profile api (#16)

Browse Source 
* update profile api

* rename

* fix realip assertion

* add waf api
This commit is contained in:
UUBulb
2024-11-23 16:22:22 +08:00
committed by GitHub
parent 22738b6244
commit 885330e948
7 changed files with 85 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
model/user.go
View File

@@ -5,3 +5,8 @@ type User struct {
Username string `json:"username,omitempty" gorm:"uniqueIndex"`
Password string `json:"password,omitempty" gorm:"type:char(72)"`
}
type Profile struct {
User
LoginIP string `json:"login_ip,omitempty"`
}

17
model/waf.go
View File

@@ -56,7 +56,22 @@ func ClearIP(db *gorm.DB, ip string) error {
if err != nil {
return err
}
return db.Delete(&WAF{}, "ip = ?", ipBinary).Error
return db.Unscoped().Delete(&WAF{}, "ip = ?", ipBinary).Error
}
func BatchClearIP(db *gorm.DB, ip []string) error {
if len(ip) < 1 {
return nil
}
ips := make([][]byte, 0, len(ip))
for _, s := range ip {
ipBinary, err := utils.IPStringToBinary(s)
if err != nil {
continue
}
ips = append(ips, ipBinary)
}
return db.Unscoped().Delete(&WAF{}, "ip in (?)", ips).Error
}
func BlockIP(db *gorm.DB, ip string, reason uint8) error {