uploadToK6Cloud

README.md

@@ -2,8 +2,15 @@
 
 Enhances Visual Studio Code with the ability to run k6 tests both on your local machine and in the Grafana Cloud.
 
+## Commands
+
 ![VS Code k6 Commands](vscode-commands.png)
 
+- Run current file
+- Run current file in k6 cloud
+- Upload current file to k6 cloud
+- Open Settings
+
 ### For more information
 
 - [k6.io](https://k6.io)

package.json

@@ -45,6 +45,10 @@
         "command": "k6.runWithK6Cloud",
         "title": "k6: Run current file in k6 cloud"
       },
+      {
+        "command": "k6.uploadToK6Cloud",
+        "title": "k6: Upload current file to k6 cloud"
+      },
       {
         "command": "k6.openSettings",
         "title": "k6: Open Settings"

src/commands/upload/index.ts

@@ -0,0 +1,61 @@
+import * as vscode from 'vscode';
+import { spawn } from 'child_process';
+import openSettings from '../settings';
+
+export default async () => {
+  const currentFile = vscode.window.activeTextEditor?.document.uri;
+  if (currentFile?.scheme !== 'file') {
+    vscode.window.showErrorMessage(
+      'The k6 extension currently only works with files saved on disk'
+    );
+    return;
+  }
+
+  if (
+    !vscode.workspace.getConfiguration('k6').get('cloudToken') &&
+    !process.env.K6_CLOUD_TOKEN
+  ) {
+    const openSettingsButton = { title: 'Open k6 settings' };
+    const action = await vscode.window.showErrorMessage(
+      'This option requires you to either have a cloud token added in your environment variables or in the k6 extension settings',
+      {},
+      openSettingsButton
+    );
+    if (action === openSettingsButton) {
+      openSettings('k6.cloudToken');
+    }
+    return;
+  }
+
+  const channel = vscode.window.createOutputChannel('k6');
+  channel.show();
+  channel.appendLine('Starting k6...');
+
+  const k6 = spawn('k6', ['cloud', '--upload-only', currentFile.fsPath], {
+    env: {
+      ...process.env,
+      K6_CLOUD_TOKEN:
+        vscode.workspace.getConfiguration('k6').get('cloudToken') ??
+        process.env.K6_CLOUD_TOKEN,
+    },
+  });
+
+  k6.stdout.on('data', (data) => channel.append(data.toString()));
+  k6.stderr.on('data', (data) => channel.append(data.toString()));
+
+  k6.on('error', (e) => {
+    if (e.message.includes('ENOENT')) {
+      channel.appendLine(
+        "Looks like you've not installed k6 into your PATH. Aborting."
+      );
+    } else {
+      channel.appendLine(
+        'Something went wrong while launching k6. See the error below for details'
+      );
+      channel.appendLine(JSON.stringify(e));
+    }
+  });
+  k6.on('exit', (code) => {
+    channel.appendLine('k6 exited with exit code ' + code);
+  });
+};

src/extension.ts

@@ -3,6 +3,7 @@
 import * as vscode from 'vscode';
 
 import runWithK6 from './commands/run';
+import uploadToK6Cloud from './commands/upload';
 import detect from './commands/detect';
 import openSettings from './commands/settings';
 
@@ -29,6 +30,7 @@ export function activate(context: vscode.ExtensionContext) {
 
   addEditorCommand(context, 'k6.runWithK6', () => runWithK6(false));
   addEditorCommand(context, 'k6.runWithK6Cloud', () => runWithK6(true));
+  addEditorCommand(context, 'k6.uploadToK6Cloud', () => uploadToK6Cloud());
   addCommand(context, 'k6.openSettings', () => openSettings());
 }