sharkey/packages/backend/src/server/api/endpoints/admin/drive-capacity-override.ts

import { Inject, Injectable } from '@nestjs/common';
import { Endpoint } from '@/server/api/endpoint-base.js';
import { UsersRepository } from '@/models/index.js';
import { ModerationLogService } from '@/core/ModerationLogService.js';
import { DI } from '@/di-symbols.js';

export const meta = {
	tags: ['admin'],

	requireCredential: true,
	requireModerator: true,
} as const;

export const paramDef = {
	type: 'object',
	properties: {
		userId: { type: 'string', format: 'misskey:id' },
		overrideMb: { type: 'number', nullable: true },
	},
	required: ['userId', 'overrideMb'],
} as const;

// eslint-disable-next-line import/no-default-export
@Injectable()
export default class extends Endpoint<typeof meta, typeof paramDef> {
	constructor(
		@Inject(DI.usersRepository)
		private usersRepository: UsersRepository,

		private moderationLogService: ModerationLogService,
	) {
		super(meta, paramDef, async (ps, me) => {
			const user = await this.usersRepository.findOneBy({ id: ps.userId });

			if (user == null) {
				throw new Error('user not found');
			}

			if (!this.userEntityService.isLocalUser(user)) {
				throw new Error('user is not local user');
			} 

			/*if (user.isAdmin) {
		throw new Error('cannot suspend admin');
	}
	if (user.isModerator) {
		throw new Error('cannot suspend moderator');
	}*/

			await this.usersRepository.update(user.id, {
				driveCapacityOverrideMb: ps.overrideMb,
			});

			this.moderationLogService.insertModerationLog(me, 'change-drive-capacity-override', {
				targetId: user.id,
			});
		});
	}
}