mirror of
https://github.com/amigan/aim-oscar-server.git
synced 2024-11-22 04:29:47 -05:00
92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
package models
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/uptrace/bun"
|
|
)
|
|
|
|
type UserStatus int16
|
|
|
|
const (
|
|
UserStatusOnline = 0
|
|
UserStatusAway = 1
|
|
UserStatusDnd = 2
|
|
UserStatusNA = 4
|
|
UserStatusOccupied = 0x10
|
|
UserStatusFree4Chat = 0x20
|
|
UserStatusInvisible = 0x100
|
|
)
|
|
|
|
type User struct {
|
|
bun.BaseModel `bun:"table:users"`
|
|
UIN int64 `bun:",pk,autoincrement"`
|
|
Email string `bun:",unique"`
|
|
ScreenName string `bun:",unique"`
|
|
Password string
|
|
Cipher string
|
|
CreatedAt time.Time `bun:",nullzero,notnull,default:current_timestamp"`
|
|
UpdatedAt time.Time `bun:",nullzero,notnull,default:current_timestamp"`
|
|
DeletedAt *time.Time `bun:",nullzero"`
|
|
Status UserStatus
|
|
Verified bool `bun:",notnull,default:false"`
|
|
Profile string
|
|
ProfileEncoding string
|
|
AwayMessage string
|
|
AwayMessageEncoding string
|
|
LastActivityAt time.Time `bin:"-"`
|
|
}
|
|
|
|
type userKey string
|
|
|
|
func (s userKey) String() string {
|
|
return "user-" + string(s)
|
|
}
|
|
|
|
var (
|
|
currentUser = userKey("user")
|
|
)
|
|
|
|
func UserByScreenName(ctx context.Context, db *bun.DB, screen_name string) (*User, error) {
|
|
user := new(User)
|
|
if err := db.NewSelect().Model(user).Where("screen_name = ?", screen_name).Scan(ctx, user); err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, nil
|
|
}
|
|
return nil, errors.Wrap(err, "could not fetch user")
|
|
}
|
|
return user, nil
|
|
}
|
|
|
|
func UserByUIN(ctx context.Context, db *bun.DB, uin int64) (*User, error) {
|
|
user := new(User)
|
|
if err := db.NewSelect().Model(user).Where("uin = ?", uin).Scan(ctx, user); err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, nil
|
|
}
|
|
return nil, errors.Wrap(err, "could not fetch user")
|
|
}
|
|
return user, nil
|
|
}
|
|
|
|
func NewContextWithUser(ctx context.Context, user *User) context.Context {
|
|
return context.WithValue(ctx, currentUser, user)
|
|
}
|
|
|
|
func UserFromContext(ctx context.Context) *User {
|
|
v := ctx.Value(currentUser)
|
|
if v == nil {
|
|
return nil
|
|
}
|
|
return v.(*User)
|
|
}
|
|
|
|
func (u *User) Update(ctx context.Context, db *bun.DB) error {
|
|
if _, err := db.NewUpdate().Model(u).WherePK("uin").Exec(ctx); err != nil {
|
|
return errors.Wrap(err, "could not update user")
|
|
}
|
|
return nil
|
|
}
|