synctv/internal/op/user.go

package op

import (
	"errors"
	"sync/atomic"

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

type User struct {
	model.User
	version uint32
}

func (u *User) Version() uint32 {
	return atomic.LoadUint32(&u.version)
}

func (u *User) CheckVersion(version uint32) bool {
	return atomic.LoadUint32(&u.version) == version
}

func (u *User) CreateRoom(name, password string, conf ...db.CreateRoomConfig) (*model.Room, error) {
	return db.CreateRoom(name, password, append(conf, db.WithCreaterID(u.ID), db.WithRelations(
		[]model.RoomUserRelation{
			{
				UserID: u.ID,
				Role:   model.RoleCreater,
			},
		},
	))...)
}

func (u *User) NewMovie(movie model.BaseMovieInfo) model.MovieInfo {
	return model.MovieInfo{
		BaseMovieInfo: movie,
		CreatorID:     u.ID,
	}
}

func (u *User) HasPermission(room *Room, permission model.Permission) bool {
	return room.HasPermission(&u.User, permission)
}

func (u *User) DeleteRoom(room *Room) error {
	if !u.HasPermission(room, model.CanDeleteRoom) {
		return errors.New("no permission")
	}
	return DeleteRoom(room)
}

func (u *User) NeedPassword() bool {
	return len(u.HashedPassword) != 0
}

func (u *User) SetPassword(password string) error {
	if u.CheckPassword(password) && u.NeedPassword() {
		return errors.New("password is the same")
	}
	var hashedPassword []byte
	if password != "" {
		var err error
		hashedPassword, err = bcrypt.GenerateFromPassword(stream.StringToBytes(password), bcrypt.DefaultCost)
		if err != nil {
			return err
		}
	}
	u.HashedPassword = hashedPassword
	atomic.AddUint32(&u.version, 1)
	return db.SetUserPassword(u.ID, hashedPassword)
}