make sure there's initial source node

Author: trangdoan982

apps/roam/src/components/FuzzySelectInput.tsx

@@ -21,6 +21,7 @@ type FuzzySelectInputProps<T extends Result = Result> = {
   placeholder?: string;
   autoFocus?: boolean;
   disabled?: boolean;
+  initialIsLocked?: boolean;
 };
 
 const FuzzySelectInput = <T extends Result = Result>({
@@ -33,9 +34,9 @@ const FuzzySelectInput = <T extends Result = Result>({
   placeholder = "Enter value",
   autoFocus,
   disabled,
+  initialIsLocked,
 }: FuzzySelectInputProps<T>) => {
-  const [isLocked, setIsLocked] = useState(false);
-  const [lockedValue, setLockedValue] = useState<T | undefined>(undefined);
+  const [isLocked, setIsLocked] = useState(initialIsLocked || false);
   const [query, setQuery] = useState<string>(() => value?.text || "");
   const [isOpen, setIsOpen] = useState(false);
   const [activeIndex, setActiveIndex] = useState(0);
@@ -57,7 +58,6 @@ const FuzzySelectInput = <T extends Result = Result>({
     (item: T) => {
       if (mode === "create" && item.uid && item.uid !== initialUid) {
         // Lock the value
-        setLockedValue(item);
         setIsLocked(true);
         setQuery(item.text);
         setValue(item);
@@ -76,7 +76,6 @@ const FuzzySelectInput = <T extends Result = Result>({
   // Handle clear locked value
   const handleClear = useCallback(() => {
     setIsLocked(false);
-    setLockedValue(undefined);
     setQuery("");
     setValue({ text: "", uid: "" } as T);
     onLockedChange?.(false);
@@ -162,12 +161,12 @@ const FuzzySelectInput = <T extends Result = Result>({
   }
 
   // Create mode: locked value display
-  if (isLocked && lockedValue) {
+  if (isLocked || initialIsLocked) {
     return (
       <div className="flex w-full items-center gap-2">
         <div className="flex flex-1 items-center gap-2 rounded border border-gray-300 bg-gray-100 px-3 py-2 dark:border-gray-600 dark:bg-gray-800">
           <span className="flex-1 text-gray-900 dark:text-gray-100">
-            {lockedValue.text}
+            {value?.text}
           </span>
           <Button
             icon="cross"

apps/roam/src/components/ModifyNodeDialog.tsx

@@ -72,7 +72,9 @@ const ModifyNodeDialog = ({
     initialReferencedNode?.uid || "",
   );
   const [isContentLocked, setIsContentLocked] = useState(false);
-  const [isReferencedNodeLocked, setIsReferencedNodeLocked] = useState(false);
+  const [isReferencedNodeLocked, setIsReferencedNodeLocked] = useState(
+    Boolean(initialReferencedNode?.uid),
+  );
   const [contentOptions, setContentOptions] = useState<Result[]>([]);
   const [referencedNodeOptions, setReferencedNodeOptions] = useState<Result[]>(
     [],
@@ -482,6 +484,7 @@ const ModifyNodeDialog = ({
                 onLockedChange={setIsReferencedNodeLocked}
                 mode={"create"}
                 initialUid={referencedNodeUid}
+                initialIsLocked={isReferencedNodeLocked}
               />
             </div>
           )}

apps/roam/src/utils/renderNodeTagPopup.tsx

@@ -1,11 +1,15 @@
 import React from "react";
 import ReactDOM from "react-dom";
 import { Button, Popover, Position } from "@blueprintjs/core";
-import { OnloadArgs } from "roamjs-components/types";
+import { OnloadArgs, PullBlock } from "roamjs-components/types";
 import getUids from "roamjs-components/dom/getUids";
 import getTextByBlockUid from "roamjs-components/queries/getTextByBlockUid";
 import { type DiscourseNode } from "./getDiscourseNodes";
 import { renderModifyNodeDialog } from "~/components/ModifyNodeDialog";
+import { getReferencedNodeInFormat } from "./formatUtils";
+import discourseNodeFormatToDatalog from "./discourseNodeFormatToDatalog";
+import compileDatalog from "./compileDatalog";
+import getPageUidByPageTitle from "roamjs-components/queries/getPageUidByPageTitle";
 
 export const renderNodeTagPopupButton = (
   parent: HTMLSpanElement,
@@ -43,7 +47,38 @@ export const renderNodeTagPopupButton = (
   const cleanedBlockText = rawBlockText.replace(textContent, "").trim();
 
   const getInitialReferencedNode = () => {
-    // TODO: Implement this in a follow-up PR
+    if (!blockUid) return { text: "", uid: "" };
+
+    const referencedNodeType = getReferencedNodeInFormat({
+      format: matchedNode.format,
+    });
+
+    if (!referencedNodeType) return { text: "", uid: "" };
+
+    try {
+      const referenced = window.roamAlphaAPI.data.fast.q(
+        `[:find (pull ?r [:node/title :block/string]) :where [?b :block/uid "${blockUid}"] (or-join [?b ?r] (and [?b :block/parents ?p] [?p :block/refs ?r]) (and [?b :block/page ?r])) ${discourseNodeFormatToDatalog(
+          {
+            freeVar: "r",
+            ...referencedNodeType,
+          },
+        )
+          .map((c) => compileDatalog(c, 0))
+          .join(" ")}]`,
+      )?.[0]?.[0] as PullBlock;
+
+      if (referenced) {
+        const title =
+          referenced[":node/title"] || referenced[":block/string"] || "";
+        if (title) {
+          const uid = getPageUidByPageTitle(title);
+          return { text: title, uid: uid };
+        }
+      }
+    } catch (error) {
+      console.error("Error getting initial referenced node:", error);
+    }
+
     return { text: "", uid: "" };
   };