From db7604c0963c1b7b2f3ec106b85c78722f81fedc Mon Sep 17 00:00:00 2001
From: David Vargas <dvargas92495@gmail.com>
Date: Fri, 23 Dec 2022 13:31:17 -0500
Subject: [PATCH] Sparql query fix

---
 src/sparql.tsx | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/sparql.tsx b/src/sparql.tsx
index fadc82f..19382f0 100644
--- a/src/sparql.tsx
+++ b/src/sparql.tsx
@@ -49,6 +49,7 @@ import apiGet from "roamjs-components/util/apiGet";
 import extractTag from "roamjs-components/util/extractTag";
 import renderOverlay from "roamjs-components/util/renderOverlay";
 import addStyle from "roamjs-components/dom/addStyle";
+import getFirstChildUidByBlockUid from "roamjs-components/queries/getFirstChildUidByBlockUid";
 
 // https://github.com/spamscanner/url-regex-safe/blob/master/src/index.js
 const protocol = `(?:https?://)`;
@@ -72,7 +73,7 @@ export type RenderProps = {
     };
   };
   parentUid: string;
-  location: { "window-id": string; "block-uid": string };
+  blockUid: string;
 };
 
 export const DEFAULT_EXPORT_LABEL = "SPARQL Import";
@@ -275,7 +276,7 @@ const SparqlQuery = ({
   onClose,
   queriesCache,
   parentUid,
-  location,
+  blockUid: _blockUid,
 }: {
   onClose: () => void;
 } & RenderProps): React.ReactElement => {
@@ -286,8 +287,8 @@ const SparqlQuery = ({
     [parentUid]
   );
   const cursorBlockUid = useMemo(
-    () => (getTextByBlockUid(parentUid) ? parentUid : location["block-uid"]),
-    [parentUid]
+    () => (getTextByBlockUid(parentUid) ? parentUid : _blockUid),
+    [parentUid, _blockUid]
   );
   const cursorBlockString = useMemo(
     () => extractTag(getTextByBlockUid(cursorBlockUid)),
@@ -738,7 +739,9 @@ const initializeSparql = () => {
         .getOpenPageOrBlockUid()
         .then((parentUid) =>
           render({
-            location: window.roamAlphaAPI.ui.getFocusedBlock(),
+            blockUid:
+              window.roamAlphaAPI.ui.getFocusedBlock()?.["block-uid"] ||
+              getFirstChildUidByBlockUid(parentUid),
             queriesCache,
             parentUid,
           })