handle integration

stripe · Aug 12, 2024 · 567c0ce · 567c0ce
1 parent 3d8d9d5
commit 567c0ce
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 6 deletions.
diff --git a/src/commands.ts b/src/commands.ts
@@ -623,12 +623,15 @@ export class Commands {
     });
   };
 
-  createStripeSample = async (stripeSamples: StripeSamples, sample?: string) => {
+  createStripeSample = async (stripeSamples: StripeSamples, sample?: string, integration?: string) => {
     this.telemetry.sendEvent('createStripeSample');
     if (typeof sample !== 'string') {
       sample = '';
     }
-    await stripeSamples.selectAndCloneSample(sample);
+    if (typeof integration !== 'string') {
+      integration = '';
+    }
+    await stripeSamples.selectAndCloneSample(sample, integration);
   };
 
   createWebhookEndpoint = async (

diff --git a/src/extension.ts b/src/extension.ts
@@ -138,7 +138,7 @@ export function activate(this: any, context: ExtensionContext) {
 
   const commandCallbackPairs: [string, (...args: any[]) => any][] = [
     ['stripe.createStripeSample',
-      (sampleName?: string) => stripeCommands.createStripeSample(stripeSamples, sampleName ?? ''),
+      (sampleName?: string, integration?: string) => stripeCommands.createStripeSample(stripeSamples, sampleName ?? '', integration ?? ''),
     ],
     ['stripe.login', () => stripeCommands.startLogin(stripeDaemon)],
     ['stripe.openCLI', stripeCommands.openCLI],
@@ -204,8 +204,10 @@ export function activate(this: any, context: ExtensionContext) {
         if (uri.path === '/createStripeSample') {
           const params = new URLSearchParams(uri.query);
           const sampleName = params.get('sample');
+          const integration = params.get('integration');
           console.log('Sample name:', sampleName);
-          vscode.commands.executeCommand('stripe.createStripeSample', sampleName);
+          console.log('Integration from URI:', integration);
+          vscode.commands.executeCommand('stripe.createStripeSample', sampleName, integration);
         }
       }
     })

diff --git a/src/stripeSamples.ts b/src/stripeSamples.ts
@@ -35,7 +35,7 @@ export class StripeSamples {
    * Show a menu with a list of Stripe samples, prompt for sample options, clone the sample, and
    * prompt to open the sample.
    */
-  selectAndCloneSample = async (sample?: string) => {
+  selectAndCloneSample = async (sample?: string, integration?: string) => {
     try {
       this.daemonClient = await this.stripeDaemon.setupClient();
     } catch (e: any) {
@@ -60,7 +60,13 @@ export class StripeSamples {
 
       const sampleName = selectedSample.sampleData.name;
 
-      const selectedIntegration = await this.promptIntegration(selectedSample);
+      const integrationsList = await this.getConfigsForSample(sampleName);
+      let selectedIntegration: SampleConfigsResponse.Integration | undefined
+      if (integration) {
+        selectedIntegration = integrationsList.find((i) => i.getIntegrationName() === integration);
+      } else {
+        selectedIntegration = await this.promptIntegration(selectedSample);
+      }
       if (!selectedIntegration) {
         return;
       }