144 lines
4 KiB
TypeScript
144 lines
4 KiB
TypeScript
/*
|
|
* SPDX-FileCopyrightText: syuilo and other misskey contributors
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
import ms from 'ms';
|
|
import { Injectable } from '@nestjs/common';
|
|
import { DB_MAX_IMAGE_COMMENT_LENGTH } from '@/const.js';
|
|
import type Logger from '@/logger.js';
|
|
import { IdentifiableError } from '@/misc/identifiable-error.js';
|
|
import { Endpoint } from '@/server/api/endpoint-base.js';
|
|
import { DriveFileEntityService } from '@/core/entities/DriveFileEntityService.js';
|
|
import { MetaService } from '@/core/MetaService.js';
|
|
import { LoggerService } from '@/core/LoggerService.js';
|
|
import { DriveService } from '@/core/DriveService.js';
|
|
import { ApiError } from '../../../error.js';
|
|
|
|
export const meta = {
|
|
tags: ['drive'],
|
|
|
|
requireCredential: true,
|
|
requireRolePolicy: 'canCreateContent',
|
|
|
|
prohibitMoved: true,
|
|
|
|
limit: {
|
|
duration: ms('1hour'),
|
|
max: 120,
|
|
},
|
|
|
|
requireFile: true,
|
|
|
|
kind: 'write:drive',
|
|
|
|
description: 'Upload a new drive file.',
|
|
|
|
res: {
|
|
type: 'object',
|
|
optional: false, nullable: false,
|
|
ref: 'DriveFile',
|
|
},
|
|
|
|
errors: {
|
|
invalidFileName: {
|
|
message: 'Invalid file name.',
|
|
code: 'INVALID_FILE_NAME',
|
|
id: 'f449b209-0c60-4e51-84d5-29486263bfd4',
|
|
},
|
|
|
|
inappropriate: {
|
|
message: 'Cannot upload the file because it has been determined that it possibly contains inappropriate content.',
|
|
code: 'INAPPROPRIATE',
|
|
id: 'bec5bd69-fba3-43c9-b4fb-2894b66ad5d2',
|
|
},
|
|
|
|
noFreeSpace: {
|
|
message: 'Cannot upload the file because you have no free space of drive.',
|
|
code: 'NO_FREE_SPACE',
|
|
id: 'd08dbc37-a6a9-463a-8c47-96c32ab5f064',
|
|
},
|
|
},
|
|
} as const;
|
|
|
|
export const paramDef = {
|
|
type: 'object',
|
|
properties: {
|
|
folderId: { type: 'string', format: 'misskey:id', nullable: true, default: null },
|
|
name: { type: 'string', nullable: true, default: null },
|
|
comment: { type: 'string', nullable: true, maxLength: DB_MAX_IMAGE_COMMENT_LENGTH, default: null },
|
|
isSensitive: { type: 'boolean', default: false },
|
|
force: { type: 'boolean', default: false },
|
|
},
|
|
required: [],
|
|
} as const;
|
|
|
|
@Injectable()
|
|
export default class extends Endpoint<typeof meta, typeof paramDef> { // eslint-disable-line import/no-default-export
|
|
private logger: Logger;
|
|
|
|
constructor(
|
|
private driveFileEntityService: DriveFileEntityService,
|
|
private metaService: MetaService,
|
|
private loggerService: LoggerService,
|
|
private driveService: DriveService,
|
|
) {
|
|
super(meta, paramDef, async (ps, me, _, file, cleanup, ip, headers) => {
|
|
// Get 'name' parameter
|
|
let name = ps.name ?? file!.name ?? null;
|
|
if (name != null) {
|
|
name = name.trim();
|
|
if (name.length === 0) {
|
|
name = null;
|
|
} else if (name === 'blob') {
|
|
name = null;
|
|
} else if (!this.driveFileEntityService.validateFileName(name)) {
|
|
throw new ApiError(meta.errors.invalidFileName);
|
|
}
|
|
}
|
|
|
|
const instance = await this.metaService.fetch();
|
|
|
|
try {
|
|
// Create file
|
|
const driveFile = await this.driveService.addFile({
|
|
user: me,
|
|
path: file!.path,
|
|
name,
|
|
comment: ps.comment,
|
|
folderId: ps.folderId,
|
|
force: ps.force,
|
|
sensitive: ps.isSensitive,
|
|
requestIp: instance.enableIpLogging ? ip : null,
|
|
requestHeaders: instance.enableIpLogging ? headers : null,
|
|
});
|
|
return await this.driveFileEntityService.pack(driveFile, me, { self: true });
|
|
} catch (e) {
|
|
this.logger.error('Failed to create drive file', { error: e });
|
|
if (e instanceof IdentifiableError) {
|
|
if (e.id === '282f77bf-5816-4f72-9264-aa14d8261a21') throw new ApiError(meta.errors.inappropriate);
|
|
if (e.id === 'c6244ed2-a39a-4e1c-bf93-f0fbd7764fa6') throw new ApiError(meta.errors.noFreeSpace);
|
|
}
|
|
const err = e as Error;
|
|
throw new ApiError(
|
|
{
|
|
message: 'Failed to create drive file.',
|
|
code: 'FAILED_TO_CREATE_DRIVE_FILE',
|
|
id: '6708863c-6791-4487-aa01-2d682c6e7db0',
|
|
},
|
|
{
|
|
e: {
|
|
message: err.message,
|
|
code: err.name,
|
|
}
|
|
}
|
|
);
|
|
} finally {
|
|
cleanup!();
|
|
}
|
|
});
|
|
|
|
this.logger = this.loggerService.getLogger('api:drive:files:create');
|
|
}
|
|
}
|