@anthusai/papyrus 5.43.0-next.2 → 5.43.0-next.21

package/components/topic-steering-workspace.tsx

@@ -94,6 +94,10 @@ import { useOptionalNewsDeskClient } from "./news-desk-client-provider";
 import { ReferenceSourcePreview } from "./reference-source-preview";
 import type { ReaderAuthSnapshot } from "./reader-auth-state";
 import { NewsDeskAccessGate } from "./newsroom-access-gate";
+import { correctReferenceCapture } from "../lib/content-actions-client";
+import { loadMessagesWithHeadStart, loadReferencesWithHeadStart, loadRelationsWithHeadStart } from "./references-prefetch";
+import { loadFeedSnapshot, saveFeedSnapshot, type FeedSnapshot } from "../lib/reference-feed-cache";
+import { referencePreviewsByLineage } from "../lib/reference-preview";
 import { NewsroomOpsOverview } from "./newsroom-ops-overview";
 import { NewsroomCurationBootstrap } from "./newsroom-curation-bootstrap";
 import { NewsroomReferencesView } from "./newsroom-references-view";
@@ -696,7 +700,7 @@ export function NewsDeskWorkspace({
   }
 
   if (dashboard?.isPublicSkeleton && session?.shell.phase !== "ready") {
-    return <NewsDeskAccessGate shell={session?.shell ?? null} showSectionTabs={showSectionTabs} />;
+    return <NewsDeskAccessGate initialTab={initialTab} shell={session?.shell ?? null} showSectionTabs={showSectionTabs} />;
   }
 
   if (dashboard) {
@@ -716,7 +720,7 @@ export function NewsDeskWorkspace({
     );
   }
 
-  return <NewsDeskAccessGate shell={session?.shell ?? null} showSectionTabs={showSectionTabs} />;
+  return <NewsDeskAccessGate initialTab={initialTab} shell={session?.shell ?? null} showSectionTabs={showSectionTabs} />;
 }
 
 function NewsDeskDashboard({
@@ -777,9 +781,16 @@ function NewsDeskDashboard({
   const [referenceAttachments, setReferenceAttachments] = useState(dashboard.referenceAttachments);
   const [messages, setMessages] = useState(dashboard.messages);
   const [semanticRelations, setSemanticRelations] = useState(dashboard.semanticRelations);
+  const [savedFeedCopy, setSavedFeedCopy] = useState<Pick<FeedSnapshot, "references" | "previews" | "decisions"> | null>(null);
+  useEffect(() => {
+    // Display only: the last view this browser saved, shown until the live read arrives.
+    if (dashboard.isDemo) return;
+    const snapshot = loadFeedSnapshot(SITE_BRAND.id);
+    if (snapshot) setSavedFeedCopy(snapshot);
+  }, [dashboard.isDemo]);
   const cyclotronPreviewFixtures = useMemo(
-    () => (isLocalPreviewEnabled() ? createLocalPreviewCyclotronFixtures(references) : null),
-    [references],
+    () => (isLocalPreviewEnabled() && !semanticRelations.some((relation) => relation.predicate === "relevance_decision_is") ? createLocalPreviewCyclotronFixtures(references) : null),
+    [references, semanticRelations],
   );
   const relevanceDecisions = useMemo(
     () => currentRelevanceDecisions(cyclotronPreviewFixtures ? [...semanticRelations, ...cyclotronPreviewFixtures.relations] : semanticRelations),
@@ -1215,7 +1226,7 @@ function NewsDeskDashboard({
   useEffect(() => {
     if (!canLoadPrivateCorpus || hasHydratedReferences) return;
     let active = true;
-    void loadEditorReferencesData()
+    void loadReferencesWithHeadStart()
       .then(({ references: nextReferences, referenceAttachments }) => {
         if (!active) return;
         setReferences(nextReferences);
@@ -1322,12 +1333,12 @@ function NewsDeskDashboard({
     refreshNewsroomSections,
   ]);
 
-  const needsReferenceReviewData = activeTab === "references" || activeTab === "assignments" || activeTab === "messages";
+  const needsReferenceReviewData = activeTab === "overview" || activeTab === "references" || activeTab === "assignments" || activeTab === "messages";
   const selectedReviewLineageId = pathname ? pathnameReferenceLineageId ?? "" : initialSelection.reference ?? "";
   const referenceReviewData = useReferenceReviewData({
     enabled: canLoadPrivateCorpus && needsReferenceReviewData && !(activeTab === "references" && selectedReviewLineageId),
-    loadRelations: loadEditorSemanticRelationsData,
-    loadMessages: loadEditorMessagesData,
+    loadRelations: loadRelationsWithHeadStart,
+    loadMessages: loadMessagesWithHeadStart,
     onRelations: setSemanticRelations,
     onMessages: setMessages,
   });
@@ -1351,6 +1362,18 @@ function NewsDeskDashboard({
     ? selectedReviewRead.lineageId === selectedReviewLineageId ? selectedReviewRead.state : "loading"
     : referenceReviewData.state;
   const predictionError = selectedReviewIsCurrent ? selectedReviewRead.error : referenceReviewData.error;
+  useEffect(() => {
+    // Keep a display copy for the next visit once everything the feed shows has been read.
+    if (dashboard.isDemo || (activeTab !== "references" && activeTab !== "overview") || !hasHydratedReferences || predictionState !== "ready" || !references.length) return;
+    const timer = window.setTimeout(() => {
+      saveFeedSnapshot(SITE_BRAND.id, {
+        references: selectCanonicalReferenceRecords(references),
+        previews: referencePreviewsByLineage(referenceAttachments),
+        decisions: relevanceDecisions,
+      });
+    }, 800);
+    return () => window.clearTimeout(timer);
+  }, [activeTab, dashboard.isDemo, hasHydratedReferences, predictionState, referenceAttachments, references, relevanceDecisions]);
   const retryReferenceReviewData = () => { referenceReviewData.retry(); setSelectedReviewAttempt((current) => current + 1); };
 
   const applyReferenceRecord = useCallback((nextReference: ReferenceRecord) => {
@@ -3035,7 +3058,7 @@ function NewsDeskDashboard({
         headerActions={(
           <>
             {dashboard.isDemo ? (
-              <Badge variant="outline">Demo</Badge>
+              dashboard.realDataPreview ? null : <Badge variant="outline">Demo</Badge>
             ) : (
               <Link className={cn(buttonVariants({ variant: "ghost", size: "sm" }))} href="/settings">Settings</Link>
             )}
@@ -3081,9 +3104,18 @@ function NewsDeskDashboard({
             assignments={assignments}
             decisions={relevanceDecisions}
             demo={Boolean(dashboard.isDemo)}
+            disabledReason={controlsDisabled ? "Editor access is required to change this reference." : null}
             humanCuratedReferenceLineageIds={humanCuratedReferenceLineageIds}
+            loading={!dashboard.isDemo && !hasHydratedReferences}
             newsroomSections={newsroomSections}
+            notificationFixture={dashboard.notificationFixture ?? null}
+            onCorrectCapture={dashboard.isDemo ? undefined : (input) => correctReferenceCapture(input)}
+            onReview={(reference, action, options) => runReferenceCurationAction(reference, action, options?.note, options?.reasonCode,
+              options?.decisionRelationId ? { decisionRelationId: options.decisionRelationId, shareable: options.shareable === true } : undefined)}
+            predictionState={referenceReviewData.state}
+            referenceAttachments={referenceAttachments}
             references={references}
+            savedCopy={savedFeedCopy}
           />
         ) : null}
         {activeTab === "search" ? (
@@ -3182,6 +3214,9 @@ function NewsDeskDashboard({
               options?.sectionId)}
             referenceAttachments={referenceAttachments}
             references={references}
+            loading={!dashboard.isDemo && !hasHydratedReferences}
+            onCorrectCapture={dashboard.isDemo ? undefined : (input) => correctReferenceCapture(input)}
+            savedCopy={savedFeedCopy}
           />
           </NewsroomCurationBootstrap>
         ) : null}

package/components/use-media-query.ts

@@ -0,0 +1,16 @@
+"use client";
+
+import { useEffect, useState } from "react";
+
+/** True while the media query matches. False on the server and before the first effect, so the first paint is the wide layout. */
+export function useMediaQuery(query: string): boolean {
+  const [matches, setMatches] = useState(false);
+  useEffect(() => {
+    const mediaQuery = window.matchMedia(query);
+    const update = () => setMatches(mediaQuery.matches);
+    update();
+    mediaQuery.addEventListener("change", update);
+    return () => mediaQuery.removeEventListener("change", update);
+  }, [query]);
+  return matches;
+}

package/components/use-reference-preview-attachments.ts

@@ -0,0 +1,101 @@
+"use client";
+
+import { useEffect, useMemo, useState } from "react";
+import type { ReferenceAttachmentRecord, ReferenceRecord } from "../lib/category-repository";
+import { referenceLineageId } from "../lib/newsroom-references";
+import { loadReferenceAttachmentsForLineageId, loadStoragePathUrl } from "./news-desk-taxonomy-client";
+
+function normalizeReferenceDetailHttpUri(value: string | null | undefined): string | null {
+  const trimmed = value?.trim();
+  if (!trimmed) return null;
+  if (/^https?:\/\//i.test(trimmed)) return trimmed;
+  return null;
+}
+
+/** The attachments of one Reference (seeded, else read), with stored files turned into links. Shared by the References detail pane and the Needs me detail pane. */
+export function useReferencePreviewAttachments(
+  reference: ReferenceRecord | null,
+  referenceAttachments: ReferenceAttachmentRecord[],
+) {
+  const lineageId = reference ? referenceLineageId(reference) : "";
+  const seededAttachments = useMemo(
+    () => referenceAttachments.filter((attachment) => attachment.referenceLineageId === lineageId),
+    [lineageId, referenceAttachments],
+  );
+  const [attachments, setAttachments] = useState<ReferenceAttachmentRecord[]>(seededAttachments);
+  const [attachmentLinksById, setAttachmentLinksById] = useState<Record<string, string>>({});
+
+  useEffect(() => {
+    setAttachments(seededAttachments);
+  }, [seededAttachments]);
+
+  useEffect(() => {
+    if (!lineageId) return;
+    if (seededAttachments.length > 0) return;
+    let active = true;
+    void loadReferenceAttachmentsForLineageId(lineageId)
+      .then((loaded) => {
+        if (!active || !loaded.length) return;
+        setAttachments(loaded);
+      })
+      .catch(() => undefined);
+    return () => {
+      active = false;
+    };
+  }, [lineageId, seededAttachments.length]);
+
+  useEffect(() => {
+    if (!attachments.length) {
+      setAttachmentLinksById({});
+      return;
+    }
+    let active = true;
+    const staticLinks = new Map<string, string>();
+    const storageLookups: Array<{ id: string; storagePath: string }> = [];
+
+    for (const attachment of attachments) {
+      if (attachment.storagePath) {
+        storageLookups.push({ id: attachment.id, storagePath: attachment.storagePath });
+        continue;
+      }
+      const sourceHref = normalizeReferenceDetailHttpUri(attachment.sourceUri);
+      if (sourceHref) staticLinks.set(attachment.id, sourceHref);
+    }
+
+    if (!storageLookups.length) {
+      setAttachmentLinksById(Object.fromEntries(staticLinks.entries()));
+      return () => {
+        active = false;
+      };
+    }
+
+    setAttachmentLinksById(Object.fromEntries(staticLinks.entries()));
+    void Promise.all(storageLookups.map(async (lookup) => {
+      const result = await loadStoragePathUrl(lookup.storagePath);
+      return { id: lookup.id, url: result.url };
+    }))
+      .then((resolved) => {
+        if (!active) return;
+        const nextLinks = Object.fromEntries(staticLinks.entries());
+        for (const entry of resolved) {
+          if (entry.url) nextLinks[entry.id] = entry.url;
+        }
+        setAttachmentLinksById(nextLinks);
+      })
+      .catch(() => undefined);
+
+    return () => {
+      active = false;
+    };
+  }, [attachments]);
+
+  const previewAttachments = useMemo(
+    () => attachments.map((attachment) => ({
+      ...attachment,
+      sourceUri: attachmentLinksById[attachment.id] ?? attachment.sourceUri,
+    })),
+    [attachmentLinksById, attachments],
+  );
+
+  return previewAttachments;
+}

package/components/use-reference-ratings.ts

@@ -0,0 +1,134 @@
+"use client";
+
+import { useCallback, useEffect, useRef, useState } from "react";
+import type { ReferenceRejectionReasonCode } from "../lib/reference-policy";
+import type { ReviewChoice } from "./reference-review-card";
+import type { RatingFeedback } from "./review-rating-grid";
+
+/** How long after the last keystroke a typed reason is saved on its own. */
+export const NOTE_AUTOSAVE_MS = 1200;
+
+export type ItemRating = {
+  /** The reason as typed, else as saved. In the store itself it stays undefined until touched, so an untouched field follows the record. */
+  note?: string;
+  reasonCode?: ReferenceRejectionReasonCode | null;
+  /** A vote that is waiting for a reason (Reject needs one) or is still saving. */
+  pending: ReviewChoice | null;
+  /** True while a Reject is waiting for its reason. */
+  askReason: boolean;
+  feedback: RatingFeedback;
+};
+
+const EMPTY: ItemRating = { pending: null, askReason: false, feedback: { phase: "idle" } };
+
+export type RatingSave = (id: string, choice: ReviewChoice, detail: { note: string; reasonCode: ReferenceRejectionReasonCode | null }) => Promise<void> | void;
+
+export type RatingController = {
+  get: (id: string) => ItemRating;
+  choose: (id: string, choice: ReviewChoice) => void;
+  setReason: (id: string, code: ReferenceRejectionReasonCode) => void;
+  setNote: (id: string, text: string) => void;
+  /** Save a typed reason now (on blur). */
+  commitNote: (id: string) => void;
+  cancelReason: (id: string) => void;
+};
+
+/**
+ * One store for the rating state of every item, shared by the grid card and the detail pane so a vote or a reason typed in either
+ * shows in both at once. `saved(id)` is the saved human answer; `seed(id)` the saved note and reason code.
+ *   vote, then reason  -> the reason is saved onto the existing decision (on blur, or after a short pause)
+ *   reason, then vote  -> the vote carries the reason
+ * A Reject needs a reason code (the backend refuses one without), so it waits as `pending` until one is chosen; a typed reason is
+ * kept in the store until it is saved and is never dropped by a failed save.
+ */
+export function useRatingStore({ save, saved, seed, needsReason }: {
+  save: RatingSave;
+  saved: (id: string) => ReviewChoice | null;
+  seed: (id: string) => { note: string; reasonCode: ReferenceRejectionReasonCode | null };
+  /** Whether a Reject on this item must wait for a reason code. Default yes (a Reference); an AI draft or an email asks nothing. */
+  needsReason?: (id: string) => boolean;
+}): RatingController {
+  const [items, setItems] = useState<Record<string, ItemRating>>({});
+  const itemsRef = useRef(items);
+  const latest = useRef({ save, saved, seed, needsReason });
+  latest.current = { save, saved, seed, needsReason };
+  const timers = useRef<Record<string, ReturnType<typeof setTimeout>>>({});
+  const inFlight = useRef<Set<string>>(new Set());
+  const dirty = useRef<Set<string>>(new Set());
+
+  const patch = useCallback((id: string, change: Partial<ItemRating>) => {
+    const next = { ...itemsRef.current, [id]: { ...(itemsRef.current[id] ?? EMPTY), ...change } };
+    itemsRef.current = next;
+    setItems(next);
+  }, []);
+
+  // What a reader of the store sees: the touched value, else what is saved. (A saved reason therefore shows on the card too.)
+  const get = useCallback((id: string): ItemRating => {
+    const item = items[id] ?? EMPTY;
+    const seeded = latest.current.seed(id);
+    return { ...item, note: item.note ?? seeded.note, reasonCode: item.reasonCode !== undefined ? item.reasonCode : seeded.reasonCode };
+  }, [items]);
+
+  const values = (id: string) => {
+    const item = itemsRef.current[id] ?? EMPTY;
+    const seeded = latest.current.seed(id);
+    return { note: item.note ?? seeded.note, reasonCode: item.reasonCode !== undefined ? item.reasonCode : seeded.reasonCode };
+  };
+
+  const commit = useCallback(async (id: string, choice: ReviewChoice) => {
+    const { note, reasonCode } = values(id);
+    if (choice === "reject" && !reasonCode && (latest.current.needsReason?.(id) ?? true)) { patch(id, { pending: "reject", askReason: true }); return; }
+    if (inFlight.current.has(id)) { dirty.current.add(id); return; }
+    inFlight.current.add(id);
+    dirty.current.delete(id);
+    const started = itemsRef.current[id] ?? EMPTY;
+    patch(id, { pending: choice, askReason: false, feedback: { phase: "saving" } });
+    try {
+      await latest.current.save(id, choice, { note: note.trim(), reasonCode: choice === "reject" ? reasonCode : null });
+      // What was saved now follows the record again (another editor's update shows); anything typed meanwhile is kept.
+      const now = itemsRef.current[id] ?? EMPTY;
+      patch(id, { pending: null, feedback: { phase: "saved" }, ...(now.note === started.note ? { note: undefined } : {}), ...(now.reasonCode === started.reasonCode ? { reasonCode: undefined } : {}) });
+    } catch (failure) {
+      // The typed reason stays in the store; only the vote is undone.
+      patch(id, { pending: null, feedback: { phase: "error", message: failure instanceof Error ? failure.message : "Your answer could not be saved. Tap it again to retry." } });
+    } finally {
+      inFlight.current.delete(id);
+    }
+    if (dirty.current.has(id)) { dirty.current.delete(id); const again = itemsRef.current[id]?.pending ?? latest.current.saved(id); if (again) void commit(id, again); }
+  }, [patch]);
+
+  const effective = (id: string): ReviewChoice | null => itemsRef.current[id]?.pending ?? latest.current.saved(id);
+
+  const choose = useCallback((id: string, choice: ReviewChoice) => {
+    if (effective(id) === choice && !itemsRef.current[id]?.askReason) return; // the chosen answer never unselects itself
+    void commit(id, choice);
+  }, [commit]);
+
+  const setReason = useCallback((id: string, code: ReferenceRejectionReasonCode) => {
+    patch(id, { reasonCode: code });
+    // Vote then reason: onto the existing (or waiting) decision. Reason then vote: stored, and the vote carries it.
+    if (effective(id) === "reject") void commit(id, "reject");
+  }, [commit, patch]);
+
+  const setNote = useCallback((id: string, text: string) => {
+    patch(id, { note: text, feedback: { phase: "idle" } });
+    dirty.current.add(`note:${id}`);
+    clearTimeout(timers.current[id]);
+    timers.current[id] = setTimeout(() => {
+      const choice = effective(id);
+      if (choice && dirty.current.delete(`note:${id}`)) void commit(id, choice);
+    }, NOTE_AUTOSAVE_MS);
+  }, [commit, patch]);
+
+  const commitNote = useCallback((id: string) => {
+    clearTimeout(timers.current[id]);
+    const choice = effective(id);
+    if (choice && dirty.current.delete(`note:${id}`)) void commit(id, choice);
+  }, [commit]);
+
+  const cancelReason = useCallback((id: string) => patch(id, { pending: null, askReason: false }), [patch]);
+
+  useEffect(() => () => { for (const timer of Object.values(timers.current)) clearTimeout(timer); }, []);
+
+  return { get, choose, setReason, setNote, commitNote, cancelReason };
+}

package/infra/amplify-app-shell.js

@@ -6,10 +6,26 @@ import { addCloudEmail } from "./cloud-email.js";
 import { parseCloudEmail } from "./cloud-email-config.js";
 import { GITHUB_OIDC_PROVIDER_HOST } from "./github-oidc-provider.js";
 import { cmsProductionBuildSpec, cmsStagingBuildSpec, readerBuildSpec } from "./build-specs.js";
-import { PROSPECTUS_PAPYRUS_SITE_ROUTES, hasCmsApp, isStagingEnabled, resolveRedirectSubDomainPrefix, resolveCmsHostName, resolveSiteOrigin, githubOidcSubjects, resolveGithubRepository, resolveProspectusAccountsEndpoint, resolveProspectusSessionSealingParameterName, resolveReaderPollSecretParameterName, resolveRevalidateSecretParameterName, resolveStackName, resolveStagingDomainName, resolveStoragePreviewPrefix } from "./site-config.js";
+import { PROSPECTUS_PAPYRUS_SITE_ROUTES, hasCmsApp, isStagingEnabled, resolveRedirectSubDomainPrefix, resolveCmsHostName, resolveSiteOrigin, githubOidcSubjects, resolveGithubRepository, resolveProspectusAccountsEndpoint, resolveProspectusSessionSealingParameterName, resolveReaderPollSecretParameterName, resolveRevalidateSecretParameterName, resolveStackName, resolveStagingDomainName, resolveChatStackName, chatExportName, CHAT_EXPORTED_RESOURCES, CHAT_ENVIRONMENT_NAMES, CHAT_PRIVATE_ASSISTANT_ID, CHAT_PRIVATE_ASSISTANT_NAME, resolveStoragePreviewPrefix } from "./site-config.js";
 function environmentVariables(variables) {
     return Object.entries(variables).map(([name, value]) => ({ name, value }));
 }
+/**
+ * The production branch's private chat environment. Only a site with an enabled `chat` block gets the flag and the
+ * resource names (imported from the chat data stack's exports, so that stack must be deployed first); every other
+ * site keeps the flag "false" and no chat variables, and its newsroom serves neither the chat route nor the UI.
+ */
+function chatEnvironment(config) {
+    const chatStackName = resolveChatStackName(config);
+    if (!chatStackName)
+        return { PAPYRUS_PRIVATE_CHAT_ENABLED: "false" };
+    return {
+        PAPYRUS_PRIVATE_CHAT_ENABLED: "true",
+        ...Object.fromEntries(CHAT_EXPORTED_RESOURCES.map((resource) => [CHAT_ENVIRONMENT_NAMES[resource], Fn.importValue(chatExportName(chatStackName, resource))])),
+        PAPYRUS_CHAT_PRIVATE_BOT_ID: CHAT_PRIVATE_ASSISTANT_ID,
+        PAPYRUS_CHAT_PRIVATE_BOT_NAME: CHAT_PRIVATE_ASSISTANT_NAME,
+    };
+}
 function readerCustomHeaders(rules) {
     if (!rules)
         return undefined;
@@ -199,7 +215,7 @@ export class AmplifyAppShellStack extends Stack {
             environmentVariables: environmentVariables({
                 ...config.cms.environment,
                 ...emailEnvironment,
-                PAPYRUS_PRIVATE_CHAT_ENABLED: config.cms.environment.PAPYRUS_PRIVATE_CHAT_ENABLED ?? "false",
+                ...(config.chat?.enabled ? chatEnvironment(config) : { PAPYRUS_PRIVATE_CHAT_ENABLED: config.cms.environment.PAPYRUS_PRIVATE_CHAT_ENABLED ?? "false" }),
                 PAPYRUS_COGNITO_DOMAIN_PREFIX: config.cms.cognitoDomainPrefix,
                 ...(config.cms.applyCognitoDomainPrefix ? { PAPYRUS_APPLY_COGNITO_DOMAIN_PREFIX: "true" } : {}),
                 PAPYRUS_OAUTH_REDIRECT_URLS: oauthRedirectUrls,

package/infra/amplify-app-shell.ts

@@ -7,12 +7,28 @@ import { addCloudEmail } from "./cloud-email";
 import { parseCloudEmail } from "./cloud-email-config";
 import { GITHUB_OIDC_PROVIDER_HOST } from "./github-oidc-provider";
 import { cmsProductionBuildSpec, cmsStagingBuildSpec, readerBuildSpec } from "./build-specs";
-import { AmplifyAppShellSiteConfig, PROSPECTUS_PAPYRUS_SITE_ROUTES, ReaderCustomHeaderRule, hasCmsApp, isStagingEnabled, resolveRedirectSubDomainPrefix, resolveCmsHostName, resolveSiteOrigin, githubOidcSubjects, resolveGithubRepository, resolveProspectusAccountsEndpoint, resolveProspectusSessionSealingParameterName, resolveReaderPollSecretParameterName, resolveRevalidateSecretParameterName, resolveStackName, resolveStagingDomainName, resolveStoragePreviewPrefix } from "./site-config";
+import { AmplifyAppShellSiteConfig, PROSPECTUS_PAPYRUS_SITE_ROUTES, ReaderCustomHeaderRule, hasCmsApp, isStagingEnabled, resolveRedirectSubDomainPrefix, resolveCmsHostName, resolveSiteOrigin, githubOidcSubjects, resolveGithubRepository, resolveProspectusAccountsEndpoint, resolveProspectusSessionSealingParameterName, resolveReaderPollSecretParameterName, resolveRevalidateSecretParameterName, resolveStackName, resolveStagingDomainName, resolveChatStackName, chatExportName, CHAT_EXPORTED_RESOURCES, CHAT_ENVIRONMENT_NAMES, CHAT_PRIVATE_ASSISTANT_ID, CHAT_PRIVATE_ASSISTANT_NAME, resolveStoragePreviewPrefix } from "./site-config";
 
 function environmentVariables(variables: Record<string, string>): amplify.CfnBranch.EnvironmentVariableProperty[] {
   return Object.entries(variables).map(([name, value]) => ({ name, value }));
 }
 
+/**
+ * The production branch's private chat environment. Only a site with an enabled `chat` block gets the flag and the
+ * resource names (imported from the chat data stack's exports, so that stack must be deployed first); every other
+ * site keeps the flag "false" and no chat variables, and its newsroom serves neither the chat route nor the UI.
+ */
+function chatEnvironment(config: AmplifyAppShellSiteConfig): Record<string, string> {
+  const chatStackName = resolveChatStackName(config);
+  if (!chatStackName) return { PAPYRUS_PRIVATE_CHAT_ENABLED: "false" };
+  return {
+    PAPYRUS_PRIVATE_CHAT_ENABLED: "true",
+    ...Object.fromEntries(CHAT_EXPORTED_RESOURCES.map((resource) => [CHAT_ENVIRONMENT_NAMES[resource], Fn.importValue(chatExportName(chatStackName, resource))])),
+    PAPYRUS_CHAT_PRIVATE_BOT_ID: CHAT_PRIVATE_ASSISTANT_ID,
+    PAPYRUS_CHAT_PRIVATE_BOT_NAME: CHAT_PRIVATE_ASSISTANT_NAME,
+  };
+}
+
 function readerCustomHeaders(rules: ReaderCustomHeaderRule[] | undefined): string | undefined {
   if (!rules) return undefined;
   const lines = ["customHeaders:"];
@@ -221,7 +237,7 @@ export class AmplifyAppShellStack extends Stack {
       environmentVariables: environmentVariables({
         ...config.cms.environment,
         ...emailEnvironment,
-        PAPYRUS_PRIVATE_CHAT_ENABLED: config.cms.environment.PAPYRUS_PRIVATE_CHAT_ENABLED ?? "false",
+        ...(config.chat?.enabled ? chatEnvironment(config) : { PAPYRUS_PRIVATE_CHAT_ENABLED: config.cms.environment.PAPYRUS_PRIVATE_CHAT_ENABLED ?? "false" }),
         PAPYRUS_COGNITO_DOMAIN_PREFIX: config.cms.cognitoDomainPrefix,
         ...(config.cms.applyCognitoDomainPrefix ? { PAPYRUS_APPLY_COGNITO_DOMAIN_PREFIX: "true" } : {}),
         PAPYRUS_OAUTH_REDIRECT_URLS: oauthRedirectUrls,