mongo distinct field values

packages/web/src/datagrid/DataFilterControl.svelte

@@ -33,6 +33,7 @@
   export let pureName = null;
   export let schemaName = null;
   export let columnName = null;
+  export let uniqueName = null;
 
   let value;
   let isError;
@@ -223,7 +224,7 @@
       multiselect: true,
       schemaName,
       pureName,
-      columnName,
+      field: columnName || uniqueName,
       onConfirm: keys => setFilter(keys.map(x => getFilterValueExpression(x)).join(',')),
     });
   }
@@ -271,7 +272,7 @@
       <InlineButton on:click={handleShowDictionary} narrow square>
         <FontIcon icon="icon dots-horizontal" />
       </InlineButton>
-    {:else if pureName && columnName}
+    {:else if (pureName && columnName) || (pureName && uniqueName && driver?.databaseEngineTypes?.includes('document'))}
       <InlineButton on:click={handleShowValuesModal} narrow square>
         <FontIcon icon="icon dots-vertical" />
       </InlineButton>

packages/web/src/datagrid/DataGridCore.svelte

@@ -1535,6 +1535,7 @@
                   onGetReference={value => (domFilterControlsRef.get()[col.uniqueName] = value)}
                   foreignKey={col.foreignKey}
                   columnName={col.uniquePath.length == 1 ? col.uniquePath[0] : null}
+                  uniqueName={col.uniqueName}
                   pureName={col.pureName}
                   schemaName={col.schemaName}
                   {conid}

packages/web/src/modals/ValueLookupModal.svelte

@@ -14,13 +14,14 @@
   import FormTextField from '../forms/FormTextField.svelte';
   import _ from 'lodash';
   import { apiCall } from '../utility/api';
+  import ErrorInfo from '../elements/ErrorInfo.svelte';
 
   export let onConfirm;
   export let conid;
   export let database;
   export let pureName;
   export let schemaName;
-  export let columnName;
+  export let field;
   export let driver;
   export let multiselect = false;
 
@@ -41,7 +42,7 @@
       search,
       schemaName,
       pureName,
-      field: columnName,
+      field,
     });
 
     isLoading = false;
@@ -59,7 +60,7 @@
 
 <FormProvider>
   <ModalBase {...$$restProps}>
-    <svelte:fragment slot="header">Choose value from {columnName}</svelte:fragment>
+    <svelte:fragment slot="header">Choose value from {field}</svelte:fragment>
 
     <!-- <FormTextField name="search" label='Search' placeholder="Search" bind:value={search} /> -->
     <div class="largeFormMarker">
@@ -71,6 +72,9 @@
     {/if}
 
     {#if !isLoading && rows}
+      {#if rows.errorMessage}
+        <ErrorInfo message={rows.errorMessage} />
+      {:else}
         <div class="tableWrapper">
           <ScrollableTableControl
             {rows}
@@ -117,6 +121,7 @@
           </ScrollableTableControl>
         </div>
       {/if}
+    {/if}
 
     <svelte:fragment slot="footer">
       {#if multiselect}

plugins/dbgate-plugin-mongo/src/backend/driver.js

@@ -277,6 +277,54 @@ const driver = {
     const db = pool.db(name);
     await db.createCollection('collection1');
   },
+
+  async loadFieldValues(pool, name, field, search) {
+    try {
+      const collection = pool.__getDatabase().collection(name.pureName);
+      // console.log('options.condition', JSON.stringify(options.condition, undefined, 2));
+
+      const pipelineMatch = [];
+
+      if (search) {
+        const tokens = _.compact(search.split(' ').map((x) => x.trim()));
+        if (tokens.length > 0) {
+          pipelineMatch.push({
+            $match: {
+              $and: tokens.map((token) => ({
+                [field]: {
+                  $regex: `.*${token}.*`,
+                  $options: 'i',
+                },
+              })),
+            },
+          });
+        }
+      }
+
+      let cursor = await collection.aggregate([
+        ...pipelineMatch,
+        {
+          $group: { _id: '$' + field },
+        },
+        {
+          $sort: { _id: 1 },
+        },
+        {
+          $limit: 100,
+        },
+      ]);
+      const rows = await cursor.toArray();
+      return _.uniqBy(
+        rows.map(transformMongoData).map(({ _id }) => {
+          if (_.isArray(_id) || _.isPlainObject(_id)) return { value: null };
+          return { value: _id };
+        }),
+        (x) => x.value
+      );
+    } catch (err) {
+      return { errorMessage: err.message };
+    }
+  },
 };
 
 module.exports = driver;