import * as path from "path";
import * as fs from "fs";
import readdirp, { EntryInfo } from "readdirp";

const inputFolder = path.resolve(process.argv[2]);

console.log(`Clearing private definitions in ${inputFolder}`);

function filterTypeScriptFiles(content: string) {
    var leadingPrivate = /^.*@private/ig;
    if (leadingPrivate.test(content)) {
        return { shouldDelete: true };
    }

    let blockCommentPrivate = /\/\*\*([^](?!\*\/))*@module([^](?!\*\/))*@private[^]*?\*\//g;
    if (blockCommentPrivate.test(content)) {
        return { shouldDelete: true };
    }

    let newContent = content;
    newContent = newContent.replace(/\/\/[\/\s]*@private[^]*?\/\/[\/\s]*?@endprivate/gm, "");

    if (newContent !== content) {
        return { shouldReplace: true, newContent: newContent };
    }

    return { shouldReplace: false, shouldDelete: false };
}

readdirp(inputFolder, {
    fileFilter: ["*.d.ts"],
    directoryFilter: function (di) { return !di.path.includes("node_modules"); }
}).on("data", (entry: EntryInfo) => {
    const { fullPath } = entry;
    const content = fs.readFileSync(fullPath, "utf8");
    const { shouldDelete, shouldReplace, newContent } = filterTypeScriptFiles(content);

    if (shouldDelete) {
        console.log("[Delete]", fullPath);
        fs.unlinkSync(fullPath);
    } else if (shouldReplace) {
        console.log("[Cleared]", fullPath);
        try {
            fs.writeFileSync(fullPath, newContent, "utf8");
        } catch (error) {
            console.log("ERROR writing file: " + fullPath, error);
            process.exit(1);
        }
    }
})
    .on("warn", (error: Error) => console.error("non-fatal error", error))
    .on("error", (error: Error) => {
        console.error("fatal error", error);
        process.exit(1);
    })
    .on("end", () => console.log("done"));