rsnext/packages/next/lib/verify-partytown-setup.ts

import { promises } from 'fs'
import chalk from 'next/dist/compiled/chalk'

import path from 'path'
import {
  hasNecessaryDependencies,
  NecessaryDependencies,
} from './has-necessary-dependencies'
import { isYarn } from './is-yarn'
import { fileExists } from './file-exists'
import { FatalError } from './fatal-error'
import { recursiveDelete } from './recursive-delete'
import * as Log from '../build/output/log'

async function missingDependencyError(dir: string) {
  throw new FatalError(
    chalk.bold.red(
      "It looks like you're trying to use Partytown with next/script but do not have the required package(s) installed."
    ) +
      '\n\n' +
      chalk.bold(`Please install Partytown by running:`) +
      '\n\n' +
      `\t${chalk.bold.cyan(
        (await isYarn(dir))
          ? 'yarn add @builder.io/partytown'
          : 'npm install @builder.io/partytown'
      )}` +
      '\n\n' +
      chalk.bold(
        `If you are not trying to use Partytown, please disable the experimental ${chalk.cyan(
          '"nextScriptWorkers"'
        )} flag in next.config.js.`
      ) +
      '\n'
  )
}

async function copyPartytownStaticFiles(
  deps: NecessaryDependencies,
  staticDir: string
) {
  const partytownLibDir = path.join(staticDir, '~partytown')
  const hasPartytownLibDir = await fileExists(partytownLibDir, 'directory')

  if (hasPartytownLibDir) {
    await recursiveDelete(partytownLibDir)
    await promises.rmdir(partytownLibDir)
  }

  const { copyLibFiles } = await Promise.resolve(
    require(path.join(deps.resolved.get('@builder.io/partytown')!, '../utils'))
  )

  await copyLibFiles(partytownLibDir)
}

export async function verifyPartytownSetup(
  dir: string,
  targetDir: string
): Promise<void> {
  try {
    const partytownDeps: NecessaryDependencies = await hasNecessaryDependencies(
      dir,
      [
        {
          file: '@builder.io/partytown',
          pkg: '@builder.io/partytown',
          exportsRestrict: false,
        },
      ]
    )

    if (partytownDeps.missing?.length > 0) {
      await missingDependencyError(dir)
    } else {
      try {
        await copyPartytownStaticFiles(partytownDeps, targetDir)
      } catch (err) {
        Log.warn(
          `Partytown library files could not be copied to the static directory. Please ensure that ${chalk.bold.cyan(
            '@builder.io/partytown'
          )} is installed as a dependency.`
        )
      }
    }
  } catch (err) {
    // Don't show a stack trace when there is an error due to missing dependencies
    if (err instanceof FatalError) {
      console.error(err.message)
      process.exit(1)
    }
    throw err
  }
}