From fc92ba11c06a84252054611c30a7571d10920e2e Mon Sep 17 00:00:00 2001
From: Andrey Avtomonov <andreybavt@gmail.com>
Date: Tue, 12 May 2026 23:47:36 +0200
Subject: [PATCH] fix(context): expose target warehouses to Notion ingest

---
 .../adapters/notion/notion.adapter.test.ts    |  8 ++++++
 .../ingest/adapters/notion/notion.adapter.ts  |  9 +++++++
 .../context/src/ingest/local-adapters.test.ts | 25 +++++++++++++++++++
 packages/context/src/ingest/local-adapters.ts |  1 +
 4 files changed, 43 insertions(+)

diff --git a/packages/context/src/ingest/adapters/notion/notion.adapter.test.ts b/packages/context/src/ingest/adapters/notion/notion.adapter.test.ts
index de989d02..76e28658 100644
--- a/packages/context/src/ingest/adapters/notion/notion.adapter.test.ts
+++ b/packages/context/src/ingest/adapters/notion/notion.adapter.test.ts
@@ -52,6 +52,14 @@ describe('NotionSourceAdapter', () => {
     expect(adapter.triageSupported).toBe(true);
   });
 
+  it('returns configured target warehouse connection ids', async () => {
+    const adapter = new NotionSourceAdapter({
+      targetConnectionIds: ['warehouse', 'warehouse', 'analytics'],
+    });
+
+    await expect(adapter.listTargetConnectionIds?.(stagedDir)).resolves.toEqual(['analytics', 'warehouse']);
+  });
+
   it('returns structural triage signals for a staged Notion page', async () => {
     await mkdir(join(stagedDir, 'pages', 'page-1'), { recursive: true });
     await writeFile(
diff --git a/packages/context/src/ingest/adapters/notion/notion.adapter.ts b/packages/context/src/ingest/adapters/notion/notion.adapter.ts
index fba68cee..b903d4b3 100644
--- a/packages/context/src/ingest/adapters/notion/notion.adapter.ts
+++ b/packages/context/src/ingest/adapters/notion/notion.adapter.ts
@@ -32,6 +32,11 @@ interface NotionPullSucceededContext {
 export interface NotionSourceAdapterDeps {
   onPullSucceeded?: (ctx: NotionPullSucceededContext) => Promise<void>;
   logger?: NotionFetchLogger;
+  targetConnectionIds?: string[];
+}
+
+function uniqueSorted(values: readonly string[] | undefined): string[] {
+  return [...new Set(values ?? [])].sort((left, right) => left.localeCompare(right));
 }
 
 export class NotionSourceAdapter implements SourceAdapter {
@@ -73,6 +78,10 @@ export class NotionSourceAdapter implements SourceAdapter {
     return describeNotionScope(stagedDir);
   }
 
+  async listTargetConnectionIds(_stagedDir: string): Promise<string[]> {
+    return uniqueSorted(this.deps.targetConnectionIds);
+  }
+
   async getTriageSignals(stagedDir: string, externalId: string): Promise<TriageSignals> {
     const metadata = await this.findMetadataByExternalId(stagedDir, externalId);
     if (!metadata) {
diff --git a/packages/context/src/ingest/local-adapters.test.ts b/packages/context/src/ingest/local-adapters.test.ts
index f8dd7da7..9fd51d01 100644
--- a/packages/context/src/ingest/local-adapters.test.ts
+++ b/packages/context/src/ingest/local-adapters.test.ts
@@ -498,6 +498,31 @@ describe('local ingest adapters', () => {
     await expect(adapter?.listTargetConnectionIds?.('/tmp/staged-dbt')).resolves.toEqual(['warehouse']);
   });
 
+  it('passes primary warehouse connection ids to the local Notion adapter', async () => {
+    const adapters = createDefaultLocalIngestAdapters(
+      projectWithConnections({
+        notion: {
+          driver: 'notion',
+          auth_token: 'secret',
+          crawl_mode: 'selected_roots',
+          root_page_ids: ['page-1'],
+        },
+        warehouse: {
+          driver: 'postgres',
+          url: 'postgresql://readonly@db.example.test/analytics',
+        },
+        docs: {
+          driver: 'dbt',
+          source_dir: './dbt',
+        },
+      } as never),
+    );
+
+    const notion = adapters.find((adapter) => adapter.source === 'notion');
+
+    await expect(notion?.listTargetConnectionIds?.('/tmp/staged-notion')).resolves.toEqual(['warehouse']);
+  });
+
   it('resolves MetricFlow auth_token_ref without writing literal tokens to config', async () => {
     const project = projectWithConnections({
       metricflow_main: {
diff --git a/packages/context/src/ingest/local-adapters.ts b/packages/context/src/ingest/local-adapters.ts
index 59daf6d1..14c6b683 100644
--- a/packages/context/src/ingest/local-adapters.ts
+++ b/packages/context/src/ingest/local-adapters.ts
@@ -108,6 +108,7 @@ export function createDefaultLocalIngestAdapters(
     }),
     new MetricflowSourceAdapter({ homeDir: join(project.projectDir, '.ktx/cache') }),
     new NotionSourceAdapter({
+      targetConnectionIds: primaryWarehouseConnectionIds(project),
       ...(options.logger ? { logger: options.logger } : {}),
     }),
   ];