beta.ts 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167
  1. #!/usr/bin/env bun
  2. interface PR {
  3. number: number
  4. headRefName: string
  5. headRefOid: string
  6. createdAt: string
  7. isDraft: boolean
  8. title: string
  9. }
  10. async function main() {
  11. console.log("Fetching open contributor PRs...")
  12. const prsResult =
  13. await $`gh pr list --label contributor --state open --json number,headRefName,headRefOid,createdAt,isDraft,title --limit 100`.nothrow()
  14. if (prsResult.exitCode !== 0) {
  15. throw new Error(`Failed to fetch PRs: ${prsResult.stderr}`)
  16. }
  17. const allPRs: PR[] = JSON.parse(prsResult.stdout)
  18. const prs = allPRs.filter((pr) => !pr.isDraft)
  19. console.log(`Found ${prs.length} open non-draft contributor PRs`)
  20. console.log("Fetching latest dev branch...")
  21. const fetchDev = await $`git fetch origin dev`.nothrow()
  22. if (fetchDev.exitCode !== 0) {
  23. throw new Error(`Failed to fetch dev branch: ${fetchDev.stderr}`)
  24. }
  25. console.log("Checking out beta branch...")
  26. const checkoutBeta = await $`git checkout -B beta origin/dev`.nothrow()
  27. if (checkoutBeta.exitCode !== 0) {
  28. throw new Error(`Failed to checkout beta branch: ${checkoutBeta.stderr}`)
  29. }
  30. const applied: number[] = []
  31. const skipped: Array<{ number: number; reason: string }> = []
  32. for (const pr of prs) {
  33. console.log(`\nProcessing PR #${pr.number}: ${pr.title}`)
  34. // Try to update PR branch via GitHub API (rebase onto base branch)
  35. console.log(` Attempting to rebase PR #${pr.number} via GitHub API...`)
  36. const updateBranch = await $`gh pr update-branch ${pr.number} --rebase`.nothrow()
  37. if (updateBranch.exitCode !== 0) {
  38. console.log(` Rebase failed for PR #${pr.number} (has conflicts)`)
  39. console.log(` Error: ${updateBranch.stderr}`)
  40. skipped.push({ number: pr.number, reason: "Rebase failed (conflicts)" })
  41. continue
  42. }
  43. console.log(` Rebase initiated for PR #${pr.number}`)
  44. // Wait for rebase to complete by polling PR state
  45. console.log(` Waiting for rebase to complete...`)
  46. let rebaseComplete = false
  47. let attempts = 0
  48. const maxAttempts = 30
  49. while (!rebaseComplete && attempts < maxAttempts) {
  50. await Bun.sleep(2000) // Wait 2 seconds
  51. attempts++
  52. const prCheck = await $`gh pr view ${pr.number} --json mergeStateStatus,headRefOid`.nothrow()
  53. if (prCheck.exitCode === 0) {
  54. const prData = JSON.parse(prCheck.stdout)
  55. // mergeStateStatus will be "clean" when rebase is complete and no conflicts
  56. if (prData.mergeStateStatus === "clean") {
  57. rebaseComplete = true
  58. console.log(` Rebase completed for PR #${pr.number}`)
  59. }
  60. }
  61. }
  62. if (!rebaseComplete) {
  63. console.log(` Timeout waiting for rebase on PR #${pr.number}`)
  64. skipped.push({ number: pr.number, reason: "Timeout waiting for rebase" })
  65. continue
  66. }
  67. // Fetch the rebased PR
  68. const fetchPR = await $`git fetch origin pull/${pr.number}/head:pr-${pr.number}`.nothrow()
  69. if (fetchPR.exitCode !== 0) {
  70. console.log(` Failed to fetch PR #${pr.number} after rebase, skipping`)
  71. skipped.push({ number: pr.number, reason: "Failed to fetch after rebase" })
  72. continue
  73. }
  74. const merge = await $`git merge --squash pr-${pr.number}`.nothrow()
  75. if (merge.exitCode !== 0) {
  76. console.log(` Squash merge failed for PR #${pr.number}`)
  77. console.log(` Error: ${merge.stderr}`)
  78. await $`git reset --hard HEAD`.nothrow()
  79. skipped.push({ number: pr.number, reason: `Squash merge failed: ${merge.stderr}` })
  80. continue
  81. }
  82. const add = await $`git add -A`.nothrow()
  83. if (add.exitCode !== 0) {
  84. console.log(` Failed to stage changes for PR #${pr.number}`)
  85. await $`git reset --hard HEAD`.nothrow()
  86. skipped.push({ number: pr.number, reason: "Failed to stage" })
  87. continue
  88. }
  89. const status = await $`git status --porcelain`.nothrow()
  90. if (status.exitCode !== 0 || !status.stdout.trim()) {
  91. console.log(` No changes to commit for PR #${pr.number}, skipping`)
  92. await $`git reset --hard HEAD`.nothrow()
  93. skipped.push({ number: pr.number, reason: "No changes to commit" })
  94. continue
  95. }
  96. const commitMsg = `Apply PR #${pr.number}: ${pr.title}`
  97. const commit = await Bun.spawn(["git", "commit", "-m", commitMsg], { stdout: "pipe", stderr: "pipe" })
  98. const commitExit = await commit.exited
  99. const commitStderr = await Bun.readableStreamToText(commit.stderr)
  100. if (commitExit !== 0) {
  101. console.log(` Failed to commit PR #${pr.number}`)
  102. console.log(` Error: ${commitStderr}`)
  103. await $`git reset --hard HEAD`.nothrow()
  104. skipped.push({ number: pr.number, reason: `Commit failed: ${commitStderr}` })
  105. continue
  106. }
  107. console.log(` Successfully applied PR #${pr.number}`)
  108. applied.push(pr.number)
  109. }
  110. console.log("\n--- Summary ---")
  111. console.log(`Applied: ${applied.length} PRs`)
  112. applied.forEach((num) => console.log(` - PR #${num}`))
  113. console.log(`Skipped: ${skipped.length} PRs`)
  114. skipped.forEach((x) => console.log(` - PR #${x.number}: ${x.reason}`))
  115. console.log("\nForce pushing beta branch...")
  116. const push = await $`git push origin beta --force`.nothrow()
  117. if (push.exitCode !== 0) {
  118. throw new Error(`Failed to push beta branch: ${push.stderr}`)
  119. }
  120. console.log("Successfully synced beta branch")
  121. }
  122. main().catch((err) => {
  123. console.error("Error:", err)
  124. process.exit(1)
  125. })
  126. function $(strings: TemplateStringsArray, ...values: unknown[]) {
  127. const cmd = strings.reduce((acc, str, i) => acc + str + (values[i] ?? ""), "")
  128. return {
  129. async nothrow() {
  130. const proc = Bun.spawn(cmd.split(" "), {
  131. stdout: "pipe",
  132. stderr: "pipe",
  133. })
  134. const exitCode = await proc.exited
  135. const stdout = await new Response(proc.stdout).text()
  136. const stderr = await new Response(proc.stderr).text()
  137. return { exitCode, stdout, stderr }
  138. },
  139. }
  140. }