From 9f85b6154dc08e03350eb3808b5b5c90dab7491d Mon Sep 17 00:00:00 2001
From: Pavel <nybkox@gmail.com>
Date: Thu, 24 Jul 2025 18:48:58 +0200
Subject: [PATCH] feat: look into cursor's firstBatch in mongosh res

---
 plugins/dbgate-plugin-mongo/src/backend/driver.js | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/plugins/dbgate-plugin-mongo/src/backend/driver.js b/plugins/dbgate-plugin-mongo/src/backend/driver.js
index 6fde7d4bf..3752634c7 100644
--- a/plugins/dbgate-plugin-mongo/src/backend/driver.js
+++ b/plugins/dbgate-plugin-mongo/src/backend/driver.js
@@ -103,6 +103,8 @@ const driver = {
       return printable;
     } else if ('documents' in printable) {
       return printable.documents;
+    } else if ('cursor' in printable && 'firstBatch' in printable.cursor) {
+      return printable.cursor.firstBatch;
     }
 
     return printable;
@@ -169,6 +171,11 @@ const driver = {
         for (const row of printable.documents) {
           options.row(row);
         }
+      } else if ('cursor' in printable && 'firstBatch' in printable.cursor) {
+        options.recordset({ __isDynamicStructure: true });
+        for (const row of printable.cursor.firstBatch) {
+          options.row(row);
+        }
       } else {
         options.info({
           printable: printable,