vite/scripts/patchTypes.js

// @ts-check
const fs = require('fs')
const path = require('path')
const chalk = require('chalk')
const { parse } = require('@babel/parser')
const MagicString = require('magic-string').default
const tempDir = path.resolve(__dirname, '../temp/node')
const typesDir = path.resolve(__dirname, '../types')

// walk through the temp dts dir, find all import/export of types/*
// and rewrite them into relative imports - so that api-extractor actually
// includes them in the rolled-up final d.ts file.
walkDir(tempDir)
console.log(chalk.green.bold(`patched types/* imports`))

function slash(p) {
  return p.replace(/\\/g, '/')
}

/**
 * @param {string} dir
 */
function walkDir(dir) {
  const files = fs.readdirSync(dir)
  for (const file of files) {
    const resolved = path.resolve(dir, file)
    const isDir = fs.statSync(resolved).isDirectory()
    if (isDir) {
      walkDir(resolved)
    } else {
      rewriteFile(resolved)
    }
  }
}

/**
 * @param {string} file
 */
function rewriteFile(file) {
  const content = fs.readFileSync(file, 'utf-8')
  const str = new MagicString(content)
  let ast
  try {
    ast = parse(content, {
      sourceType: 'module',
      plugins: ['typescript', 'classProperties']
    })
  } catch (e) {
    console.log(chalk.red(`failed to parse ${file}`))
    throw e
  }
  for (const statement of ast.program.body) {
    if (
      (statement.type === 'ImportDeclaration' ||
        statement.type === 'ExportNamedDeclaration' ||
        statement.type === 'ExportAllDeclaration') &&
      statement.source &&
      statement.source.value.startsWith('types/')
    ) {
      const source = statement.source
      const absoluteTypePath = path.resolve(typesDir, source.value.slice(6))
      const relativeTypePath = slash(
        path.relative(path.dirname(file), absoluteTypePath)
      )
      // @ts-ignore
      str.overwrite(source.start, source.end, JSON.stringify(relativeTypePath))
    }
  }
  fs.writeFileSync(file, str.toString())
}