import fs from "fs-extra";
import path from "path";

function main() {
    if (process.argv.length !== 3) {
        console.error("More than one path was supplied as argument. Aborting.");
        process.exit(1);
    }

    const basePath = process.argv[2];

    if (!fs.existsSync(basePath)) {
        console.error(`Supplied path '${basePath}' does not exist. Aborting.`)
        process.exit(1);
    }

    cleanupNodeModules(basePath);
}

function cleanupNodeModules(basePath: string) {
    const nodeModulesDirPath = path.join(basePath, "node_modules");
    const nodeModulesContent = fs.readdirSync(nodeModulesDirPath, { recursive: true, withFileTypes: true });
    //const libDir = fs.readdirSync(path.join(basePath, "./libraries"), { recursive: true, withFileTypes: true });

    /**
     * Delete unnecessary folders
     */
    const filterableDirs = new Set([
        "demo",
        "demos",
        "doc",
        "docs",
        "example",
        "examples",
        "test",
        "tests"
    ]);

    nodeModulesContent
        .filter(el => el.isDirectory() && filterableDirs.has(el.name))
        .forEach(dir => fs.removeSync(path.join(dir.parentPath, dir.name)));

    /**
     * Delete unnecessary files based on file extension
     * TODO filter out useless (README).md files
     */
    const filterableFileExt = new Set([
        "ts",
        "map"
    ]);

    nodeModulesContent
        // TriliumNextTODO: check if we can improve this naive file ext matching, without introducing any additional dependency
        .filter(el => el.isFile() && filterableFileExt.has(el.name.split(".").at(-1) || ""))
        .forEach(file => fs.removeSync(path.join(file.parentPath, file.name)));


    /**
     * Delete specific unnecessary folders
     * TODO: check if we want removeSync to throw an error, if path does not exist anymore -> currently it will silently fail
     */
    const extraFoldersDelete = new Set([
        path.join(nodeModulesDirPath, "@excalidraw", "excalidraw", "dist", "dev"),
        path.join(nodeModulesDirPath, "boxicons", "svg"),
        path.join(nodeModulesDirPath, "boxicons", "node_modules"),
        path.join(nodeModulesDirPath, "boxicons", "src"),
        path.join(nodeModulesDirPath, "boxicons", "iconjar"),
        path.join(nodeModulesDirPath, "@jimp", "plugin-print", "fonts"),
        path.join(nodeModulesDirPath, "jimp", "dist", "browser") // missing "@" in front of jimp is not a typo here
    ]);

    nodeModulesContent
        .filter(el => el.isDirectory() && extraFoldersDelete.has(path.join(el.parentPath, el.name)))
        .forEach(dir => fs.removeSync(path.join(dir.parentPath, dir.name)))

}

main()