From 2ce0678475231797ae20e45e0f00d74ca77600b2 Mon Sep 17 00:00:00 2001
From: Jakub Bednar <jakub.bednar@gmail.com>
Date: Thu, 13 Jul 2023 17:13:40 +0200
Subject: [PATCH] chore(client_libs): update Node.js wizard to use new v3
 client

---
 .../OptionAccordion/QueryDataAccordion.tsx    |  1 +
 .../containers/NodejsWizard.tsx               | 56 +++++++++++++++++--
 2 files changed, 53 insertions(+), 4 deletions(-)

diff --git a/src/homepageExperience/components/OptionAccordion/QueryDataAccordion.tsx b/src/homepageExperience/components/OptionAccordion/QueryDataAccordion.tsx
index 597f7a7a6c..b70f0a080f 100644
--- a/src/homepageExperience/components/OptionAccordion/QueryDataAccordion.tsx
+++ b/src/homepageExperience/components/OptionAccordion/QueryDataAccordion.tsx
@@ -47,6 +47,7 @@ export const QueryDataAccordion: FC = () => {
     'C#': 'new-user-setup/csharp',
     Go: 'new-user-setup/golang',
     Java: 'new-user-setup/java',
+    'Node.js': 'new-user-setup/nodejs',
     Python: 'new-user-setup/python',
   }
 
diff --git a/src/homepageExperience/containers/NodejsWizard.tsx b/src/homepageExperience/containers/NodejsWizard.tsx
index 5c69325c29..62eb1db0af 100644
--- a/src/homepageExperience/containers/NodejsWizard.tsx
+++ b/src/homepageExperience/containers/NodejsWizard.tsx
@@ -29,7 +29,7 @@ import {event} from 'src/cloud/utils/reporting'
 import {
   scrollNextPageIntoView,
   HOMEPAGE_NAVIGATION_STEPS,
-  HOMEPAGE_NAVIGATION_STEPS_WRITE_ONLY,
+  HOMEPAGE_NAVIGATION_STEPS_SQL,
 } from 'src/homepageExperience/utils'
 import {isFlagEnabled} from 'src/shared/utils/featureFlag'
 
@@ -67,7 +67,7 @@ export class NodejsWizard extends PureComponent<null, State> {
   }
 
   subwayNavSteps = isFlagEnabled('ioxOnboarding')
-    ? HOMEPAGE_NAVIGATION_STEPS_WRITE_ONLY
+    ? HOMEPAGE_NAVIGATION_STEPS_SQL
     : HOMEPAGE_NAVIGATION_STEPS
 
   handleNextClick = () => {
@@ -131,7 +131,53 @@ export class NodejsWizard extends PureComponent<null, State> {
     scrollNextPageIntoView()
   }
 
-  renderStep = () => {
+  renderSqlStep = () => {
+    switch (this.state.currentStep) {
+      case 1: {
+        return <Overview wizard="nodejsWizard" />
+      }
+      case 2: {
+        return <InstallDependencies />
+      }
+      case 3: {
+        return (
+          <Tokens
+            wizardEventName="nodejsWizard"
+            setTokenValue={this.setTokenValue}
+            tokenValue={this.state.tokenValue}
+          />
+        )
+      }
+      case 4: {
+        return <InitializeClient />
+      }
+      case 5: {
+        return <WriteData onSelectBucket={this.handleSelectBucket} />
+      }
+      case 6: {
+        return <ExecuteQuery bucket={this.state.selectedBucket} />
+      }
+      case 7: {
+        return <ExecuteAggregateQuery bucket={this.state.selectedBucket} />
+      }
+      case 8: {
+        return (
+          <Finish
+            wizardEventName="nodejsWizard"
+            markStepAsCompleted={this.handleMarkStepAsCompleted}
+            finishStepCompleted={this.state.finishStepCompleted}
+            finalFeedback={this.state.finalFeedback}
+            setFinalFeedback={this.setFinalFeedback}
+          />
+        )
+      }
+      default: {
+        return <Overview wizard="nodejsWizard" />
+      }
+    }
+  }
+
+  renderFluxStep = () => {
     switch (this.state.currentStep) {
       case 1: {
         return <Overview wizard="nodejsWizard" />
@@ -219,7 +265,9 @@ export class NodejsWizard extends PureComponent<null, State> {
                 )}
               >
                 <WriteDataDetailsContextProvider>
-                  {this.renderStep()}
+                  {isFlagEnabled('ioxOnboarding')
+                    ? this.renderSqlStep()
+                    : this.renderFluxStep()}
                 </WriteDataDetailsContextProvider>
               </div>