jake/vOOice
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #305 from nulinspiratie/fix/clipboard-context

Browse Source 
Fix: Enable Clipboard context with custom AI enhancers
This commit is contained in:
Prakash Joshi Pax 2025-09-24 07:44:41 +05:45 committed by GitHub
commit 154748c1aa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 26 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
VoiceInk/Services/AIEnhancementService.swift
View File

@ -132,6 +132,7 @@ class AIEnhancementService: ObservableObject {
}
private func getSystemMessage(for mode: EnhancementPrompt) -> String {
let clipboardSnapshot = NSPasteboard.general.string(forType: .string)
let selectedText = SelectedTextService.fetchSelectedText()
if let activePrompt = activePrompt,
@ -149,7 +150,7 @@ class AIEnhancementService: ObservableObject {
}
let clipboardContext = if useClipboardContext,
let clipboardText = NSPasteboard.general.string(forType: .string),
let clipboardText = clipboardSnapshot ?? NSPasteboard.general.string(forType: .string),
!clipboardText.isEmpty {
"\n\n<CLIPBOARD_CONTEXT>\n\(clipboardText)\n</CLIPBOARD_CONTEXT>"
} else {

42
VoiceInk/Services/SelectedTextService.swift
View File

@ -1,8 +1,6 @@
import Foundation
import AppKit
class SelectedTextService {
// Private pasteboard type to avoid clipboard history pollution
private static let privatePasteboardType = NSPasteboard.PasteboardType("com.prakashjoshipax.VoiceInk.transient")
@ -14,10 +12,15 @@ class SelectedTextService {
}
let pasteboard = NSPasteboard.general
// Save original clipboard content
let originalClipboardText = pasteboard.string(forType: .string)
// Save original clipboard content (all UTIs with their data)
let originalPasteboardItems = pasteboard.pasteboardItems?.map { item in
(item.types, item.data(forType: item.types.first ?? .string))
item.types.reduce(into: [NSPasteboard.PasteboardType: Data]()) { acc, type in
if let data = item.data(forType: type) {
acc[type] = data
}
}
}
// Clear clipboard to prepare for selection detection
@ -43,24 +46,27 @@ class SelectedTextService {
// Read the copied text
let selectedText = pasteboard.string(forType: .string)
// Restore original clipboard content
pasteboard.clearContents()
if let originalItems = originalPasteboardItems {
for (types, data) in originalItems {
if let data = data {
let pasteboardItem = NSPasteboardItem()
pasteboardItem.setData(data, forType: types.first ?? .string)
pasteboard.writeObjects([pasteboardItem])
if let originalItems = originalPasteboardItems, !originalItems.isEmpty {
let restoredItems: [NSPasteboardItem] = originalItems.compactMap { dataMap in
guard !dataMap.isEmpty else { return nil }
let item = NSPasteboardItem()
for (type, data) in dataMap {
item.setData(data, forType: type)
}
return item
}
if !restoredItems.isEmpty {
pasteboard.writeObjects(restoredItems)
} else if let originalClipboardText {
_ = pasteboard.setString(originalClipboardText, forType: .string)
}
} else if let originalClipboardText {
_ = pasteboard.setString(originalClipboardText, forType: .string)
}
// Clear clipboard history by writing transient data
let transientItem = NSPasteboardItem()
transientItem.setString("", forType: privatePasteboardType)
pasteboard.writeObjects([transientItem])
return selectedText
}
}
}