feat: output generated code

Includes a few fixes that were needed to make it work.

Author: sorccu

Diff for: packages/cli/src/commands/import/plan.ts

@@ -1,3 +1,4 @@
+import { writeFileSync } from 'node:fs'
 import { Flags, ux } from '@oclif/core'
 import prompts from 'prompts'
 import chalk from 'chalk'
@@ -10,6 +11,8 @@ import commonMessages from '../../messages/common-messages'
 import { splitConfigFilePath } from '../../services/util'
 import { loadChecklyConfig } from '../../services/checkly-config-loader'
 import { ImportPlan } from '../../rest/projects'
+import { Program, Output } from '../../sourcegen'
+import { codegen } from '../../constructs/construct.codegen'
 
 export default class ImportPlanCommand extends AuthCommand {
   static coreCommand = true
@@ -50,8 +53,11 @@ export default class ImportPlanCommand extends AuthCommand {
       ux.action.start('Creating a new plan', undefined, { stdout: true })
     }
 
+    let plan: ImportPlan
+
     try {
-      const { data: plan } = await api.projects.createImportPlan(logicalId)
+      const { data } = await api.projects.createImportPlan(logicalId)
+      plan = data
 
       if (this.fancy) {
         ux.action.stop('✅ ')
@@ -74,16 +80,35 @@ export default class ImportPlanCommand extends AuthCommand {
       throw err
     }
 
+    this.#generateCode(plan)
+  }
+
+  #generateCode (plan: ImportPlan) {
     if (this.fancy) {
       ux.action.start('Generating Checkly constructs for imported resources', undefined, { stdout: true })
     }
 
     try {
-      this.log('TODO') // TODO
+      const program = new Program()
+      if (plan.changes) {
+        for (const resource of plan.changes.resources) {
+          codegen(program, resource as any)
+        }
+      }
 
       if (this.fancy) {
         ux.action.stop('✅ ')
       }
+
+      const output = new Output()
+      program.render(output)
+
+      // TODO: file structure
+      const filename = './generated_resources.ts'
+
+      writeFileSync(filename, output.finalize())
+
+      this.log(`${logSymbols.success} Generated code can be found in ${chalk.bold(filename)}`)
     } catch (err) {
       if (this.fancy) {
         ux.action.stop('❌')

Diff for: packages/cli/src/constructs/construct.codegen.ts

@@ -0,0 +1,69 @@
+import { Program } from '../sourcegen'
+
+import { codegen as alertChannelCodegen } from './alert-channel.codegen'
+import { codegen as apiCheckCodegen } from './api-check.codegen'
+import { codegen as browserCheckCodegen } from './browser-check.codegen'
+import { codegen as checkGroupCodegen } from './check-group.codegen'
+import { codegen as dashboardCodegen } from './dashboard.codegen'
+import { codegen as heartbeatCheckCodegen } from './heartbeat-check.codegen'
+import { codegen as maintenanceWindowCodegen } from './maintenance-window.codegen'
+import { codegen as multiStepCheckCodegen } from './multi-step-check.codegen'
+import { codegen as privateLocationCheckCodegen } from './private-location.codegen'
+import { codegen as tcpCheckCodegen } from './tcp-check.codegen'
+
+export type ResourceType =
+  'alert-channel-subscription' |
+  'alert-channel' |
+  'check-group' |
+  'check' |
+  'dashboard' |
+  'maintenance-window' |
+  'private-location-check-assignment' |
+  'private-location-group-assignment' |
+  'private-location'
+
+interface Resource {
+  type: ResourceType
+  logicalId: string
+  payload: any
+}
+
+export function codegen (program: Program, resource: Resource): void {
+  switch (resource.type) {
+    case 'alert-channel-subscription':
+      return // Skip temporarily
+    case 'alert-channel':
+      return alertChannelCodegen(program, resource.logicalId, resource.payload)
+    case 'check-group':
+      return checkGroupCodegen(program, resource.logicalId, resource.payload)
+    case 'check': {
+      const checkType = resource.payload.checkType
+      switch (checkType) {
+        case 'BROWSER':
+          return browserCheckCodegen(program, resource.logicalId, resource.payload)
+        case 'API':
+          return apiCheckCodegen(program, resource.logicalId, resource.payload)
+        case 'TCP':
+          return tcpCheckCodegen(program, resource.logicalId, resource.payload)
+        case 'MULTI_STEP':
+          return multiStepCheckCodegen(program, resource.logicalId, resource.payload)
+        case 'HEARTBEAT':
+          return heartbeatCheckCodegen(program, resource.logicalId, resource.payload)
+        default:
+          throw new Error(`Unable to generate for for unsupported check type '${checkType}'.`)
+      }
+    }
+    case 'dashboard':
+      return dashboardCodegen(program, resource.logicalId, resource.payload)
+    case 'maintenance-window':
+      return maintenanceWindowCodegen(program, resource.logicalId, resource.payload)
+    case 'private-location-check-assignment':
+      return // Skip temporarily
+    case 'private-location-group-assignment':
+      return // Skip temporarily
+    case 'private-location':
+      return privateLocationCheckCodegen(program, resource.logicalId, resource.payload)
+    default:
+      throw new Error(`Unable to generate for for unsupported resource type '${resource.type}'.`)
+  }
+}

Diff for: packages/cli/src/rest/projects.ts

@@ -37,7 +37,7 @@ export interface ProjectDeployResponse {
 }
 
 export interface ImportPlanChanges {
-  resources: ResourceSync
+  resources: ResourceSync[]
 }
 
 export interface ImportPlan {

Diff for: packages/cli/src/sourcegen/index.ts

@@ -25,6 +25,7 @@ export { NullValue } from './null'
 export { NumberValue } from './number'
 export { ObjectValue } from './object'
 export { object, ObjectValueBuilder } from './objectbuilder'
+export { Output } from './output'
 export { Program } from './program'
 export { StringValue } from './string'
 export { UndefinedValue } from './undefined'