feat(W-17752806): Pipeline creation must support the generation property

packages/cli/src/commands/pipelines/create.ts

@@ -2,9 +2,9 @@ import color from '@heroku-cli/color'
 import {Command, flags} from '@heroku-cli/command'
 import {StageCompletion} from '@heroku-cli/command/lib/completions'
 import {Args, ux} from '@oclif/core'
-import {prompt} from 'inquirer'
+import {prompt, type Answers, type InputQuestion, type ListQuestion} from 'inquirer'
 
-import {createCoupling, createPipeline, getAccountInfo, getTeam} from '../../lib/api'
+import {createCoupling, createPipeline, getAccountInfo, getTeam, Owner} from '../../lib/api'
 import infer from '../../lib/pipelines/infer'
 import {inferrableStageNames as stages} from '../../lib/pipelines/stages'
 
@@ -22,6 +22,12 @@ export default class Create extends Command {
 
   static flags = {
     app: flags.app({required: true}),
+    generation: flags.string({
+      description: 'generation of the the app execution environment',
+      completion: ['fir', 'cedar'],
+      default: 'cedar',
+      options: ['fir', 'cedar'],
+    }),
     remote: flags.remote(),
     stage: flags.string({
       name: 'stage',
@@ -46,9 +52,9 @@ export default class Create extends Command {
 
     let name
     let stage
-    let owner: any
+    let owner: Owner
     const guesses = infer(flags.app)
-    const questions: any = []
+    const questions: (InputQuestion | ListQuestion)[] = []
 
     const app = flags.app
 
@@ -79,18 +85,18 @@ export default class Create extends Command {
     const ownerType = teamName ? 'team' : 'user'
 
     // If team or org is not specified, we assign ownership to the user creating
-    owner = teamName ? await getTeam(this.heroku, teamName) : await getAccountInfo(this.heroku)
-    owner = owner.body
+    const response = teamName ? await getTeam(this.heroku, teamName) : await getAccountInfo(this.heroku)
+    owner = response.body
     const ownerID = owner.id
 
     owner = {id: ownerID, type: ownerType}
 
-    const answers: any = await prompt(questions)
+    const answers: Answers = await prompt(questions)
     if (answers.name) name = answers.name
     if (answers.stage) stage = answers.stage
 
     ux.action.start(`Creating ${name} pipeline`)
-    const {body: pipeline}: any = await createPipeline(this.heroku, name, owner)
+    const {body: pipeline} = await createPipeline(this.heroku, name, owner, flags.generation)
     ux.action.stop()
 
     ux.action.start(`Adding ${color.app(app)} to ${color.pipeline(pipeline.name)} pipeline as ${stage}`)

packages/cli/src/lib/api.ts

@@ -8,6 +8,8 @@ export const FILTERS_HEADER = `${V3_HEADER}.filters`
 export const PIPELINES_HEADER = `${V3_HEADER}.pipelines`
 const CI_HEADER = `${V3_HEADER}.ci`
 
+export type Owner = Pick<Heroku.Account, 'id' | 'type'> | Pick<Heroku.Team, 'id' | 'type'>
+
 export function createAppSetup(heroku: APIClient, body: {body: any}) {
   return heroku.post<Heroku.AppSetup>('/app-setups', {body})
 }
@@ -22,11 +24,11 @@ export function createCoupling(heroku: APIClient, pipeline: any, app: string, st
   return postCoupling(heroku, pipeline.id, app, stage)
 }
 
-export function createPipeline(heroku: APIClient, name: any, owner: any) {
+export function createPipeline(heroku: APIClient, name: string, owner: Owner, generationName: string = 'cedar') {
   return heroku.request<Heroku.Pipeline>('/pipelines', {
     method: 'POST',
     headers: {Accept: PIPELINES_HEADER},
-    body: {name, owner},
+    body: {name, owner, generation: {name: generationName}},
   })
 }