chore: process.mainModule -> require.main (#38477)

chore: process.mainModule -> require.main Co-authored-by: trop[bot] <37223003+trop[bot]@users.noreply.github.com> Co-authored-by: Shelley Vohr <shelley.vohr@gmail.com>
electron · May 29, 2023 · 6a5bd8d · 6a5bd8d
1 parent b78c30a
commit 6a5bd8d
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/script/yarn.js b/script/yarn.js
@@ -3,13 +3,9 @@ const fs = require('fs');
 const path = require('path');
 
 const YARN_VERSION = /'yarn_version': '(.+?)'/.exec(fs.readFileSync(path.resolve(__dirname, '../DEPS'), 'utf8'))[1];
+const NPX_CMD = process.platform === 'win32' ? 'npx.cmd' : 'npx';
 
-exports.YARN_VERSION = YARN_VERSION;
-
-// If we are running "node script/yarn" run as the yarn CLI
-if (process.mainModule === module) {
-  const NPX_CMD = process.platform === 'win32' ? 'npx.cmd' : 'npx';
-
+if (require.main === module) {
   const child = cp.spawn(NPX_CMD, [`yarn@${YARN_VERSION}`, ...process.argv.slice(2)], {
     stdio: 'inherit',
     env: {
@@ -20,3 +16,5 @@ if (process.mainModule === module) {
 
   child.on('exit', code => process.exit(code));
 }
+
+exports.YARN_VERSION = YARN_VERSION;