openshain 0.1.0

package/src/tui/lines.ts

@@ -0,0 +1,75 @@
+import { displayWidth } from "../format.ts";
+import { logoSegments, type Segment } from "./banner.ts";
+import type { Entry, EntryKind } from "./controller.ts";
+
+export interface ScreenLine {
+  kind: EntryKind | "blank";
+  text: string;
+  /** Colored pieces of a logo row; the other rows are one color. */
+  segments?: Segment[];
+}
+
+/** What starts a line of each kind. The continuation lines of a wrapped entry are indented to match. */
+const MARKERS: Record<EntryKind, string> = {
+  user: "> ",
+  assistant: "⏺ ",
+  progress: "  ⎿ ",
+  notice: "! ",
+  question: "? ",
+  line: "  ",
+  logo: "",
+  banner: "",
+};
+
+/** Breaks text into lines no wider than `width` display columns, counting East Asian wide characters as two. */
+export function wrapText(text: string, width: number): string[] {
+  const out: string[] = [];
+  for (const paragraph of text.split("\n")) {
+    let line = "";
+    let used = 0;
+    for (const ch of paragraph) {
+      const w = displayWidth(ch);
+      if (used + w > width && line !== "") {
+        out.push(line);
+        line = "";
+        used = 0;
+      }
+      line += ch;
+      used += w;
+    }
+    out.push(line);
+  }
+  return out;
+}
+
+/** A blank row goes before an entry that starts something new: a message, a reply, a notice, a question. */
+function startsBlock(kind: EntryKind, previous: EntryKind | undefined): boolean {
+  if (previous === undefined) return false;
+  if (kind === "logo" || kind === "banner") return false;
+  if (kind === "progress") return previous === "user";
+  if (kind === "line") return previous !== "line";
+  return true;
+}
+
+/** The rows the screen shows for the entries, wrapped to the width, with markers and blank rows between blocks. */
+export function screenLines(entries: readonly Entry[], width: number): ScreenLine[] {
+  const lines: ScreenLine[] = [];
+  let previous: EntryKind | undefined;
+  for (const entry of entries) {
+    if (startsBlock(entry.kind, previous)) lines.push({ kind: "blank", text: "" });
+    if (entry.kind === "logo") {
+      // Never wrapped: a cut row of the wordmark reads better than a broken one.
+      lines.push({ kind: "logo", text: entry.text, segments: logoSegments(entry.text) });
+      previous = entry.kind;
+      continue;
+    }
+    const marker = MARKERS[entry.kind];
+    const indent = " ".repeat(displayWidth(marker));
+    const body = wrapText(entry.text, Math.max(8, width - displayWidth(marker)));
+    for (const [i, text] of body.entries()) {
+      lines.push({ kind: entry.kind, text: (i === 0 ? marker : indent) + text });
+    }
+    previous = entry.kind;
+  }
+  return lines;
+}

package/src/usage.ts

@@ -0,0 +1,44 @@
+import { countToolCalls } from "@openshain/agent";
+import type { AnyEvent, Event } from "@openshain/core";
+
+export interface UsageSummary {
+  modelCalls: number;
+  toolCalls: number;
+  inputTokens: number;
+  /** The part of inputTokens a prompt cache served. */
+  cachedInputTokens: number;
+  outputTokens: number;
+}
+
+/** Totals over a work's events: calls and tokens. */
+export function summarizeUsage(events: AnyEvent[]): UsageSummary {
+  const summary: UsageSummary = {
+    modelCalls: 0,
+    toolCalls: 0,
+    inputTokens: 0,
+    cachedInputTokens: 0,
+    outputTokens: 0,
+  };
+  for (const event of events) {
+    if (event.type === "model.requested") summary.modelCalls += 1;
+    if (event.type === "usage.recorded") {
+      const { payload } = event as Event<"usage.recorded">;
+      if (payload.kind === "model_inference") {
+        summary.inputTokens += payload.usage.inputTokens;
+        summary.cachedInputTokens += payload.usage.cachedInputTokens ?? 0;
+        summary.outputTokens += payload.usage.outputTokens;
+      }
+    }
+  }
+  summary.toolCalls = countToolCalls(events);
+  return summary;
+}
+
+export function formatUsage(summary: UsageSummary): string {
+  if (summary.modelCalls === 0 && summary.inputTokens === 0 && summary.outputTokens === 0) {
+    return `model 呼び出し 0 回、Tool 呼び出し ${summary.toolCalls} 回。model の使用量の記録はない`;
+  }
+  const cached =
+    summary.cachedInputTokens > 0 ? `(うちキャッシュ ${summary.cachedInputTokens})` : "";
+  return `model 呼び出し ${summary.modelCalls} 回、Tool 呼び出し ${summary.toolCalls} 回、入力 ${summary.inputTokens} トークン${cached}、出力 ${summary.outputTokens} トークン`;
+}

package/src/workspace.ts

@@ -0,0 +1,23 @@
+import { access } from "node:fs/promises";
+import { dirname, join, resolve } from "node:path";
+import { CONFIG_FILE_NAME, OpenshainError } from "@openshain/core";
+
+/** The nearest directory at or above `start` that holds openshain.yaml. */
+export async function findWorkspace(start: string): Promise<string> {
+  let dir = resolve(start);
+  for (;;) {
+    try {
+      await access(join(dir, CONFIG_FILE_NAME));
+      return dir;
+    } catch {
+      const parent = dirname(dir);
+      if (parent === dir) {
+        throw new OpenshainError(
+          "config",
+          `${CONFIG_FILE_NAME} が見つかりません。${resolve(start)} から上に向かって探しました。openshain init で作れます。`,
+        );
+      }
+      dir = parent;
+    }
+  }
+}