gouldingken · December 1, 2022 21:30
diff --git a/readme.md b/readme.md
diff --git a/pnpm-post.js b/pnpm-post.js
 const fs = require('fs');
 const path = require("path");

 const runScript = (script) => {
    const child_process = require('child_process');
    child_process.execSync(script, {stdio: [0, 1, 2]});
 };

 const processDir = async (dir) => {
    let files = await fs.promises.readdir(dir, {withFileTypes: true});
    for (let f of files) {
        let fullPath = path.join(dir, f.name);
        if (f.isDirectory()) {
            if (f.name === '.git' || f.name === '.idea' || f.name === '_dev' || f.name === 'node_modules') {
                continue;
            }
            await processDir(fullPath);
        } else {
            if (f.name === 'package.json') {
                console.log('FOUND package.json', fullPath);
                const packageStr = await fs.promises.readFile(fullPath);
                const packageData = JSON.parse(packageStr);
                if (packageData.stashedGlobalDependencies) {
                    if (!packageData.dependencies) packageData.dependencies = {};
                    console.log('Setting dir', dir);
                    process.chdir(dir);
                    //I think it's ok to just run these and not wait for a result because this is a post-install
                    Object.keys(packageData.stashedGlobalDependencies).forEach((depKey) => {
                        console.log('Relinking:', depKey);
                        runScript(`pnpm link -g ${depKey}`);
                        packageData.dependencies[depKey] = packageData.stashedGlobalDependencies[depKey];
                    });
                    delete packageData.stashedGlobalDependencies;
                    //NOTE: this overwrites the update that was just written via pnpm link so version changes could happen...
                    //there may be a better approach that waits for that install to complete, then checks the version etc
                    await fs.promises.writeFile(fullPath, JSON.stringify(packageData, null, 2));
                }
            }
        }
    }
 };

 let baseDir = path.resolve(__dirname,'../');
 console.log('RUNNING POST INSTALL...', baseDir);

 processDir(baseDir);
diff --git a/pnpm-pre.js b/pnpm-pre.js
 const fs = require('fs');
 const path = require('path');

 console.log('RUNNING PREINSTALL');

 const isIncluded = (f) => {
  //NOTE write any custom inclusion rules for folders here so it doesn't have to scan EVERY folder
  return true;
 }

 const saveJson = async (filePath, data) => {
    await fs.promises.writeFile(filePath, JSON.stringify(data, null, 2));
    console.log('Saved ' + filePath);
 }

 const processNodeModules = async (dir, prefix = '') => {
    // console.log('processNodeModules', dir);
    let files = await fs.promises.readdir(dir, {withFileTypes: true});
    const ans = [];
    for (let f of files) {
        if (f.isSymbolicLink()) {
            let full_path = path.join(dir, f.name);
            // console.log(full_path);
            try {
                const linkString = await fs.promises.readlink(full_path);
                if (linkString.indexOf('.pnpm') < 0) {
                    console.log('global LINKED', dir, linkString, full_path);
                    ans.push({name: prefix + f.name, path: full_path});
                }
            } catch (err) {
                console.error(err);
            }
        }
        if (isIncluded(f)) {
            let full_path = path.join(dir, f.name);
            ans.push(...await processNodeModules(full_path, f.name + '/'));
        }
    }
    return ans;

 };

 const isWorkspaceRef = (versionString) => {
    return versionString.indexOf('*') > -1;
 };

 const stashGlobalDependencies = async (packageInfo) => {
    const packageStr = await fs.promises.readFile(packageInfo.packageJson);
    const packageData = JSON.parse(packageStr);

    packageInfo.globalDependencies.forEach((dep, i) => {
        let versionString = packageData.dependencies[dep.name];
        if (versionString && !isWorkspaceRef(versionString)) {
            if (!packageData.stashedGlobalDependencies) packageData.stashedGlobalDependencies = {};
            packageData.stashedGlobalDependencies[dep.name] = versionString;
            delete packageData.dependencies[dep.name];
        }
    });

    await saveJson(packageInfo.packageJson, packageData);
 };

 const processDir = async (dir) => {
    let files = await fs.promises.readdir(dir, {withFileTypes: true});
    const packageInfo = {globalDependencies: []};
    for (let f of files) {
        let full_path = path.join(dir, f.name);
        if (f.isDirectory()) {
            if (f.name === '.git' || f.name === '.idea' || f.name === '_dev') {
                continue;
            }
            if (f.name === 'node_modules') {
                packageInfo.globalDependencies.push(...await processNodeModules(full_path));
            } else {
                await processDir(full_path);
            }
        } else {
            if (f.name === 'package.json') {
                // console.log('FOUND package.json', path.join(dir, f.name));
                packageInfo.packageJson = path.join(dir, f.name);
            }
        }
    }
    if (packageInfo.globalDependencies.length > 0) {
        // console.log('PACKAGE INFO: ', dir, JSON.stringify(packageInfo));
        await stashGlobalDependencies(packageInfo);
    }
 };

 const findLinks = async () => {
    //find all package.json
    //look through node_modules
    //find all links that don't have a .pnpm or node_modules
    //these are the global dependencies that should be deleted from the package.json
    //we store these in a 'stashedGlobalDependencies' custom field on package.json
    //the post install will run `pnpm install -g` on all these stashedGlobalDependencies

    let baseDir = path.resolve(__dirname, '../');
    console.log('PROCESSING...', baseDir);
    await processDir(baseDir);
    console.log('DONE PROCESSING ', baseDir);
    process.exit();
 };
 findLinks();
	const fs = require('fs');
	const path = require("path");

	const runScript = (script) => {
	const child_process = require('child_process');
	child_process.execSync(script, {stdio: [0, 1, 2]});
	};

	const processDir = async (dir) => {
	let files = await fs.promises.readdir(dir, {withFileTypes: true});
	for (let f of files) {
	let fullPath = path.join(dir, f.name);
	if (f.isDirectory()) {
	if (f.name === '.git' \|\| f.name === '.idea' \|\| f.name === '_dev' \|\| f.name === 'node_modules') {
	continue;
	}
	await processDir(fullPath);
	} else {
	if (f.name === 'package.json') {
	console.log('FOUND package.json', fullPath);
	const packageStr = await fs.promises.readFile(fullPath);
	const packageData = JSON.parse(packageStr);
	if (packageData.stashedGlobalDependencies) {
	if (!packageData.dependencies) packageData.dependencies = {};
	console.log('Setting dir', dir);
	process.chdir(dir);
	//I think it's ok to just run these and not wait for a result because this is a post-install
	Object.keys(packageData.stashedGlobalDependencies).forEach((depKey) => {
	console.log('Relinking:', depKey);
	runScript(`pnpm link -g ${depKey}`);
	packageData.dependencies[depKey] = packageData.stashedGlobalDependencies[depKey];
	});
	delete packageData.stashedGlobalDependencies;
	//NOTE: this overwrites the update that was just written via pnpm link so version changes could happen...
	//there may be a better approach that waits for that install to complete, then checks the version etc
	await fs.promises.writeFile(fullPath, JSON.stringify(packageData, null, 2));
	}
	}
	}
	}
	};

	let baseDir = path.resolve(__dirname,'../');
	console.log('RUNNING POST INSTALL...', baseDir);

	processDir(baseDir);
	const fs = require('fs');
	const path = require('path');

	console.log('RUNNING PREINSTALL');

	const isIncluded = (f) => {
	//NOTE write any custom inclusion rules for folders here so it doesn't have to scan EVERY folder
	return true;
	}

	const saveJson = async (filePath, data) => {
	await fs.promises.writeFile(filePath, JSON.stringify(data, null, 2));
	console.log('Saved ' + filePath);
	}

	const processNodeModules = async (dir, prefix = '') => {
	// console.log('processNodeModules', dir);
	let files = await fs.promises.readdir(dir, {withFileTypes: true});
	const ans = [];
	for (let f of files) {
	if (f.isSymbolicLink()) {
	let full_path = path.join(dir, f.name);
	// console.log(full_path);
	try {
	const linkString = await fs.promises.readlink(full_path);
	if (linkString.indexOf('.pnpm') < 0) {
	console.log('global LINKED', dir, linkString, full_path);
	ans.push({name: prefix + f.name, path: full_path});
	}
	} catch (err) {
	console.error(err);
	}
	}
	if (isIncluded(f)) {
	let full_path = path.join(dir, f.name);
	ans.push(...await processNodeModules(full_path, f.name + '/'));
	}
	}
	return ans;

	};

	const isWorkspaceRef = (versionString) => {
	return versionString.indexOf('*') > -1;
	};

	const stashGlobalDependencies = async (packageInfo) => {
	const packageStr = await fs.promises.readFile(packageInfo.packageJson);
	const packageData = JSON.parse(packageStr);

	packageInfo.globalDependencies.forEach((dep, i) => {
	let versionString = packageData.dependencies[dep.name];
	if (versionString && !isWorkspaceRef(versionString)) {
	if (!packageData.stashedGlobalDependencies) packageData.stashedGlobalDependencies = {};
	packageData.stashedGlobalDependencies[dep.name] = versionString;
	delete packageData.dependencies[dep.name];
	}
	});

	await saveJson(packageInfo.packageJson, packageData);
	};

	const processDir = async (dir) => {
	let files = await fs.promises.readdir(dir, {withFileTypes: true});
	const packageInfo = {globalDependencies: []};
	for (let f of files) {
	let full_path = path.join(dir, f.name);
	if (f.isDirectory()) {
	if (f.name === '.git' \|\| f.name === '.idea' \|\| f.name === '_dev') {
	continue;
	}
	if (f.name === 'node_modules') {
	packageInfo.globalDependencies.push(...await processNodeModules(full_path));
	} else {
	await processDir(full_path);
	}
	} else {
	if (f.name === 'package.json') {
	// console.log('FOUND package.json', path.join(dir, f.name));
	packageInfo.packageJson = path.join(dir, f.name);
	}
	}
	}
	if (packageInfo.globalDependencies.length > 0) {
	// console.log('PACKAGE INFO: ', dir, JSON.stringify(packageInfo));
	await stashGlobalDependencies(packageInfo);
	}
	};

	const findLinks = async () => {
	//find all package.json
	//look through node_modules
	//find all links that don't have a .pnpm or node_modules
	//these are the global dependencies that should be deleted from the package.json
	//we store these in a 'stashedGlobalDependencies' custom field on package.json
	//the post install will run `pnpm install -g` on all these stashedGlobalDependencies

	let baseDir = path.resolve(__dirname, '../');
	console.log('PROCESSING...', baseDir);
	await processDir(baseDir);
	console.log('DONE PROCESSING ', baseDir);
	process.exit();
	};
	findLinks();