chat: implement first part of automention and also fix bug with restoring selection

Author: williamstein

src/packages/frontend/chat/message.tsx

@@ -768,6 +768,17 @@ export default function Message(props: Readonly<Props>) {
       return;
     }
     const replyDate = -getThreadRootDate({ date, messages });
+    let input;
+    if (isLLMThread) {
+      input = "";
+    } else {
+      const replying_to = message.get("history")?.first()?.get("author_id");
+      if (!replying_to || replying_to == props.account_id) {
+        input = "";
+      } else {
+        input = `<span class="user-mention" account-id=${replying_to} >@${editor_name}</span> `;
+      }
+    }
     return (
       <div style={{ marginLeft: mode === "standalone" ? "30px" : "0" }}>
         <ChatInput
@@ -778,7 +789,7 @@ export default function Message(props: Readonly<Props>) {
             height: "auto" /* for some reason the default 100% breaks things */,
           }}
           cacheId={`${props.path}${props.project_id}${date}-reply`}
-          input={""}
+          input={input}
           submitMentionsRef={replyMentionsRef}
           on_send={sendReply}
           height={"auto"}

src/packages/frontend/editors/markdown-input/multimode.tsx

@@ -204,10 +204,10 @@ export default function MultiMarkdownInput(props: Props) {
 
   const editBar2 = useRef<JSX.Element | undefined>(undefined);
 
+  const getKey = () => `${project_id}${path}:${cacheId}`;
+
   function getCache() {
-    return cacheId === undefined
-      ? undefined
-      : multimodeStateCache.get(`${project_id}${path}:${cacheId}`);
+    return cacheId == null ? undefined : multimodeStateCache.get(getKey());
   }
 
   const [mode, setMode0] = useState<Mode>(
@@ -248,23 +248,34 @@ export default function MultiMarkdownInput(props: Props) {
   } | null>(null);
 
   useEffect(() => {
-    if (cacheId == null) return;
+    if (cacheId == null) {
+      return;
+    }
     const cache = getCache();
     if (cache?.[mode] != null && selectionRef.current != null) {
       // restore selection on mount.
       try {
         selectionRef.current.setSelection(cache?.[mode]);
       } catch (_err) {
-        // console.warn(_err);  // definitely don't need this.
-        // This is expected to fail, since the selection from last
-        // use will be invalid now if another user changed the
-        // document, etc., or you did in a different mode, possibly.
+        // it might just be that the document isn't initialized yet
+        setTimeout(() => {
+          try {
+            selectionRef.current?.setSelection(cache?.[mode]);
+          } catch (_err2) {
+            //  console.warn(_err2); // definitely don't need this.
+            // This is expected to fail, since the selection from last
+            // use will be invalid now if another user changed the
+            // document, etc., or you did in a different mode, possibly.
+          }
+        }, 100);
       }
     }
     return () => {
-      if (selectionRef.current == null || cacheId == null) return;
+      if (selectionRef.current == null || cacheId == null) {
+        return;
+      }
       const selection = selectionRef.current.getSelection();
-      multimodeStateCache.set(`${project_id}${path}:${cacheId}`, {
+      multimodeStateCache.set(getKey(), {
         ...getCache(),
         [mode]: selection,
       });