Add script to restore Github integrations (#9700)

Signed-off-by: Artem Savchenko <armisav@gmail.com>
2026-09-27 20:14:57 +02:00 · 2025-08-20 12:13:29 +07:00
parent 47402753ff
commit b438971926
3 changed files with 134 additions and 188 deletions
@@ -0,0 +1,128 @@
+import { type IntegrationKind, type PersonId, type WorkspaceUuid, systemAccountUuid } from '@hcengineering/core'
+import { getAccountClient } from '@hcengineering/server-client'
+import { generateToken } from '@hcengineering/server-token'
+import { type Integration } from '@hcengineering/account-client'
+import { getDBClient } from '@hcengineering/postgres'
+
+const GITHUB_INTEGRATION: IntegrationKind = 'github' as any
+
+interface IntegrationSetting {
+  workspaceId: WorkspaceUuid
+  createdBy: PersonId
+  installationIds: number[]
+}
+
+export async function restoreGithubIntegrations (dbUrl: string, dryrun: boolean): Promise<void> {
+  try {
+    const pg = getDBClient(dbUrl)
+    const pgClient = await pg.getClient()
+    const token = generateToken(systemAccountUuid, undefined, { service: 'admin', admin: 'true' })
+    const accountClient = getAccountClient(token)
+
+    const integrationSettings = await pgClient<
+    { workspaceId: WorkspaceUuid, createdBy: PersonId, installationId: number }[]
+    >`
+      SELECT "workspaceId", "createdBy", data -> 'installationId' "installationId"
+      FROM github
+      WHERE _class='github:class:GithubIntegration'
+    `
+
+    // Group by workspace and createdBy, collecting installationIds
+    const uniqueSettings = groupIntegrationSettings(integrationSettings)
+    console.info('Start restoring GitHub installation IDs, count: ', uniqueSettings.length)
+
+    let createdCount = 0
+    let updatedCount = 0
+    for (const setting of uniqueSettings) {
+      try {
+        const existingIntegration = await accountClient.getIntegration({
+          workspaceUuid: setting.workspaceId,
+          socialId: setting.createdBy,
+          kind: GITHUB_INTEGRATION
+        })
+
+        // Determine the installationId to use
+        const installationId =
+          setting.installationIds.length === 1 ? setting.installationIds[0] : setting.installationIds // Use array if multiple values
+
+        if (existingIntegration != null) {
+          const existingInstallationId = existingIntegration?.data?.installationId
+          const isSame = Array.isArray(installationId)
+            ? Array.isArray(existingInstallationId) &&
+              installationId.length === existingInstallationId.length &&
+              installationId.every((id) => existingInstallationId.includes(id))
+            : existingInstallationId === installationId
+
+          if (isSame) {
+            if (dryrun) {
+              console.info('Dry run: skip existing integration', existingIntegration)
+            }
+            continue
+          }
+
+          const updatedData = {
+            ...(existingIntegration.data ?? {}),
+            installationId
+          }
+          const updatedIntegration: Integration = {
+            ...existingIntegration,
+            data: updatedData
+          }
+          if (dryrun) {
+            console.info('Dry run: would update integration', existingIntegration, updatedIntegration)
+            continue
+          }
+          await accountClient.updateIntegration(updatedIntegration)
+          updatedCount++
+        } else {
+          const integration: Integration = {
+            workspaceUuid: setting.workspaceId,
+            socialId: setting.createdBy,
+            kind: GITHUB_INTEGRATION,
+            data: {
+              installationId
+            }
+          }
+          if (dryrun) {
+            console.info('Dry run: would create integration', integration)
+            continue
+          }
+          await accountClient.createIntegration(integration)
+          createdCount++
+        }
+      } catch (e: any) {
+        console.error('Error restoring GitHub integration', setting.createdBy, setting.workspaceId, e)
+      }
+    }
+    console.info(`Finished restoring GitHub integrations, updated: ${updatedCount}, created: ${createdCount}`)
+  } catch (e) {
+    console.error('Failed to restore GitHub integrations', e)
+  }
+}
+
+function groupIntegrationSettings (
+  settings: Array<{ workspaceId: WorkspaceUuid, createdBy: PersonId, installationId: number }>
+): IntegrationSetting[] {
+  const groupedSettings = new Map<string, IntegrationSetting>()
+
+  for (const setting of settings) {
+    if (setting.installationId == null) continue
+
+    const key = `${setting.workspaceId}-${setting.createdBy}`
+    const existing = groupedSettings.get(key)
+
+    if (existing != null) {
+      if (!existing.installationIds.includes(setting.installationId)) {
+        existing.installationIds.push(setting.installationId)
+      }
+    } else {
+      groupedSettings.set(key, {
+        workspaceId: setting.workspaceId,
+        createdBy: setting.createdBy,
+        installationIds: [setting.installationId]
+      })
+    }
+  }
+
+  return Array.from(groupedSettings.values())
+}