diff --git a/packages/app/e2e/performance/README.md b/packages/app/e2e/performance/README.md index ce868d573bb..675057366ba 100644 --- a/packages/app/e2e/performance/README.md +++ b/packages/app/e2e/performance/README.md @@ -56,6 +56,32 @@ Benchmarks do not assert machine-dependent performance budgets. Streaming proces Committed smoke and regression tests continue to own correctness coverage for pagination, tab paint, context resize, collapse state, and composer spacing. +## Desktop profiler + +The desktop profiler launches the existing production build directly, creates isolated desktop state, chooses an available CDP port, and writes reports under the OS temporary directory by default. + +```sh +bun run profile:desktop --help +``` + +Create a private partial snapshot from the default local database and run Home once: + +```sh +bun run profile:desktop --partial-snapshot-out /tmp/opencode-perf.db \ + --window-end 2026-08-04T06:14:26.878Z \ + --scenarios home,calibration --skip-build +``` + +Repeat against the immutable partial snapshot: + +```sh +bun run profile:desktop --mode partial-snapshot --db /tmp/opencode-perf.db \ + --window-end 2026-08-04T06:14:26.878Z \ + --scenarios home,calibration --runs 3 --skip-build +``` + +Partial snapshots contain private application data and must not be committed or shared. The profiler copies each partial snapshot to a per-run working database and remaps selected project paths to temporary workspaces, leaving the source snapshot unchanged. `PROFILE_SUMMARY` is the compact comparison output; `PROFILE_REPORT` points to the complete JSON report with the database hash, invocation parameters, raw runs, and attribution data. + ## Chrome traces Set `OPENCODE_PERFORMANCE_TRACE_DIR` to emit a standard Chrome DevTools trace for every benchmark page automatically: diff --git a/packages/app/e2e/performance/desktop-profile/corpus.ts b/packages/app/e2e/performance/desktop-profile/corpus.ts new file mode 100644 index 00000000000..fe7cf155622 --- /dev/null +++ b/packages/app/e2e/performance/desktop-profile/corpus.ts @@ -0,0 +1,137 @@ +import { Database } from "bun:sqlite" +import { mkdir, rm } from "node:fs/promises" +import path from "node:path" +import { progress } from "./progress" +import type { Options, Target } from "./types" + +export async function createPartialSnapshot(source: string, destination: string, options: Options, targets: Target[]) { + await mkdir(path.dirname(destination), { recursive: true }) + await rm(destination, { force: true }) + const input = new Database(source, { readonly: true }) + const schema = input + .query( + `SELECT type, name, sql FROM sqlite_schema + WHERE sql IS NOT NULL AND name NOT LIKE 'sqlite_%' + ORDER BY CASE type WHEN 'table' THEN 0 WHEN 'index' THEN 1 ELSE 2 END, name`, + ) + .all() as { type: string; name: string; sql: string }[] + input.close() + + const output = new Database(destination, { create: true }) + output.run("PRAGMA foreign_keys = OFF") + schema.filter((item) => item.type === "table").forEach((item) => output.run(item.sql)) + output.run("ATTACH DATABASE ? AS source", source) + const selected = [...new Set(targets.map((target) => target.id))] + const placeholders = selected.map(() => "?").join(",") + + for (const table of schema.filter((item) => item.type === "table").map((item) => item.name)) { + progress("copying partial snapshot table", { table }) + if (table === "event") continue + if (table === "message") { + output.run( + `INSERT INTO main.message SELECT * FROM source.message + WHERE (time_created >= ? AND time_created < ? AND session_id IN ( + SELECT id FROM source.session WHERE parent_id IS NULL + )) OR session_id IN (${placeholders})`, + options.windowStart, + options.windowEnd, + ...selected, + ) + continue + } + if (table === "part") { + output.run("INSERT INTO main.part SELECT * FROM source.part WHERE message_id IN (SELECT id FROM main.message)") + continue + } + if (["session_context_epoch", "session_input", "session_message", "session_share", "todo"].includes(table)) { + output.run( + `INSERT INTO main."${table}" SELECT * FROM source."${table}" WHERE session_id IN (${placeholders})`, + ...selected, + ) + continue + } + output.run(`INSERT INTO main."${table}" SELECT * FROM source."${table}"`) + } + output.run("DETACH DATABASE source") + schema.filter((item) => item.type !== "table").forEach((item) => output.run(item.sql)) + output.close() +} + +export async function fingerprint(file: string) { + const input = Bun.file(file) + const hasher = new Bun.CryptoHasher("sha256") + for await (const chunk of input.stream()) hasher.update(chunk) + return { bytes: input.size, sha256: hasher.digest("hex") } +} + +export function loadCorpus(options: Options) { + const database = new Database(options.database, { readonly: true }) + database.run("PRAGMA query_only = ON") + const sessions = database + .query( + `SELECT id, project_id AS projectID, directory, title + FROM session AS candidate + WHERE parent_id IS NULL + AND EXISTS ( + SELECT 1 FROM message + WHERE session_id = candidate.id AND time_created >= ? AND time_created < ? + )`, + ) + .all(options.windowStart, options.windowEnd) as { id: string; projectID: string; directory: string; title: string }[] + const messageRows = database.query( + `SELECT id, data FROM message + WHERE session_id = ? AND time_created >= ? AND time_created < ? + ORDER BY time_created, id`, + ) + const partRows = database.query(`SELECT data FROM part WHERE message_id = ? ORDER BY id`) + const ranked = sessions + .map((session) => { + const messages = messageRows.all(session.id, options.windowStart, options.windowEnd) as { + id: string + data: string + }[] + const parts = messages.flatMap((message) => partRows.all(message.id) as { data: string }[]) + return { + ...session, + bytes: + messages.reduce((sum, message) => sum + Buffer.byteLength(message.data), 0) + + parts.reduce((sum, part) => sum + Buffer.byteLength(part.data), 0), + messages: messages.length, + parts: parts.length, + userTurns: messages.filter((message) => JSON.parse(message.data).role === "user").length, + } + }) + .filter((session) => session.messages > 0) + .sort((a, b) => a.bytes - b.bytes || a.id.localeCompare(b.id)) + if (ranked.length === 0) throw new Error("No sessions found in the profile window") + const select = (label: Target["label"], percentile: number) => ({ + label, + ...ranked[Math.max(0, Math.ceil(ranked.length * percentile) - 1)]!, + }) + const targets = [select("p50", 0.5), select("p95", 0.95), select("max", 1)] satisfies Target[] + const typingText = loadTypingText(database, partRows, messageRows, targets[2]!, options) + const projectIDs = [...new Set(ranked.map((session) => session.projectID))] + database.close() + return { targets, typingText, projectIDs } +} + +function loadTypingText( + database: Database, + partRows: ReturnType, + messageRows: ReturnType, + target: Target, + options: Options, +) { + const messages = messageRows.all(target.id, options.windowStart, options.windowEnd) as { id: string; data: string }[] + const text = messages + .filter((message) => JSON.parse(message.data).role === "user") + .flatMap((message) => + (partRows.all(message.id) as { data: string }[]).flatMap((part) => { + const data = JSON.parse(part.data) + return data.type === "text" && typeof data.text === "string" ? [data.text] : [] + }), + ) + .sort((a, b) => b.length - a.length)[0] + if (!text) throw new Error("No real user prompt found for composer profiling") + return text +} diff --git a/packages/app/e2e/performance/desktop-profile/desktop-state.ts b/packages/app/e2e/performance/desktop-profile/desktop-state.ts new file mode 100644 index 00000000000..a9df97b1860 --- /dev/null +++ b/packages/app/e2e/performance/desktop-profile/desktop-state.ts @@ -0,0 +1,116 @@ +import { Database } from "bun:sqlite" +import { mkdir } from "node:fs/promises" +import path from "node:path" +import type { Options } from "./types" + +export async function prepareDesktopState( + options: Options, + databasePath: string, + userData: string, + run: number, + projectIDs: string[], +) { + const database = new Database(databasePath) + const projects = database.query("SELECT id, worktree, sandboxes FROM project ORDER BY id").all() as { + id: string + worktree: string + sandboxes: string + }[] + const selected = new Set(projectIDs) + const profileProjects = projects.filter((project) => selected.has(project.id)) + const worktrees = + options.mode === "partial-snapshot" + ? await remapDirectories(database, profileProjects, path.join(options.output, "workspaces", String(run))) + : profileProjects.map((project) => project.worktree) + database.close() + + await Bun.write( + path.join(userData, "opencode.settings"), + JSON.stringify({ firstLaunchOnboardingComplete: true, oldLayoutEligible: true, tauriMigrated: true }), + ) + await Bun.write( + path.join(userData, "opencode.global.dat"), + JSON.stringify({ + server: JSON.stringify({ + list: [], + projects: { local: worktrees.map((worktree) => ({ worktree, expanded: true })) }, + lastProject: worktrees[0] ? { local: worktrees[0] } : {}, + recentlyClosed: {}, + }), + }), + ) +} + +async function remapDirectories( + database: Database, + projects: { id: string; worktree: string; sandboxes: string }[], + root: string, +) { + await mkdir(root, { recursive: true }) + const mappings = new Map() + const worktrees = await Promise.all( + projects.map(async (project, index) => { + const worktree = path.join(root, `project-${String(index + 1).padStart(3, "0")}`) + await mkdir(worktree, { recursive: true }) + mappings.set(project.worktree, worktree) + const sandboxes = JSON.parse(project.sandboxes) as string[] + const nextSandboxes = await Promise.all( + sandboxes.map(async (sandbox, sandboxIndex) => { + const next = path.join(worktree, `sandbox-${sandboxIndex + 1}`) + await mkdir(next, { recursive: true }) + mappings.set(sandbox, next) + return next + }), + ) + database.run("UPDATE project SET worktree = ?, sandboxes = ? WHERE id = ?", worktree, JSON.stringify(nextSandboxes), project.id) + return worktree + }), + ) + const byProject = new Map(projects.map((project, index) => [project.id, worktrees[index]!])) + const sessions = database.query("SELECT id, project_id, directory FROM session").all() as { + id: string + project_id: string + directory: string + }[] + const directories = database.query("SELECT * FROM project_directory").all() as { + project_id: string + directory: string + type: string | null + strategy: string | null + time_created: number + }[] + const selected = new Set(projects.map((project) => project.id)) + const nextDirectories = await Promise.all( + directories.filter((item) => selected.has(item.project_id)).map(async (item, index) => { + const directory = + mappings.get(item.directory) ?? path.join(byProject.get(item.project_id) ?? root, `directory-${index + 1}`) + await mkdir(directory, { recursive: true }) + return { ...item, directory } + }), + ) + database.transaction(() => { + sessions.filter((session) => selected.has(session.project_id)).forEach((session) => + database.run( + "UPDATE session SET directory = ? WHERE id = ?", + mappings.get(session.directory) ?? byProject.get(session.project_id) ?? worktrees[0]!, + session.id, + ), + ) + database.run( + `DELETE FROM project_directory WHERE project_id IN (${projects.map(() => "?").join(",")})`, + ...projects.map((project) => project.id), + ) + nextDirectories.forEach((item) => + database.run( + `INSERT INTO project_directory (project_id, directory, type, strategy, time_created) + VALUES (?, ?, ?, ?, ?)`, + item.project_id, + item.directory, + item.type, + item.strategy, + item.time_created, + ), + ) + })() + return worktrees +} diff --git a/packages/app/e2e/performance/desktop-profile/options.test.ts b/packages/app/e2e/performance/desktop-profile/options.test.ts new file mode 100644 index 00000000000..dda79da1636 --- /dev/null +++ b/packages/app/e2e/performance/desktop-profile/options.test.ts @@ -0,0 +1,54 @@ +import { Database } from "bun:sqlite" +import { afterAll, expect, test } from "bun:test" +import { mkdir, rm } from "node:fs/promises" +import path from "node:path" +import { createPartialSnapshot, fingerprint } from "./corpus" +import { parseOptions } from "./options" + +const directory = path.join(import.meta.dir, `.tmp-${process.pid}`) +const source = path.join(directory, "source.db") +const partialSnapshot = path.join(directory, "partial-snapshot.db") +await mkdir(directory, { recursive: true }) +const database = new Database(source, { create: true }) +database.run("CREATE TABLE sample (value TEXT NOT NULL)") +database.run("INSERT INTO sample VALUES ('repeatable')") +database.close() + +afterAll(() => rm(directory, { recursive: true, force: true })) + +test("parses a portable fixed-window partial snapshot invocation", () => { + const options = parseOptions([ + "--mode", + "partial-snapshot", + "--db", + source, + "--window-end", + "2026-08-04T06:14:26.878Z", + "--window-hours", + "24", + "--scenarios", + "home,calibration", + "--runs", + "3", + "--skip-build", + ])! + + expect(options.database).toBe(source) + expect(options.windowEnd).toBe(1_785_824_066_878) + expect(options.windowStart).toBe(1_785_737_666_878) + expect(options.scenarios).toEqual(["home", "calibration"]) + expect(options.runs).toBe(3) + expect(options.build).toBe(false) +}) + +test("creates a consistent private partial database snapshot", async () => { + const options = parseOptions(["--db", source, "--window-end", "2026-08-04T06:14:26.878Z"])! + await createPartialSnapshot(source, partialSnapshot, options, []) + const copy = new Database(partialSnapshot, { readonly: true }) + expect(copy.query("SELECT value FROM sample").get()).toEqual({ value: "repeatable" }) + copy.close() + expect(await fingerprint(partialSnapshot)).toEqual({ + bytes: expect.any(Number), + sha256: expect.stringMatching(/^[a-f0-9]{64}$/), + }) +}) diff --git a/packages/app/e2e/performance/desktop-profile/options.ts b/packages/app/e2e/performance/desktop-profile/options.ts new file mode 100644 index 00000000000..fd5d6496759 --- /dev/null +++ b/packages/app/e2e/performance/desktop-profile/options.ts @@ -0,0 +1,88 @@ +import { Global } from "@opencode-ai/core/global" +import { existsSync } from "node:fs" +import { tmpdir } from "node:os" +import path from "node:path" +import { scenarios, type Options, type Scenario } from "./types" + +const help = `Desktop renderer profiler + +Usage: + bun run profile:desktop [options] + +Options: + --mode local|partial-snapshot + Local corpus or fixed partial snapshot (default: local) + --db SQLite database (default: opencode data directory) + --partial-snapshot-out + Copy the benchmark corpus to a private partial snapshot + --output Report directory (default: OS temp directory) + --window-end End of corpus window (default: now; required for partial snapshot) + --window-hours Corpus window size (default: 24) + --scenarios Comma list: ${scenarios.join(",")} (default: all) + --runs Restart Electron and repeat (default: 1) + --skip-build Use the existing desktop production build + --diagnostics Capture Chrome traces + --cpu Capture sampled CPU summaries + --response-urls Attribute Response.text durations by URL + --help Show this message + +Partial snapshots contain private application data. Do not commit or share them. +` + +export function parseOptions(args: string[], now = Date.now()): Options | undefined { + if (args.includes("--help")) { + console.log(help) + return + } + + const value = (name: string) => { + const index = args.indexOf(name) + if (index === -1) return + const result = args[index + 1] + if (!result || result.startsWith("--")) throw new Error(`${name} requires a value`) + return result + } + const mode = value("--mode") ?? "local" + if (mode !== "local" && mode !== "partial-snapshot") throw new Error(`Unsupported mode: ${mode}`) + const endValue = value("--window-end") + if (mode === "partial-snapshot" && !endValue) + throw new Error("--window-end is required in partial-snapshot mode") + const windowEnd = endValue ? parseTime(endValue) : now + const windowHours = number(value("--window-hours") ?? "24", "--window-hours") + const selected = (value("--scenarios")?.split(",") ?? [...scenarios]).map((item) => item.trim()) + if (selected.some((item) => !scenarios.includes(item as Scenario))) + throw new Error(`--scenarios must contain only: ${scenarios.join(", ")}`) + const database = path.resolve(value("--db") ?? path.join(Global.Path.data, "opencode.db")) + if (!existsSync(database)) throw new Error(`Database does not exist: ${database}`) + + return { + mode, + database, + output: path.resolve( + value("--output") ?? path.join(tmpdir(), "opencode-performance", new Date(windowEnd).toISOString().replace(/[:.]/g, "-")), + ), + windowStart: windowEnd - windowHours * 60 * 60 * 1_000, + windowEnd, + scenarios: selected as Scenario[], + runs: number(value("--runs") ?? "1", "--runs"), + build: !args.includes("--skip-build"), + diagnostics: args.includes("--diagnostics"), + cpu: args.includes("--cpu"), + responseURLs: args.includes("--response-urls"), + partialSnapshotOut: value("--partial-snapshot-out") + ? path.resolve(value("--partial-snapshot-out")!) + : undefined, + } +} + +function parseTime(value: string) { + const result = /^\d+$/.test(value) ? Number(value) : Date.parse(value) + if (!Number.isFinite(result)) throw new Error(`Invalid --window-end: ${value}`) + return result +} + +function number(value: string, option: string) { + const result = Number(value) + if (!Number.isFinite(result) || result <= 0) throw new Error(`${option} must be greater than zero`) + return result +} diff --git a/packages/app/e2e/performance/desktop-profile/probe.ts b/packages/app/e2e/performance/desktop-profile/probe.ts new file mode 100644 index 00000000000..70b907c6227 --- /dev/null +++ b/packages/app/e2e/performance/desktop-profile/probe.ts @@ -0,0 +1,192 @@ +import type { Page } from "@playwright/test" +import type { Options, ProbeResult } from "./types" + +export async function installProbe(page: Page, options: Options) { + await page.addInitScript((attributeResponses) => { + const state = { + longTasks: [] as number[], + animationFrames: [] as ProbeResult["animationFrames"], + frameGaps: [] as number[], + responseText: [] as ProbeResult["responseText"], + } + ;(window as Window & { __opencodeRendererProfile?: typeof state }).__opencodeRendererProfile = state + if (PerformanceObserver.supportedEntryTypes.includes("longtask")) { + new PerformanceObserver((list) => + state.longTasks.push(...list.getEntries().map((entry) => entry.duration)), + ).observe({ type: "longtask" }) + } + if (PerformanceObserver.supportedEntryTypes.includes("long-animation-frame")) { + new PerformanceObserver((list) => + state.animationFrames.push( + ...list.getEntries().map((entry) => { + const frame = entry as PerformanceEntry & { + blockingDuration: number + scripts?: { + duration: number + forcedStyleAndLayoutDuration?: number + sourceFunctionName?: string + sourceURL?: string + sourceCharPosition?: number + invoker?: string + invokerType?: string + }[] + } + return { + duration: frame.duration, + blockingDuration: frame.blockingDuration, + forcedStyleAndLayoutDuration: + frame.scripts?.reduce((sum, script) => sum + (script.forcedStyleAndLayoutDuration ?? 0), 0) ?? 0, + scripts: + frame.scripts?.map((script) => ({ + function: script.sourceFunctionName || "(anonymous)", + source: script.sourceURL?.split("/").at(-1) || "(document)", + position: script.sourceCharPosition ?? -1, + invoker: script.invoker ?? "(unknown)", + invokerType: script.invokerType ?? "(unknown)", + duration: script.duration, + forcedStyleAndLayoutDuration: script.forcedStyleAndLayoutDuration ?? 0, + })) ?? [], + } + }), + ), + ).observe({ type: "long-animation-frame" }) + } + let previous = performance.now() + const frame = (now: number) => { + const gap = now - previous + if (gap > 20) state.frameGaps.push(gap) + previous = now + requestAnimationFrame(frame) + } + requestAnimationFrame(frame) + if (!attributeResponses) return + const responseText = Response.prototype.text + Response.prototype.text = function () { + const started = performance.now() + const url = this.url + return responseText.call(this).then((text) => { + state.responseText.push({ url, duration: performance.now() - started }) + return text + }) + } + }, options.responseURLs) +} + +export async function resetProbe(page: Page) { + await page.evaluate(() => { + const state = (window as Window & { __opencodeRendererProfile?: ProbeResult }).__opencodeRendererProfile + if (!state) return + state.longTasks.length = 0 + state.animationFrames.length = 0 + state.frameGaps.length = 0 + state.responseText.length = 0 + }) +} + +export async function collectProbe(page: Page) { + return page.evaluate( + () => (window as Window & { __opencodeRendererProfile?: ProbeResult }).__opencodeRendererProfile!, + ) +} + +export function summarizeProbe(probe: ProbeResult) { + const scripts = new Map< + string, + { + function: string + source: string + position: number + invoker: string + invokerType: string + durationMs: number + forcedStyleAndLayoutMs: number + } + >() + probe.animationFrames + .flatMap((frame) => frame.scripts) + .forEach((script) => { + const key = `${script.source}:${script.position}:${script.invoker}` + const current = scripts.get(key) ?? { + function: script.function, + source: script.source, + position: script.position, + invoker: script.invoker, + invokerType: script.invokerType, + durationMs: 0, + forcedStyleAndLayoutMs: 0, + } + current.durationMs += script.duration + current.forcedStyleAndLayoutMs += script.forcedStyleAndLayoutDuration + scripts.set(key, current) + }) + return { + longTasks: { + count: probe.longTasks.length, + totalMs: sum(probe.longTasks), + maxMs: Math.max(0, ...probe.longTasks), + }, + longAnimationFrames: { + count: probe.animationFrames.length, + totalBlockingMs: sum(probe.animationFrames.map((frame) => frame.blockingDuration)), + maxDurationMs: Math.max(0, ...probe.animationFrames.map((frame) => frame.duration)), + forcedStyleAndLayoutMs: sum(probe.animationFrames.map((frame) => frame.forcedStyleAndLayoutDuration)), + scripts: [...scripts.values()].sort((a, b) => b.durationMs - a.durationMs).slice(0, 15), + }, + frameGaps: { + count: probe.frameGaps.length, + maxMs: Math.max(0, ...probe.frameGaps), + }, + responseText: probe.responseText + .map((item) => ({ path: responsePath(item.url), durationMs: item.duration })) + .sort((a, b) => b.durationMs - a.durationMs), + } +} + +export async function startCPUProfile(page: Page, enabled: boolean) { + if (!enabled) return { stop: async () => [] } + const session = await page.context().newCDPSession(page) + await session.send("Profiler.enable") + await session.send("Profiler.setSamplingInterval", { interval: 1_000 }) + await session.send("Profiler.start") + return { + async stop() { + const result = await session.send("Profiler.stop") + await session.detach() + const self = new Map() + result.profile.samples?.forEach((id, index) => { + self.set(id, (self.get(id) ?? 0) + (result.profile.timeDeltas?.[index] ?? 0) / 1_000) + }) + return result.profile.nodes + .map((node) => ({ + function: node.callFrame.functionName || "(anonymous)", + source: sourceName(node.callFrame.url), + line: node.callFrame.lineNumber + 1, + selfMs: self.get(node.id) ?? 0, + })) + .filter((node) => node.selfMs >= 1) + .sort((a, b) => b.selfMs - a.selfMs) + .slice(0, 40) + }, + } +} + +function responsePath(value: string) { + try { + return new URL(value).pathname + } catch { + return value + } +} + +function sourceName(value: string) { + if (!value) return "(native)" + try { + return new URL(value).pathname.split("/").at(-1) || "(document)" + } catch { + return value.split(/[\\/]/).at(-1) || value + } +} + +function sum(values: number[]) { + return values.reduce((total, value) => total + value, 0) +} diff --git a/packages/app/e2e/performance/desktop-profile/progress.ts b/packages/app/e2e/performance/desktop-profile/progress.ts new file mode 100644 index 00000000000..4799e5b14b5 --- /dev/null +++ b/packages/app/e2e/performance/desktop-profile/progress.ts @@ -0,0 +1,7 @@ +const started = performance.now() + +export function progress(message: string, details?: Record) { + const elapsed = ((performance.now() - started) / 1_000).toFixed(1) + const suffix = details ? ` ${JSON.stringify(details)}` : "" + console.error(`[desktop-profile +${elapsed}s] ${message}${suffix}`) +} diff --git a/packages/app/e2e/performance/desktop-profile/runtime.ts b/packages/app/e2e/performance/desktop-profile/runtime.ts new file mode 100644 index 00000000000..d6379696eb3 --- /dev/null +++ b/packages/app/e2e/performance/desktop-profile/runtime.ts @@ -0,0 +1,156 @@ +import { chromium, type Page } from "@playwright/test" +import { copyFile, mkdir, rm } from "node:fs/promises" +import path from "node:path" +import { prepareDesktopState } from "./desktop-state" +import { progress } from "./progress" +import type { Options } from "./types" + +export async function withDesktop( + options: Options, + desktop: string, + run: number, + projectIDs: string[], + use: (page: Page) => Promise, +) { + const port = availablePort() + const endpoint = `http://127.0.0.1:${port}` + const userData = path.join(options.output, `user-data-${run}`) + const database = + options.mode === "partial-snapshot" ? path.join(options.output, `working-database-${run}.db`) : options.database + await rm(userData, { recursive: true, force: true }) + await mkdir(userData, { recursive: true }) + if (database !== options.database) await copyFile(options.database, database) + await prepareDesktopState(options, database, userData, run, projectIDs) + const electron = path.join( + desktop, + "node_modules", + "electron", + "dist", + (await Bun.file(path.join(desktop, "node_modules", "electron", "path.txt")).text()).trim(), + ) + progress("launching Electron", { run, port }) + const child = Bun.spawn([electron, "."], { + cwd: desktop, + env: { + ...process.env, + OPENCODE_DB: database, + OPENCODE_CHANNEL: "dev", + OPENCODE_PROFILE_LOAF: "1", + OPENCODE_PROFILE_CDP_PORT: String(port), + OPENCODE_PROFILE_USER_DATA: userData, + OPENCODE_PERFORMANCE_TRACE_DIR: options.diagnostics ? path.join(options.output, "traces", String(run)) : "", + OPENCODE_PERFORMANCE_RUN_ID: `desktop-${run}`, + }, + stdout: "pipe", + stderr: "pipe", + }) + const stdout = drain(child.stdout, "stdout") + const stderr = drain(child.stderr, "stderr") + let browser: Awaited> | undefined + + try { + progress("waiting for CDP", { run }) + await waitForCDP(endpoint, child, stdout, stderr) + progress("connecting Playwright", { run }) + browser = await chromium.connectOverCDP(endpoint) + progress("waiting for renderer", { run }) + const page = await waitForRenderer(browser) + progress("waiting for desktop API", { run }) + await page.waitForFunction(() => typeof window.api === "object", undefined, { timeout: 60_000 }) + progress("desktop ready", { run }) + return await use(page) + } finally { + progress("stopping Electron", { run }) + await browser?.close().catch(() => {}) + await killTree(child.pid) + await Promise.allSettled([stdout, stderr]) + if (database !== options.database) { + await Bun.sleep(500) + await rm(database, { force: true }).catch(() => undefined) + } + } +} + +export async function run(command: string[], cwd: string, database: string) { + const child = Bun.spawn(command, { + cwd, + env: { ...process.env, OPENCODE_DB: database, OPENCODE_CHANNEL: "dev" }, + stdout: "inherit", + stderr: "inherit", + }) + const code = await child.exited + if (code !== 0) throw new Error(`${command.join(" ")} exited with ${code}`) +} + +function availablePort() { + const server = Bun.serve({ hostname: "127.0.0.1", port: 0, fetch: () => new Response() }) + const port = server.port + server.stop(true) + return port +} + +async function waitForCDP( + endpoint: string, + child: ReturnType, + stdout: Promise, + stderr: Promise, +) { + const timeout = Date.now() + 5 * 60_000 + let heartbeat = Date.now() + 10_000 + while (Date.now() < timeout) { + const ready = await fetch(`${endpoint}/json/version`) + .then((response) => response.ok) + .catch(() => false) + if (ready) return + if (child.exitCode !== null) + throw new Error(`Desktop exited before CDP was ready (${child.exitCode})\n${await stdout}\n${await stderr}`) + if (Date.now() >= heartbeat) { + progress("still waiting for CDP") + heartbeat = Date.now() + 10_000 + } + await Bun.sleep(250) + } + throw new Error("Timed out waiting for desktop CDP") +} + +async function waitForRenderer(browser: Awaited>) { + const timeout = Date.now() + 60_000 + let heartbeat = Date.now() + 10_000 + while (Date.now() < timeout) { + const page = browser + .contexts() + .flatMap((context) => context.pages()) + .find((candidate) => candidate.url().startsWith("oc://renderer")) + if (page) return page + if (Date.now() >= heartbeat) { + progress("still waiting for renderer") + heartbeat = Date.now() + 10_000 + } + await Bun.sleep(100) + } + throw new Error("Desktop renderer target was not found") +} + +async function drain(stream: ReadableStream, label: string) { + const decoder = new TextDecoder() + let output = "" + let pending = "" + for await (const chunk of stream) { + const text = decoder.decode(chunk, { stream: true }) + output = (output + text).slice(-50_000) + const lines = (pending + text).split(/\r?\n/) + pending = lines.pop() ?? "" + lines.filter(Boolean).forEach((line) => progress(`Electron ${label}`, { line: line.slice(0, 500) })) + } + if (pending) progress(`Electron ${label}`, { line: pending.slice(0, 500) }) + return output + decoder.decode() +} + +async function killTree(pid: number) { + if (process.platform !== "win32") { + process.kill(pid, "SIGTERM") + return + } + const child = Bun.spawn(["taskkill", "/pid", String(pid), "/T", "/F"], { stdout: "ignore", stderr: "ignore" }) + await child.exited +} diff --git a/packages/app/e2e/performance/desktop-profile/scenario-utils.ts b/packages/app/e2e/performance/desktop-profile/scenario-utils.ts new file mode 100644 index 00000000000..9285b8ee8ed --- /dev/null +++ b/packages/app/e2e/performance/desktop-profile/scenario-utils.ts @@ -0,0 +1,75 @@ +import type { Page } from "@playwright/test" +import { progress } from "./progress" + +export async function setDesktopRoute(page: Page, route: string) { + await page.evaluate(async (value) => { + const api = window.api as typeof window.api & { getWindowID?: () => Promise } + const id = (await api.getWindowID?.()) ?? "browser" + localStorage.setItem(`opencode.desktop.window.${id}.last-active-url`, value) + }, route) +} + +export async function waitForQuietDOM(page: Page) { + progress("waiting for DOM to settle") + await page.evaluate( + () => + new Promise((resolve) => { + let settled = false + let timer = setTimeout(done, 750) + const deadline = setTimeout(done, 30_000) + const observer = new MutationObserver(() => { + clearTimeout(timer) + timer = setTimeout(done, 750) + }) + observer.observe(document.body, { childList: true, subtree: true, characterData: true }) + function done() { + if (settled) return + settled = true + clearTimeout(deadline) + observer.disconnect() + requestAnimationFrame(() => requestAnimationFrame(() => resolve())) + } + }), + ) + progress("DOM settled") +} + +export async function waitForSelector(page: Page, selector: string, label: string) { + progress("waiting for UI", { label }) + try { + await page.waitForSelector(selector, { timeout: 30_000 }) + } catch (error) { + progress("UI wait failed", { + label, + url: page.url(), + body: (await page.locator("body").innerText().catch(() => "")).replace(/\s+/g, " ").slice(0, 500), + }) + throw error + } + progress("UI ready", { label }) +} + +export async function domCounts(page: Page, review = false) { + return page.evaluate((review) => ({ + elements: document.getElementsByTagName("*").length, + ...(review + ? { + diffViewers: document.querySelectorAll('[data-component="file"][data-mode="diff"]').length, + diffLines: document.querySelectorAll("[data-line]").length, + } + : { + timelineRows: document.querySelectorAll("[data-timeline-row]").length, + messageRows: document.querySelectorAll("[data-message-id]").length, + markdownRoots: document.querySelectorAll('[data-component="markdown"]').length, + diffViewers: document.querySelectorAll('[data-component="file"][data-mode="diff"]').length, + }), + }), review) +} + +export function sum(values: number[]) { + return values.reduce((total, value) => total + value, 0) +} + +export function percentile(values: number[], quantile: number) { + return values.toSorted((a, b) => a - b)[Math.max(0, Math.ceil(values.length * quantile) - 1)] ?? 0 +} diff --git a/packages/app/e2e/performance/desktop-profile/scenarios.ts b/packages/app/e2e/performance/desktop-profile/scenarios.ts new file mode 100644 index 00000000000..3f05fe0d06b --- /dev/null +++ b/packages/app/e2e/performance/desktop-profile/scenarios.ts @@ -0,0 +1,178 @@ +import type { Page } from "@playwright/test" +import { base64Encode } from "@opencode-ai/core/util/encode" +import { startChromeTrace } from "../chrome-trace" +import { collectProbe, resetProbe, startCPUProfile, summarizeProbe } from "./probe" +import { progress } from "./progress" +import { domCounts, percentile, setDesktopRoute, sum, waitForQuietDOM, waitForSelector } from "./scenario-utils" +import type { Options, Target } from "./types" + +export async function runScenarios(page: Page, options: Options, targets: Target[], typingText: string) { + const results: unknown[] = [] + if (options.scenarios.includes("home")) results.push(await profileHome(page, options)) + if (options.scenarios.includes("calibration")) results.push(await profileCalibration(page)) + if (options.scenarios.includes("session")) { + for (const target of targets) results.push(await profileSession(page, options, target)) + } + if (options.scenarios.some((scenario) => ["composer", "history", "review"].includes(scenario))) { + await openSession(page, targets[2]!) + } + if (options.scenarios.includes("composer")) results.push(await profileComposer(page, options, typingText)) + if (options.scenarios.includes("history")) results.push(await profileHistory(page, options, targets[2]!)) + if (options.scenarios.includes("review")) { + const review = await profileReview(page, options) + if (review) results.push(review) + } + return results +} + +async function profileHome(page: Page, options: Options) { + const measured = await measure(page, options, "home", async () => { + await setDesktopRoute(page, "/") + await page.reload({ waitUntil: "domcontentloaded", timeout: 60_000 }) + await waitForSelector(page, '[data-component="home-session-row"]', "Home session rows") + await waitForQuietDOM(page) + }) + return { ...measured, dom: await domCounts(page) } +} + +async function profileCalibration(page: Page) { + await resetProbe(page) + await page.evaluate( + () => + new Promise((resolve) => { + setTimeout(function opencodeProfileCalibration() { + const end = performance.now() + 80 + while (performance.now() < end) { + // Deliberate benchmark-only main-thread block. + } + requestAnimationFrame(() => setTimeout(resolve, 100)) + }) + }), + ) + return { name: "attribution-calibration", ...summarizeProbe(await collectProbe(page)) } +} + +async function profileSession(page: Page, options: Options, target: Target) { + await prepareHome(page) + const measured = await measure(page, options, `session-${target.label}`, async () => { + await navigateSession(page, target) + await waitForSelector(page, '[data-component="prompt-input"]', "session composer") + await waitForQuietDOM(page) + }) + return { ...measured, context: targetContext(target), dom: await domCounts(page) } +} + +async function profileComposer(page: Page, options: Options, typingText: string) { + const editor = page.locator('[data-component="prompt-input"][contenteditable="true"]').first() + await editor.click() + await page.keyboard.press("Control+A") + await page.keyboard.press("Backspace") + const printable = [...typingText].filter((character) => !["\r", "\n", "\t"].includes(character)) + const measuredText = printable.slice(-120).join("") + const prefix = printable.slice(0, -measuredText.length).join("") + if (prefix) await page.keyboard.insertText(prefix) + await waitForQuietDOM(page) + const durations: number[] = [] + const measured = await measure(page, options, "composer-typing", async () => { + for (const character of measuredText) { + const started = performance.now() + await page.keyboard.type(character) + durations.push(performance.now() - started) + } + await waitForQuietDOM(page) + }) + await page.keyboard.press("Control+A") + await page.keyboard.press("Backspace") + return { + ...measured, + context: { promptCharacters: printable.length, measuredCharacters: measuredText.length }, + typing: { + totalMs: sum(durations), + meanMs: sum(durations) / durations.length, + p50Ms: percentile(durations, 0.5), + p95Ms: percentile(durations, 0.95), + maxMs: Math.max(...durations), + }, + } +} + +async function profileHistory(page: Page, options: Options, target: Target) { + await waitForSelector(page, '[data-component="prompt-input"]', "history session composer") + await waitForQuietDOM(page) + let requests = 0 + const onResponse = (response: { url(): string }) => { + if (/\/session\/[^/]+\/message(?:\?|$)/.test(response.url())) requests++ + } + page.on("response", onResponse) + const measured = await measure(page, options, "session-max-history-boundary", async () => { + const scroller = page.locator(".scroll-view__viewport", { has: page.locator("[data-timeline-row]") }).first() + await scroller.evaluate((element) => { + element.scrollTop = 0 + element.dispatchEvent(new WheelEvent("wheel", { deltaY: -10_000, bubbles: true })) + element.dispatchEvent(new Event("scroll", { bubbles: true })) + }) + const timeout = Date.now() + 60_000 + while (requests === 0 && Date.now() < timeout) await page.waitForTimeout(50) + if (requests === 0) throw new Error("History boundary did not request a page") + await waitForQuietDOM(page) + }) + page.off("response", onResponse) + return { ...measured, context: targetContext(target), messageRequests: requests } +} + +async function profileReview(page: Page, options: Options) { + const button = page.getByRole("button", { name: "Toggle review" }) + if (!(await button.isVisible().catch(() => false))) return + const panel = page.locator("#review-panel") + if (await panel.isVisible().catch(() => false)) { + await button.click() + await panel.waitFor({ state: "hidden", timeout: 60_000 }) + await waitForQuietDOM(page) + } + const measured = await measure(page, options, "review-open", async () => { + await button.click() + await panel.waitFor({ state: "visible", timeout: 60_000 }) + await waitForQuietDOM(page) + }) + return { ...measured, dom: await domCounts(page, true) } +} + +async function measure(page: Page, options: Options, name: string, action: () => Promise) { + progress("scenario started", { name }) + await resetProbe(page) + const stopTrace = options.diagnostics ? await startChromeTrace(page, name) : undefined + const cpu = await startCPUProfile(page, options.cpu) + const started = performance.now() + await action() + const result = { + name, + elapsedMs: performance.now() - started, + ...summarizeProbe(await collectProbe(page)), + cpu: await cpu.stop(), + trace: await stopTrace?.(), + } + progress("scenario completed", { name, elapsedMs: Math.round(result.elapsedMs), longTasks: result.longTasks.count }) + return result +} + +async function openSession(page: Page, target: Target) { + await navigateSession(page, target) + await waitForSelector(page, '[data-component="prompt-input"]', "session composer") + await waitForQuietDOM(page) +} + +async function prepareHome(page: Page) { + await setDesktopRoute(page, "/") + await page.reload({ waitUntil: "domcontentloaded", timeout: 60_000 }) + await waitForSelector(page, '[data-component="home-session-row"]', "Home session rows") + await waitForQuietDOM(page) +} + +async function navigateSession(page: Page, target: Target) { + await setDesktopRoute(page, `/server/${base64Encode("sidecar")}/session/${target.id}`) + await page.reload({ waitUntil: "domcontentloaded", timeout: 60_000 }) +} + +function targetContext(target: Target) { + return { serializedBytes: target.bytes, messages: target.messages, parts: target.parts, userTurns: target.userTurns } +} diff --git a/packages/app/e2e/performance/desktop-profile/types.ts b/packages/app/e2e/performance/desktop-profile/types.ts new file mode 100644 index 00000000000..b3d728b2887 --- /dev/null +++ b/packages/app/e2e/performance/desktop-profile/types.ts @@ -0,0 +1,50 @@ +export const scenarios = ["home", "calibration", "session", "composer", "history", "review"] as const + +export type Scenario = (typeof scenarios)[number] + +export type Options = { + mode: "local" | "partial-snapshot" + database: string + output: string + windowStart: number + windowEnd: number + scenarios: Scenario[] + runs: number + build: boolean + diagnostics: boolean + cpu: boolean + responseURLs: boolean + partialSnapshotOut?: string +} + +export type Target = { + label: "p50" | "p95" | "max" + id: string + projectID: string + directory: string + title: string + bytes: number + messages: number + parts: number + userTurns: number +} + +export type ProbeResult = { + longTasks: number[] + animationFrames: { + duration: number + blockingDuration: number + forcedStyleAndLayoutDuration: number + scripts: { + function: string + source: string + position: number + invoker: string + invokerType: string + duration: number + forcedStyleAndLayoutDuration: number + }[] + }[] + frameGaps: number[] + responseText: { url: string; duration: number }[] +} diff --git a/packages/app/e2e/performance/profile-desktop.ts b/packages/app/e2e/performance/profile-desktop.ts new file mode 100644 index 00000000000..896c6c76f9a --- /dev/null +++ b/packages/app/e2e/performance/profile-desktop.ts @@ -0,0 +1,107 @@ +import { mkdir } from "node:fs/promises" +import path from "node:path" +import { createPartialSnapshot, fingerprint, loadCorpus } from "./desktop-profile/corpus" +import { parseOptions } from "./desktop-profile/options" +import { installProbe } from "./desktop-profile/probe" +import { progress } from "./desktop-profile/progress" +import { withDesktop, run } from "./desktop-profile/runtime" +import { runScenarios } from "./desktop-profile/scenarios" + +const root = path.resolve(import.meta.dir, "../../../..") +const desktop = path.join(root, "packages/desktop") +const options = parseOptions(process.argv.slice(2)) +if (!options) process.exit(0) + +await mkdir(options.output, { recursive: true }) +progress("loading corpus", { mode: options.mode }) +let corpus = loadCorpus(options) +if (options.partialSnapshotOut) { + progress("creating partial snapshot") + await createPartialSnapshot(options.database, options.partialSnapshotOut, options, corpus.targets) + options.database = options.partialSnapshotOut + options.mode = "partial-snapshot" + corpus = loadCorpus(options) +} +if (options.build) { + progress("building desktop production bundle") + await run(["bun", "run", "build"], desktop, options.database) +} + +progress("corpus ready", { targets: corpus.targets.map((target) => target.label), runs: options.runs }) +const runs = [] +for (let index = 1; index <= options.runs; index++) { + runs.push( + await withDesktop(options, desktop, index, corpus.projectIDs, async (page) => { + await installProbe(page, options) + await page.evaluate(() => { + const settings = JSON.parse(localStorage.getItem("settings.v3") ?? "{}") + localStorage.setItem( + "settings.v3", + JSON.stringify({ ...settings, general: { ...settings.general, newLayoutDesigns: true } }), + ) + }) + return runScenarios(page, options, corpus.targets, corpus.typingText) + }), + ) +} + +const report = { + schemaVersion: 2, + source: options.mode === "partial-snapshot" ? "partial-database-snapshot" : "local-opencode-db", + command: process.argv.slice(2), + diagnostics: options.diagnostics, + profileCPU: options.cpu, + database: await fingerprint(options.database), + window: { + start: new Date(options.windowStart).toISOString(), + end: new Date(options.windowEnd).toISOString(), + }, + revision: (await Bun.$`git rev-parse HEAD`.cwd(root).text()).trim(), + targets: corpus.targets.map(({ id: _, projectID: __, directory: ___, title: ____, ...target }) => target), + summary: summarize(runs), + runs: runs.map((results, index) => ({ index: index + 1, results })), +} +const file = path.join(options.output, "renderer-profile.json") +await Bun.write(file, JSON.stringify(report, null, 2)) +console.log(`PROFILE_REPORT ${file}`) +console.log(`PROFILE_SUMMARY ${JSON.stringify(report.summary)}`) +console.log(JSON.stringify(report, null, 2)) + +function summarize(runs: unknown[][]) { + type Result = { + name: string + elapsedMs?: number + longTasks: { count: number; totalMs: number; maxMs: number } + longAnimationFrames: { totalBlockingMs: number } + typing?: { p50Ms: number; p95Ms: number; maxMs: number } + } + return Object.fromEntries( + [...Map.groupBy(runs.flat() as Result[], (result) => result.name)].map(([name, samples]) => [ + name, + { + samples: samples.length, + elapsedMedianMs: median(samples.flatMap((sample) => sample.elapsedMs ?? [])), + longTasks: { + maxCount: Math.max(...samples.map((sample) => sample.longTasks.count)), + maxTotalMs: Math.max(...samples.map((sample) => sample.longTasks.totalMs)), + maxTaskMs: Math.max(...samples.map((sample) => sample.longTasks.maxMs)), + }, + maxBlockingMs: Math.max(...samples.map((sample) => sample.longAnimationFrames.totalBlockingMs)), + ...(samples[0]?.typing + ? { + typingMedianMs: { + p50: median(samples.flatMap((sample) => sample.typing?.p50Ms ?? [])), + p95: median(samples.flatMap((sample) => sample.typing?.p95Ms ?? [])), + max: median(samples.flatMap((sample) => sample.typing?.maxMs ?? [])), + }, + } + : {}), + }, + ]), + ) +} + +function median(values: number[]) { + if (values.length === 0) return + return values.toSorted((a, b) => a - b)[Math.floor(values.length / 2)] +} diff --git a/packages/app/e2e/performance/real-desktop-profile.ts b/packages/app/e2e/performance/real-desktop-profile.ts new file mode 100644 index 00000000000..cd7a85fd2b2 --- /dev/null +++ b/packages/app/e2e/performance/real-desktop-profile.ts @@ -0,0 +1 @@ +import "./profile-desktop" diff --git a/packages/app/e2e/regression/session-timeline-history-root.spec.ts b/packages/app/e2e/regression/session-timeline-history-root.spec.ts index e5ef7998ea7..4ea64c3155c 100644 --- a/packages/app/e2e/regression/session-timeline-history-root.spec.ts +++ b/packages/app/e2e/regression/session-timeline-history-root.spec.ts @@ -18,7 +18,7 @@ import { installSseTransport } from "../utils/sse-transport" import { expectSessionTitle } from "../utils/waits" const initialPageSize = 20 -const historyPageSize = 200 +const historyPageSize = 50 const assistants = Array.from({ length: initialPageSize + 1 }, (_, index) => assistantMessage([textPart(`prt_history_root_${index}`, `Assistant response ${index}`)], { id: `msg_${String(index + 1001).padStart(4, "0")}_history_root_assistant`, diff --git a/packages/app/package.json b/packages/app/package.json index e2e97abe5c7..bc046c57c5e 100644 --- a/packages/app/package.json +++ b/packages/app/package.json @@ -28,7 +28,8 @@ "test:e2e:ui": "playwright test --ui", "test:e2e:report": "playwright show-report e2e/playwright-report", "test:stability": "bun test ./e2e/performance/unit/visual-stability.test.ts && playwright test --config e2e/performance/timeline-stability/playwright.config.ts", - "test:bench": "bun test ./e2e/performance/unit && playwright test --config e2e/performance/playwright.config.ts" + "test:bench": "bun test ./e2e/performance/unit && playwright test --config e2e/performance/playwright.config.ts", + "profile:desktop": "bun run e2e/performance/profile-desktop.ts" }, "license": "MIT", "devDependencies": { diff --git a/packages/app/src/app.tsx b/packages/app/src/app.tsx index 996f912a73b..696a2c9a73a 100644 --- a/packages/app/src/app.tsx +++ b/packages/app/src/app.tsx @@ -4,7 +4,6 @@ import { I18nProvider } from "@opencode-ai/ui/context" import type { UiI18n } from "@opencode-ai/ui/context/i18n" import { DialogProvider } from "@opencode-ai/ui/context/dialog" import { FileComponentProvider } from "@opencode-ai/ui/context/file" -import { File } from "@opencode-ai/session-ui/file" import { Font } from "@opencode-ai/ui/font" import { Splash } from "@opencode-ai/ui/logo" import { ThemeProvider } from "@opencode-ai/ui/theme/context" @@ -59,18 +58,24 @@ import { TabsProvider, useTabs, type DraftTab } from "@/context/tabs" import { SDKProvider, useSDK } from "@/context/sdk" import { WslServersProvider } from "@/wsl/context" import DirectoryLayout, { DirectoryDataProvider } from "@/pages/directory-layout" -import LegacyLayout from "@/pages/layout" -import NewLayout from "@/pages/layout-new" import { ErrorPage } from "./pages/error" import { useCheckServerHealth } from "./utils/server-health" import { legacySessionHref, legacySessionServer, requireServerKey, sessionHref } from "./utils/session-route" import { createSessionLineage } from "@/pages/session/session-lineage" -import { SessionPage, SessionRouteErrorBoundary, TargetSessionRouteContent } from "@/pages/session" -import { NewHome } from "@/pages/home" -import { LegacyHome } from "@/pages/home/legacy-home" - const NewSession = lazy(() => import("@/pages/new-session")) +const NewLayout = lazy(() => import("@/pages/layout-new")) +const NewHome = lazy(() => import("@/pages/home").then((module) => ({ default: module.NewHome }))) +const LegacyLayout = lazy(() => import("@/pages/layout")) +const LegacyHome = lazy(() => import("@/pages/home/legacy-home").then((module) => ({ default: module.LegacyHome }))) +const File = lazy(() => import("@opencode-ai/session-ui/file").then((module) => ({ default: module.File }))) +const SessionPage = lazy(() => import("@/pages/session").then((module) => ({ default: module.SessionPage }))) +const SessionRouteErrorBoundary = lazy(() => + import("@/pages/session").then((module) => ({ default: module.SessionRouteErrorBoundary })), +) +const TargetSessionRouteContent = lazy(() => + import("@/pages/session").then((module) => ({ default: module.TargetSessionRouteContent })), +) const SessionRoute = () => { const settings = useSettings() diff --git a/packages/app/src/components/settings-models.tsx b/packages/app/src/components/settings-models.tsx index f3d9e1522ff..c29acd0b8ea 100644 --- a/packages/app/src/components/settings-models.tsx +++ b/packages/app/src/components/settings-models.tsx @@ -7,6 +7,7 @@ import { TextField } from "@opencode-ai/ui/text-field" import { type Component, For, Show } from "solid-js" import { useLanguage } from "@/context/language" import { useModels } from "@/context/models" +import { useServerSync } from "@/context/server-sync" import { popularProviders } from "@/hooks/use-providers" import { SettingsList } from "./settings-list" import { SettingsServerPicker, SettingsServerScope } from "./settings-server-picker" @@ -43,6 +44,7 @@ export const SettingsModels: Component = () => { const SettingsModelsContent: Component = () => { const language = useLanguage() const models = useModels() + useServerSync()().loadProviders() const list = useFilteredList({ items: (_filter) => models.list(), diff --git a/packages/app/src/components/settings-v2/models.tsx b/packages/app/src/components/settings-v2/models.tsx index 482e8cbffbd..7007cbef64a 100644 --- a/packages/app/src/components/settings-v2/models.tsx +++ b/packages/app/src/components/settings-v2/models.tsx @@ -9,6 +9,7 @@ import { createStore } from "solid-js/store" import { useLanguage } from "@/context/language" import { useModels } from "@/context/models" import { useServerSDK } from "@/context/server-sdk" +import { useServerSync } from "@/context/server-sync" import { popularProviders } from "@/hooks/use-providers" import { Persist, persisted } from "@/utils/persist" import { SettingsListV2 } from "./parts/list" @@ -23,6 +24,7 @@ export const SettingsModelsV2: Component = () => { const language = useLanguage() const models = useModels() const serverSdk = useServerSDK() + useServerSync()().loadProviders() const [store, setStore] = persisted( Persist.serverGlobal(serverSdk().scope, "settings-v2.models.providers"), createStore({ collapsed: {} as Record }), diff --git a/packages/app/src/context/global-sync/bootstrap.ts b/packages/app/src/context/global-sync/bootstrap.ts index 0f3e4738164..e4b437b5dca 100644 --- a/packages/app/src/context/global-sync/bootstrap.ts +++ b/packages/app/src/context/global-sync/bootstrap.ts @@ -152,12 +152,6 @@ export async function bootstrapGlobal(input: { queryClient: QueryClient }) { const slow = [ - () => input.queryClient.fetchQuery(loadGlobalConfigQuery(input.scope, input.serverSDK, input.protocol)), - () => - input.queryClient.fetchQuery( - loadProvidersQuery(input.scope, null, input.serverAPI, input.serverSDK, input.protocol), - ), - () => input.queryClient.fetchQuery(loadPathQuery(input.scope, null, input.serverSDK, input.protocol)), () => input.queryClient .fetchQuery(loadProjectsQuery(input.scope, input.serverAPI.project)) @@ -524,17 +518,6 @@ export async function bootstrapDirectory(input: { input.queryClient.fetchQuery( loadMcpResourcesQuery(input.scope, input.directory, input.api.mcp, input.sdk, input.protocol), )), - () => - input.queryClient - .fetchQuery(loadProvidersQuery(input.scope, input.directory, input.api, input.sdk, input.protocol)) - .catch((err) => { - const project = getFilename(input.directory) - showToast({ - variant: "error", - title: input.translate("toast.project.reloadFailed.title", { project }), - description: formatServerError(err, input.translate), - }) - }), ].filter(Boolean) as (() => Promise)[] await waitForPaint() diff --git a/packages/app/src/context/global-sync/child-store.test.ts b/packages/app/src/context/global-sync/child-store.test.ts index e05b7c39a96..05d7831d403 100644 --- a/packages/app/src/context/global-sync/child-store.test.ts +++ b/packages/app/src/context/global-sync/child-store.test.ts @@ -263,10 +263,13 @@ describe("createChildStoreManager", () => { manager.child("/project") expect(queries[0]?.().enabled).toBe(true) expect(queries[3]?.().enabled).toBe(true) - expect(queries[4]?.().enabled).toBe(true) + expect(queries[4]?.().enabled).toBe(false) expect(queries[5]?.().enabled).toBe(true) expect(bootstraps).toEqual(["/project"]) + manager.enableProviders("/project") + expect(queries[4]?.().enabled).toBe(true) + manager.child("/project", { bootstrap: false }) expect(queries[0]?.().enabled).toBe(true) } finally { diff --git a/packages/app/src/context/global-sync/child-store.ts b/packages/app/src/context/global-sync/child-store.ts index 4eaa785789f..bfea6237d74 100644 --- a/packages/app/src/context/global-sync/child-store.ts +++ b/packages/app/src/context/global-sync/child-store.ts @@ -47,6 +47,7 @@ export function createChildStoreManager(input: { const mcpToggles = new Map void>() const activeDirectories = new Set() const activationToggles = new Map void>() + const providerToggles = new Map void>() const markKey = (key: DirectoryKey) => { if (!key) return @@ -122,6 +123,7 @@ export function createChildStoreManager(input: { mcpToggles.delete(key) activeDirectories.delete(key) activationToggles.delete(key) + providerToggles.delete(key) const dispose = disposers.get(key) if (dispose) { dispose() @@ -187,6 +189,7 @@ export function createChildStoreManager(input: { const initialIcon = icon[0].value const [mcpEnabled, setMcpEnabled] = createSignal(false) const [instanceQueriesEnabled, setInstanceQueriesEnabled] = createSignal(false) + const [providerEnabled, setProviderEnabled] = createSignal(false) const pathQuery = useQuery(() => ({ ...input.queryOptions.path(key), enabled: instanceQueriesEnabled() })) const mcpQuery = useQuery(() => ({ ...input.queryOptions.mcp(key), enabled: mcpEnabled() })) @@ -194,7 +197,7 @@ export function createChildStoreManager(input: { const lspQuery = useQuery(() => ({ ...input.queryOptions.lsp(key), enabled: instanceQueriesEnabled() })) const providerQuery = useQuery(() => ({ ...input.queryOptions.providers(key), - enabled: instanceQueriesEnabled(), + enabled: providerEnabled(), })) const referenceQuery = useQuery(() => ({ ...input.queryOptions.references(key), @@ -206,7 +209,7 @@ export function createChildStoreManager(input: { projectMeta: initialMeta, icon: initialIcon, get provider_ready() { - return instanceQueriesEnabled() && !providerQuery.isLoading + return providerEnabled() && !providerQuery.isLoading }, get provider() { const EMPTY = { all: new Map(), connected: [], default: {} } @@ -263,6 +266,7 @@ export function createChildStoreManager(input: { disposers.set(key, dispose) mcpToggles.set(key, setMcpEnabled) activationToggles.set(key, setInstanceQueriesEnabled) + providerToggles.set(key, setProviderEnabled) const onPersistedInit = (init: Promise | string | null, run: () => void) => { if (!(init instanceof Promise)) return @@ -329,6 +333,12 @@ export function createChildStoreManager(input: { if (childStore[0].status !== "loading") input.onMcp(directory, childStore[1]) } + function enableProviders(directory: string) { + const key = directoryKey(directory) + ensureChild(directory) + providerToggles.get(key)?.(true) + } + // Passive Home/project metadata reads must not initialize the directory. // A real directory access enables these queries once for the store lifetime. // TODO(v2): After Home switches to v2.project.list and root-filtered, @@ -387,6 +397,7 @@ export function createChildStoreManager(input: { mcp: (directory: string) => mcpDirectories.has(directoryKey(directory)), active: (directory: string) => activeDirectories.has(directoryKey(directory)), disableMcp, + enableProviders, disposeDirectory, runEviction, vcsCache, diff --git a/packages/app/src/context/global-sync/home-session-index.ts b/packages/app/src/context/global-sync/home-session-index.ts index 03a085e34d5..e5f8422f0c5 100644 --- a/packages/app/src/context/global-sync/home-session-index.ts +++ b/packages/app/src/context/global-sync/home-session-index.ts @@ -22,6 +22,7 @@ export const homeSessionIndexKey = (server: string) => ["home", "session-index", export const homeSessionEventsKey = (server: string) => ["home", "session-events", server] as const type HomeSessionPage = { data?: V2SessionListResponse } +type ProjectedHomeSessionPage = { data?: { data: Session[]; cursor: { next?: string } } } export async function loadHomeSessionIndex( list: ( @@ -31,7 +32,30 @@ export async function loadHomeSessionIndex( eventSequence = 0, signal?: AbortSignal, ) { - const data: SessionV2Info[] = [] + return loadHomeSessionPages(list, parseHomeSessionIndex, eventSequence, signal) +} + +export async function loadProjectedHomeSessionIndex( + list: ( + input: { limit: number; order: "desc"; cursor?: string }, + options: { signal?: AbortSignal }, + ) => Promise, + eventSequence = 0, + signal?: AbortSignal, +) { + return loadHomeSessionPages(list, (sessions) => sessions, eventSequence, signal) +} + +async function loadHomeSessionPages( + list: ( + input: { limit: number; order: "desc"; cursor?: string }, + options: { signal?: AbortSignal }, + ) => Promise<{ data?: { data: T[]; cursor: { next?: string } } }>, + project: (sessions: T[]) => Session[], + eventSequence: number, + signal?: AbortSignal, +) { + const data: T[] = [] let cursor: string | undefined for (;;) { @@ -46,7 +70,7 @@ export async function loadHomeSessionIndex( const page = response.data! data.push(...page.data) if (page.data.length < HOME_V2_SESSION_PAGE_LIMIT || !page.cursor.next) - return { sessions: parseHomeSessionIndex(data), eventSequence } + return { sessions: project(data), eventSequence } cursor = page.cursor.next } } diff --git a/packages/app/src/context/server-sdk.tsx b/packages/app/src/context/server-sdk.tsx index 1d192d406ed..c459e18c5a4 100644 --- a/packages/app/src/context/server-sdk.tsx +++ b/packages/app/src/context/server-sdk.tsx @@ -15,6 +15,8 @@ import { detectServerProtocol, type ServerProtocol } from "@/utils/server-protoc import { createCompatibleApi, type CompatibleApi } from "@/utils/server-compat" import { Worktree } from "@/utils/worktree" import { WorkspaceOperation } from "@/utils/workspace-operation" +import { decodeVcsDiff } from "@/utils/vcs-diff-decoder" +import { decodeSessionList } from "./session-message-decoder" const isAbortError = (error: unknown) => error !== null && typeof error === "object" && "name" in error && error.name === "AbortError" @@ -430,7 +432,7 @@ function createServerSdkContextBase(server: ServerConnection.Any, scope: ServerS throwOnError: true, directory, }) - const api = createCompatibleApi({ protocol, current: currentApi, legacy }) + const api = createCompatibleApi({ protocol, current: currentApi, legacy, decodeVcsDiff, decodeSessionList }) return { server, @@ -516,6 +518,8 @@ function createDirSdkContext(directory: string, serverSDK: ServerSDKBase) { current: serverSDK.currentApi, legacy: (next) => serverSDK.createClient({ directory: next ?? directory, throwOnError: true }), directory, + decodeVcsDiff, + decodeSessionList, }), createApi(next: string) { return createCompatibleApi({ @@ -523,6 +527,8 @@ function createDirSdkContext(directory: string, serverSDK: ServerSDKBase) { current: serverSDK.currentApi, legacy: (target) => serverSDK.createClient({ directory: target ?? next, throwOnError: true }), directory: next, + decodeVcsDiff, + decodeSessionList, }) }, event: emitter, diff --git a/packages/app/src/context/server-session.ts b/packages/app/src/context/server-session.ts index 3565f58a35a..05c064af61f 100644 --- a/packages/app/src/context/server-session.ts +++ b/packages/app/src/context/server-session.ts @@ -22,16 +22,23 @@ import { compareMessages, messageKey, normalizeSessionMessages } from "@/utils/s import { dropSessionCaches, pickSessionCacheEvictions, SESSION_CACHE_LIMIT } from "./global-sync/session-cache" import { createV2SessionReducer, type V2SessionReduction } from "./server-session-v2-reducer" import type { ServerApi } from "@/utils/server" +import type { DecodedLegacyMessagePage } from "./session-message-decode" type MessageApi = ServerApi["message"] const cmp = (a: string, b: string) => (a < b ? -1 : a > b ? 1 : 0) const SKIP_PARTS = new Set(["patch", "step-start", "step-finish"]) const initialMessagePageSize = 20 -const historyMessagePageSize = 200 +const historyMessagePageSize = 50 const sessionInfoLimit = 2_048 const emptyIDs: ReadonlySet = new Set() +function yieldToMain() { + const scheduler = (globalThis as { scheduler?: { yield: () => Promise } }).scheduler + if (scheduler) return scheduler.yield() + return new Promise((resolve) => setTimeout(resolve, 0)) +} + function needsOlderTurnRoot(source: readonly SessionMessageInfo[]) { const boundary = source.find( (message) => @@ -183,7 +190,11 @@ function reconcileFetched( return options.compare ? items.sort(options.compare) : items } -type ServerSessionOptions = { retry?: typeof retry; protocol?: Promise<"v1" | "v2"> } +type ServerSessionOptions = { + retry?: typeof retry + protocol?: Promise<"v1" | "v2"> + decodeMessages?: (buffer: ArrayBuffer) => Promise +} export function createServerSession( client: OpencodeClient, @@ -551,6 +562,7 @@ export function createServerSession( if (!response.data.length) break } const response = pages.at(-1)! + await yieldToMain() const source = pages.flatMap((page) => page.data).toReversed() const normalized = normalizeSessionMessages(sessionID, source) return { @@ -565,10 +577,21 @@ export function createServerSession( complete: response.data.length === 0, } } - const response = await (options?.retry ?? retry)(() => { + const response = await (options?.retry ?? retry)(async () => { onAttempt?.() - return client.session.messages({ sessionID, limit, before }) + if (!options?.decodeMessages) return client.session.messages({ sessionID, limit, before }) + const response = await client.session.messages({ sessionID, limit, before }, { parseAs: "arrayBuffer" }) + if (!(response.data instanceof ArrayBuffer)) throw new Error("Session messages response is not an ArrayBuffer") + return { response, decoded: await options.decodeMessages(response.data) } }) + await yieldToMain() + if ("decoded" in response) + return { + ...response.decoded, + sourceMode: before ? ("older" as const) : ("latest" as const), + cursor: response.response.response.headers.get("x-next-cursor") ?? undefined, + complete: !response.response.response.headers.get("x-next-cursor"), + } const items = (response.data ?? []).filter((item) => !!item?.info?.id) return { session: items.map((item) => cleanMessage(item.info)).sort(compareMessages), diff --git a/packages/app/src/context/server-sync.tsx b/packages/app/src/context/server-sync.tsx index 9dd10ab3525..90934195086 100644 --- a/packages/app/src/context/server-sync.tsx +++ b/packages/app/src/context/server-sync.tsx @@ -8,7 +8,7 @@ import type { } from "@opencode-ai/sdk/v2/client" import { showToast } from "@/utils/toast" import { getFilename } from "@opencode-ai/core/util/path" -import { type Accessor, batch, createMemo, getOwner, onCleanup, onMount, untrack } from "solid-js" +import { type Accessor, batch, createMemo, createSignal, getOwner, onCleanup, onMount, untrack } from "solid-js" import { createStore, produce, reconcile } from "solid-js/store" import { useLanguage } from "@/context/language" import type { InitError } from "../pages/error" @@ -85,6 +85,7 @@ import type { } from "@opencode-ai/client/promise" import { toggleMcp } from "./global-sync/mcp" import { createServerSession, type ServerSession } from "./server-session" +import { decodeSessionMessages } from "./session-message-decoder" type GlobalStore = { ready: boolean @@ -252,6 +253,7 @@ export function createServerSyncContextInner(serverSDK: ServerSDK) { const session = createServerSession(serverSDK.client, serverSDK.api.session, serverSDK.api.message, { protocol: serverSDK.protocol, + decodeMessages: decodeSessionMessages, }) const queryOptionsApi = makeQueryOptionsApi( serverSDK.scope, @@ -261,31 +263,40 @@ export function createServerSyncContextInner(serverSDK: ServerSDK) { serverSDK.protocol, ) + const [providersEnabled, setProvidersEnabled] = createSignal(false) + const [backgroundEnabled, setBackgroundEnabled] = createSignal(false) const [configQuery, providerQuery, pathQuery] = useQueries(() => ({ - queries: [queryOptionsApi.globalConfig(), queryOptionsApi.providers(null), queryOptionsApi.path(null)], + queries: [ + { ...queryOptionsApi.globalConfig(), enabled: backgroundEnabled() }, + { ...queryOptionsApi.providers(null), enabled: providersEnabled() }, + { ...queryOptionsApi.path(null), enabled: backgroundEnabled() }, + ], })) const activeSessionsQuery = useQuery(() => - loadActiveSessionsQuery(serverSDK.scope, { - active: async () => { - if ((await serverSDK.protocol) === "v1") { - const statuses = (await serverSDK.client.session.status()).data ?? {} - seedActiveSessionStatuses(session, statuses) - for (const sessionID of Object.keys(statuses)) { + ({ + ...loadActiveSessionsQuery(serverSDK.scope, { + active: async () => { + if ((await serverSDK.protocol) === "v1") { + const statuses = (await serverSDK.client.session.status()).data ?? {} + seedActiveSessionStatuses(session, statuses) + for (const sessionID of Object.keys(statuses)) { + void session.resolve(sessionID).catch(() => undefined) + } + return Object.fromEntries( + Object.entries(statuses).flatMap(([sessionID, status]) => + status.type === "idle" ? [] : [[sessionID, { type: "running" as const }]], + ), + ) + } + const active = await serverSDK.api.session.active() + seedActiveSessionStatuses(session, active) + for (const sessionID of Object.keys(active)) { void session.resolve(sessionID).catch(() => undefined) } - return Object.fromEntries( - Object.entries(statuses).flatMap(([sessionID, status]) => - status.type === "idle" ? [] : [[sessionID, { type: "running" as const }]], - ), - ) - } - const active = await serverSDK.api.session.active() - seedActiveSessionStatuses(session, active) - for (const sessionID of Object.keys(active)) { - void session.resolve(sessionID).catch(() => undefined) - } - return active - }, + return active + }, + }), + enabled: backgroundEnabled(), }), ) @@ -325,10 +336,36 @@ export function createServerSyncContextInner(serverSDK: ServerSDK) { let bootingRoot = false let eventFrame: number | undefined let eventTimer: ReturnType | undefined + let providerFrame: number | undefined + let providerIdle: number | undefined + let providerTimer: ReturnType | undefined + + onMount(() => { + providerFrame = requestAnimationFrame(() => { + providerFrame = requestAnimationFrame(() => { + providerFrame = undefined + providerTimer = setTimeout(() => { + providerTimer = undefined + if ("requestIdleCallback" in window) { + providerIdle = requestIdleCallback(() => { + setProvidersEnabled(true) + setBackgroundEnabled(true) + }, { timeout: 5_000 }) + return + } + setProvidersEnabled(true) + setBackgroundEnabled(true) + }, 10_000) + }) + }) + }) onCleanup(() => { if (eventFrame !== undefined) cancelAnimationFrame(eventFrame) if (eventTimer !== undefined) clearTimeout(eventTimer) + if (providerFrame !== undefined) cancelAnimationFrame(providerFrame) + if (providerIdle !== undefined) cancelIdleCallback(providerIdle) + if (providerTimer !== undefined) clearTimeout(providerTimer) }) const setProjects = (next: Project[] | ((draft: Project[]) => Project[])) => { @@ -740,7 +777,9 @@ export function createServerSyncContextInner(serverSDK: ServerSDK) { child: children.child, peek: children.peek, disableMcp: children.disableMcp, + enableProviders: children.enableProviders, queryOptions: queryOptionsApi, + loadProviders: () => setProvidersEnabled(true), refreshProviders, // bootstrap, updateConfig: updateConfigMutation.mutateAsync, diff --git a/packages/app/src/context/session-message-decode.test.ts b/packages/app/src/context/session-message-decode.test.ts new file mode 100644 index 00000000000..494c3f6b861 --- /dev/null +++ b/packages/app/src/context/session-message-decode.test.ts @@ -0,0 +1,64 @@ +import { expect, test } from "bun:test" +import type { Message, Part, Session, SessionV2Info } from "@opencode-ai/sdk/v2/client" +import { decodeHomeSessionPage, decodeLegacyMessagePage, decodeLegacySessionList } from "./session-message-decode" + +test("decodes and projects a legacy message page", () => { + const info = { + id: "message", + sessionID: "session", + role: "user", + time: { created: 1 }, + agent: "build", + model: { providerID: "provider", modelID: "model" }, + } as Message + const part = { + id: "part", + sessionID: "session", + messageID: info.id, + type: "text", + text: "hello", + } as Part + const result = decodeLegacyMessagePage(new TextEncoder().encode(JSON.stringify([{ info, parts: [part] }])).buffer) + + expect(result.session).toEqual([info]) + expect(result.part).toEqual([{ id: info.id, part: [part] }]) + expect(result.source).toEqual([{ id: info.id, type: "user", text: "hello", time: info.time }]) +}) + +test("decodes and projects a legacy session list", () => { + const session = { + id: "session", + projectID: "project", + directory: "/repo", + title: "Session", + version: "1", + time: { created: 1, updated: 1 }, + } as Session + const result = decodeLegacySessionList(new TextEncoder().encode(JSON.stringify([session])).buffer) + + expect(result).toEqual([ + expect.objectContaining({ id: session.id, title: session.title, location: { directory: "/repo" } }), + ]) +}) + +test("bounds Home sessions by directory before returning from the decoder", () => { + const session = (id: string, directory: string, updated: number) => + ({ + id, + projectID: "project", + location: { directory }, + subpath: "", + title: id, + time: { created: updated, updated }, + }) as SessionV2Info + const page = { + data: [session("old", "/repo", 1), session("new", "/repo", 2), session("other", "/other", 3)], + cursor: {}, + } + const result = decodeHomeSessionPage(new TextEncoder().encode(JSON.stringify(page)).buffer, { + directories: ["/repo"], + limit: 1, + }) + + expect(result.data.map((item) => item.id)).toEqual(["new"]) +}) diff --git a/packages/app/src/context/session-message-decode.ts b/packages/app/src/context/session-message-decode.ts new file mode 100644 index 00000000000..f99c25d4cc1 --- /dev/null +++ b/packages/app/src/context/session-message-decode.ts @@ -0,0 +1,94 @@ +import type { SessionInfo, SessionMessageInfo } from "@opencode-ai/client/promise" +import type { Message, Part, Session, V2SessionListResponse } from "@opencode-ai/sdk/v2/client" +import { message as cleanMessage } from "@/utils/diffs" +import { pathKey } from "@/utils/path-key" +import { parseHomeSessionIndex } from "./global-sync/home-session-index" +import { takeRecentSessions } from "./global-sync/session-trim" + +export type DecodedLegacyMessagePage = { + session: Message[] + part: { id: string; part: Part[] }[] + source: SessionMessageInfo[] +} + +export function decodeLegacyMessagePage(buffer: ArrayBuffer): DecodedLegacyMessagePage { + const text = new TextDecoder().decode(buffer) + const items = (text ? (JSON.parse(text) as { info?: Message; parts?: Part[] }[]) : []).filter( + (item): item is { info: Message; parts: Part[] } => !!item.info?.id && Array.isArray(item.parts), + ) + return { + session: items.map((item) => cleanMessage(item.info)).sort((a, b) => compare(a.id, b.id)), + part: items.map((item) => ({ + id: item.info.id, + part: item.parts.filter((part) => !!part?.id).sort((a, b) => compare(a.id, b.id)), + })), + source: items + .slice() + .sort((a, b) => compare(a.info.id, b.info.id)) + .map((item) => + item.info.role === "user" + ? { + id: item.info.id, + type: "user" as const, + text: item.parts.flatMap((part) => (part.type === "text" ? [part.text] : [])).join("\n"), + time: item.info.time, + } + : { + id: item.info.id, + type: "assistant" as const, + agent: item.info.agent ?? item.info.mode, + model: { id: item.info.modelID, providerID: item.info.providerID, variant: item.info.variant }, + content: [], + time: item.info.time, + }, + ), + } +} + +export function decodeLegacySessionList(buffer: ArrayBuffer) { + const text = new TextDecoder().decode(buffer) + return (text ? (JSON.parse(text) as Session[]) : []).map(legacySessionInfo) +} + +export function decodeHomeSessionPage(buffer: ArrayBuffer, options?: { directories: string[]; limit: number }) { + const text = new TextDecoder().decode(buffer) + const page = (text ? JSON.parse(text) : { data: [], cursor: {} }) as V2SessionListResponse + const sessions = parseHomeSessionIndex(page.data) + if (!options) return { data: sessions, cursor: page.cursor } + const directories = new Set(options.directories.map(pathKey)) + return { + data: [...Map.groupBy(sessions, (session) => pathKey(session.directory))] + .filter(([directory]) => directories.has(directory)) + .flatMap(([, items]) => takeRecentSessions(items, options.limit, Number.NEGATIVE_INFINITY)), + cursor: page.cursor, + } +} + +export function legacySessionInfo(session: Session): SessionInfo { + return { + id: session.id, + parentID: session.parentID, + projectID: session.projectID, + agent: session.agent, + model: session.model && { + id: session.model.id, + providerID: session.model.providerID, + variant: session.model.variant, + }, + cost: session.cost ?? 0, + tokens: session.tokens ?? { input: 0, output: 0, reasoning: 0, cache: { read: 0, write: 0 } }, + time: session.time, + title: session.title, + location: { directory: session.directory, workspaceID: session.workspaceID }, + subpath: session.path, + revert: session.revert && { + messageID: session.revert.messageID, + partID: session.revert.partID, + snapshot: session.revert.snapshot, + }, + } +} + +function compare(a: string, b: string) { + return a < b ? -1 : a > b ? 1 : 0 +} diff --git a/packages/app/src/context/session-message-decoder.ts b/packages/app/src/context/session-message-decoder.ts new file mode 100644 index 00000000000..30e7c27747b --- /dev/null +++ b/packages/app/src/context/session-message-decoder.ts @@ -0,0 +1,56 @@ +import type { DecodedLegacyMessagePage } from "./session-message-decode" +import type { SessionInfo } from "@opencode-ai/client/promise" +import type { Session } from "@opencode-ai/sdk/v2/client" + +type Response = { id: number; data?: unknown; error?: string } + +let worker: Worker | undefined +let nextID = 0 +const pending = new Map void; reject: (error: Error) => void }>() + +export function decodeSessionMessages(buffer: ArrayBuffer) { + return decode("messages", buffer) +} + +export function decodeSessionList(buffer: ArrayBuffer) { + return decode("sessions", buffer) +} + +export function decodeHomeSessionPage(buffer: ArrayBuffer, options: { directories: string[]; limit: number }) { + return decode<{ data: Session[]; cursor: { next?: string } }>("homeSessions", buffer, options) +} + +function decode( + type: "messages" | "sessions" | "homeSessions", + buffer: ArrayBuffer, + options?: { directories: string[]; limit: number }, +) { + const id = ++nextID + return new Promise((resolve, reject) => { + pending.set(id, { resolve: (value) => resolve(value as T), reject }) + getWorker().postMessage({ id, type, buffer, options }, [buffer]) + }) +} + +function getWorker() { + if (worker) return worker + worker = new Worker(new URL("./session-message-decoder.worker.ts", import.meta.url), { type: "module" }) + worker.onmessage = (event: MessageEvent) => { + const request = pending.get(event.data.id) + if (!request) return + pending.delete(event.data.id) + if (event.data.error) { + request.reject(new Error(event.data.error)) + return + } + request.resolve(event.data.data) + } + worker.onerror = (event) => { + const error = new Error(event.message) + pending.forEach((request) => request.reject(error)) + pending.clear() + worker?.terminate() + worker = undefined + } + return worker +} diff --git a/packages/app/src/context/session-message-decoder.worker.ts b/packages/app/src/context/session-message-decoder.worker.ts new file mode 100644 index 00000000000..b98411d7e71 --- /dev/null +++ b/packages/app/src/context/session-message-decoder.worker.ts @@ -0,0 +1,25 @@ +import { decodeHomeSessionPage, decodeLegacyMessagePage, decodeLegacySessionList } from "./session-message-decode" + +type DecoderRequest = { + id: number + type: "messages" | "sessions" | "homeSessions" + buffer: ArrayBuffer + options?: { directories: string[]; limit: number } +} + +self.onmessage = (event: MessageEvent) => { + try { + self.postMessage({ + id: event.data.id, + data: (() => { + if (event.data.type === "messages") return decodeLegacyMessagePage(event.data.buffer) + if (event.data.type === "sessions") return decodeLegacySessionList(event.data.buffer) + return decodeHomeSessionPage(event.data.buffer, event.data.options) + })(), + }) + } catch (error) { + self.postMessage({ id: event.data.id, error: error instanceof Error ? error.message : String(error) }) + } +} + +export {} diff --git a/packages/app/src/hooks/use-providers.ts b/packages/app/src/hooks/use-providers.ts index 60ad01c8674..d1bf5d84ad5 100644 --- a/packages/app/src/hooks/use-providers.ts +++ b/packages/app/src/hooks/use-providers.ts @@ -21,6 +21,12 @@ export function useProviders(directory: Accessor) { const serverSync = useServerSync() const params = useParams() const dir = () => (directory ? directory() : decode64(params.dir)) + createEffect(() => { + const value = dir() + if (value) { + serverSync().enableProviders(value) + } + }) const providers = () => { const value = dir() const projectStore = value ? serverSync().child(value)[0] : undefined diff --git a/packages/app/src/pages/home/home-sessions-controller.tsx b/packages/app/src/pages/home/home-sessions-controller.tsx index 85cc47a9a9b..bcdd4e73833 100644 --- a/packages/app/src/pages/home/home-sessions-controller.tsx +++ b/packages/app/src/pages/home/home-sessions-controller.tsx @@ -1,16 +1,16 @@ import type { Session } from "@opencode-ai/sdk/v2/client" -import { preloadMarkdown } from "@opencode-ai/session-ui/markdown-cache" import { useDialog } from "@opencode-ai/ui/context/dialog" import { useQuery } from "@tanstack/solid-query" import { DateTime } from "luxon" -import { type Accessor, createEffect, createMemo, createRoot, type JSX, startTransition } from "solid-js" +import { type Accessor, createEffect, createMemo, createSignal, onCleanup, type JSX, startTransition } from "solid-js" import { produce } from "solid-js/store" import { useCommand } from "@/context/command" import { - loadHomeSessionIndex, - retainHomeSessions, + loadProjectedHomeSessionIndex, type HomeSessionEvents, } from "@/context/global-sync/home-session-index" +import { takeRecentSessions } from "@/context/global-sync/session-trim" +import { decodeHomeSessionPage } from "@/context/session-message-decoder" import type { LocalProject } from "@/context/layout" import { useLanguage } from "@/context/language" import { ServerConnection } from "@/context/server" @@ -25,6 +25,7 @@ import { archiveHomeSession } from "../home-session-archive" import type { HomeController } from "./home-controller" const HOME_SESSION_LIMIT = 64 +const HOME_SESSION_RENDER_BATCH = 4 export type HomeSessionRecord = { session: Session project: LocalProject @@ -67,8 +68,17 @@ export function createHomeSessionsController(home: HomeController) { if (!ctx) return { sessions: [], eventSequence: 0 } const cache = homeSessions() const eventSequence = cache.eventSequence() - const index = await loadHomeSessionIndex( - (input, options) => ctx.sdk.client.v2.session.list(input, options), + const index = await loadProjectedHomeSessionIndex( + async (input, options) => { + const response = await ctx.sdk.client.v2.session.list(input, { ...options, parseAs: "arrayBuffer" }) + if (!(response.data instanceof ArrayBuffer)) throw new Error("Home session response is not an ArrayBuffer") + return { + data: await decodeHomeSessionPage(response.data, { + directories: projectDirectories(), + limit: HOME_SESSION_LIMIT, + }), + } + }, eventSequence, signal, ) @@ -80,13 +90,16 @@ export function createHomeSessionsController(home: HomeController) { refetchOnMount: true, refetchOnReconnect: true, })) - const indexedSessions = createMemo(() => - retainHomeSessions( - homeSessions().sessions(sessionLoad.data, sessionEventLoad.data), + const indexedSessions = createMemo(() => { + const directories = new Set(projectDirectories().map(pathKey)) + return takeRecentSessions( + homeSessions() + .sessions(sessionLoad.data, sessionEventLoad.data) + .filter((session) => directories.has(pathKey(session.directory))), HOME_SESSION_LIMIT, - Date.now(), - ), - ) + Number.NEGATIVE_INFINITY, + ) + }) const allRecords = createMemo(() => buildHomeSessionRecords({ sessions: indexedSessions, @@ -95,43 +108,21 @@ export function createHomeSessionsController(home: HomeController) { projectByID, }), ) - const records = createMemo(() => allRecords().slice(0, HOME_SESSION_LIMIT)) - const groups = createMemo(() => groupSessions(records(), language)) - const prefetched = new Set() - + const [visible, setVisible] = createSignal(HOME_SESSION_RENDER_BATCH) + let revealFrame: number | undefined createEffect(() => { - const ctx = home.server.focusedContext() - const conn = home.server.focused() - if (!ctx || !conn) return - records() - .slice(0, 2) - .forEach((record) => { - const key = `${ServerConnection.key(conn)}\0${record.session.id}` - if (prefetched.has(key)) return - prefetched.add(key) - createRoot((dispose) => { - try { - void ctx.sync.session - .sync(record.session.id) - .then(() => - Promise.all( - (ctx.sync.session.data.message[record.session.id] ?? []).flatMap((message) => - (ctx.sync.session.data.part[message.id] ?? []).flatMap((part) => { - if (part.type !== "text" || !part.text) return [] - return preloadMarkdown(part.text, part.id) - }), - ), - ), - ) - .catch(() => {}) - .finally(dispose) - } catch { - dispose() - } - }) - }) + const count = Math.min(allRecords().length, HOME_SESSION_LIMIT) + if (visible() >= count || revealFrame !== undefined) return + revealFrame = requestAnimationFrame(() => { + revealFrame = undefined + setVisible((current) => Math.min(current + HOME_SESSION_RENDER_BATCH, count)) + }) }) - + onCleanup(() => { + if (revealFrame !== undefined) cancelAnimationFrame(revealFrame) + }) + const records = createMemo(() => allRecords().slice(0, visible())) + const groups = createMemo(() => groupSessions(records(), language)) command.register("home.palette", () => [ { id: "command.palette", diff --git a/packages/app/src/pages/layout-new.tsx b/packages/app/src/pages/layout-new.tsx index 6d76ca84017..3c658413aab 100644 --- a/packages/app/src/pages/layout-new.tsx +++ b/packages/app/src/pages/layout-new.tsx @@ -1,11 +1,15 @@ -import { createEffect, Suspense, type ParentProps } from "solid-js" +import { createEffect, lazy, Suspense, type ParentProps } from "solid-js" import { createStore } from "solid-js/store" import { DebugBar } from "@/components/debug-bar" -import { TabsInfoPopup } from "@/components/help-button" -import { Titlebar, type TitlebarUpdate } from "@/components/titlebar" +import type { TitlebarUpdate } from "@/components/titlebar" import { usePlatform } from "@/context/platform" import { setV2Toast, ToastRegion } from "@/utils/toast" +const Titlebar = lazy(() => import("@/components/titlebar").then((module) => ({ default: module.Titlebar }))) +const TabsInfoPopup = lazy(() => + import("@/components/help-button").then((module) => ({ default: module.TabsInfoPopup })), +) + export default function NewLayout(props: ParentProps) { const platform = usePlatform() const [state, setState] = createStore({ debugTools: true }) @@ -30,19 +34,23 @@ export default function NewLayout(props: ParentProps) { "padding-bottom": "env(safe-area-inset-bottom, 0px)", }} > - setState("debugTools", (value) => !value) } - : undefined - } - /> + }> + setState("debugTools", (value) => !value) } + : undefined + } + /> +
{props.children}
{import.meta.env.DEV && state.debugTools && } - + + + ) diff --git a/packages/app/src/pages/layout.tsx b/packages/app/src/pages/layout.tsx index cc8cb2c61b7..3894819ab25 100644 --- a/packages/app/src/pages/layout.tsx +++ b/packages/app/src/pages/layout.tsx @@ -635,7 +635,7 @@ export default function LegacyLayout(props: ParentProps) { running: number } - const prefetchChunk = 200 + const prefetchChunk = 50 const prefetchConcurrency = 2 const prefetchPendingLimit = 10 const span = 4 @@ -778,18 +778,11 @@ export default function LegacyLayout(props: ParentProps) { } createEffect(() => { + if (params.id) return const sessions = currentSessions() if (sessions.length === 0) return - - const index = params.id ? sessions.findIndex((s) => s.id === params.id) : 0 - if (index === -1) return - - if (!params.id) { - const first = sessions[index] - if (first) prefetchSession(first, "high") - } - - warm(sessions, index) + const first = sessions[0] + if (first) prefetchSession(first, "high") }) function navigateSessionByOffset(offset: number) { diff --git a/packages/app/src/pages/session.tsx b/packages/app/src/pages/session.tsx index 19ba0cb1eea..b27e088f3c9 100644 --- a/packages/app/src/pages/session.tsx +++ b/packages/app/src/pages/session.tsx @@ -694,8 +694,12 @@ export default function Page() { queryFn: mode ? () => sdk() - .api.vcs.diff({ location: { directory: sdk().directory }, mode: mode === "git" ? "working" : mode }) - .then((result) => result.data) + .api.vcs.diff({ + location: { directory: sdk().directory }, + mode: mode === "git" ? "working" : mode, + context: 0, + }) + .then((result) => result.data.map((diff) => ({ ...diff, patch: "" }))) .catch((error) => { console.debug("[session-review] failed to load vcs diff", { mode, error }) return [] diff --git a/packages/app/src/utils/server-compat.test.ts b/packages/app/src/utils/server-compat.test.ts index 52e5ec6e3be..854ca8c4158 100644 --- a/packages/app/src/utils/server-compat.test.ts +++ b/packages/app/src/utils/server-compat.test.ts @@ -1,6 +1,8 @@ import { describe, expect, test } from "bun:test" import { createApiForServer, createSdkForServer } from "./server" import { createCompatibleApi } from "./server-compat" +import { decodeVcsDiffData } from "./vcs-diff-data" +import { decodeLegacySessionList } from "@/context/session-message-decode" function setup( protocol: "v1" | "v2" | Promise<"v1" | "v2">, @@ -48,6 +50,8 @@ function setup( current: createApiForServer({ server, fetch: fetcher }), legacy: (directory) => createSdkForServer({ server, fetch: fetcher, directory, throwOnError: true }), directory: "/repo", + decodeVcsDiff: async (buffer) => decodeVcsDiffData(buffer), + decodeSessionList: async (buffer) => decodeLegacySessionList(buffer), }) return { api, requests } } diff --git a/packages/app/src/utils/server-compat.ts b/packages/app/src/utils/server-compat.ts index 1df1338b71e..9e1b62614d2 100644 --- a/packages/app/src/utils/server-compat.ts +++ b/packages/app/src/utils/server-compat.ts @@ -1,7 +1,8 @@ import type { ServerApi } from "./server" import type { ServerProtocol } from "./server-protocol" -import type { AgentPartInput, FilePartInput, OpencodeClient, Session, TextPartInput } from "@opencode-ai/sdk/v2/client" +import type { AgentPartInput, FilePartInput, OpencodeClient, TextPartInput } from "@opencode-ai/sdk/v2/client" import type { + FileDiffInfo, Project, ProjectCurrent, SessionApi, @@ -15,6 +16,7 @@ import type { SessionShellInput, SessionShellOutput, } from "@opencode-ai/client/promise" +import { legacySessionInfo } from "@/context/session-message-decode" type LegacyClient = OpencodeClient type LegacyFor = (directory?: string) => LegacyClient @@ -51,6 +53,8 @@ type CompatibleInput = { current: ServerApi legacy: LegacyFor directory?: string + decodeVcsDiff: (buffer: ArrayBuffer) => Promise + decodeSessionList: (buffer: ArrayBuffer) => Promise } function mime(uri: string) { @@ -58,31 +62,6 @@ function mime(uri: string) { return match?.[1] ?? "application/octet-stream" } -function sessionInfo(session: Session): SessionInfo { - return { - id: session.id, - parentID: session.parentID, - projectID: session.projectID, - agent: session.agent, - model: session.model && { - id: session.model.id, - providerID: session.model.providerID, - variant: session.model.variant, - }, - cost: session.cost ?? 0, - tokens: session.tokens ?? { input: 0, output: 0, reasoning: 0, cache: { read: 0, write: 0 } }, - time: session.time, - title: session.title, - location: { directory: session.directory, workspaceID: session.workspaceID }, - subpath: session.path, - revert: session.revert && { - messageID: session.revert.messageID, - partID: session.revert.partID, - snapshot: session.revert.snapshot, - }, - } -} - export function createCompatibleApi(input: CompatibleInput): CompatibleApi { const v1 = createV1Api(input) return lazyApi( @@ -148,29 +127,34 @@ function createV1Api(input: CompatibleInput): CompatibleApi { search: value.search, limit: value.limit, }, - options, + { ...options, parseAs: "arrayBuffer" }, ) - return { data: (result.data ?? []).map(sessionInfo), cursor: {} } + if (!(result.data instanceof ArrayBuffer)) throw new Error("Session list response is not an ArrayBuffer") + return { data: await input.decodeSessionList(result.data), cursor: {} } } - const result = await legacy({ directory: value?.directory }).session.list({ - directory: value?.directory, - roots: value?.parentID === null ? true : undefined, - search: value?.search, - limit: value?.limit, - }) - return { data: (result.data ?? []).map(sessionInfo), cursor: {} } + const result = await legacy({ directory: value?.directory }).session.list( + { + directory: value?.directory, + roots: value?.parentID === null ? true : undefined, + search: value?.search, + limit: value?.limit, + }, + { parseAs: "arrayBuffer" }, + ) + if (!(result.data instanceof ArrayBuffer)) throw new Error("Session list response is not an ArrayBuffer") + return { data: await input.decodeSessionList(result.data), cursor: {} } }, async create(value?: Parameters[0]) { const result = await legacy(value?.location ?? undefined).session.create({ directory: directory(value?.location ?? undefined), }) if (!result.data) throw new Error("Failed to create session") - return sessionInfo(result.data) + return legacySessionInfo(result.data) }, async get(value: Parameters[0]) { const result = await legacy().session.get(value) if (!result.data) throw new Error(`Session not found: ${value.sessionID}`) - return sessionInfo(result.data) + return legacySessionInfo(result.data) }, async active() { const result = await legacy().session.status() @@ -192,7 +176,7 @@ function createV1Api(input: CompatibleInput): CompatibleApi { async fork(value: Parameters[0]) { const result = await legacy().session.fork(value) if (!result.data) throw new Error("Failed to fork session") - return sessionInfo(result.data) + return legacySessionInfo(result.data) }, async interrupt(value: Parameters[0]) { await legacy().session.abort(value) @@ -341,20 +325,15 @@ function createV1Api(input: CompatibleInput): CompatibleApi { return located(result.data ?? [], value?.location) }, async diff(value: Parameters[0]) { - const result = await legacy(value.location).vcs.diff({ - mode: value.mode === "working" ? "git" : value.mode, - context: value.context, - }) - return located( - (result.data ?? []).map((file) => ({ - file: file.file, - patch: file.patch ?? "", - additions: file.additions, - deletions: file.deletions, - status: file.status ?? "modified", - })), - value.location, + const result = await legacy(value.location).vcs.diff( + { + mode: value.mode === "working" ? "git" : value.mode, + context: value.context, + }, + { parseAs: "arrayBuffer" }, ) + if (!(result.data instanceof ArrayBuffer)) throw new Error("VCS diff response is not an ArrayBuffer") + return located(await input.decodeVcsDiff(result.data), value.location) }, }, file: { diff --git a/packages/app/src/utils/vcs-diff-data.ts b/packages/app/src/utils/vcs-diff-data.ts new file mode 100644 index 00000000000..7664ddafd65 --- /dev/null +++ b/packages/app/src/utils/vcs-diff-data.ts @@ -0,0 +1,20 @@ +import type { FileDiffInfo } from "@opencode-ai/client/promise" + +export function decodeVcsDiffData(buffer: ArrayBuffer): FileDiffInfo[] { + const text = new TextDecoder().decode(buffer) + return (text ? JSON.parse(text) : []).map( + (file: { + file: string + patch?: string + additions: number + deletions: number + status?: "added" | "deleted" | "modified" + }) => ({ + file: file.file, + patch: file.patch ?? "", + additions: file.additions, + deletions: file.deletions, + status: file.status ?? "modified", + }), + ) +} diff --git a/packages/app/src/utils/vcs-diff-decoder.ts b/packages/app/src/utils/vcs-diff-decoder.ts new file mode 100644 index 00000000000..5e65abe2f93 --- /dev/null +++ b/packages/app/src/utils/vcs-diff-decoder.ts @@ -0,0 +1,60 @@ +import type { FileDiffInfo } from "@opencode-ai/client/promise" + +type Response = { id: number; data?: FileDiffInfo[]; error?: string } + +let worker: Worker | undefined +let nextID = 0 +const pending = new Map void; reject: (error: Error) => void }>() +let lastInput = 0 +document.addEventListener( + "beforeinput", + () => { + lastInput = performance.now() + }, + { capture: true }, +) + +export function decodeVcsDiff(buffer: ArrayBuffer) { + const id = ++nextID + return new Promise((resolve, reject) => { + pending.set(id, { resolve, reject }) + getWorker().postMessage({ id, buffer }, [buffer]) + }) +} + +function getWorker() { + if (worker) return worker + worker = new Worker(new URL("./vcs-diff-decoder.worker.ts", import.meta.url), { type: "module" }) + worker.onmessage = (event: MessageEvent) => { + const request = pending.get(event.data.id) + if (!request) return + pending.delete(event.data.id) + if (event.data.error) { + request.reject(new Error(event.data.error)) + return + } + resolveWhenInputIdle(request.resolve, event.data.data ?? []) + } + worker.onerror = (event) => { + const error = new Error(event.message) + pending.forEach((request) => request.reject(error)) + pending.clear() + worker?.terminate() + worker = undefined + } + return worker +} + +function resolveWhenInputIdle(resolve: (value: FileDiffInfo[]) => void, value: FileDiffInfo[], initial = true) { + const active = document.activeElement + const editing = + active instanceof HTMLInputElement || + active instanceof HTMLTextAreaElement || + (active instanceof HTMLElement && active.isContentEditable) + const delay = Math.max(lastInput + 100 - performance.now(), initial && editing ? 100 : 0) + if (delay <= 0) { + resolve(value) + return + } + setTimeout(() => resolveWhenInputIdle(resolve, value, false), delay) +} diff --git a/packages/app/src/utils/vcs-diff-decoder.worker.ts b/packages/app/src/utils/vcs-diff-decoder.worker.ts new file mode 100644 index 00000000000..27b90384568 --- /dev/null +++ b/packages/app/src/utils/vcs-diff-decoder.worker.ts @@ -0,0 +1,13 @@ +import { decodeVcsDiffData } from "./vcs-diff-data" + +type Request = { id: number; buffer: ArrayBuffer } + +self.onmessage = (event: MessageEvent) => { + try { + self.postMessage({ id: event.data.id, data: decodeVcsDiffData(event.data.buffer) }) + } catch (error) { + self.postMessage({ id: event.data.id, error: error instanceof Error ? error.message : String(error) }) + } +} + +export {} diff --git a/packages/desktop/src/main/index.ts b/packages/desktop/src/main/index.ts index 183fc634db0..f588648379c 100644 --- a/packages/desktop/src/main/index.ts +++ b/packages/desktop/src/main/index.ts @@ -142,8 +142,11 @@ const main = Effect.gen(function* () { app.setAppUserModelId(appId) app.setPath( "userData", - onboardingTestRoot ? join(onboardingTestRoot, "desktop") : join(app.getPath("appData"), appId), + process.env.OPENCODE_PROFILE_USER_DATA ?? + (onboardingTestRoot ? join(onboardingTestRoot, "desktop") : join(app.getPath("appData"), appId)), ) + if (process.env.OPENCODE_PROFILE_USER_DATA) + app.setPath("sessionData", join(process.env.OPENCODE_PROFILE_USER_DATA, "session")) if (onboardingTestRoot) app.setPath("sessionData", join(onboardingTestRoot, "session")) initializeOldLayoutEligibility(app.getPath("userData")) logger = initLogging() @@ -191,9 +194,16 @@ const main = Effect.gen(function* () { ensureLoopbackNoProxy() useEnvProxy() app.commandLine.appendSwitch("proxy-bypass-list", "<-loopback>") - const features = app.commandLine.getSwitchValue("enable-features") - app.commandLine.appendSwitch("enable-features", features ? `${jsCallStackFeature},${features}` : jsCallStackFeature) - if (!app.isPackaged) app.commandLine.appendSwitch("remote-debugging-port", "9222") + const features = [ + jsCallStackFeature, + process.env.OPENCODE_PROFILE_LOAF === "1" ? "AlwaysLogLOAFURL" : "", + app.commandLine.getSwitchValue("enable-features"), + ] + .filter(Boolean) + .join(",") + app.commandLine.appendSwitch("enable-features", features) + if (!app.isPackaged) + app.commandLine.appendSwitch("remote-debugging-port", process.env.OPENCODE_PROFILE_CDP_PORT ?? "9222") if (!app.requestSingleInstanceLock()) { app.quit() diff --git a/packages/desktop/src/renderer/i18n/index.ts b/packages/desktop/src/renderer/i18n/index.ts index 552acffb705..d533f2ef49e 100644 --- a/packages/desktop/src/renderer/i18n/index.ts +++ b/packages/desktop/src/renderer/i18n/index.ts @@ -6,72 +6,11 @@ import { } from "../../../../app/src/i18n/desktop-native" import { dict as desktopEn } from "./en" -import { dict as desktopZh } from "./zh" -import { dict as desktopZht } from "./zht" -import { dict as desktopKo } from "./ko" -import { dict as desktopDe } from "./de" -import { dict as desktopEs } from "./es" -import { dict as desktopFr } from "./fr" -import { dict as desktopDa } from "./da" -import { dict as desktopJa } from "./ja" -import { dict as desktopPl } from "./pl" -import { dict as desktopRu } from "./ru" -import { dict as desktopUk } from "./uk" -import { dict as desktopAr } from "./ar" -import { dict as desktopNo } from "./no" -import { dict as desktopBr } from "./br" -import { dict as desktopBs } from "./bs" -import { dict as desktopTr } from "./tr" -import { dict as desktopHi } from "./hi" -import { dict as desktopNl } from "./nl" -import { dict as desktopId } from "./id" -import { dict as desktopVi } from "./vi" -import { dict as desktopIt } from "./it" -import { dict as desktopUr } from "./ur" -import { dict as desktopPa } from "./pa" -import { dict as desktopAz } from "./az" -import { dict as desktopFi } from "./fi" -import { dict as desktopSv } from "./sv" -import { dict as desktopTh } from "./th" - -import { dict as desktopAm } from "./am" -import { dict as desktopBg } from "./bg" -import { dict as desktopBn } from "./bn" -import { dict as desktopCa } from "./ca" -import { dict as desktopCs } from "./cs" -import { dict as desktopDv } from "./dv" -import { dict as desktopDz } from "./dz" -import { dict as desktopEl } from "./el" -import { dict as desktopEt } from "./et" -import { dict as desktopFa } from "./fa" -import { dict as desktopFo } from "./fo" -import { dict as desktopHr } from "./hr" -import { dict as desktopHu } from "./hu" -import { dict as desktopHy } from "./hy" -import { dict as desktopIs } from "./is" -import { dict as desktopKa } from "./ka" -import { dict as desktopKm } from "./km" -import { dict as desktopLo } from "./lo" -import { dict as desktopLt } from "./lt" -import { dict as desktopLv } from "./lv" -import { dict as desktopMk } from "./mk" -import { dict as desktopMn } from "./mn" -import { dict as desktopMs } from "./ms" -import { dict as desktopMy } from "./my" -import { dict as desktopNe } from "./ne" -import { dict as desktopRo } from "./ro" -import { dict as desktopSi } from "./si" -import { dict as desktopSk } from "./sk" -import { dict as desktopSl } from "./sl" -import { dict as desktopSq } from "./sq" -import { dict as desktopSr } from "./sr" -import { dict as desktopTg } from "./tg" -import { dict as desktopTk } from "./tk" -import { dict as desktopUz } from "./uz" +import { dict as appEn } from "../../../../app/src/i18n/en" export type Locale = DesktopNativeLocale -type RawDictionary = typeof desktopEn +type RawDictionary = typeof appEn & typeof desktopEn type Dictionary = Record, string> function detectLocale(): Locale { @@ -111,71 +50,76 @@ function pickLocale(value: unknown): Locale | null { return parseLocale(record.locale) } -const base = i18n.flatten(desktopEn) +const base = i18n.flatten({ ...appEn, ...desktopEn }) -function build(locale: Locale): Dictionary { +const loaders = { + zh: () => Promise.all([import("../../../../app/src/i18n/zh"), import("./zh")]), + zht: () => Promise.all([import("../../../../app/src/i18n/zht"), import("./zht")]), + ko: () => Promise.all([import("../../../../app/src/i18n/ko"), import("./ko")]), + de: () => Promise.all([import("../../../../app/src/i18n/de"), import("./de")]), + es: () => Promise.all([import("../../../../app/src/i18n/es"), import("./es")]), + fr: () => Promise.all([import("../../../../app/src/i18n/fr"), import("./fr")]), + da: () => Promise.all([import("../../../../app/src/i18n/da"), import("./da")]), + ja: () => Promise.all([import("../../../../app/src/i18n/ja"), import("./ja")]), + pl: () => Promise.all([import("../../../../app/src/i18n/pl"), import("./pl")]), + ru: () => Promise.all([import("../../../../app/src/i18n/ru"), import("./ru")]), + uk: () => Promise.all([import("../../../../app/src/i18n/uk"), import("./uk")]), + ar: () => Promise.all([import("../../../../app/src/i18n/ar"), import("./ar")]), + no: () => Promise.all([import("../../../../app/src/i18n/no"), import("./no")]), + br: () => Promise.all([import("../../../../app/src/i18n/br"), import("./br")]), + bs: () => Promise.all([import("../../../../app/src/i18n/bs"), import("./bs")]), + tr: () => Promise.all([import("../../../../app/src/i18n/tr"), import("./tr")]), + hi: () => Promise.all([import("../../../../app/src/i18n/hi"), import("./hi")]), + nl: () => Promise.all([import("../../../../app/src/i18n/nl"), import("./nl")]), + id: () => Promise.all([import("../../../../app/src/i18n/id"), import("./id")]), + vi: () => Promise.all([import("../../../../app/src/i18n/vi"), import("./vi")]), + it: () => Promise.all([import("../../../../app/src/i18n/it"), import("./it")]), + ur: () => Promise.all([import("../../../../app/src/i18n/ur"), import("./ur")]), + pa: () => Promise.all([import("../../../../app/src/i18n/pa"), import("./pa")]), + az: () => Promise.all([import("../../../../app/src/i18n/az"), import("./az")]), + fi: () => Promise.all([import("../../../../app/src/i18n/fi"), import("./fi")]), + sv: () => Promise.all([import("../../../../app/src/i18n/sv"), import("./sv")]), + th: () => Promise.all([import("../../../../app/src/i18n/th"), import("./th")]), + am: () => Promise.all([import("../../../../app/src/i18n/am"), import("./am")]), + bg: () => Promise.all([import("../../../../app/src/i18n/bg"), import("./bg")]), + bn: () => Promise.all([import("../../../../app/src/i18n/bn"), import("./bn")]), + ca: () => Promise.all([import("../../../../app/src/i18n/ca"), import("./ca")]), + cs: () => Promise.all([import("../../../../app/src/i18n/cs"), import("./cs")]), + dv: () => Promise.all([import("../../../../app/src/i18n/dv"), import("./dv")]), + dz: () => Promise.all([import("../../../../app/src/i18n/dz"), import("./dz")]), + el: () => Promise.all([import("../../../../app/src/i18n/el"), import("./el")]), + et: () => Promise.all([import("../../../../app/src/i18n/et"), import("./et")]), + fa: () => Promise.all([import("../../../../app/src/i18n/fa"), import("./fa")]), + fo: () => Promise.all([import("../../../../app/src/i18n/fo"), import("./fo")]), + hr: () => Promise.all([import("../../../../app/src/i18n/hr"), import("./hr")]), + hu: () => Promise.all([import("../../../../app/src/i18n/hu"), import("./hu")]), + hy: () => Promise.all([import("../../../../app/src/i18n/hy"), import("./hy")]), + is: () => Promise.all([import("../../../../app/src/i18n/is"), import("./is")]), + ka: () => Promise.all([import("../../../../app/src/i18n/ka"), import("./ka")]), + km: () => Promise.all([import("../../../../app/src/i18n/km"), import("./km")]), + lo: () => Promise.all([import("../../../../app/src/i18n/lo"), import("./lo")]), + lt: () => Promise.all([import("../../../../app/src/i18n/lt"), import("./lt")]), + lv: () => Promise.all([import("../../../../app/src/i18n/lv"), import("./lv")]), + mk: () => Promise.all([import("../../../../app/src/i18n/mk"), import("./mk")]), + mn: () => Promise.all([import("../../../../app/src/i18n/mn"), import("./mn")]), + ms: () => Promise.all([import("../../../../app/src/i18n/ms"), import("./ms")]), + my: () => Promise.all([import("../../../../app/src/i18n/my"), import("./my")]), + ne: () => Promise.all([import("../../../../app/src/i18n/ne"), import("./ne")]), + ro: () => Promise.all([import("../../../../app/src/i18n/ro"), import("./ro")]), + si: () => Promise.all([import("../../../../app/src/i18n/si"), import("./si")]), + sk: () => Promise.all([import("../../../../app/src/i18n/sk"), import("./sk")]), + sl: () => Promise.all([import("../../../../app/src/i18n/sl"), import("./sl")]), + sq: () => Promise.all([import("../../../../app/src/i18n/sq"), import("./sq")]), + sr: () => Promise.all([import("../../../../app/src/i18n/sr"), import("./sr")]), + tg: () => Promise.all([import("../../../../app/src/i18n/tg"), import("./tg")]), + tk: () => Promise.all([import("../../../../app/src/i18n/tk"), import("./tk")]), + uz: () => Promise.all([import("../../../../app/src/i18n/uz"), import("./uz")]), +} + +async function build(locale: Locale): Promise { if (locale === "en") return base - if (locale === "zh") return { ...base, ...i18n.flatten(desktopZh) } - if (locale === "zht") return { ...base, ...i18n.flatten(desktopZht) } - if (locale === "de") return { ...base, ...i18n.flatten(desktopDe) } - if (locale === "es") return { ...base, ...i18n.flatten(desktopEs) } - if (locale === "fr") return { ...base, ...i18n.flatten(desktopFr) } - if (locale === "da") return { ...base, ...i18n.flatten(desktopDa) } - if (locale === "ja") return { ...base, ...i18n.flatten(desktopJa) } - if (locale === "pl") return { ...base, ...i18n.flatten(desktopPl) } - if (locale === "ru") return { ...base, ...i18n.flatten(desktopRu) } - if (locale === "uk") return { ...base, ...i18n.flatten(desktopUk) } - if (locale === "ar") return { ...base, ...i18n.flatten(desktopAr) } - if (locale === "no") return { ...base, ...i18n.flatten(desktopNo) } - if (locale === "br") return { ...base, ...i18n.flatten(desktopBr) } - if (locale === "bs") return { ...base, ...i18n.flatten(desktopBs) } - if (locale === "tr") return { ...base, ...i18n.flatten(desktopTr) } - if (locale === "hi") return { ...base, ...i18n.flatten(desktopHi) } - if (locale === "nl") return { ...base, ...i18n.flatten(desktopNl) } - if (locale === "id") return { ...base, ...i18n.flatten(desktopId) } - if (locale === "vi") return { ...base, ...i18n.flatten(desktopVi) } - if (locale === "it") return { ...base, ...i18n.flatten(desktopIt) } - if (locale === "ur") return { ...base, ...i18n.flatten(desktopUr) } - if (locale === "pa") return { ...base, ...i18n.flatten(desktopPa) } - if (locale === "az") return { ...base, ...i18n.flatten(desktopAz) } - if (locale === "fi") return { ...base, ...i18n.flatten(desktopFi) } - if (locale === "sv") return { ...base, ...i18n.flatten(desktopSv) } - if (locale === "th") return { ...base, ...i18n.flatten(desktopTh) } - if (locale === "am") return { ...base, ...i18n.flatten(desktopAm) } - if (locale === "bg") return { ...base, ...i18n.flatten(desktopBg) } - if (locale === "bn") return { ...base, ...i18n.flatten(desktopBn) } - if (locale === "ca") return { ...base, ...i18n.flatten(desktopCa) } - if (locale === "cs") return { ...base, ...i18n.flatten(desktopCs) } - if (locale === "dv") return { ...base, ...i18n.flatten(desktopDv) } - if (locale === "dz") return { ...base, ...i18n.flatten(desktopDz) } - if (locale === "el") return { ...base, ...i18n.flatten(desktopEl) } - if (locale === "et") return { ...base, ...i18n.flatten(desktopEt) } - if (locale === "fa") return { ...base, ...i18n.flatten(desktopFa) } - if (locale === "fo") return { ...base, ...i18n.flatten(desktopFo) } - if (locale === "hr") return { ...base, ...i18n.flatten(desktopHr) } - if (locale === "hu") return { ...base, ...i18n.flatten(desktopHu) } - if (locale === "hy") return { ...base, ...i18n.flatten(desktopHy) } - if (locale === "is") return { ...base, ...i18n.flatten(desktopIs) } - if (locale === "ka") return { ...base, ...i18n.flatten(desktopKa) } - if (locale === "km") return { ...base, ...i18n.flatten(desktopKm) } - if (locale === "lo") return { ...base, ...i18n.flatten(desktopLo) } - if (locale === "lt") return { ...base, ...i18n.flatten(desktopLt) } - if (locale === "lv") return { ...base, ...i18n.flatten(desktopLv) } - if (locale === "mk") return { ...base, ...i18n.flatten(desktopMk) } - if (locale === "mn") return { ...base, ...i18n.flatten(desktopMn) } - if (locale === "ms") return { ...base, ...i18n.flatten(desktopMs) } - if (locale === "my") return { ...base, ...i18n.flatten(desktopMy) } - if (locale === "ne") return { ...base, ...i18n.flatten(desktopNe) } - if (locale === "ro") return { ...base, ...i18n.flatten(desktopRo) } - if (locale === "si") return { ...base, ...i18n.flatten(desktopSi) } - if (locale === "sk") return { ...base, ...i18n.flatten(desktopSk) } - if (locale === "sl") return { ...base, ...i18n.flatten(desktopSl) } - if (locale === "sq") return { ...base, ...i18n.flatten(desktopSq) } - if (locale === "sr") return { ...base, ...i18n.flatten(desktopSr) } - if (locale === "tg") return { ...base, ...i18n.flatten(desktopTg) } - if (locale === "tk") return { ...base, ...i18n.flatten(desktopTk) } - if (locale === "uz") return { ...base, ...i18n.flatten(desktopUz) } - return { ...base, ...i18n.flatten(desktopKo) } + const dictionaries = await loaders[locale]() + return { ...base, ...i18n.flatten(dictionaries[0].dict), ...i18n.flatten(dictionaries[1].dict) } } const state = { @@ -184,8 +128,6 @@ const state = { init: undefined as Promise | undefined, } -state.dict = build(state.locale) - const translate = i18n.translator(() => state.dict, i18n.resolveTemplate) export function t(key: keyof Dictionary, params?: Record) { @@ -202,7 +144,7 @@ export function initI18n(): Promise { const next = pickLocale(value) ?? state.locale state.locale = next - state.dict = build(next) + state.dict = await build(next) return next })().catch(() => state.locale) diff --git a/packages/desktop/src/renderer/index.tsx b/packages/desktop/src/renderer/index.tsx index 496060e0d66..3d90569bd69 100644 --- a/packages/desktop/src/renderer/index.tsx +++ b/packages/desktop/src/renderer/index.tsx @@ -22,7 +22,7 @@ import { createMemoryHistory, MemoryRouter, type BaseRouterProps } from "@solidj import { createEffect, createMemo, createResource, createSignal, onCleanup, Show } from "solid-js" import { render } from "solid-js/web" import pkg from "../../package.json" -import { t } from "./i18n" +import { initI18n, t } from "./i18n" import { initializationData } from "./initialization" import { DesktopFirstLaunchOnboarding } from "./onboarding" import { resetZoom, setPinchZoomEnabled, webviewZoom, zoomIn, zoomOut } from "./webview-zoom" @@ -60,6 +60,8 @@ if (import.meta.env.VITE_SENTRY_DSN) { }) } +void initI18n() + const [updaterState, setUpdaterState] = createSignal({ status: "disabled" }) void window.api.updater.subscribe(setUpdaterState) diff --git a/packages/session-ui/src/components/markdown-stream.test.ts b/packages/session-ui/src/components/markdown-stream.test.ts index baa0803f44b..ce18bc628f1 100644 --- a/packages/session-ui/src/components/markdown-stream.test.ts +++ b/packages/session-ui/src/components/markdown-stream.test.ts @@ -158,6 +158,12 @@ describe("markdown stream", () => { expect(final.blocks[2]).toEqual({ raw: "- final item", src: "- final item", mode: "full" }) }) + test("splits completed markdown into bounded top-level blocks", () => { + const result = project(undefined, "# Plan\n\nFirst paragraph.\n\nSecond paragraph.", false) + + expect(result.blocks.map((block) => block.raw)).toEqual(["# Plan", "First paragraph.", "Second paragraph."]) + }) + test("catches up paced text before finalizing", () => { const live = project(undefined, "# Plan\n\nFinished paragraph.\n\n- final", true) const final = project(live, `${live.text} item`, false) diff --git a/packages/session-ui/src/components/markdown-stream.ts b/packages/session-ui/src/components/markdown-stream.ts index 18cf4157beb..efebf4ba710 100644 --- a/packages/session-ui/src/components/markdown-stream.ts +++ b/packages/session-ui/src/components/markdown-stream.ts @@ -51,7 +51,7 @@ function heal(text: string) { } export function stream(text: string, live: boolean): Block[] { - if (!live) return completedProjection(text).blocks + if (!live) return completedBlocks(text) if (refs(text)) return [{ raw: text, src: heal(text), mode: "live" }] satisfies Block[] const tokens = marked.lexer(text) const tail = tokens.findLastIndex((token) => token.type !== "space") @@ -85,6 +85,17 @@ export function stream(text: string, live: boolean): Block[] { return [...result, { raw, src: openCode(code.raw), mode: "code", language: language(code.lang) }] } +function completedBlocks(text: string) { + if (refs(text)) return completedProjection(text).blocks + const tokens = marked.lexer(text) + return tokens.flatMap((token): Block[] => { + if (token.type === "space") return [] + if (token.type !== "code") return [{ raw: token.raw, src: token.raw, mode: "full" }] + const code = token as Tokens.Code + return [{ raw: code.raw, src: code.text, mode: "code", language: language(code.lang), complete: true }] + }) +} + export function project(previous: Projection | undefined, text: string, live: boolean): Projection { if (!live) { const current = @@ -93,7 +104,7 @@ export function project(previous: Projection | undefined, text: string, live: bo : previous && text.startsWith(previous.text) ? project(previous, text, true) : undefined - if (!current) return completedProjection(text) + if (!current) return { text, blocks: completedBlocks(text) } return { text, blocks: current.blocks.map((block) => { diff --git a/packages/session-ui/src/components/markdown.tsx b/packages/session-ui/src/components/markdown.tsx index 95894ba2712..b0ac55cbb12 100644 --- a/packages/session-ui/src/components/markdown.tsx +++ b/packages/session-ui/src/components/markdown.tsx @@ -491,6 +491,8 @@ export function Markdown( ) let copyCleanup: (() => void) | undefined + let renderFrame: number | undefined + let renderGeneration = 0 createEffect(() => { const container = root() @@ -499,6 +501,9 @@ export function Markdown( const content = local.text ? pendingBlocks(result, projected, local.cacheKey, owner) : [] if (!container) return if (isServer) return + const generation = ++renderGeneration + if (renderFrame !== undefined) cancelAnimationFrame(renderFrame) + renderFrame = undefined if (content.length === 0) { disposeCopyButtons(container) container.innerHTML = "" @@ -515,24 +520,40 @@ export function Markdown( }) activeCodeKeys.clear() nextCodeKeys.forEach((key) => activeCodeKeys.add(key)) - content.forEach((block, index) => updateBlock(container, index, block, labels)) - while (container.children.length > content.length) { - const child = container.lastElementChild - if (!child) break - disposeCopyButtons(child) - child.remove() + let index = 0 + const update = () => { + renderFrame = undefined + if (generation !== renderGeneration) return + const deadline = performance.now() + 8 + while (index < content.length && performance.now() < deadline) { + updateBlock(container, index, content[index]!, labels) + index += 1 + } + if (index < content.length) { + renderFrame = requestAnimationFrame(update) + return + } + while (container.children.length > content.length) { + const child = container.lastElementChild + if (!child) break + disposeCopyButtons(child) + child.remove() + } + container + .querySelectorAll('[data-slot="markdown-copy-button"]') + .forEach((button) => setCopyState(button, labels, button.dataset.copied === "true")) + if (!copyCleanup) + copyCleanup = setupCodeCopy(container, () => ({ + copy: i18n.t("ui.message.copy"), + copied: i18n.t("ui.message.copied"), + })) } - container - .querySelectorAll('[data-slot="markdown-copy-button"]') - .forEach((button) => setCopyState(button, labels, button.dataset.copied === "true")) - if (!copyCleanup) - copyCleanup = setupCodeCopy(container, () => ({ - copy: i18n.t("ui.message.copy"), - copied: i18n.t("ui.message.copied"), - })) + update() }) onCleanup(() => { + renderGeneration += 1 + if (renderFrame !== undefined) cancelAnimationFrame(renderFrame) if (copyCleanup) copyCleanup() disposeMarkdownProjection(owner) activeCodeKeys.forEach(disposeCode) diff --git a/packages/session-ui/src/pierre/worker.ts b/packages/session-ui/src/pierre/worker.ts index 1c1bebb40b5..9456bddc70a 100644 --- a/packages/session-ui/src/pierre/worker.ts +++ b/packages/session-ui/src/pierre/worker.ts @@ -24,7 +24,7 @@ function createPool(lineDiffType: "none" | "word-alt") { { theme: "OpenCode", lineDiffType, - preferredHighlighter: "shiki-wasm", + preferredHighlighter: "shiki-js", }, ) diff --git a/packages/session-ui/src/v2/components/prompt-input/index.tsx b/packages/session-ui/src/v2/components/prompt-input/index.tsx index c5358992c70..0c54bd32417 100644 --- a/packages/session-ui/src/v2/components/prompt-input/index.tsx +++ b/packages/session-ui/src/v2/components/prompt-input/index.tsx @@ -53,9 +53,11 @@ export function PromptInputV2(props: PromptInputV2Props) { const view = props.controller.view let editor: HTMLDivElement | undefined let localInput = false - const updateCursor = () => { + const updateCursor = (event: KeyboardEvent | PointerEvent) => { if (!editor || !window.getSelection()?.isCollapsed) return - props.controller.onCursor(promptInputV2Cursor(editor)) + if (event instanceof KeyboardEvent && !["ArrowLeft", "ArrowRight", "ArrowUp", "ArrowDown", "Home", "End"].includes(event.key)) + return + props.controller.onCursor(parsePromptInputV2Editor(editor).cursor) } const mode = createMemo(() => state.mode) const buttons = createMemo(() => ({ @@ -164,8 +166,7 @@ export function PromptInputV2(props: PromptInputV2Props) { class="relative z-10 block min-h-[60px] max-h-[180px] w-full overflow-y-auto whitespace-pre-wrap bg-transparent px-4 pt-4 pb-2 text-[13px] font-[440] leading-5 text-v2-text-text-base focus:outline-none empty:before:content-['\200B'] [&_[data-mention=file]]:text-syntax-property [&_[data-mention=agent]]:text-syntax-type [&_[data-mention=reference]]:text-syntax-keyword" classList={{ "font-mono!": state.mode === "shell", "opacity-50": props.disabled }} onInput={(event) => { - const cursor = promptInputV2Cursor(event.currentTarget) - const prompt = parsePromptInputV2Editor(event.currentTarget) + const { prompt, cursor } = parsePromptInputV2Editor(event.currentTarget) const images = props.controller.parts().filter((part) => part.type === "image") localInput = true props.controller.onInput(prompt.map((part) => part.content).join(""), [...prompt, ...images], cursor) @@ -302,8 +303,13 @@ function renderPromptInputV2Editor(editor: HTMLDivElement, prompt: PromptInputV2 function parsePromptInputV2Editor(editor: HTMLDivElement) { const parts: Exclude[] = [] + const selection = window.getSelection() + const anchorNode = selection && editor.contains(selection.anchorNode) ? selection.anchorNode : undefined + const anchorOffset = anchorNode ? selection!.anchorOffset : 0 let buffer = "" let position = 0 + let cursor: number | undefined + const offset = () => position + buffer.length const flush = () => { if (!buffer) return @@ -338,43 +344,42 @@ function parsePromptInputV2Editor(editor: HTMLDivElement) { } const visit = (node: Node) => { if (node.nodeType === Node.TEXT_NODE) { + if (node === anchorNode) cursor = offset() + Math.min(anchorOffset, node.textContent?.length ?? 0) buffer += node.textContent ?? "" return } if (!(node instanceof HTMLElement)) return if (node.dataset.mention) { + if (node === anchorNode) cursor = offset() + (anchorOffset > 0 ? (node.textContent?.length ?? 0) : 0) mention(node) return } if (node.tagName === "BR") { + if (node === anchorNode) cursor = offset() + (anchorOffset > 0 ? 1 : 0) buffer += "\n" return } - Array.from(node.childNodes).forEach(visit) + Array.from(node.childNodes).forEach((child, index) => { + if (node === anchorNode && anchorOffset === index) cursor = offset() + visit(child) + }) + if (node === anchorNode && anchorOffset >= node.childNodes.length) cursor = offset() } Array.from(editor.childNodes).forEach((node, index, nodes) => { + if (editor === anchorNode && anchorOffset === index) cursor = offset() visit(node) if (node instanceof HTMLElement && ["DIV", "P"].includes(node.tagName) && index < nodes.length - 1) buffer += "\n" }) + if (editor === anchorNode && anchorOffset >= editor.childNodes.length) cursor = offset() flush() - if ( - parts.every((part) => part.type === "text") && - parts.every((part) => part.content.replace(/[\n\u200B]/g, "") === "") - ) { - return [{ type: "text" as const, content: "", start: 0, end: 0 }] - } - if (parts.length > 0) return parts - return [{ type: "text" as const, content: "", start: 0, end: 0 }] -} - -function promptInputV2Cursor(editor: HTMLDivElement) { - const selection = window.getSelection() - if (!selection?.rangeCount || !editor.contains(selection.anchorNode)) return editor.textContent?.length ?? 0 - const range = selection.getRangeAt(0).cloneRange() - range.selectNodeContents(editor) - range.setEnd(selection.anchorNode!, selection.anchorOffset) - return range.toString().length + const result = + parts.length === 0 || + (parts.every((part) => part.type === "text") && + parts.every((part) => part.content.replace(/[\n\u200B]/g, "") === "")) + ? [{ type: "text" as const, content: "", start: 0, end: 0 }] + : parts + return { prompt: result, cursor: cursor ?? offset() } } export function PromptInputV2Attachments(props: {