Albirew/nyaa-pantsu
Archivé
1
0
Bifurcation 0
Ce dépôt a été archivé le 2022-05-07. Vous pouvez voir ses fichiers ou le cloner, mais pas ouvrir de ticket ou de demandes d'ajout, ni soumettre de changements.
nyaa-pantsu/model/user.go

118 lignes
3,1 Kio
Go
Brut Vue normale Historique

package model
import (
"time"
"github.com/NyaaPantsu/nyaa/config"
)
const (
UserStatusBanned = -1
UserStatusMember = 0
UserStatusTrusted = 1
UserStatusModerator = 2
)
type User struct {
ID uint `gorm:"column:user_id;primary_key"`
Username string `gorm:"column:username"`
Password string `gorm:"column:password"`
Email string `gorm:"column:email"`
Status int `gorm:"column:status"`
CreatedAt time.Time `gorm:"column:created_at"`
UpdatedAt time.Time `gorm:"column:updated_at"`
ApiToken string `gorm:"column:api_token"`
ApiTokenExpiry time.Time `gorm:"column:api_token_expiry"`
Language string `gorm:"column:language"`
2017-05-09 12:14:12 +02:00
// TODO: move this to PublicUser
Likings []User // Don't work `gorm:"foreignkey:user_id;associationforeignkey:follower_id;many2many:user_follows"`
Liked []User // Don't work `gorm:"foreignkey:follower_id;associationforeignkey:user_id;many2many:user_follows"`
2017-05-09 12:14:12 +02:00
2017-05-10 13:57:31 +02:00
MD5 string `json:"md5" gorm:"column:md5"` // Hash of email address, used for Gravatar
Torrents []Torrent `gorm:"ForeignKey:UploaderID"`
2017-05-20 20:53:05 +02:00
2017-05-21 00:02:57 +02:00
UnreadNotifications int `gorm:"-"` // We don't want to loop every notifications when accessing user unread notif
2017-05-20 20:53:05 +02:00
Notifications []Notification `gorm:"ForeignKey:UserID"`
2017-05-10 10:27:17 +02:00
}
2017-05-10 21:09:37 +02:00
type UserJSON struct {
ID uint `json:"user_id"`
Username string `json:"username"`
Status int `json:"status"`
CreatedAt string `json:"created_at"`
LikingCount int `json:"liking_count"`
LikedCount int `json:"liked_count"`
}
2017-05-10 10:27:17 +02:00
// Returns the total size of memory recursively allocated for this struct
func (u User) Size() (s int) {
s += 4 + // ints
6*2 + // string pointers
4*3 + //time.Time
3*2 + // arrays
// string arrays
2017-05-12 12:40:31 +02:00
len(u.Username) + len(u.Password) + len(u.Email) + len(u.ApiToken) + len(u.MD5) + len(u.Language)
2017-05-10 10:27:17 +02:00
s *= 8
// Ignoring foreign key users. Fuck them.
return
}
func (u User) IsBanned() bool {
return u.Status == UserStatusBanned
}
func (u User) IsMember() bool {
return u.Status == UserStatusMember
}
func (u User) IsTrusted() bool {
return u.Status == UserStatusTrusted
}
func (u User) IsModerator() bool {
return u.Status == UserStatusModerator
}
2017-05-20 20:53:05 +02:00
func (u User) GetUnreadNotifications() int {
if u.UnreadNotifications == 0 {
for _, notif := range u.Notifications {
if !notif.Read {
u.UnreadNotifications++
}
}
}
return u.UnreadNotifications
}
2017-05-08 19:26:29 +02:00
type PublicUser struct {
2017-05-10 10:27:17 +02:00
User *User
}
2017-05-09 17:14:13 +02:00
// different users following eachother
type UserFollows struct {
2017-05-09 12:14:12 +02:00
UserID uint `gorm:"column:user_id"`
FollowerID uint `gorm:"column:following"`
2017-05-09 07:21:14 +02:00
}
type UserUploadsOld struct {
Username string `gorm:"column:username"`
TorrentId uint `gorm:"column:torrent_id"`
}
func (c UserUploadsOld) TableName() string {
// is this needed here?
return config.UploadsOldTableName
}
2017-05-10 21:09:37 +02:00
func (u *User) ToJSON() UserJSON {
json := UserJSON{
ID: u.ID,
Username: u.Username,
Status: u.Status,
CreatedAt: u.CreatedAt.Format(time.RFC3339),
2017-05-20 20:53:05 +02:00
LikingCount: len(u.Likings),
LikedCount: len(u.Liked),
2017-05-10 21:09:37 +02:00
}
return json
}