import type { Project, UserMessage } from "@opencode-ai/sdk/v2" import { useDialog } from "@opencode-ai/ui/context/dialog" import { createQuery, skipToken, useMutation, useQueryClient } from "@tanstack/solid-query" import { batch, onCleanup, Show, Match, Switch, createMemo, createEffect, createComputed, on, onMount, untrack, } from "solid-js" import { makeEventListener } from "@solid-primitives/event-listener" import { createMediaQuery } from "@solid-primitives/media" import { createResizeObserver } from "@solid-primitives/resize-observer" import { debounce } from "@solid-primitives/scheduled" import { useLocal } from "@/context/local" import { selectionFromLines, useFile, type FileSelection, type SelectedLineRange } from "@/context/file" import { createStore } from "solid-js/store" import { ResizeHandle } from "@opencode-ai/ui/resize-handle" import { Select } from "@opencode-ai/ui/select" import { Tabs } from "@opencode-ai/ui/tabs" import { createAutoScroll } from "@opencode-ai/ui/hooks" import { previewSelectedLines } from "@opencode-ai/ui/pierre/selection-bridge" import { Button } from "@opencode-ai/ui/button" import { showToast } from "@/utils/toast" import { checksum } from "@opencode-ai/core/util/encode" import { useLocation, useSearchParams } from "@solidjs/router" import { NewSessionView, SessionHeader } from "@/components/session" import { useComments } from "@/context/comments" import { useServerSync } from "@/context/server-sync" import { useLanguage } from "@/context/language" import { useLayout } from "@/context/layout" import { usePrompt } from "@/context/prompt" import { usePlatform } from "@/context/platform" import { useSDK } from "@/context/sdk" import { useServerSDK } from "@/context/server-sdk" import { useSettings } from "@/context/settings" import { useSync } from "@/context/sync" import { useTerminal } from "@/context/terminal" import { type FollowupDraft, sendFollowupDraft } from "@/components/prompt-input/submit" import { createSessionComposerState, SessionComposerRegion } from "@/pages/session/composer" import { createOpenReviewFile, createSessionTabs, createSizing, focusTerminalById, shouldFocusTerminalOnKeyDown, shouldShowFileTree, } from "@/pages/session/helpers" import { MessageTimeline } from "@/pages/session/timeline/message-timeline" import { createTimelineModel } from "@/pages/session/timeline/model" import { type DiffStyle, SessionReviewTab, type SessionReviewTabProps } from "@/pages/session/review-tab" import { useSessionLayout } from "@/pages/session/session-layout" import { useServer } from "@/context/server" import { syncSessionModel } from "@/pages/session/session-model-helpers" import { SessionSidePanel } from "@/pages/session/session-side-panel" import { TerminalPanel } from "@/pages/session/terminal-panel" import { useSessionCommands } from "@/pages/session/use-session-commands" import { useSessionHashScroll } from "@/pages/session/use-session-hash-scroll" import { Identifier } from "@/utils/id" import { diffs as list } from "@/utils/diffs" import { Persist, persisted } from "@/utils/persist" import { extractPromptFromParts } from "@/utils/prompt" import { formatServerError } from "@/utils/server-errors" import { useUsageExceededDialogs } from "./session/usage-exceeded-dialogs" type FollowupItem = FollowupDraft & { id: string } type FollowupEdit = Pick const emptyFollowups: FollowupItem[] = [] type ChangeMode = "git" | "branch" | "turn" type VcsMode = "git" | "branch" export default function Page() { const serverSync = useServerSync() const layout = useLayout() const local = useLocal() const file = useFile() const sync = useSync() const queryClient = useQueryClient() const dialog = useDialog() const language = useLanguage() const sdk = useSDK() const serverSDK = useServerSDK() const settings = useSettings() const platform = usePlatform() const prompt = usePrompt() const comments = useComments() const terminal = useTerminal() const server = useServer() const [searchParams, setSearchParams] = useSearchParams<{ prompt?: string }>() const location = useLocation() const { params, sessionKey, workspaceKey, tabs, view } = useSessionLayout() const newSessionDesign = createMemo(() => settings.general.newLayoutDesigns()) createEffect(() => { if (!prompt.ready()) return untrack(() => { if (params.id) return const text = searchParams.prompt if (!text) return prompt.set([{ type: "text", content: text, start: 0, end: text.length }], text.length) setSearchParams({ ...searchParams, prompt: undefined }) }) }) const [ui, setUi] = createStore({ pendingMessage: undefined as string | undefined, reviewSnap: false, scrollGesture: 0, scroll: { overflow: false, bottom: true, jump: false, }, }) const composer = createSessionComposerState() const workspaceTabs = createMemo(() => layout.tabs(workspaceKey)) createEffect( on( () => params.id, (id, prev) => { if (!id) return if (prev) return const pending = layout.handoff.tabs() if (!pending) return if (Date.now() - pending.at > 60_000) { layout.handoff.clearTabs() return } if (pending.scope !== server.scope()) return if (pending.id !== id) return layout.handoff.clearTabs() if (pending.dir !== (params.dir ?? "")) return const from = workspaceTabs().tabs() if (from.all.length === 0 && !from.active) return const current = tabs().tabs() if (current.all.length > 0 || current.active) return const all = normalizeTabs(from.all) const active = from.active ? normalizeTab(from.active) : undefined tabs().setAll(all) tabs().setActive(active && all.includes(active) ? active : all[0]) workspaceTabs().setAll([]) workspaceTabs().setActive(undefined) }, { defer: true }, ), ) const isDesktop = createMediaQuery("(min-width: 768px)") const size = createSizing() const desktopReviewOpen = createMemo(() => isDesktop() && view().reviewPanel.opened()) const desktopFileTreeOpen = createMemo( () => isDesktop() && shouldShowFileTree({ visible: settings.visibility.fileTree(), opened: layout.fileTree.opened(), }), ) const desktopSidePanelOpen = createMemo(() => desktopReviewOpen() || desktopFileTreeOpen()) const sessionPanelWidth = createMemo(() => { if (!desktopSidePanelOpen()) return "100%" if (desktopReviewOpen()) return `${layout.session.width()}px` return `calc(100% - ${layout.fileTree.width()}px)` }) const centered = createMemo(() => isDesktop() && !desktopReviewOpen()) function normalizeTab(tab: string) { if (!tab.startsWith("file://")) return tab return file.tab(tab) } function normalizeTabs(list: string[]) { const seen = new Set() const next: string[] = [] for (const item of list) { const value = normalizeTab(item) if (seen.has(value)) continue seen.add(value) next.push(value) } return next } const openReviewPanel = () => { if (!view().reviewPanel.opened()) view().reviewPanel.open() } const info = createMemo(() => (params.id ? sync().session.get(params.id) : undefined)) const isChildSession = createMemo(() => !!info()?.parentID) const diffs = createMemo(() => (params.id ? list(sync().data.session_diff[params.id]) : [])) const canReview = createMemo(() => !!sync().project) const reviewTab = createMemo(() => isDesktop()) const tabState = createSessionTabs({ tabs, pathFromTab: file.pathFromTab, normalizeTab, review: reviewTab, hasReview: canReview, }) const activeTab = tabState.activeTab const activeFileTab = tabState.activeFileTab const revertMessageID = createMemo(() => info()?.revert?.messageID) const timeline = createTimelineModel({ sessionID: () => params.id, revertMessageID }) const historyLoading = timeline.history.loading const historyMore = timeline.history.more const lastUserMessage = timeline.lastUserMessage const messages = timeline.messages const messagesReady = timeline.ready const sessionSync = timeline.resource const userMessages = timeline.userMessages const visibleUserMessages = timeline.visibleUserMessages createEffect(() => { const tab = activeFileTab() if (!tab) return const path = file.pathFromTab(tab) if (path) void file.load(path) }) createEffect( on( () => lastUserMessage()?.id, () => { const msg = lastUserMessage() if (!msg) return syncSessionModel(local, msg) }, ), ) createEffect( on( () => ({ dir: params.dir, id: params.id }), (next, prev) => { if (!prev) return if (next.dir === prev.dir && next.id === prev.id) return if (prev.id && !next.id) local.session.reset() }, { defer: true }, ), ) const [store, setStore] = createStore({ messageId: undefined as string | undefined, mobileTab: "session" as "session" | "changes", changes: "git" as ChangeMode, newSessionWorktree: "main", deferRender: false, }) const [followup, setFollowup] = persisted( Persist.serverWorkspace(serverSDK().scope, sdk().directory, "followup", ["followup.v1"]), createStore<{ items: Record failed: Record paused: Record edit: Record }>({ items: {}, failed: {}, paused: {}, edit: {}, }), ) createComputed((prev) => { const key = sessionKey() if (key !== prev) { setStore("deferRender", true) requestAnimationFrame(() => { setTimeout(() => setStore("deferRender", false), 0) }) } return key }, sessionKey()) let reviewFrame: number | undefined let todoFrame: number | undefined let todoTimer: number | undefined let diffFrame: number | undefined let diffTimer: number | undefined createComputed((prev) => { const open = desktopReviewOpen() if (prev === undefined || prev === open) return open if (reviewFrame !== undefined) cancelAnimationFrame(reviewFrame) setUi("reviewSnap", true) reviewFrame = requestAnimationFrame(() => { reviewFrame = undefined setUi("reviewSnap", false) }) return open }, desktopReviewOpen()) const turnDiffs = createMemo(() => list(lastUserMessage()?.summary?.diffs)) const nogit = createMemo(() => { const project = sync().project return !!project && project.vcs !== "git" }) const changesOptions = createMemo(() => { const list: ChangeMode[] = [] const project = sync().project const vcs = sync().data.vcs if (project?.vcs === "git") list.push("git") if (project?.vcs === "git" && vcs?.branch && vcs?.default_branch && vcs.branch !== vcs.default_branch) { list.push("branch") } list.push("turn") return list }) const mobileChanges = createMemo(() => !isDesktop() && store.mobileTab === "changes") const wantsReview = createMemo(() => isDesktop() ? desktopFileTreeOpen() || (desktopReviewOpen() && activeTab() === "review") : store.mobileTab === "changes", ) const vcsMode = createMemo(() => { if (store.changes === "git" || store.changes === "branch") return store.changes }) const vcsKey = createMemo( () => ["session-vcs", sdk().directory, sync().data.vcs?.branch ?? "", sync().data.vcs?.default_branch ?? ""] as const, ) const vcsQuery = createQuery(() => { const mode = vcsMode() const enabled = wantsReview() && sync().project?.vcs === "git" return { queryKey: [...vcsKey(), mode] as const, enabled, queryFn: mode ? () => sdk() .client.vcs.diff({ mode }) .then((result) => list(result.data)) .catch((error) => { console.debug("[session-review] failed to load vcs diff", { mode, error }) return [] }) : skipToken, } }) const refreshVcs = debounce(() => void queryClient.invalidateQueries({ queryKey: vcsKey() }), 100) const reviewDiffs = () => { if (store.changes === "git" || store.changes === "branch") // avoids suspense return vcsQuery.isFetched ? (vcsQuery.data ?? []) : [] return turnDiffs() } const reviewCount = () => reviewDiffs().length const hasReview = () => reviewCount() > 0 const reviewReady = () => { if (store.changes === "git" || store.changes === "branch") return !vcsQuery.isPending return true } const newSessionWorktree = createMemo(() => { if (store.newSessionWorktree === "create") return "create" const project = sync().project if (project && sdk().directory !== project.worktree) return sdk().directory return "main" }) const setActiveMessage = (message: UserMessage | undefined) => { messageMark = scrollMark setStore("messageId", message?.id) } const anchor = (id: string) => `message-${id}` const cursor = () => { const root = scroller if (!root) return store.messageId const box = root.getBoundingClientRect() const line = box.top + 100 const list = [...root.querySelectorAll("[data-message-id]")] .map((el) => { const id = el.dataset.messageId if (!id) return const rect = el.getBoundingClientRect() return { id, top: rect.top, bottom: rect.bottom } }) .filter((item): item is { id: string; top: number; bottom: number } => !!item) const shown = list.filter((item) => item.bottom > box.top && item.top < box.bottom) const hit = shown.find((item) => item.top <= line && item.bottom >= line) if (hit) return hit.id const near = [...shown].sort((a, b) => { const da = Math.abs(a.top - line) const db = Math.abs(b.top - line) if (da !== db) return da - db return a.top - b.top })[0] if (near) return near.id return list.filter((item) => item.top <= line).at(-1)?.id ?? list[0]?.id ?? store.messageId } function navigateMessageByOffset(offset: number) { const msgs = visibleUserMessages() if (msgs.length === 0) return const current = store.messageId && messageMark === scrollMark ? store.messageId : cursor() const base = current ? msgs.findIndex((m) => m.id === current) : msgs.length const currentIndex = base === -1 ? msgs.length : base const targetIndex = currentIndex + offset if (targetIndex < 0 || targetIndex > msgs.length) return if (targetIndex === msgs.length) { resumeScroll() return } autoScroll.pause() scrollToMessage(msgs[targetIndex], "auto") } function upsert(next: Project) { const list = serverSync().data.project sync().set("project", next.id) const idx = list.findIndex((item) => item.id === next.id) if (idx >= 0) { serverSync().set( "project", list.map((item, i) => (i === idx ? { ...item, ...next } : item)), ) return } const at = list.findIndex((item) => item.id > next.id) if (at >= 0) { serverSync().set("project", [...list.slice(0, at), next, ...list.slice(at)]) return } serverSync().set("project", [...list, next]) } const gitMutation = useMutation(() => ({ mutationFn: () => sdk().client.project.initGit(), onSuccess: (x) => { if (!x.data) return upsert(x.data) }, onError: (err) => { showToast({ variant: "error", title: language.t("common.requestFailed"), description: formatServerError(err, language.t), }) }, })) function initGit() { if (gitMutation.isPending) return gitMutation.mutate() } let inputRef!: HTMLDivElement let promptDock: HTMLDivElement | undefined let dockHeight = 0 let scroller: HTMLDivElement | undefined let content: HTMLDivElement | undefined let revealMessage = (_id: string) => {} let scrollToEnd = () => {} let scrollMark = 0 let messageMark = 0 const scrollGestureWindowMs = 250 const markScrollGesture = (target?: EventTarget | null) => { const root = scroller if (!root) return const el = target instanceof Element ? target : undefined const nested = el?.closest("[data-scrollable]") if (nested && nested !== root) return setUi("scrollGesture", Date.now()) } const hasScrollGesture = () => Date.now() - ui.scrollGesture < scrollGestureWindowMs createEffect( on( () => { const id = params.id return [ sdk().directory, id, id ? (sync().data.session_status[id]?.type ?? "idle") : "idle", id ? composer.blocked() : false, ] as const }, ([dir, id, status, blocked]) => { if (todoFrame !== undefined) cancelAnimationFrame(todoFrame) if (todoTimer !== undefined) window.clearTimeout(todoTimer) todoFrame = undefined todoTimer = undefined if (!id) return if (status === "idle" && !blocked) return const cached = untrack( () => sync().data.todo[id] !== undefined || serverSync().data.session_todo[id] !== undefined, ) todoFrame = requestAnimationFrame(() => { todoFrame = undefined todoTimer = window.setTimeout(() => { todoTimer = undefined if (sdk().directory !== dir || params.id !== id) return untrack(() => { void sync().session.todo(id, cached ? { force: true } : undefined) }) }, 0) }) }, { defer: true }, ), ) createEffect( on( () => visibleUserMessages().at(-1)?.id, (lastId, prevLastId) => { if (lastId && prevLastId && lastId > prevLastId) { setStore("messageId", undefined) } }, { defer: true }, ), ) createEffect( on( sessionKey, () => { setStore("messageId", undefined) setStore("changes", "git") setUi("pendingMessage", undefined) }, { defer: true }, ), ) const stopVcs = sdk().event.listen((evt) => { if (evt.details.type !== "file.watcher.updated") return const props = typeof evt.details.properties === "object" && evt.details.properties ? (evt.details.properties as Record) : undefined const file = typeof props?.file === "string" ? props.file : undefined if (!file || file.startsWith(".git/")) return refreshVcs() }) onCleanup(stopVcs) createEffect( on( () => params.dir, (dir) => { if (!dir) return setStore("newSessionWorktree", "main") }, { defer: true }, ), ) const selectionPreview = (path: string, selection: FileSelection) => { const content = file.get(path)?.content?.content if (!content) return undefined return previewSelectedLines(content, { start: selection.startLine, end: selection.endLine }) } const addCommentToContext = (input: { file: string selection: SelectedLineRange comment: string preview?: string origin?: "review" | "file" }) => { const selection = selectionFromLines(input.selection) const preview = input.preview ?? selectionPreview(input.file, selection) const saved = comments.add({ file: input.file, selection: input.selection, comment: input.comment, }) prompt.context.add({ type: "file", path: input.file, selection, comment: input.comment, commentID: saved.id, commentOrigin: input.origin, preview, }) } const updateCommentInContext = (input: { id: string file: string selection: SelectedLineRange comment: string preview?: string }) => { comments.update(input.file, input.id, input.comment) prompt.context.updateComment(input.file, input.id, { comment: input.comment, ...(input.preview ? { preview: input.preview } : {}), }) } const removeCommentFromContext = (input: { id: string; file: string }) => { comments.remove(input.file, input.id) prompt.context.removeComment(input.file, input.id) } const reviewCommentActions = createMemo(() => ({ moreLabel: language.t("common.moreOptions"), editLabel: language.t("common.edit"), deleteLabel: language.t("common.delete"), saveLabel: language.t("common.save"), })) const isEditableTarget = (target: EventTarget | null | undefined) => { if (!(target instanceof HTMLElement)) return false return /^(INPUT|TEXTAREA|SELECT|BUTTON)$/.test(target.tagName) || target.isContentEditable } const deepActiveElement = () => { let current: Element | null = document.activeElement while (current instanceof HTMLElement && current.shadowRoot?.activeElement) { current = current.shadowRoot.activeElement } return current instanceof HTMLElement ? current : undefined } const handleKeyDown = (event: KeyboardEvent) => { const path = event.composedPath() const target = path.find((item): item is HTMLElement => item instanceof HTMLElement) const activeElement = deepActiveElement() const protectedTarget = path.some( (item) => item instanceof HTMLElement && item.closest("[data-prevent-autofocus]") !== null, ) if (protectedTarget || isEditableTarget(target)) return if (activeElement) { const isProtected = activeElement.closest("[data-prevent-autofocus]") const isInput = isEditableTarget(activeElement) if (isProtected || isInput) return } if (dialog.active) return if (activeElement === inputRef) { if (event.key === "Escape") inputRef?.blur() return } // Prefer the open terminal over the composer when it can take focus if (view().terminal.opened()) { const id = terminal.active() if (id && shouldFocusTerminalOnKeyDown(event) && focusTerminalById(id)) return } // Only treat explicit scroll keys as potential "user scroll" gestures. if (event.key === "PageUp" || event.key === "PageDown" || event.key === "Home" || event.key === "End") { markScrollGesture() return } if (event.key.length === 1 && event.key !== "Unidentified" && !(event.ctrlKey || event.metaKey)) { if (composer.blocked() || isChildSession()) return inputRef?.focus() } } createEffect(() => { const list = changesOptions() if (list.includes(store.changes)) return const next = list[0] if (!next) return setStore("changes", next) }) createEffect( on( () => sync().data.session_status[params.id ?? ""]?.type, (next, prev) => { if (next !== "idle" || prev === undefined || prev === "idle") return refreshVcs() }, { defer: true }, ), ) const fileTreeTab = () => layout.fileTree.tab() const setFileTreeTab = (value: "changes" | "all") => layout.fileTree.setTab(value) const [tree, setTree] = createStore({ reviewScroll: undefined as HTMLDivElement | undefined, pendingDiff: undefined as string | undefined, activeDiff: undefined as string | undefined, }) createEffect( on( sessionKey, () => { setTree({ reviewScroll: undefined, pendingDiff: undefined, activeDiff: undefined, }) }, { defer: true }, ), ) const showAllFiles = () => { if (fileTreeTab() !== "changes") return setFileTreeTab("all") } const focusInput = () => { if (isChildSession()) return inputRef?.focus() } useSessionCommands({ navigateMessageByOffset, setActiveMessage, focusInput, review: reviewTab, }) const openReviewFile = createOpenReviewFile({ showAllFiles, tabForPath: file.tab, openTab: tabs().open, setActive: tabs().setActive, loadFile: file.load, }) const changesTitle = () => { if (!canReview()) { return null } const label = (option: ChangeMode) => { if (option === "git") return language.t("ui.sessionReview.title.git") if (option === "branch") return language.t("ui.sessionReview.title.branch") return language.t("ui.sessionReview.title.lastTurn") } return (