From 616e25ac27e5a410693d4f0fe29914174523108f Mon Sep 17 00:00:00 2001
From: Andrew Kingston <andrew@kingston.dev>
Date: Wed, 15 Mar 2023 09:53:32 +0000
Subject: [PATCH] Add functional relationship cell

---
 .../components/sheet/KeyboardManager.svelte   |  11 +-
 .../sheet/cells/AttachmentCell.svelte         |   1 +
 .../sheet/cells/RelationshipCell.svelte       | 240 ++++++++++++++++--
 3 files changed, 224 insertions(+), 28 deletions(-)

diff --git a/packages/frontend-core/src/components/sheet/KeyboardManager.svelte b/packages/frontend-core/src/components/sheet/KeyboardManager.svelte
index bbf7bfe9a9..3e85261155 100644
--- a/packages/frontend-core/src/components/sheet/KeyboardManager.svelte
+++ b/packages/frontend-core/src/components/sheet/KeyboardManager.svelte
@@ -31,9 +31,11 @@
 
     // Pass the key event to the selected cell and let it decide whether to
     // capture it or not
-    const handled = api?.onKeyDown?.(e)
-    if (handled) {
-      return
+    if (!api?.isReadonly()) {
+      const handled = api?.onKeyDown?.(e)
+      if (handled) {
+        return
+      }
     }
     e.preventDefault()
 
@@ -120,6 +122,9 @@
   }, 100)
 
   const focusSelectedCell = () => {
+    if ($selectedCellAPI?.isReadonly()) {
+      return
+    }
     $selectedCellAPI?.focus?.()
   }
 
diff --git a/packages/frontend-core/src/components/sheet/cells/AttachmentCell.svelte b/packages/frontend-core/src/components/sheet/cells/AttachmentCell.svelte
index ed94423083..da42254b13 100644
--- a/packages/frontend-core/src/components/sheet/cells/AttachmentCell.svelte
+++ b/packages/frontend-core/src/components/sheet/cells/AttachmentCell.svelte
@@ -128,6 +128,7 @@
     text-transform: uppercase;
     font-weight: 600;
     font-size: 10px;
+    user-select: none;
   }
   img {
     height: calc(var(--cell-height) - 12px);
diff --git a/packages/frontend-core/src/components/sheet/cells/RelationshipCell.svelte b/packages/frontend-core/src/components/sheet/cells/RelationshipCell.svelte
index 3898426c12..de850f0e0f 100644
--- a/packages/frontend-core/src/components/sheet/cells/RelationshipCell.svelte
+++ b/packages/frontend-core/src/components/sheet/cells/RelationshipCell.svelte
@@ -1,48 +1,150 @@
 <script>
   import { getColor } from "../utils"
-  import { onMount } from "svelte"
+  import { onMount, getContext } from "svelte"
+  import { Icon, Input } from "@budibase/bbui"
+  import { debounce } from "../../../utils/utils"
 
   export let value
   export let api
   export let readonly
   export let selected
+  export let schema
+  export let onChange
+
+  const { API } = getContext("sheet")
 
   let isOpen = false
   let searchResults
+  let searchString
+  let definition
+  let primaryDisplay
+  let candidateIndex
 
   $: editable = selected && !readonly
+  $: search(searchString)
+  $: results = getResults(searchResults, value)
+  $: lookupMap = buildLookupMap(value, isOpen)
   $: {
     if (!selected) {
       close()
     }
   }
-  $: orderedResults = orderResults(searchResults, value)
 
-  const orderResults = () => {
-    let results = []
-    if (value?.length) {
-      results = results.concat(value)
+  const buildLookupMap = (value, isOpen) => {
+    let map = {}
+    if (!isOpen || !value?.length) {
+      return map
     }
-    if (searchResults?.length) {
-      results = results.concat(
-        searchResults.filter(result => {
-          return !value.some(x => x._id === result._id)
-        })
-      )
+    for (let i = 0; i < value.length; i++) {
+      map[value[i]._id] = true
     }
-    return results
+    return map
   }
 
-  const open = () => {
+  const isRowSelected = row => {
+    if (!row?._id) {
+      return false
+    }
+    return lookupMap?.[row._id] === true
+  }
+
+  const search = debounce(async searchString => {
+    if (!searchString || !schema?.tableId || !isOpen) {
+      candidateIndex = null
+      searchResults = []
+      return
+    }
+
+    const results = await API.searchTable({
+      paginate: false,
+      tableId: schema.tableId,
+      limit: 20,
+      query: {
+        string: {
+          [`1:${primaryDisplay}`]: searchString,
+        },
+      },
+    })
+    searchResults = results.rows?.map(row => ({
+      ...row,
+      primaryDisplay: row[primaryDisplay],
+    }))
+    candidateIndex = searchResults?.length ? 0 : null
+  }, 500)
+
+  const sortRows = rows => {
+    if (!rows?.length) {
+      return []
+    }
+    return rows.slice().sort((a, b) => {
+      return a.primaryDisplay < b.primaryDisplay ? -1 : 1
+    })
+  }
+
+  const getResults = (searchResults, value) => {
+    return searchString ? sortRows(searchResults) : sortRows(value)
+  }
+
+  const open = async () => {
     isOpen = true
+
+    // Fetch definition if required
+    if (!definition) {
+      definition = await API.fetchTableDefinition(schema.tableId)
+      primaryDisplay =
+        definition?.primaryDisplay || definition?.schema?.[0]?.name
+    }
   }
 
   const close = () => {
     isOpen = false
+    searchString = null
+    searchResults = []
+    candidateIndex = null
   }
 
-  const onKeyDown = () => {
-    return isOpen
+  const onKeyDown = e => {
+    if (!isOpen) {
+      return false
+    }
+    if (e.key === "ArrowDown") {
+      e.preventDefault()
+      if (candidateIndex == null) {
+        candidateIndex = 0
+      } else {
+        candidateIndex = Math.min(results.length - 1, candidateIndex + 1)
+      }
+    } else if (e.key === "ArrowUp") {
+      e.preventDefault()
+      if (candidateIndex === 0) {
+        candidateIndex = null
+      } else if (candidateIndex != null) {
+        candidateIndex = Math.max(0, candidateIndex - 1)
+      }
+    } else if (e.key === "Enter") {
+      if (candidateIndex != null && results[candidateIndex] != null) {
+        toggleRow(results[candidateIndex])
+      }
+    }
+    return true
+  }
+
+  const toggleRow = row => {
+    if (value?.some(x => x._id === row._id)) {
+      const newValue = value.filter(x => x._id !== row._id)
+      if (!newValue.length) {
+        candidateIndex = null
+      } else {
+        candidateIndex = Math.min(candidateIndex, newValue.length - 1)
+      }
+      onChange(newValue)
+    } else {
+      lookupMap[row._id] = true
+      onChange(sortRows([...(value || []), row]))
+      candidateIndex = null
+    }
+    searchString = null
+    searchResults = []
   }
 
   onMount(() => {
@@ -59,18 +161,54 @@
     {#if relationship.primaryDisplay}
       <div class="badge" style="--color: {getColor(idx)}">
         {relationship.primaryDisplay}
+        {relationship.primaryDisplay}
       </div>
     {/if}
   {/each}
 </div>
 
 {#if isOpen}
-  <div class="dropdown">
-    {#each orderedResults as result, idx}
-      <div class="badge" style="--color: {getColor(idx)}">
-        {result.primaryDisplay}
+  <div class="dropdown" on:wheel|stopPropagation>
+    <div class="search">
+      <Input autofocus quiet type="text" bind:value={searchString} />
+    </div>
+    {#if !searchResults?.length}
+      {#if primaryDisplay}
+        <div class="info">
+          Search for {definition.name} rows by {primaryDisplay}
+        </div>
+      {/if}
+    {:else}
+      <div class="info">
+        {searchResults.length} row{searchResults.length === 1 ? "" : "s"} found
       </div>
-    {/each}
+    {/if}
+    {#if results?.length}
+      <div class="results">
+        {#each results as row, idx}
+          <div
+            class="result"
+            on:click={() => toggleRow(row)}
+            class:candidate={idx === candidateIndex}
+            on:mouseenter={() => (candidateIndex = idx)}
+          >
+            <div class="badge" style="--color: {getColor(idx)}">
+              {row.primaryDisplay}
+              {row.primaryDisplay}
+              {row.primaryDisplay}
+              {row.primaryDisplay}
+            </div>
+            {#if isRowSelected(row)}
+              <Icon
+                size="S"
+                name="Checkmark"
+                color="var(--spectrum-global-color-blue-400)"
+              />
+            {/if}
+          </div>
+        {/each}
+      </div>
+    {/if}
   </div>
 {/if}
 
@@ -95,20 +233,72 @@
     background: var(--color);
     border-radius: var(--cell-padding);
     user-select: none;
+    overflow: hidden;
+    white-space: nowrap;
+    text-overflow: ellipsis;
+  }
+  .result .badge {
+    max-width: calc(100% - 24px);
   }
   .dropdown {
     position: absolute;
     top: -1px;
     left: -1px;
-    width: calc(100% + 2px);
-    height: 300px;
+    min-width: calc(100% + 2px);
+    max-width: calc(100% + 240px);
+    max-height: calc(var(--cell-height) + 240px);
     background: var(--cell-background);
     border: var(--cell-border);
     box-shadow: 0 0 8px 4px rgba(0, 0, 0, 0.15);
-    padding: var(--cell-padding);
     display: flex;
     flex-direction: column;
+    align-items: stretch;
+    background-color: var(--cell-background-hover);
+  }
+  .results {
+    overflow-y: auto;
+    overflow-x: hidden;
+    display: flex;
+    flex-direction: column;
+    align-items: stretch;
+  }
+  .result {
+    padding: 0 var(--cell-padding);
+    flex: 0 0 var(--cell-height);
+    display: flex;
     gap: var(--cell-spacing);
-    align-items: flex-start;
+    justify-content: space-between;
+    align-items: center;
+  }
+  .result.candidate {
+    background-color: var(--spectrum-global-color-gray-200);
+    cursor: pointer;
+  }
+
+  .search {
+    flex: 0 0 calc(var(--cell-height) - 1px);
+    display: flex;
+    align-items: center;
+    margin: 0 var(--cell-padding);
+    width: calc(100% - 2 * var(--cell-padding));
+  }
+  .search :global(.spectrum-Textfield) {
+    min-width: 0;
+    width: 100%;
+  }
+  .search :global(.spectrum-Form-item) {
+    flex: 1 1 auto;
+  }
+
+  .info {
+    color: var(--spectrum-global-color-gray-600);
+    font-size: 12px;
+    padding: var(--cell-padding);
+    flex: 0 0 var(--cell-height);
+    display: flex;
+    align-items: center;
+    white-space: nowrap;
+    text-overflow: ellipsis;
+    overflow: hidden;
   }
 </style>