pi-one-ui 0.1.1 → 0.2.0

package/extensions/layouts/editor/factory.ts

@@ -0,0 +1,159 @@
+import type {
+  ExtensionAPI,
+  ExtensionContext,
+  KeybindingsManager,
+  Theme,
+} from "@earendil-works/pi-coding-agent";
+import type { EditorTheme, TUI } from "@earendil-works/pi-tui";
+import type { PolishedTuiConfig } from "../../app/config/shell.ts";
+import { modelLabelFor, type FooterState } from "../footer/index.ts";
+import { markAccentRailLayoutEditor } from "./accent-rail-layout-patch.ts";
+import { PolishedEditor, WrappedPolishedEditor } from "./ui.ts";
+import {
+  markEditorFactory,
+  markWrappedEditorFactory,
+  type EditorFactory,
+  type ZentuiEditorFactory,
+} from "./ownership.ts";
+
+export type EditorFactoryRuntime = {
+  readonly ownerToken: symbol;
+  readonly sessionTheme: Theme;
+  readonly getConfig: () => PolishedTuiConfig;
+  readonly getState: () => FooterState;
+  readonly getThinkingLevel: () => ReturnType<ExtensionAPI["getThinkingLevel"]>;
+  readonly getContextWindow: (ctx: ExtensionContext) => number | undefined;
+  readonly getContextPercent: (ctx: ExtensionContext) => number | undefined;
+  readonly getAgentDurationMs: () => number;
+  readonly isAgentActive: () => boolean;
+  readonly getProjectRoot: () => string | undefined;
+  readonly onRender: (requestRender: () => void) => void;
+  readonly onDecorationActive: (active: boolean) => void;
+  readonly isAccentRailActive: () => boolean;
+};
+
+/**
+ * Creates the native polished Editor factory for one Pi session.
+ *
+ * @param ctx Active Pi extension context.
+ * @param runtime Runtime selectors and lifecycle callbacks used by the Editor.
+ * @returns An owned Editor factory.
+ */
+export function createEditorFactory(
+  ctx: ExtensionContext,
+  runtime: EditorFactoryRuntime,
+): ZentuiEditorFactory {
+  const factory = ((
+    tui: TUI,
+    theme: EditorTheme,
+    keybindings: KeybindingsManager,
+  ) => {
+    runtime.onRender(() => tui.requestRender());
+    const editor = new PolishedEditor(
+      tui,
+      theme,
+      keybindings,
+      runtime.sessionTheme,
+      runtime.getConfig,
+      () => editorMetadata(ctx, runtime),
+      runtime.getThinkingLevel,
+      () => editorDecoration(ctx, runtime),
+      runtime.onDecorationActive,
+    );
+    markAccentRailLayoutEditor(
+      editor,
+      runtime.ownerToken,
+      runtime.isAccentRailActive,
+    );
+    return editor;
+  }) as ZentuiEditorFactory;
+  return markEditorFactory(factory, runtime.ownerToken);
+}
+
+/**
+ * Creates an Editor factory that decorates a third-party editor.
+ *
+ * @param ctx Active Pi extension context.
+ * @param baseFactory Third-party factory retained for restoration.
+ * @param runtime Runtime selectors and lifecycle callbacks used by the Editor.
+ * @returns An owned wrapper factory.
+ */
+export function createWrappedEditorFactory(
+  ctx: ExtensionContext,
+  baseFactory: EditorFactory,
+  runtime: EditorFactoryRuntime,
+): ZentuiEditorFactory {
+  const factory = ((
+    tui: TUI,
+    theme: EditorTheme,
+    keybindings: KeybindingsManager,
+  ) => {
+    runtime.onRender(() => tui.requestRender());
+    const editor = new WrappedPolishedEditor(
+      baseFactory(tui, theme, keybindings),
+      runtime.sessionTheme,
+      runtime.getConfig,
+      () => editorMetadata(ctx, runtime),
+      runtime.getThinkingLevel,
+      () => editorDecoration(ctx, runtime),
+      runtime.onDecorationActive,
+    );
+    markAccentRailLayoutEditor(
+      editor,
+      runtime.ownerToken,
+      runtime.isAccentRailActive,
+    );
+    return editor;
+  }) as ZentuiEditorFactory;
+  return markWrappedEditorFactory(factory, baseFactory, runtime.ownerToken);
+}
+
+/**
+ * Selects the model and session metadata shown by the Editor.
+ *
+ * @param ctx Active Pi extension context.
+ * @param runtime Runtime selectors for shared state.
+ * @returns Current Editor metadata.
+ */
+function editorMetadata(ctx: ExtensionContext, runtime: EditorFactoryRuntime) {
+  const state = runtime.getState();
+  const config = runtime.getConfig();
+  return {
+    modelLabel: modelLabelFor(state, config.components.editor.modelLabel),
+    modelId: state.modelId,
+    modelName: state.modelName,
+    providerLabel: state.providerLabel,
+    sessionName: ctx.sessionManager.getSessionName() ?? "",
+  };
+}
+
+/**
+ * Selects the dynamic project and usage metadata shown by the Editor.
+ *
+ * @param ctx Active Pi extension context.
+ * @param runtime Runtime selectors for shared state and timers.
+ * @returns Current Editor decoration data.
+ */
+function editorDecoration(
+  ctx: ExtensionContext,
+  runtime: EditorFactoryRuntime,
+) {
+  const state = runtime.getState();
+  const config = runtime.getConfig();
+  return {
+    cwd: ctx.cwd,
+    projectRoot: runtime.getProjectRoot(),
+    branch: state.branch,
+    dirty: state.dirty,
+    ahead: state.ahead,
+    behind: state.behind,
+    costLabel: state.costLabel,
+    modelLabel: modelLabelFor(state, config.components.editor.modelLabel),
+    thinkingLevel: runtime.getThinkingLevel(),
+    contextPercent: runtime.getContextPercent(ctx),
+    contextWindow: runtime.getContextWindow(ctx),
+    sessionName: ctx.sessionManager.getSessionName() ?? "",
+    agentDurationMs: runtime.getAgentDurationMs(),
+    agentActive: runtime.isAgentActive(),
+  };
+}

package/extensions/layouts/editor/index.ts

@@ -0,0 +1,16 @@
+export {
+  EditorLayoutController,
+  type EditorChangeResult,
+  type EditorLayoutControllerContext,
+} from "./controller.ts";
+export {
+  installHostAccentRailLayoutPatch,
+  markAccentRailLayoutEditor,
+  retainAccentRailLayoutPatchInstallation,
+  type AccentRailLayoutPatchDiagnostic,
+} from "./accent-rail-layout-patch.ts";
+export {
+  replaceEditorComponentWithExpandedText,
+  type EditorTransferFailureReason,
+} from "./editor-transfer.ts";
+export { PolishedEditor, WrappedPolishedEditor } from "./ui.ts";

package/extensions/{shell → layouts/editor}/minimalist-editor.ts

@@ -1,7 +1,7 @@
 import { basename, isAbsolute, relative, sep } from "node:path";
 import type { Theme } from "@earendil-works/pi-coding-agent";
 import { truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
-import type { ZentuiConfig } from "../app/config/shell.ts";
+import type { ZentuiConfig } from "../../app/config/shell.ts";
 import { sanitizeEditorMetadataText } from "./editor-metadata-format.ts";
 import {
   buildContextGauge,
@@ -9,9 +9,9 @@ import {
   formatCount,
   formatCwdLabel,
   formatElapsedDuration,
-} from "./format.ts";
+} from "../../shared/format.ts";
 
-export { formatElapsedDuration } from "./format.ts";
+export { formatElapsedDuration } from "../../shared/format.ts";
 
 import {
   EDITOR_ACCENT_FALLBACK,
@@ -19,7 +19,7 @@ import {
   renderStyleForSource,
   renderStyleForSourceOrFallback,
   safeThemeFg,
-} from "./style.ts";
+} from "../../shared/style.ts";
 
 const MINIMALIST_MODEL_FALLBACK = {
   theme: "syntaxKeyword",

package/extensions/layouts/editor/ownership.ts

@@ -0,0 +1,103 @@
+import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
+
+/** Symbol used to identify factories created by the Editor layout. */
+export const ZENTUI_EDITOR_FACTORY = Symbol.for("pi-zentui.editor-factory");
+
+/** Symbol used to retain the wrapped third-party editor factory. */
+export const ZENTUI_EDITOR_BASE_FACTORY = Symbol.for(
+  "pi-zentui.editor-base-factory",
+);
+
+/** Symbol used to identify the active Editor layout owner. */
+export const ZENTUI_EDITOR_OWNER = Symbol.for("pi-zentui.editor-owner");
+
+export type EditorFactory = NonNullable<
+  Parameters<ExtensionContext["ui"]["setEditorComponent"]>[0]
+>;
+
+export type ZentuiEditorFactory = EditorFactory & {
+  [ZENTUI_EDITOR_FACTORY]?: true;
+  [ZENTUI_EDITOR_BASE_FACTORY]?: EditorFactory;
+  [ZENTUI_EDITOR_OWNER]?: symbol;
+};
+
+/**
+ * Reports whether a factory was marked as an Editor layout factory.
+ *
+ * @param factory Candidate host editor factory.
+ * @returns Whether the factory carries the Editor layout marker.
+ */
+export function isZentuiEditorFactory(
+  factory: EditorFactory | undefined,
+): boolean {
+  return Boolean(
+    (factory as ZentuiEditorFactory | undefined)?.[ZENTUI_EDITOR_FACTORY],
+  );
+}
+
+/**
+ * Returns the third-party factory wrapped by an Editor layout factory.
+ *
+ * @param factory Candidate host editor factory.
+ * @returns The retained base factory, when present.
+ */
+export function getZentuiEditorBaseFactory(
+  factory: EditorFactory | undefined,
+): EditorFactory | undefined {
+  return (factory as ZentuiEditorFactory | undefined)?.[
+    ZENTUI_EDITOR_BASE_FACTORY
+  ];
+}
+
+/**
+ * Reports whether a factory belongs to the supplied Editor layout owner.
+ *
+ * @param factory Candidate host editor factory.
+ * @param ownerToken Owner token to compare.
+ * @returns Whether the owner marker matches.
+ */
+export function isOwnedEditorFactory(
+  factory: EditorFactory | undefined,
+  ownerToken: symbol,
+): boolean {
+  return (
+    (factory as ZentuiEditorFactory | undefined)?.[ZENTUI_EDITOR_OWNER] ===
+    ownerToken
+  );
+}
+
+/**
+ * Marks a factory as owned by the Editor layout.
+ *
+ * @param factory Factory to mark.
+ * @param ownerToken Owner token written to the factory.
+ * @returns The marked factory.
+ */
+export function markEditorFactory<T extends EditorFactory>(
+  factory: T,
+  ownerToken: symbol,
+): T {
+  const marked = factory as T & ZentuiEditorFactory;
+  marked[ZENTUI_EDITOR_FACTORY] = true;
+  marked[ZENTUI_EDITOR_OWNER] = ownerToken;
+  return factory;
+}
+
+/**
+ * Retains a base factory on an Editor wrapper for safe restoration.
+ *
+ * @param factory Wrapper factory to mark.
+ * @param baseFactory Third-party factory wrapped by the layout.
+ * @param ownerToken Owner token written to the wrapper.
+ * @returns The marked wrapper.
+ */
+export function markWrappedEditorFactory<T extends EditorFactory>(
+  factory: T,
+  baseFactory: EditorFactory,
+  ownerToken: symbol,
+): T {
+  const marked = markEditorFactory(factory, ownerToken) as T &
+    ZentuiEditorFactory;
+  marked[ZENTUI_EDITOR_BASE_FACTORY] = baseFactory;
+  return factory;
+}

package/extensions/{shell → layouts/editor}/ui.ts

@@ -17,7 +17,7 @@ import {
   renderAccentRailEditorFrame,
 } from "./accent-rail-editor.ts";
 import { renderCompletionPalette } from "./completion-menu.ts";
-import type { EditorStyle, ZentuiConfig } from "../app/config/shell.ts";
+import type { EditorStyle, ZentuiConfig } from "../../app/config/shell.ts";
 import { renderEditorMetadataFormat } from "./editor-metadata-format.ts";
 import {
   type MinimalistEditorMetadata,
@@ -28,7 +28,7 @@ import {
   EDITOR_BORDER_FALLBACK,
   renderStyleForSourceOrFallback,
   safeThemeFg,
-} from "./style.ts";
+} from "../../shared/style.ts";
 
 const LEGACY_SPLIT_POLISHED_FRAME = Symbol.for("pi-zentui.polished-frame");