82 lines
2.6 KiB
TypeScript
82 lines
2.6 KiB
TypeScript
import { Inject, Injectable } from '@nestjs/common';
|
|
import type { DriveFilesRepository, DriveFoldersRepository } from '@/models/index.js';
|
|
import { Endpoint } from '@/server/api/endpoint-base.js';
|
|
import { DriveFileEntityService } from '@/core/entities/DriveFileEntityService.js';
|
|
import { GlobalEventService } from '@/core/GlobalEventService.js';
|
|
import { DI } from '@/di-symbols.js';
|
|
import { RoleService } from '@/core/RoleService.js';
|
|
import { ApiError } from '../../../error.js';
|
|
|
|
// eslint-disable-next-line import/no-default-export
|
|
@Injectable()
|
|
export default class extends Endpoint<'drive/files/update'> {
|
|
name = 'drive/files/update' as const;
|
|
constructor(
|
|
@Inject(DI.driveFilesRepository)
|
|
private driveFilesRepository: DriveFilesRepository,
|
|
|
|
@Inject(DI.driveFoldersRepository)
|
|
private driveFoldersRepository: DriveFoldersRepository,
|
|
|
|
private driveFileEntityService: DriveFileEntityService,
|
|
private roleService: RoleService,
|
|
private globalEventService: GlobalEventService,
|
|
) {
|
|
super(async (ps, me) => {
|
|
const file = await this.driveFilesRepository.findOneBy({ id: ps.fileId });
|
|
const alwaysMarkNsfw = (await this.roleService.getUserPolicies(me.id)).alwaysMarkNsfw;
|
|
if (file == null) {
|
|
throw new ApiError(this.meta.errors.noSuchFile);
|
|
}
|
|
|
|
if (!await this.roleService.isModerator(me) && (file.userId !== me.id)) {
|
|
throw new ApiError(this.meta.errors.accessDenied);
|
|
}
|
|
|
|
if (ps.name) file.name = ps.name;
|
|
if (!this.driveFileEntityService.validateFileName(file.name)) {
|
|
throw new ApiError(this.meta.errors.invalidFileName);
|
|
}
|
|
|
|
if (ps.comment !== undefined) file.comment = ps.comment;
|
|
|
|
if (ps.isSensitive !== undefined && ps.isSensitive !== file.isSensitive && alwaysMarkNsfw && !ps.isSensitive) {
|
|
throw new ApiError(this.meta.errors.restrictedByRole);
|
|
}
|
|
|
|
if (ps.isSensitive !== undefined) file.isSensitive = ps.isSensitive;
|
|
|
|
if (ps.folderId !== undefined) {
|
|
if (ps.folderId === null) {
|
|
file.folderId = null;
|
|
} else {
|
|
const folder = await this.driveFoldersRepository.findOneBy({
|
|
id: ps.folderId,
|
|
userId: me.id,
|
|
});
|
|
|
|
if (folder == null) {
|
|
throw new ApiError(this.meta.errors.noSuchFolder);
|
|
}
|
|
|
|
file.folderId = folder.id;
|
|
}
|
|
}
|
|
|
|
await this.driveFilesRepository.update(file.id, {
|
|
name: file.name,
|
|
comment: file.comment,
|
|
folderId: file.folderId,
|
|
isSensitive: file.isSensitive,
|
|
});
|
|
|
|
const fileObj = await this.driveFileEntityService.pack(file, { self: true });
|
|
|
|
// Publish fileUpdated event
|
|
this.globalEventService.publishDriveStream(me.id, 'fileUpdated', fileObj);
|
|
|
|
return fileObj;
|
|
});
|
|
}
|
|
}
|