adds context id to role and role helpers

This commit is contained in:
Joseph Ditton 2022-01-07 10:19:34 -07:00
parent 5d9cf51c10
commit de84113066
3 changed files with 59 additions and 0 deletions

View File

@ -0,0 +1,18 @@
import { MigrationInterface, QueryRunner, TableColumn } from 'typeorm';
export class AddContextIdToUserRole1641570023672 implements MigrationInterface {
public async up(queryRunner: QueryRunner): Promise<void> {
await queryRunner.addColumn(
'user_role',
new TableColumn({
name: 'contextId',
type: 'text',
default: "'root'", // default values must include single quotes for text
}),
);
}
public async down(queryRunner: QueryRunner): Promise<void> {
await queryRunner.dropColumn('user_role', 'contextId');
}
}

View File

@ -13,6 +13,9 @@ export class UserRole {
@Column()
userId: number;
@Column()
contextId: string;
@ManyToOne(() => Role, (role) => role.userRoles)
role: Role;

View File

@ -3,12 +3,19 @@ import { InjectRepository } from '@nestjs/typeorm';
import { Repository } from 'typeorm';
import * as bcrypt from 'bcrypt';
import { User } from '../../entities/user.entity';
import { Role, RoleKey } from 'server/entities/role.entity';
import { UserRole } from 'server/entities/user_role.entity';
import { intersection, isEmpty } from 'lodash';
@Injectable()
export class UsersService {
constructor(
@InjectRepository(User)
private usersRespository: Repository<User>,
@InjectRepository(UserRole)
private userRolesRepository: Repository<UserRole>,
@InjectRepository(Role)
private rolesRepository: Repository<Role>,
) {}
findAll(relations: string[] = []) {
@ -33,4 +40,35 @@ export class UsersService {
const verified: boolean = await bcrypt.compare(password, user.passwordHash);
return { verified, user: verified ? user : null };
}
addUserToRoleInContext(userId: number, contextId: string, ...roleKeys: RoleKey[]) {
return Promise.all(
roleKeys.map(async (key) => {
const role = await this.rolesRepository.findOne({ key });
const userRole = new UserRole();
userRole.userId = userId;
userRole.contextId = contextId;
userRole.role = role;
await this.userRolesRepository.save(userRole);
}),
);
}
addUserToRootRole(userId: number, ...roleKeys: RoleKey[]) {
return this.addUserToRoleInContext(userId, 'root', ...roleKeys);
}
// if multiple roles are passed then will return true if user has any of the listed roles.
async hasRoleInContext(userId: number, contextId: string, ...roleKeys: RoleKey[]) {
const userRoles = await this.userRolesRepository.find({
where: { userId, contextId },
relations: ['role'],
});
const usersRoleKeys = userRoles.map((userRole) => userRole.role.key);
return !isEmpty(intersection(roleKeys, usersRoleKeys));
}
async hasRootRole(userId: number, ...roleKeys: RoleKey[]) {
return this.hasRoleInContext(userId, 'root', ...roleKeys);
}
}