fes.js/packages/fes-plugin-model/src/index.js

import { readFileSync } from 'fs';
import { join } from 'path';

const namespace = 'plugin-model';

export default (api) => {
    const {
        paths,
        utils: { Mustache }
    } = api;

    const { lodash, winPath } = require('@fesjs/utils');
    const { getModels } = require('./utils/getModels');
    const { getTmpFile } = require('./utils/getTmpFile');

    function getModelDir() {
        return api.config.singular ? 'model' : 'models';
    }

    function getModelsPath() {
        return join(paths.absSrcPath, getModelDir());
    }

    function getAllModels() {
        const srcModelsPath = getModelsPath();
        return lodash.uniq([
            ...getModels(srcModelsPath)
        ]);
    }

    const absCoreFilePath = join(namespace, 'core.js');
    const absInitialStateFilePath = join(namespace, 'models/initialState.js');

    api.register({
        key: 'addExtraModels',
        fn: () => [{
            absPath: winPath(join(paths.absTmpPath, absInitialStateFilePath)),
            namespace: '@@initialState'
        }]
    });

    api.onGenerateFiles(async () => {
        const files = getAllModels();

        const additionalModels = await api.applyPlugins({
            key: 'addExtraModels',
            type: api.ApplyPluginsType.add,
            initialValue: []
        });

        const tmpFiles = getTmpFile(files, additionalModels, paths.absSrcPath);

        api.writeTmpFile({
            path: absCoreFilePath,
            content: Mustache.render(readFileSync(join(__dirname, 'runtime/core.tpl'), 'utf-8'), {
                ...tmpFiles
            })
        });

        api.copyTmpFiles({
            namespace,
            path: join(__dirname, 'runtime'),
            ignore: ['.tpl']
        });
    });

    api.addPluginExports(() => [
        {
            specifiers: ['useModel'],
            source: absCoreFilePath
        }
    ]);
};