wip: desktop work

2 files changed, 73 insertions, 53 deletions

diff --git a/packages/desktop/src/components/message-progress.tsx b/packages/desktop/src/components/message-progress.tsx
index fd66c5caf..5533ae413 100644
--- a/packages/desktop/src/components/message-progress.tsx
+++ b/packages/desktop/src/components/message-progress.tsx
@@ -1,7 +1,8 @@
-import { For, JSXElement, Match, Switch, createEffect, createMemo, createSignal, onCleanup } from "solid-js"
-import { Part } from "@opencode-ai/ui"
+import { For, JSXElement, Match, Show, Switch, createEffect, createMemo, createSignal, onCleanup } from "solid-js"
+import { Markdown, Part } from "@opencode-ai/ui"
 import { useSync } from "@/context/sync"
 import type { AssistantMessage as AssistantMessageType, Part as PartType, ToolPart } from "@opencode-ai/sdk"
+import { Spinner } from "./spinner"
 
 export function MessageProgress(props: { assistantMessages: () => AssistantMessageType[]; done?: boolean }) {
   const sync = useSync()
@@ -22,37 +23,42 @@ export function MessageProgress(props: { assistantMessages: () => AssistantMessa
       ) as ToolPart,
   )
 
-  const eligibleItems = createMemo(() => {
-    let allParts = parts()
+  const resolvedParts = createMemo(() => {
+    let resolved = parts()
     const task = currentTask()
     if (task && task.state && "metadata" in task.state && task.state.metadata?.sessionId) {
       const messages = sync.data.message[task.state.metadata.sessionId as string]?.filter((m) => m.role === "assistant")
-      allParts = messages?.flatMap((m) => sync.data.part[m.id]) ?? parts()
+      resolved = messages?.flatMap((m) => sync.data.part[m.id]) ?? parts()
     }
-    return allParts.filter(
-      (p) =>
-        p?.type === "text" ||
-        (p?.type === "reasoning" && p.time?.end) ||
-        (p?.type === "tool" && p.state.status === "completed"),
-    )
+    return resolved
+  })
+  const currentText = createMemo(
+    () =>
+      resolvedParts().findLast((p) => p?.type === "text")?.text ||
+      resolvedParts().findLast((p) => p?.type === "reasoning")?.text,
+  )
+  const eligibleItems = createMemo(() => {
+    return resolvedParts().filter((p) => p?.type === "tool" && p.state.status === "completed")
   })
   const finishedItems = createMemo<(JSXElement | PartType)[]>(() => [
-    "",
-    "",
-    <div class="text-text-diff-add-base">Loading...</div>,
+    <div class="h-8 w-full" />,
+    <div class="h-8 w-full" />,
+    <div class="flex items-center gap-x-5 pl-3 text-text-base">
+      <Spinner /> <span class="text-12-medium">Thinking...</span>
+    </div>,
     ...eligibleItems(),
-    ...(done() ? ["", "", ""] : []),
+    ...(done() ? [<div class="h-8 w-full" />, <div class="h-8 w-full" />, <div class="h-8 w-full" />] : []),
   ])
 
-  const MINIMUM_DELAY = 400
-  const [visibleCount, setVisibleCount] = createSignal(1)
+  const delay = createMemo(() => (done() ? 220 : 400))
+  const [visibleCount, setVisibleCount] = createSignal(eligibleItems().length)
 
   createEffect(() => {
     const total = finishedItems().length
     if (total > visibleCount()) {
       const timer = setTimeout(() => {
         setVisibleCount((prev) => prev + 1)
-      }, MINIMUM_DELAY)
+      }, delay())
       onCleanup(() => clearTimeout(timer))
     } else if (total < visibleCount()) {
       setVisibleCount(total)
@@ -66,43 +72,57 @@ export function MessageProgress(props: { assistantMessages: () => AssistantMessa
   })
 
   return (
-    <div
-      class="h-30 overflow-hidden pointer-events-none 
-             mask-alpha mask-t-from-33% mask-t-from-background-base mask-t-to-transparent
-             mask-b-from-90% mask-b-from-background-base mask-b-to-transparent"
-    >
+    <div class="flex flex-col gap-3">
       <div
-        class="w-full flex flex-col items-start self-stretch gap-2 py-8
-               transform transition-transform duration-500 ease-[cubic-bezier(0.22,1,0.36,1)]"
-        style={{ transform: `translateY(${translateY()})` }}
+        class="h-30 overflow-hidden pointer-events-none pb-1 
+               mask-alpha mask-t-from-33% mask-t-from-background-base mask-t-to-transparent
+               mask-b-from-95% mask-b-from-background-base mask-b-to-transparent"
       >
-        <For each={finishedItems()}>
-          {(part) => {
-            if (part && typeof part === "object" && "type" in part) {
-              const message = createMemo(() => sync.data.message[part.sessionID].find((m) => m.id === part.messageID))
-              return (
-                <div class="h-8 flex items-center w-full">
-                  <Switch>
-                    <Match when={part.type === "text" && part}>
-                      {(p) => (
-                        <div
-                          textContent={p().text}
-                          class="text-12-regular text-text-base whitespace-nowrap truncate w-full"
-                        />
-                      )}
-                    </Match>
-                    <Match when={part.type === "reasoning" && part}>
-                      {(p) => <Part message={message()!} part={p()} />}
-                    </Match>
-                    <Match when={part.type === "tool" && part}>{(p) => <Part message={message()!} part={p()} />}</Match>
-                  </Switch>
-                </div>
-              )
-            }
-            return <div class="h-8 flex items-center w-full">{part}</div>
-          }}
-        </For>
+        <div
+          class="w-full flex flex-col items-start self-stretch gap-2 py-8
+                 transform transition-transform duration-500 ease-[cubic-bezier(0.22,1,0.36,1)]"
+          style={{ transform: `translateY(${translateY()})` }}
+        >
+          <For each={finishedItems()}>
+            {(part) => {
+              if (part && typeof part === "object" && "type" in part) {
+                const message = createMemo(() => sync.data.message[part.sessionID].find((m) => m.id === part.messageID))
+                return (
+                  <div class="h-8 flex items-center w-full">
+                    <Switch>
+                      <Match when={part.type === "text" && part}>
+                        {(p) => (
+                          <div
+                            textContent={p().text}
+                            class="text-12-regular text-text-base whitespace-nowrap truncate w-full"
+                          />
+                        )}
+                      </Match>
+                      <Match when={part.type === "reasoning" && part}>
+                        {(p) => <Part message={message()!} part={p()} />}
+                      </Match>
+                      <Match when={part.type === "tool" && part}>
+                        {(p) => <Part message={message()!} part={p()} />}
+                      </Match>
+                    </Switch>
+                  </div>
+                )
+              }
+              return <div class="h-8 flex items-center w-full">{part}</div>
+            }}
+          </For>
+        </div>
       </div>
+      <Show when={currentText()}>
+        {(text) => (
+          <div
+            class="max-h-36 flex flex-col justify-end overflow-hidden py-3
+                   mask-alpha mask-t-from-80% mask-t-from-background-base mask-t-to-transparent"
+          >
+            <Markdown text={text()} class="w-full shrink-0 overflow-visible" />
+          </div>
+        )}
+      </Show>
     </div>
   )
 }
diff --git a/packages/desktop/src/components/prompt-input.tsx b/packages/desktop/src/components/prompt-input.tsx
index 10487612b..9b2c10df7 100644
--- a/packages/desktop/src/components/prompt-input.tsx
+++ b/packages/desktop/src/components/prompt-input.tsx
@@ -334,7 +334,7 @@ export const PromptInput: Component<PromptInputProps> = (props) => {
         onSubmit={handleSubmit}
         classList={{
           "bg-surface-raised-stronger-non-alpha border border-border-strong-base": true,
-          "rounded-2xl overflow-clip focus-within:shadow-xs-border-selected": true,
+          "rounded-2xl overflow-clip focus-within:border-transparent focus-within:shadow-xs-border-select": true,
           [props.class ?? ""]: !!props.class,
         }}
       >