From 4a2a6f0bfcd84187879f8b31911ab063b04e221e Mon Sep 17 00:00:00 2001
From: Jonathon Herbert <jonathon.herbert@guardian.co.uk>
Date: Sun, 14 Jul 2024 14:46:17 +0100
Subject: [PATCH] Add date input, focusing when a valid date is not present in
 the chip value

---
 .../src/cqlInput/TypeaheadPopover.ts          | 55 +++++++++++++++----
 prosemirror-client/src/cqlInput/plugin.ts     |  2 +-
 2 files changed, 46 insertions(+), 11 deletions(-)

diff --git a/prosemirror-client/src/cqlInput/TypeaheadPopover.ts b/prosemirror-client/src/cqlInput/TypeaheadPopover.ts
index f73f5a6..b965419 100644
--- a/prosemirror-client/src/cqlInput/TypeaheadPopover.ts
+++ b/prosemirror-client/src/cqlInput/TypeaheadPopover.ts
@@ -3,7 +3,7 @@ import { TypeaheadSuggestion } from "../services/CqlService";
 import { findNodeAt } from "./utils";
 import { EditorView } from "prosemirror-view";
 import { chip, schema } from "./schema";
-import { TextSelection } from "prosemirror-state";
+import { Selection, TextSelection } from "prosemirror-state";
 
 type MenuItem = {
   label: string;
@@ -48,6 +48,7 @@ export class TypeaheadPopover {
       suggestions.length &&
       this.view.state.selection.from === this.view.state.selection.to
     ) {
+      const currentState = this.view.state;
       const mappedSuggestions = suggestions.map((suggestion) => {
         const start = mapping.map(suggestion.from, -1);
         const end = mapping.map(suggestion.to);
@@ -58,15 +59,15 @@ export class TypeaheadPopover {
 
       const suggestionThatCoversSelection = mappedSuggestions.find(
         ({ from, to, suggestions }) =>
-          this.view.state.selection.from >= from &&
-          this.view.state.selection.to <= to &&
+          currentState.selection.from >= from &&
+          currentState.selection.to <= to &&
           Object.keys(suggestions).length
       );
 
       if (suggestionThatCoversSelection) {
         this.currentSuggestion = suggestionThatCoversSelection;
-        const { from, suggestions } = suggestionThatCoversSelection;
-        const chipPos = findNodeAt(from, this.view.state.doc, chip);
+        const { from, to, suggestions } = suggestionThatCoversSelection;
+        const chipPos = findNodeAt(from, currentState.doc, chip);
         const domSelectionAnchor = this.view.nodeDOM(chipPos) as HTMLElement;
 
         if (!domSelectionAnchor) {
@@ -82,7 +83,9 @@ export class TypeaheadPopover {
         }
 
         if (suggestions.DateSuggestion) {
-          this.renderDateSuggestion();
+          const value = this.view.state.doc.textBetween(from, to);
+
+          this.renderDateSuggestion(value);
         }
 
         this.popoverEl.showPopover?.();
@@ -115,15 +118,23 @@ export class TypeaheadPopover {
       return;
     }
 
+    this.applyValueToInput(suggestion.value);
+  };
+
+  private applyValueToInput = (value: string) => {
+    if (!this.currentSuggestion) {
+      return;
+    }
+
     const { from, to } = this.currentSuggestion;
     const tr = this.view.state.tr;
 
     this.view.dispatch(
-      tr.replaceRangeWith(from, to, schema.text(suggestion.value)).setSelection(
+      tr.replaceRangeWith(from, to, schema.text(value)).setSelection(
         // +1 to tip the selection into the next available node's text
         // position after applying the suggestion e.g. key -> value, value ->
         // searchText
-        TextSelection.near(tr.doc.resolve(from + suggestion.value.length + 1))
+        TextSelection.near(tr.doc.resolve(from + value.length + 1))
       )
     );
   };
@@ -147,8 +158,32 @@ export class TypeaheadPopover {
       .join("");
   }
 
-  private renderDateSuggestion() {
-    this.popoverEl.innerHTML = `Date goes here`;
+  private renderDateSuggestion(value: string) {
+    this.popoverEl.innerHTML = "";
+    const dateInput = document.createElement("input");
+    dateInput.classList.add("Cql_typeahead-date");
+    dateInput.setAttribute("value", value);
+    dateInput.setAttribute("type", "date");
+    dateInput.addEventListener("keydown", (e) => {
+      switch (e.key) {
+        case "Enter":
+          const value = (e.target as HTMLInputElement).value;
+          this.applyValueToInput(value);
+          this.view.dom.focus();
+          break;
+        case "Escape":
+          this.view.dom.focus();
+          break;
+      }
+    });
+
+    this.popoverEl.appendChild(dateInput);
+    const maybeValidDate = new Date(value);
+    const isDateValid =
+      maybeValidDate instanceof Date && isFinite(maybeValidDate.getTime());
+    if (!isDateValid) {
+      dateInput.focus();
+    }
   }
 
   private updateDebugSuggestions = (suggestions: TypeaheadSuggestion[]) => {
diff --git a/prosemirror-client/src/cqlInput/plugin.ts b/prosemirror-client/src/cqlInput/plugin.ts
index b2498e2..3745c52 100644
--- a/prosemirror-client/src/cqlInput/plugin.ts
+++ b/prosemirror-client/src/cqlInput/plugin.ts
@@ -308,7 +308,7 @@ export const createCqlPlugin = ({
             mapping,
           } = cqlPluginKey.getState(view.state)!;
 
-          typeaheadPopover?.updateItemsFromSuggestions(suggestions, mapping);
+          typeaheadPopover?.updateItemsFromSuggestions(suggestions, mapping, prevState.selection);
 
           if (prevQuery.trim() === currentQuery.trim()) {
             return;