From e7a15fc34ced1f817def9ba556ad33f9a171ec68 Mon Sep 17 00:00:00 2001
From: Gustavo Henke <guhenke@gmail.com>
Date: Sat, 31 Aug 2024 14:26:28 +1000
Subject: [PATCH] Expand `node:<script>` and `node --run` comands

Closes #475
---
 .../expand-npm-shortcut.spec.ts               |  38 ++---
 src/command-parser/expand-npm-shortcut.ts     |   7 +-
 .../expand-npm-wildcard.spec.ts               | 142 +++++++++---------
 src/command-parser/expand-npm-wildcard.ts     |   6 +-
 4 files changed, 101 insertions(+), 92 deletions(-)

diff --git a/src/command-parser/expand-npm-shortcut.spec.ts b/src/command-parser/expand-npm-shortcut.spec.ts
index 3f7a39c5..eb6ac202 100644
--- a/src/command-parser/expand-npm-shortcut.spec.ts
+++ b/src/command-parser/expand-npm-shortcut.spec.ts
@@ -13,24 +13,28 @@ it('returns same command if no npm: prefix is present', () => {
     expect(parser.parse(commandInfo)).toBe(commandInfo);
 });
 
-for (const npmCmd of ['npm', 'yarn', 'pnpm', 'bun']) {
-    describe(`with ${npmCmd}: prefix`, () => {
-        it(`expands to "${npmCmd} run <script> <args>"`, () => {
-            const commandInfo = createCommandInfo(`${npmCmd}:foo -- bar`, 'echo');
-            expect(parser.parse(commandInfo)).toEqual({
-                ...commandInfo,
-                name: 'echo',
-                command: `${npmCmd} run foo -- bar`,
-            });
+describe.each([
+    ['npm', 'run'],
+    ['yarn', 'run'],
+    ['pnpm', 'run'],
+    ['bun', 'run'],
+    ['node', '--run'],
+])(`with %s: prefix`, (npmCmd, runCmd) => {
+    it(`expands to "${npmCmd} ${runCmd} <script> <args>"`, () => {
+        const commandInfo = createCommandInfo(`${npmCmd}:foo -- bar`, 'echo');
+        expect(parser.parse(commandInfo)).toEqual({
+            ...commandInfo,
+            name: 'echo',
+            command: `${npmCmd} ${runCmd} foo -- bar`,
         });
+    });
 
-        it('sets name to script name if none', () => {
-            const commandInfo = createCommandInfo(`${npmCmd}:foo -- bar`);
-            expect(parser.parse(commandInfo)).toEqual({
-                ...commandInfo,
-                name: 'foo',
-                command: `${npmCmd} run foo -- bar`,
-            });
+    it('sets name to script name if none', () => {
+        const commandInfo = createCommandInfo(`${npmCmd}:foo -- bar`);
+        expect(parser.parse(commandInfo)).toEqual({
+            ...commandInfo,
+            name: 'foo',
+            command: `${npmCmd} ${runCmd} foo -- bar`,
         });
     });
-}
+});
diff --git a/src/command-parser/expand-npm-shortcut.ts b/src/command-parser/expand-npm-shortcut.ts
index 3300e6ca..bc681582 100644
--- a/src/command-parser/expand-npm-shortcut.ts
+++ b/src/command-parser/expand-npm-shortcut.ts
@@ -2,20 +2,21 @@ import { CommandInfo } from '../command';
 import { CommandParser } from './command-parser';
 
 /**
- * Expands commands prefixed with `npm:`, `yarn:`, `pnpm:`, or `bun:` into the full version `npm run <command>` and so on.
+ * Expands commands prefixed with `node:`, `npm:`, `yarn:`, `pnpm:`, or `bun:` into the full version `npm run <command>` and so on.
  */
 export class ExpandNpmShortcut implements CommandParser {
     parse(commandInfo: CommandInfo) {
         const [, npmCmd, cmdName, args] =
-            commandInfo.command.match(/^(npm|yarn|pnpm|bun):(\S+)(.*)/) || [];
+            commandInfo.command.match(/^(node|npm|yarn|pnpm|bun):(\S+)(.*)/) || [];
         if (!cmdName) {
             return commandInfo;
         }
 
+        const runCmd = npmCmd === 'node' ? '--run' : 'run';
         return {
             ...commandInfo,
             name: commandInfo.name || cmdName,
-            command: `${npmCmd} run ${cmdName}${args}`,
+            command: `${npmCmd} ${runCmd} ${cmdName}${args}`,
         };
     }
 }
diff --git a/src/command-parser/expand-npm-wildcard.spec.ts b/src/command-parser/expand-npm-wildcard.spec.ts
index 8fb0c469..f0a57285 100644
--- a/src/command-parser/expand-npm-wildcard.spec.ts
+++ b/src/command-parser/expand-npm-wildcard.spec.ts
@@ -67,82 +67,86 @@ it('expands to nothing if no scripts exist in package.json', () => {
     expect(parser.parse(createCommandInfo('npm run foo-*-baz qux'))).toEqual([]);
 });
 
-for (const npmCmd of ['npm', 'yarn', 'pnpm', 'bun']) {
-    describe(`with an ${npmCmd}: prefix`, () => {
-        it('expands to all scripts matching pattern', () => {
-            readPkg.mockReturnValue({
-                scripts: {
-                    'foo-bar-baz': '',
-                    'foo--baz': '',
-                },
-            });
-
-            expect(parser.parse(createCommandInfo(`${npmCmd} run foo-*-baz qux`))).toEqual([
-                { name: 'bar', command: `${npmCmd} run foo-bar-baz qux` },
-                { name: '', command: `${npmCmd} run foo--baz qux` },
-            ]);
+describe.each([
+    ['npm', 'run'],
+    ['yarn', 'run'],
+    ['pnpm', 'run'],
+    ['bun', 'run'],
+    ['node', '--run'],
+])(`with a %s: prefix`, (npmCmd, runCmd) => {
+    it('expands to all scripts matching pattern', () => {
+        readPkg.mockReturnValue({
+            scripts: {
+                'foo-bar-baz': '',
+                'foo--baz': '',
+            },
         });
 
-        it('uses wildcard match of script as command name', () => {
-            readPkg.mockReturnValue({
-                scripts: {
-                    'watch-js': '',
-                    'watch-css': '',
-                },
-            });
-
-            expect(
-                parser.parse({
-                    name: '',
-                    command: `${npmCmd} run watch-*`,
-                }),
-            ).toEqual([
-                { name: 'js', command: `${npmCmd} run watch-js` },
-                { name: 'css', command: `${npmCmd} run watch-css` },
-            ]);
-        });
+        expect(parser.parse(createCommandInfo(`${npmCmd} ${runCmd} foo-*-baz qux`))).toEqual([
+            { name: 'bar', command: `${npmCmd} ${runCmd} foo-bar-baz qux` },
+            { name: '', command: `${npmCmd} ${runCmd} foo--baz qux` },
+        ]);
+    });
 
-        it('uses existing command name as prefix to the wildcard match', () => {
-            readPkg.mockReturnValue({
-                scripts: {
-                    'watch-js': '',
-                    'watch-css': '',
-                },
-            });
-
-            expect(
-                parser.parse({
-                    name: 'w:',
-                    command: `${npmCmd} run watch-*`,
-                }),
-            ).toEqual([
-                { name: 'w:js', command: `${npmCmd} run watch-js` },
-                { name: 'w:css', command: `${npmCmd} run watch-css` },
-            ]);
+    it('uses wildcard match of script as command name', () => {
+        readPkg.mockReturnValue({
+            scripts: {
+                'watch-js': '',
+                'watch-css': '',
+            },
         });
 
-        it('allows negation', () => {
-            readPkg.mockReturnValue({
-                scripts: {
-                    'lint:js': '',
-                    'lint:ts': '',
-                    'lint:fix:js': '',
-                    'lint:fix:ts': '',
-                },
-            });
-
-            expect(parser.parse(createCommandInfo(`${npmCmd} run lint:*(!fix)`))).toEqual([
-                { name: 'js', command: `${npmCmd} run lint:js` },
-                { name: 'ts', command: `${npmCmd} run lint:ts` },
-            ]);
+        expect(
+            parser.parse({
+                name: '',
+                command: `${npmCmd} ${runCmd} watch-*`,
+            }),
+        ).toEqual([
+            { name: 'js', command: `${npmCmd} ${runCmd} watch-js` },
+            { name: 'css', command: `${npmCmd} ${runCmd} watch-css` },
+        ]);
+    });
+
+    it('uses existing command name as prefix to the wildcard match', () => {
+        readPkg.mockReturnValue({
+            scripts: {
+                'watch-js': '',
+                'watch-css': '',
+            },
         });
 
-        it('caches scripts upon calls', () => {
-            readPkg.mockReturnValue({});
-            parser.parse(createCommandInfo(`${npmCmd} run foo-*-baz qux`));
-            parser.parse(createCommandInfo(`${npmCmd} run foo-*-baz qux`));
+        expect(
+            parser.parse({
+                name: 'w:',
+                command: `${npmCmd} ${runCmd} watch-*`,
+            }),
+        ).toEqual([
+            { name: 'w:js', command: `${npmCmd} ${runCmd} watch-js` },
+            { name: 'w:css', command: `${npmCmd} ${runCmd} watch-css` },
+        ]);
+    });
 
-            expect(readPkg).toHaveBeenCalledTimes(1);
+    it('allows negation', () => {
+        readPkg.mockReturnValue({
+            scripts: {
+                'lint:js': '',
+                'lint:ts': '',
+                'lint:fix:js': '',
+                'lint:fix:ts': '',
+            },
         });
+
+        expect(parser.parse(createCommandInfo(`${npmCmd} ${runCmd} lint:*(!fix)`))).toEqual([
+            { name: 'js', command: `${npmCmd} ${runCmd} lint:js` },
+            { name: 'ts', command: `${npmCmd} ${runCmd} lint:ts` },
+        ]);
     });
-}
+
+    it('caches scripts upon calls', () => {
+        readPkg.mockReturnValue({});
+        parser.parse(createCommandInfo(`${npmCmd} run foo-*-baz qux`));
+        parser.parse(createCommandInfo(`${npmCmd} run foo-*-baz qux`));
+
+        expect(readPkg).toHaveBeenCalledTimes(1);
+    });
+});
diff --git a/src/command-parser/expand-npm-wildcard.ts b/src/command-parser/expand-npm-wildcard.ts
index 45200b03..dc55451b 100644
--- a/src/command-parser/expand-npm-wildcard.ts
+++ b/src/command-parser/expand-npm-wildcard.ts
@@ -25,8 +25,8 @@ export class ExpandNpmWildcard implements CommandParser {
     constructor(private readonly readPackage = ExpandNpmWildcard.readPackage) {}
 
     parse(commandInfo: CommandInfo) {
-        const [, npmCmd, cmdName, args] =
-            commandInfo.command.match(/(npm|yarn|pnpm|bun) run (\S+)([^&]*)/) || [];
+        const [, npmCmd, runCmd, cmdName, args] =
+            commandInfo.command.match(/(node|npm|yarn|pnpm|bun) ((?:--)?run) (\S+)([^&]*)/) || [];
         const wildcardPosition = (cmdName || '').indexOf('*');
 
         // If the regex didn't match an npm script, or it has no wildcard,
@@ -63,7 +63,7 @@ export class ExpandNpmWildcard implements CommandParser {
                 if (match) {
                     return {
                         ...commandInfo,
-                        command: `${npmCmd} run ${script}${args}`,
+                        command: `${npmCmd} ${runCmd} ${script}${args}`,
                         // Will use an empty command name if no prefix has been specified and
                         // the wildcard match is empty, e.g. if `npm:watch-*` matches `npm run watch-`.
                         name: prefix + match[1],