n8n/packages/cli/src/databases/repositories/role.repository.ts

import { Service } from 'typedi';
import { DataSource, In, Repository } from 'typeorm';
import type { RoleNames, RoleScopes } from '../entities/Role';
import { Role } from '../entities/Role';
import { User } from '../entities/User';

@Service()
export class RoleRepository extends Repository<Role> {
	constructor(dataSource: DataSource) {
		super(Role, dataSource.manager);
	}

	async findRole(scope: RoleScopes, name: RoleNames) {
		return this.findOne({ where: { scope, name } });
	}

	/**
	 * Counts the number of users in each role, e.g. `{ admin: 2, member: 6, owner: 1 }`
	 */
	async countUsersByRole() {
		type Row = { role_name: string; count: number | string };

		const rows: Row[] = await this.createQueryBuilder('role')
			.select('role.name')
			.addSelect('COUNT(user.id)', 'count')
			.innerJoin(User, 'user', 'role.id = user.globalRoleId')
			.groupBy('role.name')
			.getRawMany();

		return rows.reduce<Record<string, number>>((acc, item) => {
			acc[item.role_name] = typeof item.count === 'number' ? item.count : parseInt(item.count, 10);
			return acc;
		}, {});
	}

	async getIdsInScopeWorkflowByNames(roleNames: RoleNames[]) {
		return this.find({
			select: ['id'],
			where: { name: In(roleNames), scope: 'workflow' },
		}).then((role) => role.map(({ id }) => id));
	}
}