aim-oscar-server/models/User.go

package models

import (
	"context"
	"database/sql"
	"time"

	"github.com/pkg/errors"
	"github.com/uptrace/bun"
)

type User struct {
	bun.BaseModel  `bun:"table:users"`
	UIN            int    `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         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 int) (*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
}