fix: map shebang executables and resolve lint error

Author: Divyapahuja31

packages/cli/test/cli.test.ts

@@ -174,4 +174,34 @@ describe('run', () => {
     expect(output).toContain('javascript')
     expect(output).toContain('python')
   })
+
+  describe('stdin', () => {
+    it('reads from stdin and guesses language', async () => {
+      const output: string[] = []
+      const content = '#!/usr/bin/env node\nconsole.log("hi")'
+
+      const mockStdin = {
+        isTTY: false,
+        on: vi.fn((event, handler) => {
+          if (event === 'data')
+            handler(new TextEncoder().encode(content))
+          if (event === 'end')
+            handler()
+          return mockStdin
+        }),
+      }
+
+      const originalStdin = process.stdin
+      Object.defineProperty(process, 'stdin', { value: mockStdin, configurable: true })
+
+      await run(['node', 'shiki'], msg => output.push(msg))
+
+      expect(output.length).toBe(1)
+      expect(output[0]).toContain('console') // Should be highlighted
+      // Since we can't easily check for colors in this test environment without more setup,
+      // we at least verify it didn't throw and produced output.
+
+      Object.defineProperty(process, 'stdin', { value: originalStdin, configurable: true })
+    })
+  })
 })

packages/core/src/utils/strings.test.ts

@@ -199,15 +199,22 @@ print("hello")
   })
 
   it('detects languages from shebangs', () => {
-    expect(guessEmbeddedLanguages('#!/usr/bin/node', undefined)).toContain('node')
+    expect(guessEmbeddedLanguages('#!/usr/bin/node', undefined)).toContain('javascript')
     expect(guessEmbeddedLanguages('#!/bin/bash', undefined)).toContain('bash')
-    expect(guessEmbeddedLanguages('#!/usr/bin/env python3', undefined)).toContain('python3')
+    expect(guessEmbeddedLanguages('#!/usr/bin/env python3', undefined)).toContain('python')
     expect(guessEmbeddedLanguages('#!/usr/bin/env -S ts-node --foo', undefined)).toContain('ts-node')
+    expect(guessEmbeddedLanguages('#!/usr/bin/env -S node --inspect', undefined)).toContain('javascript')
+    expect(guessEmbeddedLanguages('#!/usr/bin/env', undefined)).toEqual([])
+    expect(guessEmbeddedLanguages('#!', undefined)).toEqual([])
+    expect(guessEmbeddedLanguages('  #!/bin/bash', undefined)).toEqual([]) // Must be at start
   })
 
   it('detects languages from comments', () => {
     expect(guessEmbeddedLanguages('<!-- language: lang-js -->', undefined)).toContain('js')
     expect(guessEmbeddedLanguages('// @lang typescript', undefined)).toContain('typescript')
     expect(guessEmbeddedLanguages('/** @lang python */', undefined)).toContain('python')
+    expect(guessEmbeddedLanguages('<!-- language: lang-  -->', undefined)).toEqual([])
+    expect(guessEmbeddedLanguages('// @lang  ', undefined)).toEqual([])
+    expect(guessEmbeddedLanguages('<!-- language: lang-JS-variant -->', undefined)).toContain('js-variant')
   })
 })

packages/core/src/utils/strings.ts

@@ -118,6 +118,7 @@ export function guessEmbeddedLanguages(
     if (parts.length > 0) {
       let lang = parts[0].split('/').pop()
       if (lang === 'env') {
+        lang = undefined
         // Find first part that doesn't start with '-'
         for (let i = 1; i < parts.length; i++) {
           if (parts[i] && !parts[i].startsWith('-')) {
@@ -126,8 +127,21 @@ export function guessEmbeddedLanguages(
           }
         }
       }
-      if (lang)
-        langs.add(lang.toLowerCase())
+
+      if (lang) {
+        lang = lang.toLowerCase()
+        // Map common executable names to Shiki aliases
+        if (lang === 'node')
+          lang = 'javascript'
+        else if (lang === 'python3')
+          lang = 'python'
+        else if (lang === 'rb')
+          lang = 'ruby'
+        else if (lang === 'sh' || lang === 'zsh')
+          lang = 'shell'
+
+        langs.add(lang)
+      }
     }
   }
 

packages/core/test/tokens.test.ts

@@ -197,3 +197,26 @@ it('colorsRendering none', async () => {
     }),
   ).toMatchSnapshot('colorsRendering none')
 })
+
+it('ansi with multiple themes', async () => {
+  using engine = await createShikiPrimitiveAsync({
+    themes: [
+      import('@shikijs/themes/vitesse-light'),
+      import('@shikijs/themes/vitesse-dark'),
+    ],
+    langs: [],
+    engine: createJavaScriptRegexEngine(),
+  })
+
+  const code = '\x1B[31mred\x1B[0m'
+  const html = codeToHtml(engine, code, {
+    lang: 'ansi',
+    themes: {
+      light: 'vitesse-light',
+      dark: 'vitesse-dark',
+    },
+  })
+
+  expect(html).toContain('<span style="color:#')
+  expect(html).toContain('--shiki-dark:#')
+})