mizzkey/packages/backend/src/core/entities/RoleEntityService.ts

import { Inject, Injectable } from '@nestjs/common';
import { DI } from '@/di-symbols.js';
import type { RoleAssignmentsRepository, RolesRepository } from '@/models/index.js';
import { awaitAll } from '@/misc/prelude/await-all.js';
import type { User } from '@/models/entities/User.js';
import type { Role } from '@/models/entities/Role.js';
import { bindThis } from '@/decorators.js';
import { DEFAULT_POLICIES } from '@/core/RoleService.js';
import { UserEntityService } from './UserEntityService.js';

@Injectable()
export class RoleEntityService {
	constructor(
		@Inject(DI.rolesRepository)
		private rolesRepository: RolesRepository,

		@Inject(DI.roleAssignmentsRepository)
		private roleAssignmentsRepository: RoleAssignmentsRepository,

		private userEntityService: UserEntityService,
	) {
	}

	@bindThis
	public async pack(
		src: Role['id'] | Role,
		me?: { id: User['id'] } | null | undefined,
		options?: {
			detail?: boolean;
		},
	) {
		const opts = Object.assign({
			detail: true,
		}, options);

		const role = typeof src === 'object' ? src : await this.rolesRepository.findOneByOrFail({ id: src });

		const assigns = await this.roleAssignmentsRepository.findBy({
			roleId: role.id,
		});

		const policies = { ...role.policies };
		for (const [k, v] of Object.entries(DEFAULT_POLICIES)) {
			if (policies[k] == null) policies[k] = {
				useDefault: true,
				priority: 0,
				value: v,
			};
		}

		return await awaitAll({
			id: role.id,
			createdAt: role.createdAt.toISOString(),
			updatedAt: role.updatedAt.toISOString(),
			name: role.name,
			description: role.description,
			color: role.color,
			iconUrl: role.iconUrl,
			target: role.target,
			condFormula: role.condFormula,
			isPublic: role.isPublic,
			isAdministrator: role.isAdministrator,
			isModerator: role.isModerator,
			asBadge: role.asBadge,
			canEditMembersByModerator: role.canEditMembersByModerator,
			policies: policies,
			usersCount: assigns.length,
			...(opts.detail ? {
				users: this.userEntityService.packMany(assigns.map(x => x.userId), me),
			} : {}),
		});
	}

	@bindThis
	public packMany(
		roles: any[],
		me: { id: User['id'] },
		options?: {
			detail?: boolean;
		},
	) {
		return Promise.all(roles.map(x => this.pack(x, me, options)));
	}
}