feat: add setting page for updating keploy yaml (#71)

Signed-off-by: Hermione Dadheech <[email protected]> Signed-off-by: Shivam Sourav Jha <[email protected]> Co-authored-by: Shivam Sourav Jha <[email protected]>
keploy · Oct 7, 2024 · 27edc34 · 27edc34
1 parent bd0e690
commit 27edc34
Show file tree

Hide file tree

Showing 4 changed files with 300 additions and 8 deletions.
diff --git a/src/Config.ts b/src/Config.ts
@@ -1,5 +1,56 @@
 import * as vscode from 'vscode';
 import { existsSync, readFileSync } from 'fs';
+import * as path from 'path';
+import * as yaml from 'js-yaml';
+import { writeFileSync } from 'fs';
+export interface PartialKeployConfig {
+  appName:string;
+  command:string;
+  containerName:string;
+  networkName: string;
+  test: {
+    delay: number;
+    apiTimeout: number;
+    mongoPassword: string;
+  };
+}
+
+// Function to update keploy.yaml
+export async function updateKeployYaml(newConfig: PartialKeployConfig) {
+  const workspaceFolders = vscode.workspace.workspaceFolders;
+  if (!workspaceFolders) {
+      vscode.window.showErrorMessage('No workspace folder found');
+      return;
+  }
+
+  const keployFilePath = path.join(workspaceFolders[0].uri.fsPath, 'keploy.yml');
+
+  if (!existsSync(keployFilePath)) {
+      vscode.window.showErrorMessage('keploy.yaml file not found');
+      return;
+  }
+
+  try {
+      const fileContents = readFileSync(keployFilePath, 'utf8');
+      const config = yaml.load(fileContents) as any;  // Load the YAML content into an object
+      console.log('config', config);
+      config.appName = newConfig.appName;
+      config.command = newConfig.command;
+      config.containerName = newConfig.containerName;
+      config.networkName = newConfig.networkName;
+      config.test.delay = newConfig.test.delay;
+      config.test.apiTimeout = newConfig.test.apiTimeout;
+      config.test.mongoPassword = newConfig.test.mongoPassword;
+
+      // Convert the updated config object back to YAML format
+      const newYamlContent = yaml.dump(config);
+      writeFileSync(keployFilePath, newYamlContent, 'utf8');
+
+      vscode.window.showInformationMessage('Keploy config updated successfully!');
+  } catch (error) {
+      vscode.window.showErrorMessage(`Failed to update keploy.yaml: ${error}`);
+  }
+}
 
 export async function handleOpenKeployConfigFile(webview: any) {
   const folderPath = vscode.workspace.workspaceFolders?.[0].uri.fsPath;

diff --git a/src/SidebarProvider.ts b/src/SidebarProvider.ts
@@ -5,12 +5,13 @@ import { getNonce } from "./Utils";
 import { startRecording, stopRecording } from "./Record";
 import { startTesting, stopTesting, displayTestCases, displayPreviousTestResults } from "./Test";
 import { existsSync } from "fs";
-import { handleInitializeKeployConfigFile, handleOpenKeployConfigFile } from "./Config";
+import { handleInitializeKeployConfigFile, handleOpenKeployConfigFile,updateKeployYaml,PartialKeployConfig } from "./Config";
 import SignInWithGitHub from "./SignIn";
 import oneClickInstall from './OneClickInstall';
 import * as path from 'path';
 import * as fs from 'fs';
 import { workspace } from 'vscode';
+const yaml = require('js-yaml');
 
 function precheckFunction(): Promise<string> {
   const workspacePath = workspace.workspaceFolders ? workspace.workspaceFolders[0].uri.fsPath : '';
@@ -152,6 +153,34 @@ export class SidebarProvider implements vscode.WebviewViewProvider {
     });
     webviewView.webview.onDidReceiveMessage(async (data) => {
       switch (data.type) {
+        case "getKeployConfig":{
+          // Load the keploy.yml config file
+          const workspaceFolders = vscode.workspace.workspaceFolders;
+          if (!workspaceFolders) {
+            vscode.window.showErrorMessage('No workspace folder found');
+            return;
+          }
+
+          const keployFilePath = path.join(workspaceFolders[0].uri.fsPath, 'keploy.yml');
+
+          if (!fs.existsSync(keployFilePath)) {
+            vscode.window.showErrorMessage('keploy.yml file not found');
+            return;
+          }
+
+          try {
+            const fileContents = fs.readFileSync(keployFilePath, 'utf8');
+            const config = yaml.load(fileContents); // Parse YAML into JS object
+
+            // Send the config data back to the webview
+            webviewView.webview.postMessage({
+              type: 'keployConfig',
+              config: config,
+            });
+          } catch (err) {
+            vscode.window.showErrorMessage(`Error reading keploy.yml: ${err}`);
+          }
+        }
         case "onInfo": {
           if (!data.value) {
             return;
@@ -542,10 +571,30 @@ export class SidebarProvider implements vscode.WebviewViewProvider {
           }
           break;
         }
+
+        case "updateKeployConfig": {
+          // Collect the updated data from the UI
+          const newConfig: PartialKeployConfig = {
+            appName: data.config.appName,
+            command: data.config.command,
+            containerName: data.config.containerName,
+            networkName: data.config.networkName,
+            test: {
+              delay: data.config.test?.delay,
+              apiTimeout: data.config.test?.apiTimeout,
+              mongoPassword: data.config.test?.mongoPassword,
+            },
+          };
+
+          // Call the function to update the YAML config
+          await updateKeployYaml(newConfig);
+          break;
+        }
 
       }
 
     });
+
   }
   private _startApiResponseUpdates() {
     this._interval = setInterval(() => {

diff --git a/webviews/components/IntegrationTest.svelte b/webviews/components/IntegrationTest.svelte
@@ -599,7 +599,7 @@
    display: flex;
    flex-direction: row;
    align-items: start;
-   padding: 4vw 5vw;
+   padding: 2.5vw 5vw;
    width: 100%; /* Full width of the container */
    justify-content: flex-end; /* Align icons to the left */
  }