import * as fs from 'fs'
import * as path from 'path'

import { normalizeVueRegistryFiles } from '../lib/registry-resolution'

function processJsonFile(filePath: string) {
  try {
    // Read the file
    const content = fs.readFileSync(filePath, 'utf8')
    const json = JSON.parse(content)

    // Normalize only after shadcn build has read the original package source files.
    for (const item of Array.isArray(json.items) ? json.items : [json]) {
      if (Array.isArray(item.files)) item.files = normalizeVueRegistryFiles(item.files)
    }

    // Convert to string to do replacement
    let stringified = JSON.stringify(json, null, 2)

    // Perform the replacement
    stringified = stringified
      .replace(/\/ui\/example\/password-based-auth/g, '')
      .replace(/\/example\/password-based-auth/g, '')
      .replaceAll(
        "import { Link } from '@/registry/default/components/ui/link'",
        "import Link from 'next/link'"
      )
      // Replace the file origin path to exclude the monorepo structure
      .replaceAll('node_modules/@supabase/vue-blocks/', '')

    // Blocks that combine() across all client variants hard-code the nextjs
    // client path in their source. Rewrite that import per-variant so each
    // generated artifact points at the client file it actually bundles.
    const variantClientMap: Record<string, string> = {
      react: 'react',
      'react-router': 'react-router',
      tanstack: 'tanstack',
    }
    const baseName = path.basename(filePath, '.json')
    for (const [suffix, clientDir] of Object.entries(variantClientMap)) {
      if (baseName.endsWith(`-${suffix}`)) {
        stringified = stringified.replaceAll(
          '@/registry/default/clients/nextjs/lib/supabase/client',
          `@/registry/default/clients/${clientDir}/lib/supabase/client`
        )
        break
      }
    }

    // Registry source routes are absent from this app's generated TanStack tree.
    // Consumers generate their own tree, where this suppression would be unused.
    stringified = stringified.replaceAll(
      '// @ts-expect-error The local generated route tree does not include this block route.\\n',
      ''
    )

    // Write back to file
    fs.writeFileSync(filePath, stringified)
    console.log(`✓ Updated ${filePath}`)
  } catch (error) {
    throw new Error(`Unable to prepare registry artifact "${filePath}"`, { cause: error })
  }
}

function processDirectory(directoryPath: string) {
  const files = fs.readdirSync(directoryPath)

  files.forEach((file) => {
    const fullPath = path.join(directoryPath, file)
    const stat = fs.statSync(fullPath)

    if (stat.isDirectory()) {
      processDirectory(fullPath)
    } else if (path.extname(file) === '.json') {
      processJsonFile(fullPath)
    }
  })
}

// Start processing from the specified directory
const targetDir = path.join(process.cwd(), 'public/r')

if (!fs.existsSync(targetDir)) {
  console.error('Target directory does not exist:', targetDir)
  process.exit(1)
}

console.log('Starting JSON file processing...')
processDirectory(targetDir)
console.log('Processing complete!')