feat: implement answer card

mxvsh · mxvsh · commit 2f1d330d5a01 · 2026-04-19T15:24:23.000+05:30
diff --git a/Wave.xcodeproj/project.pbxproj b/Wave.xcodeproj/project.pbxproj
@@ -184,6 +184,23 @@
 		};
 /* End PBXProject section */
 
+/* Begin PBXResourcesBuildPhase section */
+		3CA5A78D2F5063F600DBB766 /* Resources */ = {
+			isa = PBXResourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		7B4E00082F60000000DBB766 /* Resources */ = {
+			isa = PBXResourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+/* End PBXResourcesBuildPhase section */
+
 /* Begin PBXShellScriptBuildPhase section */
 		7B4E000C2F60000000DBB766 /* Generate whisper dSYM */ = {
 			isa = PBXShellScriptBuildPhase;
@@ -208,23 +225,6 @@
 		};
 /* End PBXShellScriptBuildPhase section */
 
-/* Begin PBXResourcesBuildPhase section */
-		3CA5A78D2F5063F600DBB766 /* Resources */ = {
-			isa = PBXResourcesBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
-		7B4E00082F60000000DBB766 /* Resources */ = {
-			isa = PBXResourcesBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
-/* End PBXResourcesBuildPhase section */
-
 /* Begin PBXSourcesBuildPhase section */
 		3CA5A78B2F5063F600DBB766 /* Sources */ = {
 			isa = PBXSourcesBuildPhase;
diff --git a/Wave/AppState.swift b/Wave/AppState.swift
@@ -362,6 +362,9 @@ final class AppState {
             return
         }
 
+        // Dismiss any visible answer card before starting a new session
+        overlayPanel?.hideAnswer()
+
         do {
             status = .recording
             showOverlay()
@@ -435,22 +438,51 @@ final class AppState {
 
         status = .idle
         overlayPanel?.setAIMode(false)
+        let wasAIMode = isAIMode
+        let hadSelection = selectedContext != nil
         isAIMode = false
         selectedContext = nil
         hideOverlayIfIdle()
 
         if let text = text, !text.isEmpty {
-            print("[wave] pasting: '\(text)'")
             historyManager.add(text)
-            try? await Task.sleep(for: .milliseconds(100))
-            PasteService.paste(text: text)
+
+            // Route:
+            // - Regular dictation → always paste
+            // - AI mode: paste only if there's an editable field to paste into.
+            //   Otherwise (reading webpage, PDF, etc.), show the answer card.
+            //   Selected text is just context for the LLM, not a routing signal.
+            let shouldShowCard = wasAIMode && !PasteService.hasEditableFocus()
+
+            if shouldShowCard {
+                print("[wave] showing answer card: '\(text)'")
+                overlayPanel?.showAnswer(
+                    text,
+                    onCopy: { [weak self] in
+                        self?.copyAnswerToClipboard(text)
+                    },
+                    onClose: { [weak self] in
+                        self?.overlayPanel?.hideAnswer()
+                    }
+                )
+            } else {
+                print("[wave] pasting: '\(text)' (hadSelection=\(hadSelection))")
+                try? await Task.sleep(for: .milliseconds(100))
+                PasteService.paste(text: text)
+            }
         } else {
             print("[wave] nothing to paste")
         }
 
         status = .idle
     }
 
+    private func copyAnswerToClipboard(_ text: String) {
+        let pb = NSPasteboard.general
+        pb.clearContents()
+        pb.setString(text, forType: .string)
+    }
+
     func verifyAndFetchGroqModels() async {
         guard !groqAPIKey.isEmpty else { groqAPIStatus = .unknown; return }
         groqAPIStatus = .checking
diff --git a/Wave/Services/PasteService.swift b/Wave/Services/PasteService.swift
@@ -1,4 +1,5 @@
 import AppKit
+import ApplicationServices
 import Carbon.HIToolbox
 
 struct PasteService {
@@ -95,4 +96,32 @@ struct PasteService {
 
         return text?.isEmpty == false ? text : nil
     }
+
+    /// Returns true if the focused UI element accepts text input
+    /// (e.g. user has cursor in a text field, text area, search bar).
+    static func hasEditableFocus() -> Bool {
+        let systemWide = AXUIElementCreateSystemWide()
+        var focusedElement: CFTypeRef?
+        guard AXUIElementCopyAttributeValue(systemWide, kAXFocusedUIElementAttribute as CFString, &focusedElement) == .success,
+              let element = focusedElement else { return false }
+        let axElement = element as! AXUIElement
+
+        // Primary signal: role suggests an editable field
+        var role: CFTypeRef?
+        if AXUIElementCopyAttributeValue(axElement, kAXRoleAttribute as CFString, &role) == .success,
+           let roleString = role as? String {
+            let editableRoles: Set<String> = [
+                "AXTextField",
+                "AXTextArea",
+                "AXComboBox",
+                "AXSearchField",
+            ]
+            if editableRoles.contains(roleString) { return true }
+        }
+
+        // Fallback: element lets us set its value (works for some browser inputs)
+        var settable: DarwinBoolean = false
+        AXUIElementIsAttributeSettable(axElement, kAXValueAttribute as CFString, &settable)
+        return settable.boolValue
+    }
 }
diff --git a/Wave/Views/AnswerCardView.swift b/Wave/Views/AnswerCardView.swift
@@ -0,0 +1,87 @@
+import SwiftUI
+
+struct AnswerCardView: View {
+    let text: String
+    let onCopy: () -> Void
+    let onClose: () -> Void
+
+    @State private var didCopy = false
+
+    private let cardWidth: CGFloat = 360
+    private let textMaxHeight: CGFloat = 220
+    // If text exceeds this many characters, use a scrolling view. Otherwise the card fits the text.
+    private let scrollThreshold = 400
+
+    var body: some View {
+        VStack(alignment: .leading, spacing: 10) {
+            if text.count > scrollThreshold {
+                ScrollView(.vertical, showsIndicators: true) {
+                    Text(text)
+                        .font(.system(size: 13))
+                        .foregroundStyle(.white.opacity(0.95))
+                        .textSelection(.enabled)
+                        .fixedSize(horizontal: false, vertical: true)
+                        .padding(.trailing, 4)
+                        .frame(maxWidth: .infinity, alignment: .leading)
+                }
+                .frame(height: textMaxHeight)
+            } else {
+                Text(text)
+                    .font(.system(size: 13))
+                    .foregroundStyle(.white.opacity(0.95))
+                    .textSelection(.enabled)
+                    .fixedSize(horizontal: false, vertical: true)
+                    .frame(maxWidth: .infinity, alignment: .leading)
+            }
+
+            HStack(spacing: 6) {
+                Button(action: {
+                    onCopy()
+                    withAnimation(.easeInOut(duration: 0.15)) { didCopy = true }
+                    DispatchQueue.main.asyncAfter(deadline: .now() + 1.2) {
+                        withAnimation(.easeInOut(duration: 0.2)) { didCopy = false }
+                    }
+                }) {
+                    HStack(spacing: 4) {
+                        Image(systemName: didCopy ? "checkmark" : "doc.on.doc")
+                            .font(.system(size: 10, weight: .semibold))
+                        Text(didCopy ? "Copied" : "Copy")
+                            .font(.system(size: 11, weight: .medium))
+                    }
+                    .foregroundStyle(.white.opacity(0.9))
+                    .padding(.horizontal, 8)
+                    .padding(.vertical, 4)
+                    .background(
+                        RoundedRectangle(cornerRadius: 6)
+                            .fill(.white.opacity(0.08))
+                    )
+                }
+                .buttonStyle(.plain)
+
+                Spacer()
+
+                Button(action: onClose) {
+                    Image(systemName: "xmark")
+                        .font(.system(size: 10, weight: .semibold))
+                        .foregroundStyle(.white.opacity(0.7))
+                        .frame(width: 22, height: 22)
+                        .background(
+                            RoundedRectangle(cornerRadius: 6)
+                                .fill(.white.opacity(0.08))
+                        )
+                }
+                .buttonStyle(.plain)
+            }
+        }
+        .padding(12)
+        .frame(width: cardWidth)
+        .background(
+            RoundedRectangle(cornerRadius: 14)
+                .fill(Color(red: 0.13, green: 0.13, blue: 0.13).opacity(0.95))
+                .overlay(
+                    RoundedRectangle(cornerRadius: 14)
+                        .stroke(.white.opacity(0.1), lineWidth: 0.5)
+                )
+        )
+    }
+}
diff --git a/Wave/Views/OverlayPanel.swift b/Wave/Views/OverlayPanel.swift
@@ -1,5 +1,6 @@
 import AppKit
 import QuartzCore
+import SwiftUI
 
 // Idle: tiny pill. Active: expanded wave container.
 private let kIdleW:    CGFloat = 44
@@ -221,6 +222,8 @@ final class OverlayPanel: NSPanel {
     private var baseCenterX: CGFloat = 0
     private var shortcutLabel: String = ""
     private var tooltipTimer: Timer?
+    private var answerHost: NSHostingView<AnswerCardView>?
+    private(set) var isShowingAnswer = false
 
     var onMouseDown: (() -> Void)? {
         get { drawView.onMouseDown }
@@ -272,6 +275,9 @@ final class OverlayPanel: NSPanel {
     }
 
     func updateStatus(_ status: AppStatus) {
+        // Don't overwrite the card while it's showing
+        if isShowingAnswer { return }
+
         drawView.status = status
 
         if !isVisible {
@@ -297,7 +303,9 @@ final class OverlayPanel: NSPanel {
         let targetFrame = NSRect(x: baseCenterX - w / 2, y: baseY, width: w, height: h)
 
         DispatchQueue.main.async { [weak self] in
-            self?.setFrame(targetFrame, display: true, animate: true)
+            // Re-check — the answer card may have taken over in the meantime
+            guard let self, !self.isShowingAnswer else { return }
+            self.setFrame(targetFrame, display: true, animate: true)
         }
     }
 
@@ -329,6 +337,61 @@ final class OverlayPanel: NSPanel {
         drawView.isAIMode = enabled
     }
 
+    // MARK: - Answer card morph
+
+    func showAnswer(_ text: String, onCopy: @escaping () -> Void, onClose: @escaping () -> Void) {
+        // Dismiss tooltip if visible
+        tooltipTimer?.invalidate()
+        tooltipTimer = nil
+        tooltipPanel.hide()
+
+        // Make sure baseCenterX/baseY are current
+        if baseCenterX == 0 { positionOnScreen() }
+
+        let card = AnswerCardView(text: text, onCopy: onCopy, onClose: onClose)
+        let host = NSHostingView(rootView: card)
+
+        // Force SwiftUI to lay out and compute an honest intrinsic size
+        host.layoutSubtreeIfNeeded()
+        let fitSize = host.fittingSize
+        let cardW: CGFloat = max(360, fitSize.width)
+        let cardH: CGFloat = min(320, max(90, fitSize.height))
+        host.frame = NSRect(x: 0, y: 0, width: cardW, height: cardH)
+
+        contentView = host
+        answerHost = host
+        isShowingAnswer = true
+        ignoresMouseEvents = false
+
+        // Anchor bottom-center; grow upward from pill's base Y
+        let targetFrame = NSRect(
+            x: baseCenterX - cardW / 2,
+            y: baseY,
+            width: cardW,
+            height: cardH
+        )
+        if !isVisible { orderFrontRegardless() }
+        setFrame(targetFrame, display: true, animate: true)
+    }
+
+    func hideAnswer() {
+        guard isShowingAnswer else { return }
+        isShowingAnswer = false
+        answerHost = nil
+
+        drawView.frame = NSRect(x: 0, y: 0, width: kIdleW, height: kIdleH)
+        drawView.alphaValue = 1
+        contentView = drawView
+
+        let targetFrame = NSRect(
+            x: baseCenterX - kIdleW / 2,
+            y: baseY,
+            width: kIdleW,
+            height: kIdleH
+        )
+        setFrame(targetFrame, display: true)
+    }
+
     @objc private func screenDidChange() { positionOnScreen() }
 
     private func positionOnScreen() {
@@ -341,6 +404,14 @@ final class OverlayPanel: NSPanel {
             ? visibleFrame.minY + 8
             : screenFrame.minY + 12
 
+        if isShowingAnswer {
+            // Re-center the card at new baseCenterX, keep its current size
+            let w = frame.width
+            let h = frame.height
+            setFrame(NSRect(x: baseCenterX - w / 2, y: baseY, width: w, height: h), display: true)
+            return
+        }
+
         let w = frame.width > kIdleW ? kActiveW : kIdleW
         let h = frame.height > kIdleH ? kActiveH : kIdleH
         setFrame(NSRect(x: baseCenterX - w / 2, y: baseY, width: w, height: h), display: true)
diff --git a/Wave/waveApp.swift b/Wave/waveApp.swift
@@ -79,11 +79,7 @@ struct WaveApp: App {
 
             Button("Settings...") {
                 appState.pendingNavSelection = .general
-                openWindow(id: "main")
-                NSApp.activate(ignoringOtherApps: true)
-                DispatchQueue.main.asyncAfter(deadline: .now() + 0.15) {
-                    NSApp.activate(ignoringOtherApps: true)
-                }
+                openMainWindow()
             }
             .keyboardShortcut(",", modifiers: .command)
 
@@ -98,6 +94,27 @@ struct WaveApp: App {
                 .renderingMode(.template)
         }
     }
+
+    /// Reliably shows the main window from MenuBarExtra.
+    /// Works around SwiftUI `Window` scene + `openWindow(id:)` re-open bugs
+    /// by finding and re-showing an existing window via AppKit first,
+    /// falling back to `openWindow` only for the very first launch.
+    private func openMainWindow() {
+        // Find an existing window for the main scene (may be hidden/closed)
+        let existing = NSApp.windows.first { window in
+            // SwiftUI-created window titles match the Scene title; also check identifier
+            window.title == "Wave" || window.identifier?.rawValue.contains("main") == true
+        }
+
+        if let window = existing {
+            if window.isMiniaturized { window.deminiaturize(nil) }
+            window.makeKeyAndOrderFront(nil)
+        } else {
+            openWindow(id: "main")
+        }
+
+        NSApp.activate(ignoringOtherApps: true)
+    }
 }
 
 final class AppDelegate: NSObject, NSApplicationDelegate {
