aim-oscar-server/models/User.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"`
	Username            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"`
	Status              UserStatus
	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 UserByUsername(ctx context.Context, db *bun.DB, username string) (*User, error) {
	user := new(User)
	if err := db.NewSelect().Model(user).Where("username = ?", username).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
}