synctv/internal/op/users.go

package op

import (
	"time"

	"github.com/synctv-org/synctv/internal/db"
	"github.com/synctv-org/synctv/internal/model"
	"github.com/zijiren233/stream"
	"golang.org/x/crypto/bcrypt"
	"gorm.io/gorm/clause"
)

func GetUserById(id uint) (*User, error) {
	i, err := userCache.Get(id)
	if err == nil {
		return i.(*User), nil
	}

	u, err := db.GetUserByID(id)
	if err != nil {
		return nil, err
	}

	u2 := &User{
		User:    *u,
		version: 1,
	}

	return u2, userCache.SetWithExpire(id, u2, time.Hour)
}

// slow
func GetUserByUsername(username string) (*User, error) {
	u, err := db.GetUserByUsername(username)
	if err != nil {
		return nil, err
	}

	u2 := &User{
		User:    *u,
		version: 1,
	}

	return u2, userCache.SetWithExpire(u.ID, u2, time.Hour)
}

var ErrInvalidUsernameOrPassword = bcrypt.ErrMismatchedHashAndPassword

func CreateUser(username, password string) (*User, error) {
	if username == "" || password == "" {
		return nil, ErrInvalidUsernameOrPassword
	}
	hashedPassword, err := bcrypt.GenerateFromPassword(stream.StringToBytes(password), bcrypt.DefaultCost)
	if err != nil {
		return nil, err
	}
	u, err := db.CreateUser(username, hashedPassword)
	if err != nil {
		return nil, err
	}

	u2 := &User{
		User:    *u,
		version: 1,
	}

	return u2, userCache.SetWithExpire(u.ID, u2, time.Hour)
}

func SetUserPassword(userID uint, password string) error {
	u, err := GetUserById(userID)
	if err != nil {
		return err
	}
	return u.SetPassword(password)
}

func DeleteUserByID(userID uint) error {
	u, err := db.LoadAndDeleteUserByID(userID, clause.Column{
		Name: "id",
	})
	if err != nil {
		return err
	}
	userCache.Remove(u.ID)
	return nil
}

func DeleteUserByUsername(username string) error {
	u, err := db.LoadAndDeleteUserByUsername(username, clause.Column{
		Name: "id",
	})
	if err != nil {
		return err
	}
	userCache.Remove(u.ID)
	return nil
}

func UpdateUser(u *model.User) error {
	userCache.Remove(u.ID)
	return db.UpdateUser(u)
}

func GetUserName(userID uint) string {
	u, err := GetUserById(userID)
	if err != nil {
		return ""
	}
	return u.Username
}