feat(cli): add stdin support and list commands (#1241)

Co-authored-by: DIVYA PAHUJA <divyapahuja@DIVYAs-MacBook-Pro.local>

Author: Divyapahuja31

packages/cli/src/cli.ts

@@ -47,12 +47,59 @@ export async function run(
     .option('--theme <theme>', 'Color theme to use', { default: 'vitesse-dark' })
     .option('--lang <lang>', 'Programming language')
     .option('--format <format>', 'Output format (ansi, html)', { default: 'ansi' })
+    .option('--list-themes', 'List all available themes')
+    .option('--list-langs', 'List all available languages')
     .help()
     .version(version)
 
   const { options, args } = cli.parse(argv)
+
+  if (options.listThemes) {
+    const { bundledThemes } = await import('shiki')
+    for (const theme of Object.keys(bundledThemes))
+      log(theme)
+    return
+  }
+
+  if (options.listLangs) {
+    const { bundledLanguages } = await import('shiki')
+    for (const lang of Object.keys(bundledLanguages))
+      log(lang)
+    return
+  }
+
   const files = args
 
+  if (files.length === 0) {
+    // If no files provided, verify if we are in a TTY environment
+    // If NOT in TTY (piped), read from stdin
+    // If in TTY, show help
+    if (!process.stdin.isTTY) {
+      const content = await new Promise<string>((resolve, reject) => {
+        let data = ''
+        process.stdin.on('data', chunk => data += chunk)
+        process.stdin.on('end', () => resolve(data))
+        process.stdin.on('error', reject)
+      })
+
+      const lang = options.lang || 'text'
+      if (options.format === 'html') {
+        const { codeToHtml } = await import('shiki')
+        log(await codeToHtml(content, {
+          lang: lang as BundledLanguage,
+          theme: options.theme,
+        }))
+      }
+      else {
+        log(await codeToANSI(content, lang as BundledLanguage, options.theme))
+      }
+      return
+    }
+
+    cli.outputHelp()
+    return
+  }
+
   const codes = await Promise.all(files.map(async (path) => {
     const { content, ext } = await readSource(path)
     const lang = options.lang || ext

packages/cli/test/cli.test.ts

@@ -143,4 +143,22 @@ describe('run', () => {
 
     vi.unstubAllGlobals()
   })
+
+  it('--list-themes', async () => {
+    const output: string[] = []
+    await run(['node', 'shiki', '--list-themes'], msg => output.push(msg))
+
+    expect(output.length).toBeGreaterThan(0)
+    expect(output).toContain('vitesse-dark')
+    expect(output).toContain('nord')
+  })
+
+  it('--list-langs', async () => {
+    const output: string[] = []
+    await run(['node', 'shiki', '--list-langs'], msg => output.push(msg))
+
+    expect(output.length).toBeGreaterThan(0)
+    expect(output).toContain('javascript')
+    expect(output).toContain('python')
+  })
 })