cli.ts

#!/usr/bin/env node

import cac from 'cac'
import { fs } from './fs'
import pathLib, { extname } from 'path'
import os from 'os'
import { logger } from './logger'
import { Is } from './utils'
import { getGlobalTaskManager } from './task-manager'
import chalk from 'chalk'
import { initDefaultCli } from './default-cli'
import { spawn, ChildProcess } from 'child_process'

const { foyFiles, registers, defaultCli } = initDefaultCli()
async function main() {
  const pkg = await fs.readJson('./package.json')
  const isESM = pkg.type === 'module'
  const deps = { ...pkg.dependencies, ...pkg.devDependencies }
  const results: ChildProcess[] = [];
  for (const foyFile of foyFiles) {
    let executor = defaultCli.options.executor
    if (!executor) {
      executor = 'node'

      if (['.ts', '.cts', '.tsx', '.ctsx'].includes(extname(foyFile))) {
        if ('tsx' in deps) {
          executor = 'tsx'
        } else if ('ts-node' in deps) {
          executor = isESM ? 'ts-node-esm' : 'ts-node'
        } else if ('@swc-node/register' in deps) {
          if (isESM) {
            registers.push('@swc-node/register/esm-register')
          } else {
            registers.push('@swc-node/register')
          }
        } else if ('swc-node' in deps) {
          executor = 'swc-node'
        } else {
          logger.error('no tsx/ts-node/swc-node or @swc-node/register found')
          process.exit(1)
        }
      }
    }
    const args = [
      ...registers.map((r) => `--${isESM ? 'import' : 'require'} ${r}`),
      foyFile,
      ...process.argv.slice(2),
    ]
    let NODE_OPTIONS = process.env.NODE_OPTIONS ?? ''
    ;[
      ['--inspect',defaultCli.options.inspect],
      ['--inspectBrk',defaultCli.options.inspectBrk]
    ].map(([inspect, inspectVal]) => {
      if (inspectVal) {
        if (typeof inspectVal === 'string') {
          inspect += '=' + inspectVal
        }
        NODE_OPTIONS += ' ' + inspect
      }
    })
    results.push(spawn(executor, args, {
      stdio: 'inherit',
      shell: process.platform === 'win32',
      cwd: process.cwd(),
      env:{
        ...process.env,
        NODE_OPTIONS,
      }
    }));
  }
  for (const p of results) {
    await new Promise<void>((resolve) => p.on('exit', () => resolve()))
    if (p.exitCode !== 0 && p.exitCode !== null) {
      process.exitCode = p.exitCode;
    }
  }
  // fix zombie process sometimes
   process.on('SIGINT', () => {
     results.forEach((p) => {
       p.kill(9)
     })
   })
}

main().catch((err) => {
  console.error(err)
  process.exitCode = 1;
})