Wget option Added along with a new dropdown menu

__tests__/text-classification.spec.js

@@ -32,12 +32,15 @@ test('text classification simple', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
-
-  await download.saveAs('./dist-tests/text-classification-simple.zip')
+  await page.getByRole('button', { name: 'Code' }).click()
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
+
+  await downloadPromise.saveAs('./dist-tests/text-classification-simple.zip')
 })
 
 test('text classification all', async () => {
@@ -85,12 +88,14 @@ test('text classification all', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
-
-  await download.saveAs('./dist-tests/text-classification-all.zip')
+  await page.getByRole('button', { name: 'Code' }).click()
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
+  await downloadPromise.saveAs('./dist-tests/text-classification-all.zip')
 })
 
 test('text classification launch', async () => {
@@ -107,12 +112,15 @@ test('text classification launch', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
-
-  await download.saveAs('./dist-tests/text-classification-launch.zip')
+  await page.getByRole('button', { name: 'Code' }).click()
+  await page.getByRole('button', { name: 'Download Zip' }).click()
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
+  await downloadPromise.saveAs('./dist-tests/text-classification-launch.zip')
 })
 
 test('text classification spawn', async () => {
@@ -130,10 +138,13 @@ test('text classification spawn', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
-
-  await download.saveAs('./dist-tests/text-classification-spawn.zip')
+  await page.getByRole('button', { name: 'Code' }).click()
+  await page.getByRole('button', { name: 'Download Zip' }).click()
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
+  await downloadPromise.saveAs('./dist-tests/text-classification-spawn.zip')
 })

__tests__/vision-classification.spec.js

@@ -32,12 +32,17 @@ test('vision classification simple', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
+  await page.getByRole('button', { name: 'Code' }).click()
+  await page.getByRole('button', { name: 'Download Zip' }).click()
 
-  await download.saveAs('./dist-tests/vision-classification-simple.zip')
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
+
+  await downloadPromise.saveAs('./dist-tests/vision-classification-simple.zip')
 })
 
 test('vision classification all', async () => {
@@ -85,12 +90,15 @@ test('vision classification all', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
-
-  await download.saveAs('./dist-tests/vision-classification-all.zip')
+  await page.getByRole('button', { name: 'Code' }).click()
+  await page.getByRole('button', { name: 'Download Zip' }).click()
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
+  await downloadPromise.saveAs('./dist-tests/vision-classification-all.zip')
 })
 
 test('vision classification launch', async () => {
@@ -107,12 +115,15 @@ test('vision classification launch', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
-
-  await download.saveAs('./dist-tests/vision-classification-launch.zip')
+  await page.getByRole('button', { name: 'Code' }).click()
+  await page.getByRole('button', { name: 'Download Zip' }).click()
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
+  await downloadPromise.saveAs('./dist-tests/vision-classification-launch.zip')
 })
 
 test('vision classification spawn', async () => {
@@ -130,10 +141,13 @@ test('vision classification spawn', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
-
-  await download.saveAs('./dist-tests/vision-classification-spawn.zip')
+  await page.getByRole('button', { name: 'Code' }).click()
+  await page.getByRole('button', { name: 'Download Zip' }).click()
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
+  await downloadPromise.saveAs('./dist-tests/vision-classification-spawn.zip')
 })

__tests__/vision-dcgan.spec.js

@@ -32,12 +32,16 @@ test('vision dcgan simple', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
-
-  await download.saveAs('./dist-tests/vision-dcgan-simple.zip')
+  await page.getByRole('button', { name: 'Code' }).click()
+  await page.getByRole('button', { name: 'Download Zip' }).click()
+
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
+  await downloadPromise.saveAs('./dist-tests/vision-dcgan-simple.zip')
 })
 
 test('vision dcgan all', async () => {
@@ -85,12 +89,17 @@ test('vision dcgan all', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
+  await page.getByRole('button', { name: 'Code' }).click()
+  await page.getByRole('button', { name: 'Download Zip' }).click()
+
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
 
-  await download.saveAs('./dist-tests/vision-dcgan-all.zip')
+  await downloadPromise.saveAs('./dist-tests/vision-dcgan-all.zip')
 })
 
 test('vision dcgan launch', async () => {
@@ -107,12 +116,17 @@ test('vision dcgan launch', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
+  await page.getByRole('button', { name: 'Code' }).click()
+  await page.getByRole('button', { name: 'Download Zip' }).click()
+
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
 
-  await download.saveAs('./dist-tests/vision-dcgan-launch.zip')
+  await downloadPromise.saveAs('./dist-tests/vision-dcgan-launch.zip')
 })
 
 test('vision dcgan spawn', async () => {
@@ -130,10 +144,14 @@ test('vision dcgan spawn', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
+  await page.getByRole('button', { name: 'Code' }).click()
+  await page.getByRole('button', { name: 'Download Zip' }).click()
 
-  await download.saveAs('./dist-tests/vision-dcgan-spawn.zip')
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
+  await downloadPromise.saveAs('./dist-tests/vision-dcgan-spawn.zip')
 })

__tests__/vision-segmentation.spec.js

@@ -32,12 +32,17 @@ test('vision segmentation simple', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
+  await page.getByRole('button', { name: 'Code' }).click()
+  await page.getByRole('button', { name: 'Download Zip' }).click()
 
-  await download.saveAs('./dist-tests/vision-segmentation-simple.zip')
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
+
+  await downloadPromise.saveAs('./dist-tests/vision-segmentation-simple.zip')
 })
 
 test('vision segmentation all', async () => {
@@ -85,12 +90,17 @@ test('vision segmentation all', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
+  await page.getByRole('button', { name: 'Code' }).click()
+  await page.getByRole('button', { name: 'Download Zip' }).click()
+
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
 
-  await download.saveAs('./dist-tests/vision-segmentation-all.zip')
+  await downloadPromise.saveAs('./dist-tests/vision-segmentation-all.zip')
 })
 
 test('vision segmentation launch', async () => {
@@ -107,12 +117,17 @@ test('vision segmentation launch', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
+  await page.getByRole('button', { name: 'Code' }).click()
+  await page.getByRole('button', { name: 'Download Zip' }).click()
 
-  await download.saveAs('./dist-tests/vision-segmentation-launch.zip')
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
+
+  await downloadPromise.saveAs('./dist-tests/vision-segmentation-launch.zip')
 })
 
 test('vision segmentation spawn', async () => {
@@ -130,10 +145,15 @@ test('vision segmentation spawn', async () => {
   await page.click('text=Loggers')
   await page.click('text=config.yaml')
 
-  const [download] = await Promise.all([
-    page.waitForEvent('download'),
-    page.click('button:has-text("Download")')
-  ])
+  await page.getByRole('button', { name: 'Code' }).click()
+  await page.getByRole('button', { name: 'Download Zip' }).click()
+
+  const downloadPromise = await page
+    .waitForEvent('download', { timeout: 2000 })
+    .catch(() => {
+      page.getByRole('button', { name: 'Download Zip' }).click()
+      return page.waitForEvent('download')
+    })
 
-  await download.saveAs('./dist-tests/vision-segmentation-spawn.zip')
+  await downloadPromise.saveAs('./dist-tests/vision-segmentation-spawn.zip')
 })

functions/code.js

@@ -0,0 +1,41 @@
+// Below imports are defined in
+// `external_node_modules` of [functions] in netlify.toml
+// They are required for this function to run
+
+import { v4 as uuidv4 } from 'uuid'
+import JSZip from 'jszip'
+import { pushToGitHub } from './utils'
+
+const nbUid = uuidv4()
+const repoOwner = process.env.VUE_APP_GH_USER
+const repo = process.env.VUE_APP_GH_REPO
+
+// This function is the one Netlify function runs on
+// https://docs.netlify.com/functions/build-with-javascript/#synchronous-function-format
+exports.handler = async function (event, _) {
+  // event is a JSON object
+  const data = JSON.parse(event.body)
+  const zip = new JSZip()
+  const code = data.code
+  const template = `ignite-${data.template}`
+
+  // As usual from Download component,
+  // we will zip the files and
+  // generate a base64 format for pushing to GitHub
+  // with Octokit.
+  for (const filename in code) {
+    zip.file(filename, code[filename])
+  }
+  const content = await zip.generateAsync({ type: 'base64' })
+  const zipRes = await pushToGitHub(
+    content,
+    `${template}.zip`,
+    nbUid,
+    repoOwner,
+    repo
+  )
+  return {
+    statusCode: 200,
+    body: zipRes
+  }
+}