45 lines
1.4 KiB
TypeScript
45 lines
1.4 KiB
TypeScript
import path from 'path';
|
|
import Database from '@nocobase/database';
|
|
import Resourcer from '@nocobase/resourcer';
|
|
import { PluginOptions } from '@nocobase/server';
|
|
|
|
import { action as uploadAction, middleware as uploadMiddleware } from './actions/upload';
|
|
import { getStorageConfig } from './storages';
|
|
import { STORAGE_TYPE_LOCAL } from './constants';
|
|
|
|
export default {
|
|
name: 'file-manager',
|
|
async load() {
|
|
const database: Database = this.app.db;
|
|
const resourcer: Resourcer = this.app.resourcer;
|
|
|
|
await database.import({
|
|
directory: path.resolve(__dirname, 'collections'),
|
|
});
|
|
|
|
// 暂时中间件只能通过 use 加进来
|
|
resourcer.use(uploadMiddleware);
|
|
resourcer.registerActionHandler('upload', uploadAction);
|
|
|
|
const { DEFAULT_STORAGE_TYPE } = process.env;
|
|
|
|
if (process.env.NOCOBASE_ENV !== 'production') {
|
|
await getStorageConfig(STORAGE_TYPE_LOCAL).middleware(this.app);
|
|
}
|
|
|
|
this.app.on('db.init', async () => {
|
|
const defaultStorageConfig = getStorageConfig(DEFAULT_STORAGE_TYPE);
|
|
if (defaultStorageConfig) {
|
|
const Storage = database.getCollection('storages');
|
|
await Storage.repository.create({
|
|
values: {
|
|
...defaultStorageConfig.defaults(),
|
|
type: DEFAULT_STORAGE_TYPE,
|
|
default: true,
|
|
},
|
|
});
|
|
}
|
|
});
|
|
},
|
|
} as PluginOptions;
|