@beyondwork/docx-react-component 1.0.106 → 1.0.108

package/src/runtime/document-heading-outline.ts

@@ -0,0 +1,147 @@
+import type {
+  DocumentHeadingSnapshot,
+  DocumentPageSnapshot,
+  EditorSurfaceSnapshot,
+  SurfaceBlockSnapshot,
+} from "../api/public-types";
+import type { CanonicalDocumentEnvelope } from "../core/state/editor-state.ts";
+import {
+  findSectionForPosition,
+  type ResolvedDocumentSection,
+} from "./document-layout.ts";
+
+/** Style IDs that map to heading levels by convention. */
+const HEADING_STYLE_PATTERN = /^heading\s*(\d+)$/i;
+
+function headingLevelFromStyleId(styleId?: string): number | null {
+  if (!styleId) return null;
+  const match = HEADING_STYLE_PATTERN.exec(styleId);
+  if (match) {
+    const level = parseInt(match[1]!, 10);
+    return level >= 1 && level <= 9 ? level : null;
+  }
+  return null;
+}
+
+export function buildHeadingOutline(
+  document: CanonicalDocumentEnvelope,
+  mainSurface: EditorSurfaceSnapshot,
+  sections: ResolvedDocumentSection[],
+  pages: DocumentPageSnapshot[],
+): DocumentHeadingSnapshot[] {
+  const headings: DocumentHeadingSnapshot[] = [];
+
+  for (const block of mainSurface.blocks) {
+    if (block.kind !== "paragraph") continue;
+
+    const outlineLevel = block.outlineLevel;
+    const styleLevel = resolveHeadingLevelFromStyle(document, block.styleId);
+    const level =
+      outlineLevel !== undefined && outlineLevel >= 0
+        ? outlineLevel + 1
+        : styleLevel;
+
+    if (level === null) continue;
+
+    const text = extractParagraphText(block);
+    if (!text.trim()) continue;
+
+    const offset = block.from;
+    const pageIndex = findPageForOffset(pages, offset);
+    const sectionIndex = findSectionForOffset(sections, offset);
+
+    headings.push({
+      headingId: `heading-${block.blockId}-${offset}`,
+      level,
+      text: text.trim(),
+      offset,
+      pageIndex,
+      sectionIndex,
+    });
+  }
+
+  return headings;
+}
+
+function findPageForOffset(
+  pages: readonly DocumentPageSnapshot[],
+  offset: number,
+): number {
+  for (let i = 0; i < pages.length; i += 1) {
+    if (offset < pages[i]!.endOffset) {
+      return i;
+    }
+  }
+  return Math.max(0, pages.length - 1);
+}
+
+function findSectionForOffset(
+  sections: readonly ResolvedDocumentSection[],
+  offset: number,
+): number {
+  return sections.length > 0 ? findSectionForPosition(sections, offset).index : 0;
+}
+
+function resolveHeadingLevelFromStyle(
+  document: CanonicalDocumentEnvelope,
+  styleId?: string,
+): number | null {
+  if (!styleId) {
+    return null;
+  }
+
+  const visited = new Set<string>();
+  let currentStyleId: string | undefined = styleId;
+  while (currentStyleId && !visited.has(currentStyleId)) {
+    visited.add(currentStyleId);
+    const style:
+      | { outlineLevel?: number; basedOn?: string }
+      | undefined = document.styles.paragraphs[currentStyleId];
+    if (!style) {
+      break;
+    }
+    if (
+      typeof style.outlineLevel === "number" &&
+      style.outlineLevel >= 0 &&
+      style.outlineLevel <= 8
+    ) {
+      return style.outlineLevel + 1;
+    }
+    currentStyleId = style.basedOn;
+  }
+
+  return headingLevelFromStyleId(styleId);
+}
+
+function extractParagraphText(
+  block: Extract<SurfaceBlockSnapshot, { kind: "paragraph" }>,
+): string {
+  let text = "";
+  for (const segment of block.segments) {
+    switch (segment.kind) {
+      case "text":
+        text += segment.text;
+        break;
+      case "tab":
+        text += "\t";
+        break;
+      case "hard_break":
+        text += "\n";
+        break;
+    }
+  }
+  if (!block.numberingPrefix) {
+    return text;
+  }
+  const trimmedText = text.trimStart();
+  if (
+    trimmedText === block.numberingPrefix ||
+    trimmedText.startsWith(`${block.numberingPrefix} `) ||
+    trimmedText.startsWith(`${block.numberingPrefix}\t`) ||
+    (block.numberingSuffix === "nothing" && trimmedText.startsWith(block.numberingPrefix))
+  ) {
+    return text;
+  }
+  const separator = block.numberingSuffix === "nothing" ? "" : " ";
+  return `${block.numberingPrefix}${separator}${text}`;
+}

package/src/runtime/document-navigation.ts

@@ -11,40 +11,22 @@
  */
 
 import type {
-  DocumentHeadingSnapshot,
   DocumentNavigationSnapshot,
   DocumentPageSnapshot,
   EditorStoryTarget,
-  EditorSurfaceSnapshot,
-  SurfaceBlockSnapshot,
 } from "../api/public-types";
 import type { CanonicalDocumentEnvelope } from "../core/state/editor-state.ts";
 import { createSelectionSnapshot } from "../core/state/editor-state.ts";
-import { MAIN_STORY_TARGET } from "../core/selection/mapping.ts";
-import { createEditorSurfaceSnapshot } from "./surface-projection.ts";
 import {
-  buildResolvedSections,
   findSectionForPosition as resolveSectionForPosition,
-  resolveSectionForStoryTarget,
   type ResolvedDocumentSection,
 } from "./document-layout.ts";
-import { buildPageStack } from "./layout/paginated-layout-engine.ts";
-import { findNoteReferencePosition } from "./view-state.ts";
+import { createLayoutEngine } from "./layout/layout-engine-instance.ts";
+export { buildHeadingOutline } from "./document-heading-outline.ts";
 
-interface NavigationBaseSnapshot {
-  mainSurface: EditorSurfaceSnapshot;
-  sections: ResolvedDocumentSection[];
-  pages: DocumentPageSnapshot[];
-  headings: DocumentHeadingSnapshot[];
-}
-
-const navigationBaseCache = new WeakMap<
-  CanonicalDocumentEnvelope["content"],
-  NavigationBaseSnapshot & {
-    styles: CanonicalDocumentEnvelope["styles"];
-    subParts: CanonicalDocumentEnvelope["subParts"];
-  }
->();
+const navigationLayoutEngine = createLayoutEngine({
+  autoUpgradeToCanvasBackend: false,
+});
 
 // ---------------------------------------------------------------------------
 // Public API
@@ -62,55 +44,11 @@ export function createDocumentNavigationSnapshot(
   selectionHead: number,
   activeStory: EditorStoryTarget,
 ): DocumentNavigationSnapshot {
-  const base = getNavigationBaseSnapshot(document);
-  const navigationContext = resolveActiveNavigationContext(
-    document,
-    base.pages,
-    base.sections,
-    base.mainSurface,
-    selectionHead,
+  return navigationLayoutEngine.getNavigationSnapshot(
+    { document },
+    createSelectionSnapshot(selectionHead, selectionHead),
     activeStory,
   );
-
-  return {
-    pageCount: base.pages.length,
-    pages: base.pages,
-    headings: base.headings,
-    activePageIndex: navigationContext.activePageIndex,
-    activeSectionIndex: navigationContext.activeSectionIndex,
-  };
-}
-
-function getNavigationBaseSnapshot(
-  document: CanonicalDocumentEnvelope,
-): NavigationBaseSnapshot {
-  const cached = navigationBaseCache.get(document.content);
-  if (
-    cached &&
-    cached.styles === document.styles &&
-    cached.subParts === document.subParts
-  ) {
-    return cached;
-  }
-
-  const mainSurface = createEditorSurfaceSnapshot(
-    document,
-    createSelectionSnapshot(0, 0),
-    MAIN_STORY_TARGET,
-  );
-  const sections = buildResolvedSections(document);
-  const pages = buildPageStack(document, sections, mainSurface);
-  const headings = buildHeadingOutline(document, mainSurface, sections, pages);
-  const next = {
-    mainSurface,
-    sections,
-    pages,
-    headings,
-    styles: document.styles,
-    subParts: document.subParts,
-  };
-  navigationBaseCache.set(document.content, next);
-  return next;
 }
 
 /**
@@ -137,176 +75,3 @@ export function findSectionForOffset(
 ): number {
   return sections.length > 0 ? resolveSectionForPosition(sections, offset).index : 0;
 }
-
-function resolveActiveNavigationContext(
-  document: CanonicalDocumentEnvelope,
-  pages: readonly DocumentPageSnapshot[],
-  sections: readonly ResolvedDocumentSection[],
-  mainSurface: EditorSurfaceSnapshot,
-  selectionHead: number,
-  activeStory: EditorStoryTarget,
-): { activePageIndex: number; activeSectionIndex: number } {
-  if (activeStory.kind === "main") {
-    const activePageIndex = findPageForOffset(pages, selectionHead);
-    return {
-      activePageIndex,
-      activeSectionIndex: pages[activePageIndex]?.sectionIndex ?? 0,
-    };
-  }
-
-  if (activeStory.kind === "header" || activeStory.kind === "footer") {
-    const section = resolveSectionForStoryTarget(document, sections, activeStory);
-    if (section) {
-      return {
-        activePageIndex: findFirstPageIndexForSection(pages, section.index),
-        activeSectionIndex: section.index,
-      };
-    }
-  }
-
-  if (activeStory.kind === "footnote" || activeStory.kind === "endnote") {
-    const referencePosition = findNoteReferencePosition(mainSurface, activeStory);
-    const activePageIndex = findPageForOffset(pages, referencePosition);
-    return {
-      activePageIndex,
-      activeSectionIndex:
-        pages[activePageIndex]?.sectionIndex ??
-        findSectionForOffset(sections, referencePosition),
-    };
-  }
-
-  return {
-    activePageIndex: 0,
-    activeSectionIndex: 0,
-  };
-}
-
-function findFirstPageIndexForSection(
-  pages: readonly DocumentPageSnapshot[],
-  sectionIndex: number,
-): number {
-  const match = pages.findIndex((page) => page.sectionIndex === sectionIndex);
-  return match >= 0 ? match : 0;
-}
-
-// ---------------------------------------------------------------------------
-// Heading outline
-// ---------------------------------------------------------------------------
-
-/** Style IDs that map to heading levels by convention. */
-const HEADING_STYLE_PATTERN = /^heading\s*(\d+)$/i;
-
-function headingLevelFromStyleId(styleId?: string): number | null {
-  if (!styleId) return null;
-  const match = HEADING_STYLE_PATTERN.exec(styleId);
-  if (match) {
-    const level = parseInt(match[1]!, 10);
-    return level >= 1 && level <= 9 ? level : null;
-  }
-  return null;
-}
-
-export function buildHeadingOutline(
-  document: CanonicalDocumentEnvelope,
-  mainSurface: EditorSurfaceSnapshot,
-  sections: ResolvedDocumentSection[],
-  pages: DocumentPageSnapshot[],
-): DocumentHeadingSnapshot[] {
-  const headings: DocumentHeadingSnapshot[] = [];
-
-  for (const block of mainSurface.blocks) {
-    if (block.kind !== "paragraph") continue;
-
-    // Check explicit outlineLevel first, then style-based heading detection
-    const outlineLevel = block.outlineLevel;
-    const styleLevel = resolveHeadingLevelFromStyle(document, block.styleId);
-    const level =
-      outlineLevel !== undefined && outlineLevel >= 0
-        ? outlineLevel + 1
-        : styleLevel;
-
-    if (level === null) continue;
-
-    const text = extractParagraphText(block);
-    if (!text.trim()) continue;
-
-    const offset = block.from;
-    const pageIndex = findPageForOffset(pages, offset);
-    const sectionIndex = findSectionForOffset(sections, offset);
-
-    headings.push({
-      headingId: `heading-${block.blockId}-${offset}`,
-      level,
-      text: text.trim(),
-      offset,
-      pageIndex,
-      sectionIndex,
-    });
-  }
-
-  return headings;
-}
-
-function resolveHeadingLevelFromStyle(
-  document: CanonicalDocumentEnvelope,
-  styleId?: string,
-): number | null {
-  if (!styleId) {
-    return null;
-  }
-
-  const visited = new Set<string>();
-  let currentStyleId: string | undefined = styleId;
-  while (currentStyleId && !visited.has(currentStyleId)) {
-    visited.add(currentStyleId);
-    const style:
-      | { outlineLevel?: number; basedOn?: string }
-      | undefined = document.styles.paragraphs[currentStyleId];
-    if (!style) {
-      break;
-    }
-    if (
-      typeof style.outlineLevel === "number" &&
-      style.outlineLevel >= 0 &&
-      style.outlineLevel <= 8
-    ) {
-      return style.outlineLevel + 1;
-    }
-    currentStyleId = style.basedOn;
-  }
-
-  return headingLevelFromStyleId(styleId);
-}
-
-function extractParagraphText(
-  block: Extract<SurfaceBlockSnapshot, { kind: "paragraph" }>,
-): string {
-  let text = "";
-  for (const segment of block.segments) {
-    switch (segment.kind) {
-      case "text":
-        text += segment.text;
-        break;
-      case "tab":
-        text += "\t";
-        break;
-      case "hard_break":
-        text += "\n";
-        break;
-    }
-  }
-  if (!block.numberingPrefix) {
-    return text;
-  }
-  const trimmedText = text.trimStart();
-  if (
-    trimmedText === block.numberingPrefix ||
-    trimmedText.startsWith(`${block.numberingPrefix} `) ||
-    trimmedText.startsWith(`${block.numberingPrefix}\t`) ||
-    (block.numberingSuffix === "nothing" && trimmedText.startsWith(block.numberingPrefix))
-  ) {
-    return text;
-  }
-  const separator = block.numberingSuffix === "nothing" ? "" : " ";
-  return `${block.numberingPrefix}${separator}${text}`;
-}