add fallback for multi-staged dockerfiles

Author: dobrac

packages/cli/src/commands/template/migrate.ts

@@ -1,14 +1,14 @@
+import { select } from '@inquirer/prompts'
+import CodeBlockWriter from 'code-block-writer'
 import * as commander from 'commander'
+import { Template, TemplateBuilder, TemplateFinal } from 'e2b'
 import * as fs from 'fs'
 import * as path from 'path'
-import { select } from '@inquirer/prompts'
-import CodeBlockWriter from 'code-block-writer'
-import { getRoot } from 'src/utils/filesystem'
+import { E2BConfig, getConfigPath, loadConfig } from 'src/config'
+import { defaultDockerfileName } from 'src/docker/constants'
 import { pathOption } from 'src/options'
-import { getConfigPath, loadConfig, E2BConfig } from 'src/config'
+import { getRoot } from 'src/utils/filesystem'
 import { asLocal, asLocalRelative, asPrimary } from 'src/utils/format'
-import { defaultDockerfileName } from 'src/docker/constants'
-import { Template, TemplateFinal } from 'e2b'
 import { getDockerfile } from './build'
 
 enum Language {
@@ -334,7 +334,30 @@ async function migrateToLanguage(
   const { dockerfileContent } = getDockerfile(root, dockerfilePath)
 
   // Parse Dockerfile using SDK
-  const baseTemplate = template.fromDockerfile(dockerfileContent)
+  let baseTemplate: TemplateBuilder
+  try {
+    baseTemplate = template.fromDockerfile(dockerfileContent)
+  } catch (error) {
+    console.warn(
+      "\n⚠️  Unfortunately, we weren't able to fully convert the template to the new SDK format."
+    )
+    console.warn(
+      '\nPlease build the Docker image manually, push it to a repository of your choice, and then reference it.'
+    )
+    console.warn("\nHere's an example of how to build the Docker image:")
+    console.warn(
+      `   ${asPrimary(
+        'docker build -f e2b.Dockerfile --platform linux/amd64 -t your-image-tag .'
+      )}`
+    )
+    console.warn(
+      '\nAfter building and pushing your image to a repository of your choice, update the generated template files to use the actual image tag.'
+    )
+    if (error instanceof Error) {
+      console.warn('\nCause:', error.message)
+    }
+    baseTemplate = template.fromImage('my-custom-image')
+  }
 
   // Apply config start/ready commands
   let parsedTemplate: TemplateFinal = baseTemplate

packages/cli/tests/commands/template/fixtures/multi-stage/e2b.Dockerfile

@@ -0,0 +1,9 @@
+FROM node:18 AS builder
+WORKDIR /app
+COPY package*.json ./
+RUN npm ci
+
+FROM node:18-slim
+WORKDIR /app
+COPY --from=builder /app/node_modules ./node_modules
+CMD ["node", "index.js"]

packages/cli/tests/commands/template/fixtures/multi-stage/e2b.toml

@@ -0,0 +1,2 @@
+template_id = "multi-stage"
+dockerfile = "e2b.Dockerfile"

packages/cli/tests/commands/template/fixtures/multi-stage/expected/python-async/build_dev.py

@@ -0,0 +1,14 @@
+import asyncio
+from e2b import AsyncTemplate
+from template import template
+
+
+async def main():
+    await AsyncTemplate.build(
+        template,
+        alias="multi-stage-dev",
+    )
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

packages/cli/tests/commands/template/fixtures/multi-stage/expected/python-async/build_prod.py

@@ -0,0 +1,14 @@
+import asyncio
+from e2b import AsyncTemplate
+from template import template
+
+
+async def main():
+    await AsyncTemplate.build(
+        template,
+        alias="multi-stage",
+    )
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

packages/cli/tests/commands/template/fixtures/multi-stage/expected/python-async/template.py

@@ -0,0 +1,6 @@
+from e2b import AsyncTemplate
+
+template = (
+    AsyncTemplate()
+    .from_image("my-custom-image")
+)

packages/cli/tests/commands/template/fixtures/multi-stage/expected/python-sync/build_dev.py

@@ -0,0 +1,8 @@
+from e2b import Template
+from template import template
+
+
+Template.build(
+    template,
+    alias="multi-stage-dev",
+)

packages/cli/tests/commands/template/fixtures/multi-stage/expected/python-sync/build_prod.py

@@ -0,0 +1,8 @@
+from e2b import Template
+from template import template
+
+
+Template.build(
+    template,
+    alias="multi-stage",
+)

packages/cli/tests/commands/template/fixtures/multi-stage/expected/python-sync/template.py

@@ -0,0 +1,6 @@
+from e2b import Template
+
+template = (
+    Template()
+    .from_image("my-custom-image")
+)

packages/cli/tests/commands/template/fixtures/multi-stage/expected/typescript/build.dev.ts

@@ -0,0 +1,6 @@
+import { Template } from 'e2b'
+import { template } from './template'
+
+await Template.build(template, {
+  alias: 'multi-stage-dev',
+})