#!/usr/bin/env bun

import { $ } from "bun"
import { tmpdir } from "os"
import { join } from "path"

const FORK_REPO = "anomalyco/zed-extensions"
const UPSTREAM_REPO = "zed-industries/extensions"
const EXTENSION_NAME = "opencode"

async function main() {
 const version = process.argv[2]
 if (!version) throw new Error("Version argument required, ex: bun script/sync-zed.ts v1.0.52")

 const token = process.env.ZED_EXTENSIONS_PAT
 if (!token) throw new Error("ZED_EXTENSIONS_PAT environment variable required")

 const prToken = process.env.ZED_PR_PAT
 if (!prToken) throw new Error("ZED_PR_PAT environment variable required")

 const cleanVersion = version.replace(/^v/, "")
 console.log(`📦 Syncing Zed extension for version ${cleanVersion}`)

 const commitSha = await $`git rev-parse ${version}`.text()
 const sha = commitSha.trim()
 console.log(`🔍 Found commit SHA: ${sha}`)

 const extensionToml = await $`git show ${version}:packages/extensions/zed/extension.toml`.text()
 const parsed = Bun.TOML.parse(extensionToml) as { version: string }
 const extensionVersion = parsed.version

 if (extensionVersion !== cleanVersion) {
 throw new Error(`Version mismatch: extension.toml has ${extensionVersion} but tag is ${cleanVersion}`)
 }
 console.log(`✅ Version ${extensionVersion} matches tag`)

 // Clone the fork to a temp directory
 const workDir = join(tmpdir(), `zed-extensions-${Date.now()}`)
 console.log(`📁 Working in ${workDir}`)

 await $`git clone https://x-access-token:${token}@github.com/${FORK_REPO}.git ${workDir}`
 process.chdir(workDir)

 // Configure git identity
 await $`git config user.name "Aiden Cline"`
 await $`git config user.email "63023139+rekram1-node@users.noreply.github.com "`

 // Sync fork with upstream (force reset to match exactly)
 console.log(`🔄 Syncing fork with upstream...`)
 await $`git remote add upstream https://github.com/${UPSTREAM_REPO}.git`
 await $`git fetch upstream`
 await $`git checkout main`
 await $`git reset --hard upstream/main`
 await $`git push origin main --force`
 console.log(`✅ Fork synced (force reset to upstream)`)

 // Create a new branch
 const branchName = `update-${EXTENSION_NAME}-${cleanVersion}`
 console.log(`🌿 Creating branch ${branchName}`)
 await $`git checkout -b ${branchName}`

 const submodulePath = `extensions/${EXTENSION_NAME}`
 console.log(`📌 Updating submodule to commit ${sha}`)
 await $`git submodule update --init ${submodulePath}`
 process.chdir(submodulePath)
 await $`git fetch`
 await $`git checkout ${sha}`
 process.chdir(workDir)
 await $`git add ${submodulePath}`

 console.log(`📝 Updating extensions.toml`)
 const extensionsTomlPath = "extensions.toml"
 const extensionsToml = await Bun.file(extensionsTomlPath).text()

 const versionRegex = new RegExp(`(\\[${EXTENSION_NAME}\\][\\s\\S]*?)version = "[^"]+"`)
 const updatedToml = extensionsToml.replace(versionRegex, `1ドルversion = "${cleanVersion}"`)

 if (updatedToml === extensionsToml) {
 throw new Error(`Failed to update version in extensions.toml - pattern not found`)
 }

 await Bun.write(extensionsTomlPath, updatedToml)
 await $`git add extensions.toml`

 const commitMessage = `Update ${EXTENSION_NAME} to v${cleanVersion}`

 await $`git commit -m ${commitMessage}`
 console.log(`✅ Changes committed`)

 // Delete any existing branches for opencode updates
 console.log(`🔍 Checking for existing branches...`)
 const branches = await $`git ls-remote --heads https://x-access-token:${token}@github.com/${FORK_REPO}.git`.text()
 const branchPattern = `refs/heads/update-${EXTENSION_NAME}-`
 const oldBranches = branches
 .split("\n")
 .filter((line) => line.includes(branchPattern))
 .map((line) => line.split("refs/heads/")[1])
 .filter(Boolean)

 if (oldBranches.length > 0) {
 console.log(`🗑️ Found ${oldBranches.length} old branch(es), deleting...`)
 for (const branch of oldBranches) {
 await $`git push https://x-access-token:${token}@github.com/${FORK_REPO}.git --delete ${branch}`
 console.log(`✅ Deleted branch ${branch}`)
 }
 }

 console.log(`🚀 Pushing to fork...`)
 await $`git push https://x-access-token:${token}@github.com/${FORK_REPO}.git ${branchName}`

 console.log(`📬 Creating pull request...`)
 const prResult =
 await $`gh pr create --repo ${UPSTREAM_REPO} --base main --head ${FORK_REPO.split("/")[0]}:${branchName} --title "Update ${EXTENSION_NAME} to v${cleanVersion}" --body "Updating OpenCode extension to v${cleanVersion}"`
 .env({ ...process.env, GH_TOKEN: prToken })
 .nothrow()

 if (prResult.exitCode !== 0) {
 console.error("stderr:", prResult.stderr.toString())
 throw new Error(`Failed with exit code ${prResult.exitCode}`)
 }

 const prUrl = prResult.stdout.toString().trim()
 console.log(`✅ Pull request created: ${prUrl}`)
 console.log(`🎉 Done!`)
}

main().catch((err) => {
 console.error("❌ Error:", err.message)
 process.exit(1)
})