@totalreclaw/totalreclaw 1.6.0 → 3.0.6

package/subgraph-search.ts

@@ -21,6 +21,7 @@
  */
 
 import { getSubgraphConfig } from './subgraph-store.js';
+import { CONFIG } from './config.js';
 
 export interface SubgraphSearchFact {
   id: string;
@@ -32,9 +33,9 @@ export interface SubgraphSearchFact {
 }
 
 /** Batch size for Phase 2 split queries. */
-const TRAPDOOR_BATCH_SIZE = parseInt(process.env.TOTALRECLAW_TRAPDOOR_BATCH_SIZE ?? '5', 10);
+const TRAPDOOR_BATCH_SIZE = CONFIG.trapdoorBatchSize;
 /** Graph Studio / Graph Network hard limit on `first` argument. */
-const PAGE_SIZE = parseInt(process.env.TOTALRECLAW_SUBGRAPH_PAGE_SIZE ?? '1000', 10);
+const PAGE_SIZE = CONFIG.pageSize;
 
 /**
  * Execute a single GraphQL query against the subgraph endpoint.
@@ -59,10 +60,18 @@ async function gqlQuery<T>(
       headers,
       body: JSON.stringify({ query, variables }),
     });
-    if (!response.ok) return null;
-    const json = await response.json() as { data?: T };
+    if (!response.ok) {
+      const body = await response.text().catch(() => '');
+      console.error(`[TotalReclaw] Subgraph query failed: HTTP ${response.status} — ${body.slice(0, 200)}`);
+      return null;
+    }
+    const json = await response.json() as { data?: T; error?: string; errors?: Array<{ message: string }> };
+    if (json.error || json.errors) {
+      console.error(`[TotalReclaw] Subgraph query error: ${json.error || json.errors?.map(e => e.message).join('; ')}`);
+    }
     return json.data ?? null;
-  } catch {
+  } catch (err) {
+    console.error(`[TotalReclaw] Subgraph query exception: ${err instanceof Error ? err.message : String(err)}`);
     return null;
   }
 }
@@ -83,6 +92,7 @@ const SEARCH_QUERY = `
         encryptedEmbedding
         decayScore
         timestamp
+        createdAt
         isActive
         contentFp
         sequenceId
@@ -107,6 +117,7 @@ const PAGINATE_QUERY = `
         encryptedBlob
         encryptedEmbedding
         timestamp
+        createdAt
         decayScore
         isActive
         contentFp
@@ -249,6 +260,102 @@ export async function searchSubgraph(
   return Array.from(allResults.values());
 }
 
+/**
+ * Broadened search: fetch recent active facts by owner without trapdoor filtering.
+ * Used as a fallback when trapdoor search returns 0 candidates (e.g., vague queries
+ * like "who am I?" where word trapdoors don't overlap with stored fact tokens).
+ */
+export async function searchSubgraphBroadened(
+  owner: string,
+  maxCandidates: number,
+  authKeyHex?: string,
+): Promise<SubgraphSearchFact[]> {
+  const config = getSubgraphConfig();
+  const subgraphUrl = `${config.relayUrl}/v1/subgraph`;
+
+  const query = `
+    query BroadenedSearch($owner: Bytes!, $first: Int!) {
+      facts(
+        where: { owner: $owner, isActive: true }
+        first: $first
+        orderBy: timestamp
+        orderDirection: desc
+      ) {
+        id
+        encryptedBlob
+        encryptedEmbedding
+        decayScore
+        timestamp
+        createdAt
+        isActive
+        contentFp
+        sequenceId
+        version
+      }
+    }
+  `;
+
+  const data = await gqlQuery<{ facts?: SubgraphSearchFact[] }>(
+    subgraphUrl,
+    query,
+    { owner, first: Math.min(maxCandidates, 1000) },
+    authKeyHex,
+  );
+
+  return (data?.facts ?? []).filter(f => f.isActive !== false);
+}
+
+/**
+ * Fetch a single fact by its client-generated UUID.
+ *
+ * Used by the pin/unpin tools to retrieve a known fact's encrypted blob and
+ * metadata for re-encryption with an updated status. Returns null if the fact
+ * is not found, has been tombstoned (isActive=false), or the owner does not
+ * match the caller's Smart Account address (defense against stale IDs from
+ * another user's recall results).
+ */
+export async function fetchFactById(
+  owner: string,
+  factId: string,
+  authKeyHex?: string,
+): Promise<(SubgraphSearchFact & { owner: string }) | null> {
+  const config = getSubgraphConfig();
+  const subgraphUrl = `${config.relayUrl}/v1/subgraph`;
+
+  const query = `
+    query GetFactById($id: ID!) {
+      fact(id: $id) {
+        id
+        owner
+        encryptedBlob
+        encryptedEmbedding
+        decayScore
+        timestamp
+        createdAt
+        isActive
+        contentFp
+        sequenceId
+        version
+      }
+    }
+  `;
+
+  const data = await gqlQuery<{ fact?: (SubgraphSearchFact & { owner: string }) | null }>(
+    subgraphUrl,
+    query,
+    { id: factId },
+    authKeyHex,
+  );
+
+  const fact = data?.fact;
+  if (!fact) return null;
+  if (fact.isActive === false) return null;
+  if (typeof fact.owner === 'string' && fact.owner.toLowerCase() !== owner.toLowerCase()) {
+    return null;
+  }
+  return fact;
+}
+
 /**
  * Get fact count from the subgraph for dynamic pool sizing.
  * Uses the globalStates entity for a lightweight single-row lookup