2022-09-17 20:27:08 +02:00
|
|
|
import { Inject, Injectable } from '@nestjs/common';
|
2021-11-12 11:47:04 +01:00
|
|
|
import ms from 'ms';
|
2022-09-17 20:27:08 +02:00
|
|
|
import { Endpoint } from '@/server/api/endpoint-base.js';
|
|
|
|
import { QueueService } from '@/core/QueueService.js';
|
|
|
|
import { DriveFilesRepository } from '@/models/index.js';
|
|
|
|
import { DI } from '@/di-symbols.js';
|
2022-02-27 03:07:39 +01:00
|
|
|
import { ApiError } from '../../error.js';
|
2021-10-19 18:11:13 +02:00
|
|
|
|
|
|
|
export const meta = {
|
|
|
|
secure: true,
|
2022-01-18 14:27:10 +01:00
|
|
|
requireCredential: true,
|
2021-10-19 18:11:13 +02:00
|
|
|
|
|
|
|
limit: {
|
|
|
|
duration: ms('1hour'),
|
|
|
|
max: 1,
|
|
|
|
},
|
|
|
|
|
|
|
|
errors: {
|
|
|
|
noSuchFile: {
|
|
|
|
message: 'No such file.',
|
|
|
|
code: 'NO_SUCH_FILE',
|
2021-12-09 15:58:30 +01:00
|
|
|
id: 'ebb53e5f-6574-9c0c-0b92-7ca6def56d7e',
|
2021-10-19 18:11:13 +02:00
|
|
|
},
|
|
|
|
|
|
|
|
unexpectedFileType: {
|
|
|
|
message: 'We need csv file.',
|
|
|
|
code: 'UNEXPECTED_FILE_TYPE',
|
2021-12-09 15:58:30 +01:00
|
|
|
id: 'b6fab7d6-d945-d67c-dfdb-32da1cd12cfe',
|
2021-10-19 18:11:13 +02:00
|
|
|
},
|
|
|
|
|
|
|
|
tooBigFile: {
|
|
|
|
message: 'That file is too big.',
|
|
|
|
code: 'TOO_BIG_FILE',
|
2021-12-09 15:58:30 +01:00
|
|
|
id: 'b7fbf0b1-aeef-3b21-29ef-fadd4cb72ccf',
|
2021-10-19 18:11:13 +02:00
|
|
|
},
|
|
|
|
|
|
|
|
emptyFile: {
|
|
|
|
message: 'That file is empty.',
|
|
|
|
code: 'EMPTY_FILE',
|
2021-12-09 15:58:30 +01:00
|
|
|
id: '6f3a4dcc-f060-a707-4950-806fbdbe60d6',
|
2021-10-19 18:11:13 +02:00
|
|
|
},
|
2021-12-09 15:58:30 +01:00
|
|
|
},
|
2022-01-18 14:27:10 +01:00
|
|
|
} as const;
|
2021-10-19 18:11:13 +02:00
|
|
|
|
2022-02-20 05:15:40 +01:00
|
|
|
export const paramDef = {
|
2022-02-19 06:05:32 +01:00
|
|
|
type: 'object',
|
|
|
|
properties: {
|
|
|
|
fileId: { type: 'string', format: 'misskey:id' },
|
|
|
|
},
|
|
|
|
required: ['fileId'],
|
|
|
|
} as const;
|
|
|
|
|
2022-01-02 18:12:50 +01:00
|
|
|
// eslint-disable-next-line import/no-default-export
|
2022-09-17 20:27:08 +02:00
|
|
|
@Injectable()
|
|
|
|
export default class extends Endpoint<typeof meta, typeof paramDef> {
|
|
|
|
constructor(
|
|
|
|
@Inject(DI.driveFilesRepository)
|
|
|
|
private driveFilesRepository: DriveFilesRepository,
|
2021-10-19 18:11:13 +02:00
|
|
|
|
2022-09-17 20:27:08 +02:00
|
|
|
private queueService: QueueService,
|
|
|
|
) {
|
|
|
|
super(meta, paramDef, async (ps, me) => {
|
|
|
|
const file = await this.driveFilesRepository.findOneBy({ id: ps.fileId });
|
2021-10-19 18:11:13 +02:00
|
|
|
|
2022-09-17 20:27:08 +02:00
|
|
|
if (file == null) throw new ApiError(meta.errors.noSuchFile);
|
|
|
|
//if (!file.type.endsWith('/csv')) throw new ApiError(meta.errors.unexpectedFileType);
|
|
|
|
if (file.size > 50000) throw new ApiError(meta.errors.tooBigFile);
|
|
|
|
if (file.size === 0) throw new ApiError(meta.errors.emptyFile);
|
|
|
|
|
|
|
|
this.queueService.createImportBlockingJob(me, file.id);
|
|
|
|
});
|
|
|
|
}
|
|
|
|
}
|