From a6b19fc8bff58ad9b179b8a50f8aa288557dad82 Mon Sep 17 00:00:00 2001
From: Siddharth Gelera <ahoy@barelyhuman.dev>
Date: Wed, 17 Sep 2025 16:04:50 +0530
Subject: [PATCH] fix: imports for ws files

---
 .../components/RequestPane/WsBody/index.js    | 75 -------------------
 .../bruno-app/src/utils/collections/index.js  | 12 ++-
 .../bruno-app/src/utils/importers/common.js   |  9 ++-
 packages/bruno-electron/src/ipc/collection.js |  2 +-
 .../bruno-filestore/src/formats/bru/index.ts  |  2 +-
 5 files changed, 19 insertions(+), 81 deletions(-)

diff --git a/packages/bruno-app/src/components/RequestPane/WsBody/index.js b/packages/bruno-app/src/components/RequestPane/WsBody/index.js
index 215d68686..818c7b511 100644
--- a/packages/bruno-app/src/components/RequestPane/WsBody/index.js
+++ b/packages/bruno-app/src/components/RequestPane/WsBody/index.js
@@ -60,49 +60,6 @@ const SingleWSMessage = ({
   };
   const onSave = () => dispatch(saveRequest(item.uid, collection.uid));
 
-  const onRegenerateMessage = async () => {
-    try {
-      const methodPath = item.draft?.request?.method || item.request?.method;
-
-      if (!methodPath) {
-        toastError(new Error('Method path not found in request'));
-        return;
-      }
-
-      // Find the method metadata from the appropriate cache
-      let methodMetadata = null;
-
-      const result = await generateGrpcSampleMessage(methodPath, content, {
-        arraySize: 2,
-        methodMetadata // Pass the method metadata to the function
-      });
-
-      if (result.success) {
-        const currentMessages = [...(body.ws || [])];
-
-        currentMessages[index] = {
-          name: name ? name : `message ${index + 1}`,
-          content: result.message
-        };
-
-        dispatch(
-          updateRequestBody({
-            content: currentMessages,
-            itemUid: item.uid,
-            collectionUid: collection.uid
-          })
-        );
-
-        toast.success('Sample message generated successfully!');
-      } else {
-        toastError(new Error(result.error || 'Failed to generate sample message'));
-      }
-    } catch (error) {
-      console.error('Error generating sample message:', error);
-      toastError(error);
-    }
-  };
-
   const onDeleteMessage = () => {
     const currentMessages = [...(body.ws || [])];
 
@@ -168,38 +125,6 @@ const SingleWSMessage = ({
             </button>
           </ToolHint>
 
-          <ToolHint text="Generate a new sample message based on schema" toolhintId={`regenerate-msg-${index}`}>
-            <button
-              onClick={onRegenerateMessage}
-              className="p-1 rounded hover:bg-zinc-200 dark:hover:bg-zinc-600 transition-colors"
-            >
-              <IconRefresh size={16} strokeWidth={1.5} className="text-zinc-700 dark:text-zinc-300" />
-            </button>
-          </ToolHint>
-
-          {canClientStream && (
-            <ToolHint
-              text={isConnectionActive ? 'Send WS message' : 'Connection not active'}
-              toolhintId={`send-msg-${index}`}
-            >
-              <button
-                onClick={onSend}
-                disabled={!isConnectionActive}
-                className={`p-1 rounded ${
-                  isConnectionActive ? 'hover:bg-zinc-200 dark:hover:bg-zinc-600' : 'opacity-50 cursor-not-allowed'
-                } transition-colors`}
-              >
-                <IconSend
-                  size={16}
-                  strokeWidth={1.5}
-                  className={`${
-                    isConnectionActive ? 'text-zinc-700 dark:text-zinc-300' : 'text-zinc-400 dark:text-zinc-500'
-                  }`}
-                />
-              </button>
-            </ToolHint>
-          )}
-
           {index > 0 && (
             <ToolHint text="Delete this message" toolhintId={`delete-msg-${index}`}>
               <button
diff --git a/packages/bruno-app/src/utils/collections/index.js b/packages/bruno-app/src/utils/collections/index.js
index 54c813198..98a2bc1c5 100644
--- a/packages/bruno-app/src/utils/collections/index.js
+++ b/packages/bruno-app/src/utils/collections/index.js
@@ -261,7 +261,8 @@ export const transformCollectionToSaveToExportAsFile = (collection, options = {}
             formUrlEncoded: copyFormUrlEncodedParams(si.request.body.formUrlEncoded),
             multipartForm: copyMultipartFormParams(si.request.body.multipartForm),
             file: copyFileParams(si.request.body.file),
-            grpc: si.request.body.grpc
+            grpc: si.request.body.grpc,
+            ws: si.request.body.ws
           },
           script: si.request.script,
           vars: si.request.vars,
@@ -423,6 +424,13 @@ export const transformCollectionToSaveToExportAsFile = (collection, options = {}
             content: replaceTabsWithSpaces(content)
           }));
         }
+
+        if (di.request.body.mode === 'ws') {
+          di.request.body.ws = di.request.body.ws.map(({ name, content }, index) => ({
+            name: name ? name : `message ${index + 1}`,
+            content: replaceTabsWithSpaces(content)
+          }));
+        }
       }
 
       if (si.type == 'folder' && si?.root) {
@@ -658,7 +666,7 @@ export const transformRequestToSaveToFilesystem = (item) => {
   if (itemToSave.request.body.mode === 'ws') {
     itemToSave.request.body = {
       ...itemToSave.request.body,
-      grpc: itemToSave.request.body.ws.map(({ name, content }, index) => ({
+      ws: itemToSave.request.body.ws.map(({ name, content }, index) => ({
         name: name ? name : `message ${index + 1}`,
         content: replaceTabsWithSpaces(content)
       }))
diff --git a/packages/bruno-app/src/utils/importers/common.js b/packages/bruno-app/src/utils/importers/common.js
index 5d3b21d72..ad67510a1 100644
--- a/packages/bruno-app/src/utils/importers/common.js
+++ b/packages/bruno-app/src/utils/importers/common.js
@@ -62,9 +62,10 @@ export const updateUidsInCollection = (_collection) => {
 export const transformItemsInCollection = (collection) => {
   const transformItems = (items = []) => {
     each(items, (item) => {
-      if (['http', 'graphql', 'grpc'].includes(item.type)) {
+      if (['http', 'graphql', 'grpc', 'ws'].includes(item.type)) {
         item.type = `${item.type}-request`;
         const isGrpcRequest = item.type === 'grpc-request';
+        const isWSRequest = item.type === 'ws-request';
 
         if (item.request.query) {
           item.request.params = item.request.query.map((queryItem) => ({
@@ -77,6 +78,11 @@ export const transformItemsInCollection = (collection) => {
         if (isGrpcRequest) {
           delete item.request.params;
         }
+        
+        if (isWSRequest) {
+          delete item.request.params;
+          delete item.request.method;
+        }
 
         delete item.request.query;
 
@@ -100,7 +106,6 @@ export const transformItemsInCollection = (collection) => {
   };
 
   transformItems(collection.items);
-
   return collection;
 };
 
diff --git a/packages/bruno-electron/src/ipc/collection.js b/packages/bruno-electron/src/ipc/collection.js
index 6675b029e..acffec8cc 100644
--- a/packages/bruno-electron/src/ipc/collection.js
+++ b/packages/bruno-electron/src/ipc/collection.js
@@ -601,7 +601,7 @@ const registerRendererEventHandlers = (mainWindow, watcher, lastOpenedCollection
       // Recursive function to parse the collection items and create files/folders
       const parseCollectionItems = (items = [], currentPath) => {
         items.forEach(async (item) => {
-          if (['http-request', 'graphql-request', 'grpc-request'].includes(item.type)) {
+          if (['http-request', 'graphql-request', 'grpc-request', 'ws-request'].includes(item.type)) {
             let sanitizedFilename = sanitizeName(item?.filename || `${item.name}.bru`);
             const content = await stringifyRequestViaWorker(item);
             const filePath = path.join(currentPath, sanitizedFilename);
diff --git a/packages/bruno-filestore/src/formats/bru/index.ts b/packages/bruno-filestore/src/formats/bru/index.ts
index 36ed8afc1..bdc8584b5 100644
--- a/packages/bruno-filestore/src/formats/bru/index.ts
+++ b/packages/bruno-filestore/src/formats/bru/index.ts
@@ -81,7 +81,7 @@ export const bruRequestToJson = (data: string | any, parsed: boolean = false): a
       transformedJson.request.auth.mode = _.get(json, 'ws.auth', 'none');
       transformedJson.request.body = _.get(json, 'body', {
         mode: 'ws',
-        grpc: _.get(json, 'body.ws', [
+        ws: _.get(json, 'body.ws', [
           {
             name: 'message 1',
             content: '{}'