124 lines
3.5 KiB
TypeScript
124 lines
3.5 KiB
TypeScript
import ms from 'ms';
|
|
import { Inject, Injectable } from '@nestjs/common';
|
|
import type { DriveFilesRepository } from '@/models/index.js';
|
|
import { DB_MAX_IMAGE_COMMENT_LENGTH } from '@/misc/hard-limits.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 { DriveService } from '@/core/DriveService.js';
|
|
import { DI } from '@/di-symbols.js';
|
|
import { ApiError } from '../../../error.js';
|
|
|
|
export const meta = {
|
|
tags: ['drive'],
|
|
|
|
requireCredential: 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;
|
|
|
|
// eslint-disable-next-line import/no-default-export
|
|
@Injectable()
|
|
export default class extends Endpoint<typeof meta, typeof paramDef> {
|
|
constructor(
|
|
@Inject(DI.driveFilesRepository)
|
|
private driveFilesRepository: DriveFilesRepository,
|
|
|
|
private driveFileEntityService: DriveFileEntityService,
|
|
private metaService: MetaService,
|
|
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 meta = 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: meta.enableIpLogging ? ip : null,
|
|
requestHeaders: meta.enableIpLogging ? headers : null,
|
|
});
|
|
return await this.driveFileEntityService.pack(driveFile, { self: true });
|
|
} catch (err) {
|
|
if (err instanceof Error || typeof err === 'string') {
|
|
console.error(err);
|
|
}
|
|
if (err instanceof IdentifiableError) {
|
|
if (err.id === '282f77bf-5816-4f72-9264-aa14d8261a21') throw new ApiError(meta.errors.inappropriate);
|
|
if (err.id === 'c6244ed2-a39a-4e1c-bf93-f0fbd7764fa6') throw new ApiError(meta.errors.noFreeSpace);
|
|
}
|
|
throw new ApiError();
|
|
} finally {
|
|
cleanup!();
|
|
}
|
|
});
|
|
}
|
|
}
|