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;