import path from 'path';

/**
 * Builds the full path to the correct Prisma Client types file.
 *
 * Normally located at `node_modules/.prisma/client/index.d.ts`
 */
export function buildTypesFilePath(
  clientOutput: string,
  overrideTarget?: string,
  schemaTarget?: string
) {
  // Just finds the default .prisma/client/index.d.ts
  if (!overrideTarget) {
    try {
      return path.resolve(
        // prisma client directory
        path.dirname(
          // We cannot directly resolve .prisma/client because pnpm uses a different directory structure,
          // so we find @prisma/client path and resolve the parent directory
          require.resolve(path.resolve(clientOutput, '../../.prisma/client'))
        ),
        'index.d.ts'
      );
    } catch {
      return path.resolve(
        // Complete filename?
        clientOutput.match(/\.(?:d\.ts|tsx?|jsx?)$/gm)
          ? // dirname
            path.dirname(clientOutput)
          : // filename
            clientOutput,
        'index.d.ts'
      );
    }
  }

  // Absolute path, just use it.
  if (path.isAbsolute(overrideTarget)) {
    return require.resolve(overrideTarget);
  }

  // Schema relative
  return path.resolve(
    // schemaTarget is the full path of the Prisma schema - we need the directory
    path.dirname(schemaTarget!),
    overrideTarget,
    // the original path may not be called index.
    overrideTarget.endsWith('.d.ts') ? '' : 'index.d.ts'
  );
}