patchright_nodejs_rebranding.js

import { Project, SyntaxKind } from "ts-morph"; // Import SyntaxKind from ts-morph
import * as fs from "fs";
import * as path from "path";

// Function to recursively find all TypeScript and JavaScript files
function getAllJsTsFiles(dir){
    let results = [];
    const list = fs.readdirSync(dir);
    list.forEach((file) => {
        const filePath = path.join(dir, file);
        const stat = fs.statSync(filePath);
        if (stat && stat.isDirectory()) {
            results = results.concat(getAllJsTsFiles(filePath));
        } else if (filePath.endsWith(".ts") || filePath.endsWith(".js") || filePath.endsWith(".mjs")) {
            results.push(filePath);
        }
    });
    return results;
}

// Main function to rename imports and exports from "playwright-core" to "patchright-core"
function renameImportsAndExportsInDirectory(directoryPath) {
    const project = new Project();

    // Get all TypeScript and JavaScript files recursively
    const jsTsFiles = getAllJsTsFiles(directoryPath);

    // Iterate over each file
    jsTsFiles.forEach((filePath) => {
        const sourceFile = project.addSourceFileAtPath(filePath);
        let modified = false;

        // Find and modify import declarations
        sourceFile.getImportDeclarations().forEach((importDecl) => {
            const moduleSpecifierValue = importDecl.getModuleSpecifierValue();

            // If the import path starts with "playwright-core", replace it with "patchright-core"
            if (moduleSpecifierValue.startsWith("playwright-core")) {
                const newModuleSpecifier = moduleSpecifierValue.replace("playwright-core", "patchright-core");
                importDecl.setModuleSpecifier(newModuleSpecifier);
                modified = true;
            } else if (moduleSpecifierValue.includes("playwright-core")) {
                const newModuleSpecifier = moduleSpecifierValue.replace(/playwright-core/g, "patchright-core");
                importDecl.setModuleSpecifier(newModuleSpecifier);
                modified = true;
            }
        });

        // Find and modify export declarations
        sourceFile.getExportDeclarations().forEach((exportDecl) => {
            const moduleSpecifierValue = exportDecl.getModuleSpecifierValue();

            // If the export path starts with "playwright-core", replace it with "patchright-core"
            if (moduleSpecifierValue && moduleSpecifierValue.startsWith("playwright-core")) {
                const newModuleSpecifier = moduleSpecifierValue.replace("playwright-core", "patchright-core");
                exportDecl.setModuleSpecifier(newModuleSpecifier);
                modified = true;
            }
        });

        // Handle export *
        const exportAllDeclarations = sourceFile.getExportDeclarations().filter(exportDecl => {
            return exportDecl.getModuleSpecifierValue() === 'playwright-core';
        });

        exportAllDeclarations.forEach(exportDecl => {
            exportDecl.setModuleSpecifier('patchright-core');
            modified = true;
        });

        // Find all require() calls
        const requireCalls = sourceFile.getDescendantsOfKind(
            SyntaxKind.CallExpression
        ).filter(call => {
            const expression = call.getExpression();
            // Check for require() and require.resolve()
            return expression.getText() === "require" || expression.getText() === "require.resolve";
        });
        // Modify any 'playwright-core' require or require.resolve
        requireCalls.forEach(call => {
            const args = call.getArguments();
            if (args.length && (args[0].getText().includes("playwright-core"))) {
                const arg = args[0];
                arg.replaceWithText(arg.getText().replace(/playwright-core/g, "patchright-core"));
                modified = true;
            } else if (args.length && (args[0].getText().includes("playwright"))) {
                const arg = args[0];
                arg.replaceWithText(arg.getText().replace(/playwright/g, "patchright"));
                modified = true;
            }
        });

        // Save if any modification was made
        if (modified) {
            sourceFile.saveSync();
            console.log(`Modified imports/exports in: ${filePath}`);
        }
    });
}


// Renaming the folders and using context managers to ensure they finished
fs.rename("packages/playwright-core", "packages/patchright-core", (err) => {
    fs.rename("packages/playwright", "packages/patchright", (err) => {
        // Write the Projects README to the README which is used in the release
        fs.readFile("../README.md", "utf8", (err, data) => {
            fs.writeFileSync("packages/patchright/README.md", data, "utf8", (err) => {});
        });
        fs.writeFileSync("packages/patchright-core/README.md", "# patchright-core\n\nThis package contains the no-browser flavor of [Patchright-NodeJS](https://github.com/Kaliiiiiiiiii-Vinyzu/patchright-nodejs).", "utf8", (err) => {});

        // Package.Json Files
        // playwright-core/package.json
        fs.readFile("packages/patchright-core/package.json", "utf8", (err, data) => {
          const packageJson = JSON.parse(data);
          packageJson.name = "patchright-core";
          //packageJson.version = "1.49.2"
          packageJson.author["name"] = "Microsoft Corportation, patched by github.com/Kaliiiiiiiiii-Vinyzu/";
          packageJson.homepage = "https://github.com/Kaliiiiiiiiii-Vinyzu/patchright-nodejs"
          packageJson.repository["url"] = "https://github.com/Kaliiiiiiiiii-Vinyzu/patchright-nodejs"
          packageJson.bin = {
            "patchright-core": "cli.js"
          }

          const updatedJsonData = JSON.stringify(packageJson, null, 4);
          fs.writeFile("packages/patchright-core/package.json", updatedJsonData, 'utf8', (err) => {
            if (err) {
                        console.log('Error writing to the file:', err);
                    } else {
                        console.log('JSON file has been updated successfully.');
                    }
          });
        });
        // playwright/package.json
        fs.readFile("packages/patchright/package.json", "utf8", (err, data) => {
          const packageJson = JSON.parse(data);
          packageJson.name = "patchright";
          //packageJson.version = "1.49.2"
          packageJson.author["name"] = "Microsoft Corportation, patched by github.com/Kaliiiiiiiiii-Vinyzu/";
          packageJson.homepage = "https://github.com/Kaliiiiiiiiii-Vinyzu/patchright-nodejs"
          packageJson.repository["url"] = "https://github.com/Kaliiiiiiiiii-Vinyzu/patchright-nodejs"
          packageJson.bin = {
            "patchright": "cli.js"
          }
          packageJson.dependencies = {
            "patchright-core": packageJson.dependencies["playwright-core"]
          }

          const updatedJsonData = JSON.stringify(packageJson, null, 4);
          fs.writeFile("packages/patchright/package.json", updatedJsonData, 'utf8', (err) => {
            if (err) {
                        console.log('Error writing to the file:', err);
                    } else {
                        console.log('JSON file has been updated successfully.');
                    }
          });
        });

        // Some random path which is easier to just replace manually than with ts-morph/AST
        fs.readFile("packages/patchright/lib/transform/esmUtils.js", "utf8", (err, data) => {
            const updatedContent = data.replace(/playwright\/lib\/transform\/esmLoader/g, 'patchright/lib/transform/esmLoader');
            fs.writeFile("packages/patchright/lib/transform/esmUtils.js", updatedContent, 'utf8', (err) => {});
        });

        // Usage example: pass the directory path as an argument
        renameImportsAndExportsInDirectory("packages/patchright");
    })
})