vaulterm/server/models/user.go

package models

import "slices"

const (
	UserRoleUser  = "user"
	UserRoleAdmin = "admin"
)

type User struct {
	Model

	Name     string `json:"name"`
	Username string `json:"username" gorm:"unique"`
	Password string `json:"-"`
	Email    string `json:"email" gorm:"unique"`
	Role     string `json:"role" gorm:"default:user;not null;index:users_role_idx;type:varchar(8)"`

	Teams []*TeamMembers `json:"teams" gorm:"foreignKey:UserID"`

	Timestamps
	SoftDeletes
}

type UserSession struct {
	ID     string `json:"id" gorm:"primarykey;type:varchar(40)"`
	UserID string `json:"userId" gorm:"type:varchar(26)"`
	User   User   `json:"user"`

	Timestamps
	SoftDeletes
}

func (u *User) IsAdmin() bool {
	return u.Role == UserRoleAdmin
}

func (u *User) GetTeamRole(teamId *string) string {
	if u.IsAdmin() {
		return TeamRoleAdmin
	}
	if teamId == nil {
		return ""
	}
	idx := slices.IndexFunc(u.Teams, func(tm *TeamMembers) bool {
		return tm.TeamID == *teamId
	})
	if idx == -1 {
		return ""
	}
	return u.Teams[idx].Role
}

func (u *User) IsInTeam(teamId *string) bool {
	role := u.GetTeamRole(teamId)
	return role != ""
}

func (u *User) TeamCanWrite(teamId *string) bool {
	role := u.GetTeamRole(teamId)
	return role == TeamRoleAdmin || role == TeamRoleOwner
}