Restore undelivered user message as draft on reconnect failure

When the app is killed after a user message is persisted but before the
service is created on the Sprite, reconnectIfNeeded would silently return
leaving a dangling user bubble with no response. Now it detects a trailing
user message and restores it to the input box as a draft instead.

Also handles the edge case where lastSessionComplete is true but a new user
message was appended before saveSession(isComplete:false) ran.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: mcintyre94

Wisp/ViewModels/ChatViewModel.swift

@@ -642,19 +642,41 @@ final class ChatViewModel {
         // If the last session completed cleanly, content is already loaded from
         // persistence — no need to hit the network at all.
         if let chat = fetchChat(modelContext: modelContext), chat.lastSessionComplete {
+            // Edge case: app killed between persistMessages and saveSession(isComplete:false).
+            // The session was previously complete but a new user message was appended and
+            // persisted before the service was created. Restore it as a draft.
+            restoreUndeliveredDraft(modelContext: modelContext)
             return
         }
 
         streamTask = Task {
             // Only reconnect if the service exists (running or stopped with logs)
             guard let serviceInfo = try? await apiClient.getServiceStatus(spriteName: spriteName, serviceName: serviceName)
-            else { return }
+            else {
+                // Service was never created (e.g. app killed before the service call).
+                // Restore any trailing user message as a draft rather than leaving a
+                // stale bubble with no response.
+                restoreUndeliveredDraft(modelContext: modelContext)
+                return
+            }
 
             let alreadyStopped = serviceInfo.state.status != "running"
             await reconnectToServiceLogs(apiClient: apiClient, modelContext: modelContext, serviceAlreadyStopped: alreadyStopped)
         }
     }
 
+    /// If the last message is a user message with no response, remove it from history
+    /// and restore its text to the input box as a draft.
+    func restoreUndeliveredDraft(modelContext: ModelContext) {
+        guard let last = messages.last, last.role == .user else { return }
+        let text = last.textContent
+        messages.removeLast()
+        persistMessages(modelContext: modelContext)
+        guard !text.isEmpty, inputText.isEmpty else { return }
+        inputText = text
+        saveDraft(modelContext: modelContext)
+    }
+
     // MARK: - Private
 
     private func executeClaudeCommand(

WispTests/ChatViewModelTests.swift

@@ -1296,4 +1296,53 @@ struct ChatViewModelTests {
         #expect(vm.attachedFiles.count == 1)
         #expect(vm.attachedFiles[0].name == "live.py")
     }
+
+    // MARK: - restoreUndeliveredDraft
+
+    @Test func restoreUndeliveredDraft_removesTrailingUserMessageAndRestoresInputText() throws {
+        let ctx = try makeModelContext()
+        let (vm, _) = makeChatViewModel(modelContext: ctx)
+
+        vm.messages = [
+            ChatMessage(role: .user, content: [.text("first message")]),
+            ChatMessage(role: .assistant, content: [.text("response")]),
+            ChatMessage(role: .user, content: [.text("unsent message")]),
+        ]
+
+        vm.restoreUndeliveredDraft(modelContext: ctx)
+
+        #expect(vm.messages.count == 2)
+        #expect(vm.inputText == "unsent message")
+    }
+
+    @Test func restoreUndeliveredDraft_isNoopWhenLastMessageIsAssistant() throws {
+        let ctx = try makeModelContext()
+        let (vm, _) = makeChatViewModel(modelContext: ctx)
+
+        vm.messages = [
+            ChatMessage(role: .user, content: [.text("hello")]),
+            ChatMessage(role: .assistant, content: [.text("hi there")]),
+        ]
+
+        vm.restoreUndeliveredDraft(modelContext: ctx)
+
+        #expect(vm.messages.count == 2)
+        #expect(vm.inputText == "")
+    }
+
+    @Test func restoreUndeliveredDraft_doesNotOverwriteExistingInputText() throws {
+        let ctx = try makeModelContext()
+        let (vm, _) = makeChatViewModel(modelContext: ctx)
+
+        vm.messages = [
+            ChatMessage(role: .user, content: [.text("unsent message")]),
+        ]
+        vm.inputText = "already typing something"
+
+        vm.restoreUndeliveredDraft(modelContext: ctx)
+
+        // Message removed but inputText NOT overwritten
+        #expect(vm.messages.count == 0)
+        #expect(vm.inputText == "already typing something")
+    }
 }