test/nocobase2
1
0
Fork 0
mirror of https://gitee.com/nocobase/nocobase.git synced 2024-12-03 04:38:15 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

feat(file-manager): add upload file api

Browse Source
This commit is contained in:
chenos 2024-08-20 11:24:35 +08:00
parent 51c6c6dcfa
commit 0e5f8b5e12
2 changed files with 37 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
packages/plugins/@nocobase/plugin-file-manager/src/server/__tests__/action.test.ts
View File

@ -97,7 +97,7 @@ describe('action', () => {
expect(model.toJSON()).toMatchObject(matcher);
});
it.only('should be custom values', async () => {
it('should be custom values', async () => {
const Plugin = app.pm.get(PluginFileManagerServer) as PluginFileManagerServer;
const model = await Plugin.createFileRecord({
collectionName: 'attachments',
@ -117,6 +117,22 @@ describe('action', () => {
expect(model.toJSON()).toMatchObject(matcher);
});
it('should be upload file', async () => {
const Plugin = app.pm.get(PluginFileManagerServer) as PluginFileManagerServer;
const data = await Plugin.uploadFile({
filePath: path.resolve(__dirname, './files/text.txt'),
documentRoot: 'storage/backups/test',
});
const matcher = {
title: 'text',
extname: '.txt',
path: '',
meta: {},
storageId: 1,
};
expect(data).toMatchObject(matcher);
});
it('upload file should be ok', async () => {
const { body } = await agent.resource('attachments').create({
[FILE_FIELD_NAME]: path.resolve(__dirname, './files/text.txt'),

25
packages/plugins/@nocobase/plugin-file-manager/src/server/server.ts
View File

@ -36,6 +36,12 @@ export type FileRecordOptions = {
values?: any;
} & Transactionable;
export type UploadFileOptions = {
filePath: string;
storageName?: string;
documentRoot?: string;
};
export default class PluginFileManagerServer extends Plugin {
storageTypes = new Registry<IStorage>();
storagesCache = new Map<number, StorageModel>();
@ -46,15 +52,21 @@ export default class PluginFileManagerServer extends Plugin {
if (!collection) {
throw new Error(`collection does not exist`);
}
const storageRepository = this.db.getRepository('storages');
const collectionRepository = this.db.getRepository(collectionName);
const name = storageName || collection.options.storage;
const data = await this.uploadFile({ storageName: name, filePath });
return await collectionRepository.create({ values: { ...data, ...values }, transaction });
}
async uploadFile(options: UploadFileOptions) {
const { storageName, filePath, documentRoot } = options;
const storageRepository = this.db.getRepository('storages');
let storageInstance;
if (name) {
if (storageName) {
storageInstance = await storageRepository.findOne({
filter: {
name,
name: storageName,
},
});
}
@ -73,6 +85,10 @@ export default class PluginFileManagerServer extends Plugin {
throw new Error('[file-manager] no linked or default storage provided');
}
if (documentRoot) {
storageInstance.options['documentRoot'] = documentRoot;
}
const storageConfig = this.storageTypes.get(storageInstance.type);
if (!storageConfig) {
@ -97,8 +113,7 @@ export default class PluginFileManagerServer extends Plugin {
});
});
const data = getFileData({ app: this.app, file, storage: storageInstance, request: { body: {} } } as any);
return await collectionRepository.create({ values: { ...data, ...values }, transaction });
return getFileData({ app: this.app, file, storage: storageInstance, request: { body: {} } } as any);
}
async loadStorages(options?: { transaction: any }) {