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

import path from 'path';
import { EOL } from 'os';
import { readFileSync } from 'fs';
import { parser, traverse, winPath } from '@fesjs/utils';

const getFileName = (name) => {
    const fileName = path.basename(name, path.extname(name));
    if (fileName.endsWith('.model') || fileName.endsWith('.models')) {
        return fileName.split('.').slice(0, -1).join('.');
    }
    return fileName;
};

export const getName = (absPath, absSrcPath) => {
    const relativePath = path.relative(absSrcPath, absPath);
    // model files with namespace
    const dirList = path.dirname(relativePath).split(path.sep);
    try {
        const validDirs = dirList.filter(
            ele => !['src', 'page', 'pages', 'model', 'models'].includes(ele)
        );
        if (validDirs && validDirs.length) {
            return `${validDirs.join('.')}.${getFileName(relativePath)}`;
        }
        return getFileName(relativePath);
    } catch (e) {
        return getFileName(relativePath);
    }
};

export const getPath = (absPath) => {
    const info = path.parse(absPath);
    return winPath(path.join(info.dir, info.name).replace(/'/, "'"));
};

export const genImports = imports => imports
    .map(
        (ele, index) => `import model${index} from "${winPath(getPath(ele))}";`
    )
    .join(EOL);

export const genExtraModels = (models = [], absSrcPath) => models.map((ele) => {
    if (typeof ele === 'string') {
        return {
            importPath: getPath(ele),
            importName: path.basename(ele).split('.')[0],
            namespace: getName(ele, absSrcPath)
        };
    }
    return {
        importPath: getPath(ele.absPath),
        importName: path.basename(ele.absPath).split('.')[0],
        namespace: ele.namespace,
        exportName: ele.exportName
    };
});

export const sort = (ns) => {
    let final = [];
    ns.forEach((item, index) => {
        if (item.use && item.use.length) {
            const itemGroup = [...item.use, item.namespace];

            const cannotUse = [item.namespace];
            for (let i = 0; i <= index; i += 1) {
                if (ns[i].use.filter(v => cannotUse.includes(v)).length) {
                    if (!cannotUse.includes(ns[i].namespace)) {
                        cannotUse.push(ns[i].namespace);
                        i = -1;
                    }
                }
            }

            const errorList = item.use.filter(v => cannotUse.includes(v));
            if (errorList.length) {
                throw Error(
                    `Circular dependencies: ${
                        item.namespace
                    } can't use ${errorList.join(', ')}`
                );
            }

            const intersection = final.filter(v => itemGroup.includes(v));
            if (intersection.length) {
                // first intersection
                const finalIndex = final.indexOf(intersection[0]);
                // replace with groupItem
                final = final
                    .slice(0, finalIndex)
                    .concat(itemGroup)
                    .concat(final.slice(finalIndex + 1));
            } else {
                final.push(...itemGroup);
            }
        }
        if (!final.includes(item.namespace)) {
            // first occurance append to the end
            final.push(item.namespace);
        }
    });

    return [...new Set(final)];
};

export const genModels = (imports, absSrcPath) => {
    const contents = imports.map(absPath => ({
        namespace: getName(absPath, absSrcPath),
        content: readFileSync(absPath).toString()
    }));
    const allUserModel = imports.map(absPath => getName(absPath, absSrcPath));

    const checkDuplicates = list => new Set(list).size !== list.length;

    const raw = contents.map((ele, index) => {
        const ast = parser.parse(ele.content, {
            sourceType: 'module',
            plugins: ['jsx', 'typescript']
        });

        const use = [];

        traverse(ast, {
            enter(astPath) {
                if (astPath.isIdentifier({ name: 'useModel' })) {
                    try {
                        // string literal
                        const ns = astPath.parentPath.node.arguments[0].value;
                        if (allUserModel.includes(ns)) {
                            use.push(ns);
                        }
                    } catch (e) {
                        // console.log(e)
                    }
                }
            }
        });

        return { namespace: ele.namespace, use, importName: `model${index}` };
    });

    const models = sort(raw);

    if (checkDuplicates(contents.map(ele => ele.namespace))) {
        throw Error('plugin-model: models 中包含重复的 namespace！');
    }
    return raw.sort(
        (a, b) => models.indexOf(a.namespace) - models.indexOf(b.namespace)
    );
};

export const isValidHook = (filePath) => {
    const content = readFileSync(filePath, { encoding: 'utf-8' }).toString();

    const ast = parser.parse(content, {
        sourceType: 'module',
        plugins: [
            'classProperties',
            'dynamicImport',
            'exportDefaultFrom',
            'exportNamespaceFrom',
            'functionBind',
            'nullishCoalescingOperator',
            'objectRestSpread',
            'optionalChaining',
            'decorators-legacy'
        ].filter(Boolean)
    });
    let valid = false;
    let identifierName = '';
    traverse(ast, {
        enter(p) {
            if (p.isExportDefaultDeclaration()) {
                const { type } = p.node.declaration;
                try {
                    if (
                        type === 'ArrowFunctionExpression'
                        || type === 'FunctionDeclaration'
                    ) {
                        valid = true;
                    } else if (type === 'Identifier') {
                        identifierName = p.node.declaration.name;
                    }
                } catch (e) {
                    console.error(e);
                }
            }
        }
    });

    try {
        if (identifierName) {
            ast.program.body.forEach((ele) => {
                if (ele.type === 'FunctionDeclaration') {
                    if (ele.id?.name === identifierName) {
                        valid = true;
                    }
                }
                if (ele.type === 'VariableDeclaration') {
                    if (
                        ele.declarations[0].id.name === identifierName
                        && ele.declarations[0].init.type
                            === 'ArrowFunctionExpression'
                    ) {
                        valid = true;
                    }
                }
            });
        }
    } catch (e) {
        valid = false;
    }

    return valid;
};

export const getValidFiles = (files, modelsDir) => files
    .map((file) => {
        const filePath = path.join(modelsDir, file);
        const valid = isValidHook(filePath);
        if (valid) {
            return filePath;
        }
        return '';
    })
    .filter(ele => !!ele);