safeImport/src/index.mjs

import assert from 'node:assert';
import { getASTAndScope } from './ast/analysis.mjs';

import { getRequireCallsAndConstantArgs } from './calls.mjs';
import { readFileSync ,realpathSync ,mkdirSync} from 'node:fs';
import { writeFile } from 'node:fs/promises';
import tsc, { Project, SyntaxKind } from 'ts-morph';
import {getSliceAndInfoSync} from 'slice-js/src/slice-code/test/helpers/utils.js';
import path, { dirname,join } from 'node:path';
// import tsc from 'typescript'
/**
 * Call parameter generation
 */
function main() {
    const FILE_PATH = './test_src/index.cjs';
    const { scopeManager, _parsedModAST } = getASTAndScope(FILE_PATH);
    assert(scopeManager.scopes.length >= 2, "expected atleast global and module scope");
    assert(scopeManager.scopes[1].type === 'function', "expected the 'module' scope to have function scope");

    const calls = getRequireCallsAndConstantArgs(scopeManager);

    logCallList(calls);

    const writePromises = [];
    for (const [moduleName, callBox] of calls) {
        if (!isRelativeModule(moduleName)) { // not relative module
            continue;
        }

        const relatedModuleNamePath = join(realpathSync(dirname(FILE_PATH)) ,moduleName);
        const fileSource = readFileSync(relatedModuleNamePath).toString('utf-8');
        const {slicedCode} = getSliceAndInfoSync(fileSource, (moduleExports) => {
            return [...callBox.entries()].flatMap(([methodName, methodArgsList])=>{
                const methodNameNormed = methodName.substring(1);
                console.log("Calls for ",methodNameNormed,methodArgsList)
                return methodArgsList.map(methodArgsList=>{
                    const methodObj = methodNameNormed===''?moduleExports:moduleExports[methodNameNormed];
                    methodObj.apply(moduleExports[methodNameNormed],methodArgsList)
                });
            })
        },relatedModuleNamePath);
        // console.log(`Sliced code ${moduleName}\n`,slicedCode);
        const writePath = path.resolve('./dist', moduleName);
        if(writePath===moduleName){
            throw Error("Will overwrite!!!!");
        }
        mkdirSync(path.dirname(writePath),{recursive: true});
        console.log(`Writing to`,writePath);

        writePromises.push(writeFile(writePath,slicedCode));

    }

    Promise.all(writePromises).then(p=>{
        console.log("write finished")
    }).catch(console.log);
}

class ImportCall{
    /**
     * @type {'import'|'importExpr'|'require'}
     */
    importType;
    /**
     * @type {string}
     */
    importSyntax;
    /**
     * 
     * @param {'import'|'importExpr'|'require'} importType 
     * @param {string} importSyntax 
     */
    constructor(importType, importSyntax){
        this.importSyntax = importSyntax;
        this.importType = importType;
    }
}

function main2() {
    const FILE_PATH = './test_src/index.cjs';

    const project = new Project({compilerOptions:{allowJs: true, checkJs: false,}});
    project.addSourceFileAtPathIfExists(FILE_PATH);

    // const project = tsc.createProgram([FILE_PATH],);
    const checker = project.getTypeChecker();

    const sourceFile = project.getSourceFile(FILE_PATH)
    
    const importDecls = sourceFile.getImportStringLiterals()
    for(const importStringDecl of importDecls){
        console.log(importStringDecl);
        const importDecl = importStringDecl.getFirstAncestor();
        if(importDecl.isKind(SyntaxKind.CallExpression)){
            // the declaration is callExpression. Verify its based an identifier aliasing import or require
            const importExpr = importDecl.getExpression();
            const type = checker.getTypeAtLocation(importExpr);
            console.log("Type of import expression",checker.compilerObject.resolveName());
            console.log(importExpr);
            if(importExpr.isKind(SyntaxKind.Identifier)){
                // import is a require or import
                const importName = importExpr.getText();
                if(importName==='require' || importName==='import'){
                    console.log("Found require/import call",importExpr);
                }
            }
            
        
        }else if(importDecl.isKind(SyntaxKind.ImportDeclaration)){
            // TODO pending extract the calls.
        }else{
            console.error("Unexpected import specifier",SyntaxKind[importDecl]);
        }
        const importThing = importStringDecl.getParent()
    }
    
    console.log(importDecls);
}

if (process.argv[1] === import.meta.filename) {
    console.log("[SafeImport] started");
    main2();
    console.log("done");
}


function logCallList(calls) {
    for (const [moduleName, callBoxes] of calls.entries()) {
        if (isRelativeModule(moduleName)) {
            console.log('Importing', moduleName, callBoxes);
        } else {
            console.log(`Module "${moduleName}" - System module. FIXME skipping`);
        }
    }
    console.log(`Call List`, calls);
}

function isRelativeModule(moduleName) {
    return moduleName.startsWith('.');
}