Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
83b127cf44 | ||
|
|
8332661504 | ||
|
|
832a18b06c | ||
|
|
6e5796e16d | ||
|
|
7e9cc5b269 | ||
|
|
a7cad2fc2f | ||
|
|
69e1a7fccb | ||
|
|
e22bcb99c4 | ||
|
|
457e303152 | ||
|
|
345f306edf | ||
|
|
45e4962ec3 | ||
|
|
5f137f89bf | ||
|
|
b998ea210e | ||
|
|
86826f7ecc | ||
|
|
969cda6f14 | ||
|
|
de642ad692 | ||
|
|
8b27263962 | ||
|
|
1961d0dc9c | ||
|
|
41cb06dcb5 | ||
|
|
52e36b5c5a | ||
|
|
a74a86d001 | ||
|
|
f50057d951 | ||
|
|
16b876807d | ||
|
|
6a2c8df180 | ||
|
|
c9447d2a8b | ||
|
|
3fdc2fca95 | ||
|
|
590be2356f | ||
|
|
af32a36a91 | ||
|
|
1327b61e2b |
@@ -25,11 +25,14 @@ No network calls, no API keys — everything runs locally.
|
||||
## Install with Homebrew
|
||||
|
||||
```bash
|
||||
brew tap TZGyn/tap
|
||||
brew trust TZGyn/tap # Homebrew 6+ requires trusting third-party taps
|
||||
brew tap TZGyn/tap https://gitea.tzgyn.dev/TZGyn/homebrew-tap.git
|
||||
brew trust TZGyn/tap
|
||||
brew install --cask kapture
|
||||
```
|
||||
|
||||
The tap is hosted on Gitea, so the full URL is required. Homebrew may print
|
||||
an "Unsupported tap" warning — it's harmless.
|
||||
|
||||
Releases are published as zips on GitHub Releases; `release.sh` packages a
|
||||
new one (`./release.sh 1.0.1 && git tag v1.0.1 && git push origin v1.0.1`).
|
||||
The cask lives in the [TZGyn/homebrew-tap](https://github.com/TZGyn/homebrew-tap)
|
||||
|
||||
@@ -21,6 +21,17 @@ final class AppDelegate: NSObject, NSApplicationDelegate {
|
||||
runOCRSelfTest()
|
||||
return
|
||||
}
|
||||
// Single instance: stale copies keep their own hotkey + overlays alive
|
||||
// after app updates and eat clicks meant for the new instance.
|
||||
let myPid = ProcessInfo.processInfo.processIdentifier
|
||||
let others = NSRunningApplication.runningApplications(withBundleIdentifier: "dev.tzgyn.kapture")
|
||||
.filter { $0.processIdentifier != myPid }
|
||||
if !others.isEmpty {
|
||||
CaptureController.log("launch: other Kapture instances pids=\(others.map { String($0.processIdentifier) }.joined(separator: ",")) — terminating them")
|
||||
for other in others {
|
||||
other.forceTerminate()
|
||||
}
|
||||
}
|
||||
HotkeyManager.shared.onTrigger = { CaptureController.shared.begin() }
|
||||
HotkeyManager.shared.register()
|
||||
}
|
||||
@@ -48,6 +59,13 @@ struct MenuView: View {
|
||||
|
||||
Divider()
|
||||
|
||||
Button("Copy Diagnostics") {
|
||||
let diag = CaptureController.shared.diagnostics()
|
||||
let pasteboard = NSPasteboard.general
|
||||
pasteboard.clearContents()
|
||||
pasteboard.setString(diag, forType: .string)
|
||||
}
|
||||
|
||||
Button(LoginItem.isEnabled ? "Disable Launch at Login" : "Launch at Login") {
|
||||
if !LoginItem.toggle() {
|
||||
let alert = NSAlert()
|
||||
@@ -72,6 +90,13 @@ enum PermissionManager {
|
||||
static func request() {
|
||||
CGRequestScreenCaptureAccess()
|
||||
}
|
||||
|
||||
static var isAccessibilityGranted: Bool { AXIsProcessTrusted() }
|
||||
|
||||
static func requestAccessibility() {
|
||||
let opts = [kAXTrustedCheckOptionPrompt.takeUnretainedValue(): true] as CFDictionary
|
||||
_ = AXIsProcessTrustedWithOptions(opts)
|
||||
}
|
||||
}
|
||||
|
||||
enum LoginItem {
|
||||
|
||||
@@ -1,15 +1,80 @@
|
||||
import AppKit
|
||||
import CoreGraphics
|
||||
import ScreenCaptureKit
|
||||
import ApplicationServices
|
||||
|
||||
final class CaptureController: NSObject {
|
||||
static let shared = CaptureController()
|
||||
|
||||
private var overlays: [NSWindow] = []
|
||||
private var overlays: [OverlayWindow] = []
|
||||
private(set) var isSelecting = false
|
||||
private var previousApp: NSRunningApplication?
|
||||
private var overlayIDs: [Int] = []
|
||||
|
||||
// Selection state (driven by the event tap; AppKit global coordinates)
|
||||
private var marqueeStart: NSPoint?
|
||||
private(set) var marqueeRect: NSRect?
|
||||
private(set) var hoverFrame: NSRect?
|
||||
private var trackedWindows: [(id: CGWindowID, frame: NSRect)] = []
|
||||
private var lastWindowScan = Date.distantPast
|
||||
private var eventTap: CFMachPort?
|
||||
|
||||
/// Session event log for debugging (dumped via Copy Diagnostics).
|
||||
/// Also appended to /tmp/kapture-debug.log so it survives relaunches.
|
||||
static var sessionLog: [String] = []
|
||||
static func log(_ line: String) {
|
||||
let stamped = String(format: "%@ %@", DateFormatter.localizedString(from: Date(), dateStyle: .none, timeStyle: .medium), line)
|
||||
sessionLog.append(stamped)
|
||||
if sessionLog.count > 200 { sessionLog.removeFirst() }
|
||||
NSLog("[Kapture] \(line)")
|
||||
let path = "/tmp/kapture-debug.log"
|
||||
if let handle = FileHandle(forWritingAtPath: path) {
|
||||
handle.seekToEndOfFile()
|
||||
handle.write((stamped + "\n").data(using: .utf8)!)
|
||||
handle.closeFile()
|
||||
} else {
|
||||
FileManager.default.createFile(atPath: path, contents: (stamped + "\n").data(using: .utf8))
|
||||
}
|
||||
}
|
||||
|
||||
/// Diagnostic snapshot for "Copy Diagnostics" menu item.
|
||||
func diagnostics() -> String {
|
||||
var lines: [String] = []
|
||||
lines.append("Kapture diagnostics — version \(Bundle.main.infoDictionary?["CFBundleShortVersionString"] ?? "?")")
|
||||
lines.append("accessibility=\(AXIsProcessTrusted()) eventTap=\(eventTap != nil)")
|
||||
lines.append("--- session log (newest last) ---")
|
||||
lines.append(contentsOf: Self.sessionLog)
|
||||
lines.append("--- screens ---")
|
||||
for (i, s) in NSScreen.screens.enumerated() {
|
||||
let scale = s.backingScaleFactor
|
||||
let cgID = (s.deviceDescription[NSDeviceDescriptionKey("NSScreenNumber")] as? NSNumber)?.uint32Value ?? 0
|
||||
lines.append("screen[\(i)]: frame=\(s.frame) scale=\(scale) backingSize=\(s.frame.width*scale)x\(s.frame.height*scale) cgID=\(cgID)")
|
||||
}
|
||||
if #available(macOS 14.0, *) {
|
||||
Task {
|
||||
if let content = try? await SCShareableContent.current {
|
||||
for (i, d) in content.displays.enumerated() {
|
||||
lines.append("SCDisplay[\(i)]: frame=\(d.frame) width=\(d.width)x\(d.height) id=\(d.displayID)")
|
||||
}
|
||||
} else {
|
||||
lines.append("SCShareableContent: unavailable (no Screen Recording permission?)")
|
||||
}
|
||||
let text = lines.joined(separator: "\n")
|
||||
let pb = NSPasteboard.general
|
||||
pb.clearContents()
|
||||
pb.setString(text, forType: .string)
|
||||
}
|
||||
} else {
|
||||
let text = lines.joined(separator: "\n")
|
||||
let pb = NSPasteboard.general
|
||||
pb.clearContents()
|
||||
pb.setString(text, forType: .string)
|
||||
}
|
||||
return lines.joined(separator: "\n")
|
||||
}
|
||||
|
||||
// MARK: - Session lifecycle
|
||||
|
||||
func begin() {
|
||||
guard !isSelecting else { return }
|
||||
guard PermissionManager.isGranted else {
|
||||
@@ -18,20 +83,34 @@ final class CaptureController: NSObject {
|
||||
}
|
||||
isSelecting = true
|
||||
previousApp = NSWorkspace.shared.frontmostApplication
|
||||
if #available(macOS 14.0, *) {
|
||||
NSApp.activate()
|
||||
} else {
|
||||
NSApp.activate(ignoringOtherApps: true)
|
||||
// Activation is async on macOS 26; wait a tick so the app is active
|
||||
// before the overlays appear (otherwise clicks pass through).
|
||||
NSApp.activate(ignoringOtherApps: true)
|
||||
CaptureController.log("begin: activate issued, active=\(NSApp.isActive)")
|
||||
DispatchQueue.main.asyncAfter(deadline: .now() + 0.05) { [weak self] in
|
||||
guard let self, self.isSelecting else { return }
|
||||
CaptureController.log("begin: active=\(NSApp.isActive) — showing overlays")
|
||||
self.overlays = NSScreen.screens.map { screen in
|
||||
let panel = OverlayWindow(screen: screen, controller: self)
|
||||
(panel.contentView as? OverlayView)?.controller = self
|
||||
return panel
|
||||
}
|
||||
self.overlays.forEach { $0.makeKeyAndOrderFront(nil) }
|
||||
self.overlayIDs = self.overlays.map { $0.windowNumber }
|
||||
CaptureController.log("begin: \(self.overlays.count) overlays, ids=\(self.overlayIDs), active=\(NSApp.isActive)")
|
||||
self.installEventTap()
|
||||
}
|
||||
overlays = NSScreen.screens.map { screen in
|
||||
OverlayWindow(
|
||||
screen: screen,
|
||||
onComplete: { [weak self] rect in self?.finishSelection(rect) },
|
||||
onCancel: { [weak self] in self?.cancelSelection() }
|
||||
)
|
||||
}
|
||||
|
||||
/// A window was picked from hover — capture it with rounded corners.
|
||||
private func finishWindowPick(_ rect: NSRect) {
|
||||
let ids = overlayIDs
|
||||
hideOverlays()
|
||||
isSelecting = false
|
||||
restorePreviousApp()
|
||||
DispatchQueue.main.async { [self] in
|
||||
capture(rect, excludingWindowIDs: ids, roundedCorners: true)
|
||||
}
|
||||
overlays.forEach { $0.makeKeyAndOrderFront(nil) }
|
||||
overlayIDs = overlays.map { $0.windowNumber }
|
||||
}
|
||||
|
||||
private func finishSelection(_ rect: NSRect) {
|
||||
@@ -53,8 +132,12 @@ final class CaptureController: NSObject {
|
||||
}
|
||||
|
||||
private func hideOverlays() {
|
||||
removeEventTap()
|
||||
overlays.forEach { $0.orderOut(nil) }
|
||||
overlays = []
|
||||
marqueeRect = nil
|
||||
hoverFrame = nil
|
||||
marqueeStart = nil
|
||||
}
|
||||
|
||||
private func restorePreviousApp() {
|
||||
@@ -62,8 +145,191 @@ final class CaptureController: NSObject {
|
||||
previousApp = nil
|
||||
}
|
||||
|
||||
private func capture(_ rect: NSRect, excludingWindowIDs ids: [Int]) {
|
||||
ScreenCapture.shared.capture(rect, excludingWindowIDs: ids) { [weak self] image in
|
||||
private func renderOverlays() {
|
||||
for overlay in overlays {
|
||||
(overlay.contentView as? OverlayView)?.needsDisplay = true
|
||||
}
|
||||
}
|
||||
|
||||
// MARK: - Event tap
|
||||
|
||||
private func installEventTap() {
|
||||
guard eventTap == nil else { return }
|
||||
guard AXIsProcessTrusted() else {
|
||||
CaptureController.log("tap: accessibility not granted — falling back to panel events")
|
||||
PermissionManager.requestAccessibility()
|
||||
return
|
||||
}
|
||||
let mask: CGEventMask = (1 << CGEventType.mouseMoved.rawValue)
|
||||
| (1 << CGEventType.leftMouseDown.rawValue)
|
||||
| (1 << CGEventType.leftMouseDragged.rawValue)
|
||||
| (1 << CGEventType.leftMouseUp.rawValue)
|
||||
| (1 << CGEventType.keyDown.rawValue)
|
||||
guard let tap = CGEvent.tapCreate(
|
||||
tap: .cghidEventTap,
|
||||
place: .headInsertEventTap,
|
||||
options: .defaultTap,
|
||||
eventsOfInterest: CGEventMask(mask),
|
||||
callback: Self.tapCallback,
|
||||
userInfo: Unmanaged.passUnretained(self).toOpaque()
|
||||
) else {
|
||||
CaptureController.log("tap: creation FAILED — falling back to panel events")
|
||||
return
|
||||
}
|
||||
eventTap = tap
|
||||
if let source = CFMachPortCreateRunLoopSource(nil, tap, 0) {
|
||||
CFRunLoopAddSource(CFRunLoopGetMain(), source, .commonModes)
|
||||
}
|
||||
CGEvent.tapEnable(tap: tap, enable: true)
|
||||
CaptureController.log("tap: installed")
|
||||
}
|
||||
|
||||
private func removeEventTap() {
|
||||
guard let tap = eventTap else { return }
|
||||
CGEvent.tapEnable(tap: tap, enable: false)
|
||||
if let source = CFMachPortCreateRunLoopSource(nil, tap, 0) {
|
||||
CFRunLoopRemoveSource(CFRunLoopGetMain(), source, .commonModes)
|
||||
}
|
||||
CFMachPortInvalidate(tap)
|
||||
eventTap = nil
|
||||
CaptureController.log("tap: removed")
|
||||
}
|
||||
|
||||
private static let tapCallback: CGEventTapCallBack = { _, type, event, refcon in
|
||||
guard let refcon else { return Unmanaged.passUnretained(event) }
|
||||
let controller = Unmanaged<CaptureController>.fromOpaque(refcon).takeUnretainedValue()
|
||||
if let result = controller.handleTap(type, event) {
|
||||
return Unmanaged.passUnretained(result)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
/// Convert a CG global point (top-left origin) to AppKit global.
|
||||
private func cgPointToAppKit(_ p: NSPoint) -> NSPoint {
|
||||
guard let mainScreen = NSScreen.screens.first else { return p }
|
||||
return NSPoint(x: p.x, y: mainScreen.frame.height - p.y)
|
||||
}
|
||||
|
||||
/// Scan on-screen windows (AppKit global frames), front to back.
|
||||
private func refreshTrackedWindows() {
|
||||
guard let list = CGWindowListCopyWindowInfo(
|
||||
[ .optionOnScreenOnly, .excludeDesktopElements ], kCGNullWindowID
|
||||
) as? [[String: Any]] else { trackedWindows = []; return }
|
||||
|
||||
let ownIDs = Set(overlayIDs)
|
||||
var found: [(id: CGWindowID, frame: NSRect)] = []
|
||||
for info in list {
|
||||
guard let cid = info[kCGWindowNumber as String] as? CGWindowID,
|
||||
let bounds = info[kCGWindowBounds as String] as? [String: CGFloat],
|
||||
let layer = info[kCGWindowLayer as String] as? Int,
|
||||
layer == 0
|
||||
else { continue }
|
||||
if ownIDs.contains(Int(cid)) { continue }
|
||||
let cgRect = NSRect(
|
||||
x: bounds["X"] ?? 0,
|
||||
y: bounds["Y"] ?? 0,
|
||||
width: bounds["Width"] ?? 0,
|
||||
height: bounds["Height"] ?? 0
|
||||
)
|
||||
guard cgRect.width >= 80, cgRect.height >= 60 else { continue }
|
||||
let rect = Self.cgToAppKit(rect: cgRect)
|
||||
found.append((cid, rect))
|
||||
}
|
||||
trackedWindows = found
|
||||
}
|
||||
|
||||
/// Convert a CGWindowList (top-left origin) frame into AppKit screen
|
||||
/// coordinates (bottom-left origin).
|
||||
private static func cgToAppKit(rect: NSRect) -> NSRect {
|
||||
guard let mainScreen = NSScreen.screens.first else { return rect }
|
||||
let y = mainScreen.frame.height - rect.origin.y - rect.height
|
||||
return NSRect(x: rect.origin.x, y: y, width: rect.width, height: rect.height)
|
||||
}
|
||||
|
||||
private func topmostWindowContaining(_ appKitPoint: NSPoint) -> NSRect? {
|
||||
for (_, frame) in trackedWindows {
|
||||
if frame.contains(appKitPoint) {
|
||||
return frame
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
/// Returns the event to pass through, or nil to consume it.
|
||||
func handleTap(_ type: CGEventType, _ event: CGEvent) -> CGEvent? {
|
||||
guard isSelecting else { return event }
|
||||
|
||||
switch type {
|
||||
case .tapDisabledByUserInput, .tapDisabledByTimeout:
|
||||
// A timed-out tap must be re-enabled; guard against a torn-down tap.
|
||||
if let tap = eventTap {
|
||||
CGEvent.tapEnable(tap: tap, enable: true)
|
||||
CaptureController.log("tap: re-enabled after \(type)")
|
||||
}
|
||||
return event
|
||||
case .keyDown:
|
||||
if event.getIntegerValueField(.keyboardEventKeycode) == 53 {
|
||||
CaptureController.log("tap: Esc — cancel")
|
||||
cancelSelection()
|
||||
return nil
|
||||
}
|
||||
return event
|
||||
case .mouseMoved, .leftMouseDragged:
|
||||
// Throttle window scans while moving
|
||||
if Date().timeIntervalSince(lastWindowScan) > 0.15 {
|
||||
lastWindowScan = Date()
|
||||
refreshTrackedWindows()
|
||||
}
|
||||
let appKitPoint = cgPointToAppKit(NSPoint(x: event.location.x, y: event.location.y))
|
||||
if type == .mouseMoved {
|
||||
let frame = topmostWindowContaining(appKitPoint)
|
||||
if frame != hoverFrame {
|
||||
hoverFrame = frame
|
||||
renderOverlays()
|
||||
}
|
||||
return event
|
||||
}
|
||||
// Drag: grow the marquee
|
||||
guard let start = marqueeStart else { return nil }
|
||||
let rect = NSRect(
|
||||
x: min(start.x, appKitPoint.x),
|
||||
y: min(start.y, appKitPoint.y),
|
||||
width: abs(appKitPoint.x - start.x),
|
||||
height: abs(appKitPoint.y - start.y)
|
||||
)
|
||||
marqueeRect = rect
|
||||
hoverFrame = nil
|
||||
renderOverlays()
|
||||
return nil
|
||||
case .leftMouseDown:
|
||||
marqueeStart = cgPointToAppKit(NSPoint(x: event.location.x, y: event.location.y))
|
||||
marqueeRect = nil
|
||||
CaptureController.log("tap: mouseDown at \(marqueeStart.map { "\($0)" } ?? "?")")
|
||||
return nil
|
||||
case .leftMouseUp:
|
||||
defer { marqueeStart = nil; marqueeRect = nil }
|
||||
let appKitPoint = cgPointToAppKit(NSPoint(x: event.location.x, y: event.location.y))
|
||||
let dragDistance = marqueeStart.map { hypot($0.x - appKitPoint.x, $0.y - appKitPoint.y) } ?? 0
|
||||
if let rect = marqueeRect, dragDistance >= 4, rect.width >= 2, rect.height >= 2 {
|
||||
CaptureController.log("tap: marquee complete \(rect)")
|
||||
finishSelection(rect)
|
||||
} else if let hover = hoverFrame {
|
||||
CaptureController.log("tap: window pick \(hover)")
|
||||
finishWindowPick(hover)
|
||||
} else {
|
||||
CaptureController.log("tap: click with no hover/marquee — cancel")
|
||||
cancelSelection()
|
||||
}
|
||||
return nil
|
||||
default:
|
||||
return event
|
||||
}
|
||||
}
|
||||
|
||||
// MARK: - Capture
|
||||
|
||||
private func capture(_ rect: NSRect, excludingWindowIDs ids: [Int], roundedCorners: Bool = false) {
|
||||
ScreenCapture.shared.capture(rect, excludingWindowIDs: ids, roundedCorners: roundedCorners) { [weak self] image in
|
||||
DispatchQueue.main.async {
|
||||
self?.isSelecting = false
|
||||
self?.restorePreviousApp()
|
||||
@@ -72,6 +338,7 @@ final class CaptureController: NSObject {
|
||||
ResultWindow.shared.show(image: ResultWindow.failureImage("Capture failed. Check Screen Recording permission."))
|
||||
return
|
||||
}
|
||||
CaptureController.log("capture done: \(image.width)x\(image.height) — showing result window")
|
||||
ResultWindow.shared.show(image: image)
|
||||
}
|
||||
}
|
||||
@@ -85,22 +352,19 @@ final class ScreenCapture {
|
||||
private init() {}
|
||||
|
||||
@available(macOS 14.0, *)
|
||||
private func crop(_ image: CGImage, to rect: NSRect, display: SCDisplay) -> CGImage? {
|
||||
// The captured image may be at any scale (retina 2x, secondary display
|
||||
// 1x, or trimmed by ScreenCaptureKit). Derive the pixel scale from the
|
||||
// image we actually got, relative to the display's point frame.
|
||||
private func crop(_ image: CGImage, to cgRect: CGRect, display: SCDisplay, roundedCorners: Bool = false) -> CGImage? {
|
||||
// cgRect is in SCDisplay space (bottom-left origin, Y flipped vs AppKit).
|
||||
// The captured image has a TOP-LEFT origin. So within the display,
|
||||
// we must flip Y: pixelY = displayHeight - (cgY - displayMinY) - cgHeight.
|
||||
let scaleX = CGFloat(image.width) / display.frame.width
|
||||
let scaleY = CGFloat(image.height) / display.frame.height
|
||||
guard scaleX > 0, scaleY > 0 else { return nil }
|
||||
let cgRect = CGRect(
|
||||
x: rect.minX,
|
||||
y: rect.minY,
|
||||
width: rect.width,
|
||||
height: rect.height
|
||||
)
|
||||
let localMinX = cgRect.minX - display.frame.minX
|
||||
let localMaxY = cgRect.maxY - display.frame.minY
|
||||
let flippedY = display.frame.height - localMaxY
|
||||
let pxRect = CGRect(
|
||||
x: (cgRect.minX - display.frame.minX) * scaleX,
|
||||
y: (cgRect.minY - display.frame.minY) * scaleY,
|
||||
x: localMinX * scaleX,
|
||||
y: flippedY * scaleY,
|
||||
width: cgRect.width * scaleX,
|
||||
height: cgRect.height * scaleY
|
||||
)
|
||||
@@ -108,15 +372,36 @@ final class ScreenCapture {
|
||||
// display's frame (e.g. secondary monitor coordinate quirks).
|
||||
let safeRect = pxRect.intersection(CGRect(x: 0, y: 0, width: CGFloat(image.width), height: CGFloat(image.height)))
|
||||
guard safeRect.width >= 1, safeRect.height >= 1 else { return nil }
|
||||
guard let cropCtx = CGContext(
|
||||
data: nil, width: Int(safeRect.width), height: Int(safeRect.height),
|
||||
bitsPerComponent: 8, bytesPerRow: 0,
|
||||
space: CGColorSpaceCreateDeviceRGB(),
|
||||
bitmapInfo: CGImageAlphaInfo.premultipliedLast.rawValue
|
||||
) else { return nil }
|
||||
cropCtx.interpolationQuality = .high
|
||||
cropCtx.draw(image, in: CGRect(x: -safeRect.minX, y: -safeRect.minY, width: CGFloat(image.width), height: CGFloat(image.height)))
|
||||
return cropCtx.makeImage()
|
||||
// Use a pixel-exact subimage crop instead of redrawing through a
|
||||
// CGContext, which resamples and loses quality on retina captures.
|
||||
let rounded = safeRect.integral
|
||||
guard rounded.width >= 1, rounded.height >= 1 else { return nil }
|
||||
guard var result = image.cropping(to: rounded) else { return nil }
|
||||
|
||||
// Apply a rounded-corner alpha mask for window captures, matching
|
||||
// macOS window chrome (≈10pt radius, scaled to image pixels).
|
||||
if roundedCorners {
|
||||
let radius = 10 * scaleX
|
||||
let maskRect = CGRect(x: 0, y: 0, width: result.width, height: result.height)
|
||||
guard let maskCtx = CGContext(
|
||||
data: nil, width: result.width, height: result.height,
|
||||
bitsPerComponent: 8, bytesPerRow: 0,
|
||||
space: CGColorSpaceCreateDeviceRGB(),
|
||||
bitmapInfo: CGImageAlphaInfo.premultipliedLast.rawValue
|
||||
) else { return result }
|
||||
maskCtx.setFillColor(CGColor(gray: 1, alpha: 1))
|
||||
maskCtx.fill(maskRect)
|
||||
maskCtx.setBlendMode(.clear)
|
||||
maskCtx.setFillColor(CGColor(gray: 0, alpha: 1))
|
||||
maskCtx.addPath(CGPath(roundedRect: maskRect, cornerWidth: radius, cornerHeight: radius, transform: nil))
|
||||
maskCtx.fillPath()
|
||||
maskCtx.setBlendMode(.normal)
|
||||
if let mask = maskCtx.makeImage(),
|
||||
let masked = result.masking(mask) {
|
||||
result = masked
|
||||
}
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
@available(macOS 14.0, *)
|
||||
@@ -129,7 +414,7 @@ final class ScreenCapture {
|
||||
return content.displays.first(where: { $0.displayID == cgID }) ?? content.displays.first
|
||||
}
|
||||
|
||||
func capture(_ rect: NSRect, excludingWindowIDs excludedIDs: [Int], completion: @escaping (CGImage?) -> Void) {
|
||||
func capture(_ rect: NSRect, excludingWindowIDs excludedIDs: [Int], roundedCorners: Bool = false, completion: @escaping (CGImage?) -> Void) {
|
||||
if #available(macOS 14.0, *) {
|
||||
Task {
|
||||
guard let content = try? await SCShareableContent.current,
|
||||
@@ -141,17 +426,35 @@ final class ScreenCapture {
|
||||
let excluded = content.windows.filter { excludedSet.contains($0.windowID) }
|
||||
let filter = SCContentFilter(display: display, excludingWindows: excluded)
|
||||
let config = SCStreamConfiguration()
|
||||
// display.width/height are pixels; request the display's full
|
||||
// native resolution so ScreenCaptureKit gives us every pixel.
|
||||
config.width = Int(display.width)
|
||||
config.height = Int(display.height)
|
||||
config.showsCursor = true
|
||||
// SCDisplay.width/height are in POINTS; SCStreamConfiguration
|
||||
// width/height are output PIXELS. Multiply by the backing scale
|
||||
// of the screen containing the selection so retina displays are
|
||||
// captured at full resolution and 1x displays stay 1x.
|
||||
guard let screen = NSScreen.screens.first(where: { $0.frame.contains(NSPoint(x: rect.midX, y: rect.midY)) }) else {
|
||||
completion(nil)
|
||||
return
|
||||
}
|
||||
let backingScale = screen.backingScaleFactor
|
||||
config.width = Int(CGFloat(display.width) * backingScale)
|
||||
config.height = Int(CGFloat(display.height) * backingScale)
|
||||
config.showsCursor = false
|
||||
|
||||
// ScreenCaptureKit uses a different origin than AppKit for
|
||||
// screens stacked around the main display: per-display,
|
||||
// SCDisplay y = appKitY - NSScreen.maxY + SCDisplay.maxY.
|
||||
let scRect = CGRect(
|
||||
x: rect.minX,
|
||||
y: rect.minY - screen.frame.maxY + display.frame.maxY,
|
||||
width: rect.width,
|
||||
height: rect.height
|
||||
)
|
||||
|
||||
guard let image = try? await SCScreenshotManager.captureImage(contentFilter: filter, configuration: config) else {
|
||||
completion(nil)
|
||||
return
|
||||
}
|
||||
completion(self.crop(image, to: rect, display: display))
|
||||
CaptureController.log("capture: rect=\(rect) screen=\(screen.frame) display=\(display.frame) backing=\(backingScale) image=\(image.width)x\(image.height) scRect=\(scRect)")
|
||||
completion(self.crop(image, to: scRect, display: display, roundedCorners: roundedCorners))
|
||||
}
|
||||
} else {
|
||||
guard let primary = NSScreen.screens.first(where: { $0.frame.origin == .zero }) ?? NSScreen.screens.first else {
|
||||
|
||||
@@ -1,17 +1,20 @@
|
||||
import AppKit
|
||||
|
||||
/// Renders the capture overlay. All interaction state lives in
|
||||
/// CaptureController (driven by the CGEventTap); this view only draws.
|
||||
final class OverlayWindow: NSPanel {
|
||||
init(screen: NSScreen, onComplete: @escaping (NSRect) -> Void, onCancel: @escaping () -> Void) {
|
||||
let content = OverlayView(onComplete: onComplete, onCancel: onCancel)
|
||||
init(screen: NSScreen, controller: CaptureController) {
|
||||
let content = OverlayView()
|
||||
super.init(
|
||||
contentRect: screen.frame,
|
||||
styleMask: [.borderless, .nonactivatingPanel],
|
||||
styleMask: [.borderless],
|
||||
backing: .buffered,
|
||||
defer: false
|
||||
)
|
||||
contentView = content
|
||||
content.controller = controller
|
||||
level = .screenSaver
|
||||
collectionBehavior = [.canJoinAllSpaces, .fullScreenAuxiliary, .stationary, .transient]
|
||||
collectionBehavior = [.canJoinAllSpaces, .fullScreenAuxiliary, .stationary]
|
||||
backgroundColor = .clear
|
||||
isOpaque = false
|
||||
hasShadow = false
|
||||
@@ -23,21 +26,7 @@ final class OverlayWindow: NSPanel {
|
||||
}
|
||||
|
||||
final class OverlayView: NSView {
|
||||
private let onComplete: (NSRect) -> Void
|
||||
private let onCancel: () -> Void
|
||||
|
||||
private var startPoint: NSPoint?
|
||||
private(set) var selection: NSRect?
|
||||
|
||||
init(onComplete: @escaping (NSRect) -> Void, onCancel: @escaping () -> Void) {
|
||||
self.onComplete = onComplete
|
||||
self.onCancel = onCancel
|
||||
super.init(frame: .zero)
|
||||
}
|
||||
|
||||
required init?(coder: NSCoder) {
|
||||
fatalError("init(coder:) has not been implemented")
|
||||
}
|
||||
weak var controller: CaptureController?
|
||||
|
||||
override var acceptsFirstResponder: Bool { true }
|
||||
|
||||
@@ -47,55 +36,56 @@ final class OverlayView: NSView {
|
||||
addCursorRect(bounds, cursor: .crosshair)
|
||||
}
|
||||
|
||||
override func mouseDown(with event: NSEvent) {
|
||||
let p = convert(event.locationInWindow, from: nil)
|
||||
startPoint = p
|
||||
selection = NSRect(origin: p, size: .zero)
|
||||
needsDisplay = true
|
||||
/// The screen frame this overlay covers, in AppKit global space.
|
||||
private var screenFrame: NSRect {
|
||||
guard let window else { return .zero }
|
||||
return window.convertToScreen(NSRect(origin: .zero, size: bounds.size))
|
||||
}
|
||||
|
||||
override func mouseDragged(with event: NSEvent) {
|
||||
guard let start = startPoint else { return }
|
||||
let p = convert(event.locationInWindow, from: nil)
|
||||
selection = NSRect(
|
||||
x: min(start.x, p.x),
|
||||
y: min(start.y, p.y),
|
||||
width: abs(p.x - start.x),
|
||||
height: abs(p.y - start.y)
|
||||
/// Convert an AppKit-global rect into this view's coordinates.
|
||||
private func viewRect(_ global: NSRect) -> NSRect {
|
||||
let sf = screenFrame
|
||||
return NSRect(
|
||||
x: global.origin.x - sf.origin.x,
|
||||
y: global.origin.y - sf.origin.y,
|
||||
width: global.width,
|
||||
height: global.height
|
||||
)
|
||||
needsDisplay = true
|
||||
}
|
||||
|
||||
override func mouseUp(with event: NSEvent) {
|
||||
guard let sel = selection else { return }
|
||||
if sel.width >= 2, sel.height >= 2 {
|
||||
onComplete(window!.convertToScreen(sel))
|
||||
} else {
|
||||
onCancel()
|
||||
}
|
||||
}
|
||||
|
||||
override func keyDown(with event: NSEvent) {
|
||||
if event.keyCode == 53 {
|
||||
onCancel()
|
||||
} else {
|
||||
super.keyDown(with: event)
|
||||
}
|
||||
}
|
||||
|
||||
override func draw(_ dirtyRect: NSRect) {
|
||||
NSColor.black.withAlphaComponent(0.35).setFill()
|
||||
NSBezierPath(rect: bounds).fill()
|
||||
|
||||
guard let sel = selection, sel.width > 0, sel.height > 0 else { return }
|
||||
guard let controller else { return }
|
||||
|
||||
// Highlight the hovered window (rounded, like real window corners)
|
||||
if let frame = controller.hoverFrame {
|
||||
let vf = viewRect(frame)
|
||||
NSGraphicsContext.current?.saveGraphicsState()
|
||||
NSGraphicsContext.current?.compositingOperation = .clear
|
||||
NSColor.black.setFill()
|
||||
NSBezierPath(roundedRect: vf, xRadius: 10, yRadius: 10).fill()
|
||||
NSGraphicsContext.current?.restoreGraphicsState()
|
||||
|
||||
let border = NSBezierPath(roundedRect: vf, xRadius: 10, yRadius: 10)
|
||||
border.lineWidth = 2
|
||||
NSColor.controlAccentColor.setStroke()
|
||||
border.stroke()
|
||||
return
|
||||
}
|
||||
|
||||
// Marquee selection
|
||||
guard let sel = controller.marqueeRect, sel.width > 0, sel.height > 0 else { return }
|
||||
let vf = viewRect(sel)
|
||||
|
||||
NSGraphicsContext.current?.saveGraphicsState()
|
||||
NSGraphicsContext.current?.compositingOperation = .clear
|
||||
NSColor.black.setFill()
|
||||
NSBezierPath(rect: sel).fill()
|
||||
NSBezierPath(rect: vf).fill()
|
||||
NSGraphicsContext.current?.restoreGraphicsState()
|
||||
|
||||
let border = NSBezierPath(rect: sel)
|
||||
let border = NSBezierPath(rect: vf)
|
||||
border.lineWidth = 1.5
|
||||
NSColor.controlAccentColor.setStroke()
|
||||
border.stroke()
|
||||
@@ -107,8 +97,8 @@ final class OverlayView: NSView {
|
||||
]
|
||||
let labelSize = label.size(withAttributes: attrs)
|
||||
let labelRect = NSRect(
|
||||
x: sel.maxX - labelSize.width - 8,
|
||||
y: sel.minY - labelSize.height - 10,
|
||||
x: vf.maxX - labelSize.width - 8,
|
||||
y: vf.minY - labelSize.height - 10,
|
||||
width: labelSize.width + 8,
|
||||
height: labelSize.height + 6
|
||||
)
|
||||
|
||||
@@ -88,6 +88,31 @@ final class CaptureModel: ObservableObject {
|
||||
@Published var extractedText: String?
|
||||
@Published var isOCRRunning = false
|
||||
@Published var statusMessage: String?
|
||||
/// Drag selection rect in IMAGE point space (nil = none)
|
||||
@Published var selection: CGRect?
|
||||
/// Freehand drawing strokes in IMAGE point space
|
||||
@Published var strokes: [Stroke] = []
|
||||
/// Redo stack of removed strokes
|
||||
@Published var redoStack: [Stroke] = []
|
||||
/// Whether the pen is active (drag draws instead of selecting)
|
||||
@Published var tool: Tool = .select
|
||||
/// Current drawing color
|
||||
@Published var drawColor: NSColor = .systemRed
|
||||
/// Stroke thickness in points (image space)
|
||||
@Published var strokeWidth: CGFloat = 3.0
|
||||
|
||||
enum Tool {
|
||||
case select
|
||||
case draw
|
||||
case redact
|
||||
}
|
||||
|
||||
struct Stroke {
|
||||
let points: [CGPoint]
|
||||
let width: CGFloat
|
||||
let color: NSColor
|
||||
var isRedaction: Bool = false
|
||||
}
|
||||
|
||||
enum CaptureMode {
|
||||
case screenshot
|
||||
@@ -98,24 +123,125 @@ final class CaptureModel: ObservableObject {
|
||||
self.image = image
|
||||
}
|
||||
|
||||
var hasSelection: Bool { selection != nil && selection!.width >= 2 && selection!.height >= 2 }
|
||||
var canUndo: Bool { !strokes.isEmpty }
|
||||
var canRedo: Bool { !redoStack.isEmpty }
|
||||
|
||||
func undoStroke() {
|
||||
guard let last = strokes.popLast() else { return }
|
||||
redoStack.append(last)
|
||||
}
|
||||
|
||||
func redoStroke() {
|
||||
guard let last = redoStack.popLast() else { return }
|
||||
strokes.append(last)
|
||||
}
|
||||
|
||||
func clearAnnotations() {
|
||||
strokes = []
|
||||
redoStack = []
|
||||
}
|
||||
|
||||
/// Image with drawn strokes flattened in, for copy/save/export.
|
||||
var annotatedImage: CGImage {
|
||||
guard !strokes.isEmpty else { return image }
|
||||
let size = NSSize(width: image.width, height: image.height)
|
||||
guard let rep = NSBitmapImageRep(
|
||||
bitmapDataPlanes: nil, pixelsWide: Int(size.width), pixelsHigh: Int(size.height),
|
||||
bitsPerSample: 8, samplesPerPixel: 4, hasAlpha: true, isPlanar: false,
|
||||
colorSpaceName: .deviceRGB, bytesPerRow: 0, bitsPerPixel: 0
|
||||
) else { return image }
|
||||
NSGraphicsContext.saveGraphicsState()
|
||||
NSGraphicsContext.current = NSGraphicsContext(bitmapImageRep: rep)
|
||||
let ctx = NSGraphicsContext.current!.cgContext
|
||||
// CGImage draws with a flipped Y in NSGraphicsContext
|
||||
ctx.draw(image, in: NSRect(origin: .zero, size: size))
|
||||
// Strokes are stored in TOP-LEFT image coordinates (from the SwiftUI
|
||||
// canvas), but the CGContext uses a BOTTOM-LEFT origin. Flip Y so the
|
||||
// strokes land where they were drawn.
|
||||
ctx.saveGState()
|
||||
ctx.translateBy(x: 0, y: size.height)
|
||||
ctx.scaleBy(x: 1, y: -1)
|
||||
ctx.setLineCap(.round)
|
||||
ctx.setLineJoin(.round)
|
||||
for stroke in strokes {
|
||||
guard stroke.points.count > 1 else { continue }
|
||||
if stroke.isRedaction {
|
||||
// Redaction bar: filled black rect covering the stroke bounds,
|
||||
// padded so it fully covers text.
|
||||
let xs = stroke.points.map { $0.x }
|
||||
let ys = stroke.points.map { $0.y }
|
||||
let bar = NSRect(
|
||||
x: xs.min()! - 2,
|
||||
y: ys.min()! - 2,
|
||||
width: (xs.max()! - xs.min()!) + 4,
|
||||
height: max(stroke.width, ys.max()! - ys.min()!) + 4
|
||||
)
|
||||
NSColor.black.setFill()
|
||||
NSBezierPath(rect: bar).fill()
|
||||
} else {
|
||||
stroke.color.setStroke()
|
||||
ctx.setLineWidth(stroke.width)
|
||||
ctx.beginPath()
|
||||
ctx.move(to: stroke.points[0])
|
||||
for point in stroke.points.dropFirst() {
|
||||
ctx.addLine(to: point)
|
||||
}
|
||||
ctx.strokePath()
|
||||
}
|
||||
}
|
||||
ctx.restoreGState()
|
||||
NSGraphicsContext.restoreGraphicsState()
|
||||
return rep.cgImage ?? image
|
||||
}
|
||||
|
||||
func addStroke(_ points: [CGPoint]) {
|
||||
guard points.count > 1 else { return }
|
||||
if tool == .redact {
|
||||
// Redaction: a filled black bar from start to end, as thick as the
|
||||
// current stroke width (drawn as a rect for clean edges).
|
||||
strokes.append(Stroke(points: points, width: max(strokeWidth, 12), color: .black, isRedaction: true))
|
||||
} else {
|
||||
strokes.append(Stroke(points: points, width: strokeWidth, color: drawColor))
|
||||
}
|
||||
redoStack = []
|
||||
}
|
||||
|
||||
var displayText: String {
|
||||
if isOCRRunning {
|
||||
return "Recognizing text…"
|
||||
}
|
||||
return extractedText ?? "Text appears here — click “Get Text”."
|
||||
return extractedText ?? "Select text in the screenshot, then click “Get Text”."
|
||||
}
|
||||
|
||||
func runOCR() {
|
||||
func cropImage(to rectInImage: CGRect) -> CGImage? {
|
||||
return image.cropping(to: rectInImage)
|
||||
}
|
||||
|
||||
func runOCR(over selectionOnly: CGRect? = nil) {
|
||||
guard !isOCRRunning else { return }
|
||||
isOCRRunning = true
|
||||
statusMessage = nil
|
||||
OCRService.shared.recognize(image) { [weak self] text in
|
||||
|
||||
let targetImage: CGImage
|
||||
if let sel = selectionOnly {
|
||||
guard let cropped = image.cropping(to: sel) else {
|
||||
self.isOCRRunning = false
|
||||
self.statusMessage = "Selection too small."
|
||||
return
|
||||
}
|
||||
targetImage = cropped
|
||||
} else {
|
||||
targetImage = image
|
||||
}
|
||||
|
||||
OCRService.shared.recognize(targetImage) { [weak self] text in
|
||||
Task { @MainActor in
|
||||
guard let self else { return }
|
||||
self.isOCRRunning = false
|
||||
self.mode = .text
|
||||
if text.isEmpty {
|
||||
self.extractedText = "No text found in the screenshot."
|
||||
self.extractedText = "No text found in the selection."
|
||||
} else {
|
||||
self.extractedText = text
|
||||
let pasteboard = NSPasteboard.general
|
||||
@@ -127,6 +253,11 @@ final class CaptureModel: ObservableObject {
|
||||
}
|
||||
}
|
||||
|
||||
func runOCRSelection() {
|
||||
guard let sel = selection, hasSelection else { return }
|
||||
runOCR(over: sel)
|
||||
}
|
||||
|
||||
func copyText() {
|
||||
guard let text = extractedText else { return }
|
||||
let pasteboard = NSPasteboard.general
|
||||
@@ -138,7 +269,7 @@ final class CaptureModel: ObservableObject {
|
||||
func copyImage() {
|
||||
let pasteboard = NSPasteboard.general
|
||||
pasteboard.clearContents()
|
||||
pasteboard.writeObjects([NSImage(cgImage: image, size: .zero)])
|
||||
pasteboard.writeObjects([NSImage(cgImage: annotatedImage, size: .zero)])
|
||||
statusMessage = "Image copied to clipboard."
|
||||
}
|
||||
|
||||
@@ -151,7 +282,7 @@ final class CaptureModel: ObservableObject {
|
||||
statusMessage = "Could not create PNG file."
|
||||
return
|
||||
}
|
||||
CGImageDestinationAddImage(dest, image, nil)
|
||||
CGImageDestinationAddImage(dest, annotatedImage, nil)
|
||||
if CGImageDestinationFinalize(dest) {
|
||||
statusMessage = "Saved PNG."
|
||||
} else {
|
||||
@@ -171,73 +302,350 @@ struct ResultView: View {
|
||||
let onClose: () -> Void
|
||||
|
||||
var body: some View {
|
||||
VStack(spacing: 10) {
|
||||
HStack {
|
||||
VStack(spacing: 0) {
|
||||
// Toolbar
|
||||
HStack(spacing: 10) {
|
||||
Picker("", selection: $model.mode) {
|
||||
Text("Screenshot").tag(CaptureModel.CaptureMode.screenshot)
|
||||
Text("Text").tag(CaptureModel.CaptureMode.text)
|
||||
if model.isOCRRunning {
|
||||
Text("…").tag(CaptureModel.CaptureMode.text)
|
||||
}
|
||||
}
|
||||
.pickerStyle(.segmented)
|
||||
.fixedSize()
|
||||
Spacer()
|
||||
Button("Copy Image") { model.copyImage() }
|
||||
Button("Save PNG…") { model.savePNG() }
|
||||
Button("Get Text") { model.runOCR() }
|
||||
.disabled(model.isOCRRunning || model.extractedText != nil)
|
||||
Button("Close") { onClose() }
|
||||
}
|
||||
.padding(.horizontal, 12)
|
||||
.padding(.top, 10)
|
||||
|
||||
if model.mode == .screenshot {
|
||||
GeometryReader { geo in
|
||||
let imageWidth = CGFloat(model.image.width)
|
||||
Image(nsImage: NSImage(cgImage: model.image, size: .zero))
|
||||
.resizable()
|
||||
// Downscale with smoothing (AA); upscale keeps pixels crisp.
|
||||
.interpolation(imageWidth >= geo.size.width ? .high : .none)
|
||||
.aspectRatio(contentMode: .fit)
|
||||
Spacer()
|
||||
|
||||
if model.hasSelection {
|
||||
Button {
|
||||
model.runOCRSelection()
|
||||
} label: {
|
||||
Label("OCR Selection", systemImage: "text.magnifyingglass")
|
||||
}
|
||||
.buttonStyle(.borderedProminent)
|
||||
.disabled(model.isOCRRunning)
|
||||
.help("Recognize text inside the selected region")
|
||||
}
|
||||
.border(Color.secondary.opacity(0.4))
|
||||
.padding(.horizontal, 12)
|
||||
|
||||
Button {
|
||||
model.runOCR()
|
||||
} label: {
|
||||
Label("Get All Text", systemImage: "text.viewfinder")
|
||||
}
|
||||
.buttonStyle(.bordered)
|
||||
.disabled(model.isOCRRunning)
|
||||
.help("Recognize all text in the screenshot")
|
||||
|
||||
Divider().frame(height: 18)
|
||||
|
||||
Button {
|
||||
model.tool = model.tool == .draw ? .select : .draw
|
||||
if model.tool != .select { model.selection = nil }
|
||||
} label: {
|
||||
Label("Draw", systemImage: model.tool == .draw ? "pencil.tip.crop.circle.fill" : "pencil.tip.crop.circle")
|
||||
}
|
||||
.buttonStyle(.bordered)
|
||||
.foregroundStyle(model.tool == .draw ? .primary : .secondary)
|
||||
.help("Draw on the screenshot")
|
||||
|
||||
Button {
|
||||
model.tool = model.tool == .redact ? .select : .redact
|
||||
if model.tool != .select { model.selection = nil }
|
||||
} label: {
|
||||
Label("Redact", systemImage: model.tool == .redact ? "rectangle.fill.on.rectangle.fill" : "rectangle.dashed")
|
||||
}
|
||||
.buttonStyle(.bordered)
|
||||
.foregroundStyle(model.tool == .redact ? .primary : .secondary)
|
||||
.help("Cover sensitive areas with black bars")
|
||||
|
||||
Button {
|
||||
model.copyImage()
|
||||
} label: {
|
||||
Label("Copy", systemImage: "doc.on.doc")
|
||||
}
|
||||
.buttonStyle(.bordered)
|
||||
.help("Copy image to clipboard")
|
||||
|
||||
Button {
|
||||
model.savePNG()
|
||||
} label: {
|
||||
Label("Save", systemImage: "square.and.arrow.down")
|
||||
}
|
||||
.buttonStyle(.bordered)
|
||||
.help("Save screenshot as PNG");
|
||||
|
||||
Button {
|
||||
onClose()
|
||||
} label: {
|
||||
Image(systemName: "xmark.circle.fill")
|
||||
.foregroundStyle(.secondary)
|
||||
}
|
||||
.buttonStyle(.plain)
|
||||
.help("Close")
|
||||
}
|
||||
.padding(.horizontal, 14)
|
||||
.padding(.vertical, 10)
|
||||
|
||||
Divider()
|
||||
|
||||
// Tool toolbar (only when a draw tool is active)
|
||||
if model.tool != .select {
|
||||
HStack(spacing: 14) {
|
||||
Label(model.tool == .redact ? "Redact" : "Draw", systemImage: model.tool == .redact ? "rectangle.fill" : "pencil.tip")
|
||||
.font(.caption)
|
||||
.foregroundStyle(.secondary)
|
||||
|
||||
Divider().frame(height: 18)
|
||||
|
||||
if model.tool == .draw {
|
||||
HStack(spacing: 6) {
|
||||
Image(systemName: "circle.fill").font(.system(size: 4))
|
||||
Image(systemName: "circle.fill").font(.system(size: 7))
|
||||
Image(systemName: "circle.fill").font(.system(size: 10))
|
||||
}
|
||||
.foregroundStyle(.secondary)
|
||||
Slider(value: $model.strokeWidth, in: 1...12, step: 0.5)
|
||||
.frame(width: 120)
|
||||
.help("Stroke thickness")
|
||||
|
||||
Divider().frame(height: 18)
|
||||
|
||||
ColorPicker("", selection: Binding(
|
||||
get: { Color(nsColor: model.drawColor) },
|
||||
set: { model.drawColor = NSColor($0) }
|
||||
), supportsOpacity: false)
|
||||
.labelsHidden()
|
||||
.frame(width: 28)
|
||||
.help("Drawing color")
|
||||
} else {
|
||||
Label("Drag to cover sensitive text", systemImage: "eye.slash")
|
||||
.font(.caption)
|
||||
.foregroundStyle(.secondary)
|
||||
}
|
||||
|
||||
Spacer()
|
||||
|
||||
Button {
|
||||
model.undoStroke()
|
||||
} label: {
|
||||
Label("Undo", systemImage: "arrow.uturn.backward")
|
||||
}
|
||||
.buttonStyle(.bordered)
|
||||
.controlSize(.small)
|
||||
.disabled(!model.canUndo)
|
||||
.keyboardShortcut("z", modifiers: .command)
|
||||
.help("Undo last stroke (⌘Z)")
|
||||
|
||||
Button {
|
||||
model.redoStroke()
|
||||
} label: {
|
||||
Label("Redo", systemImage: "arrow.uturn.forward")
|
||||
}
|
||||
.buttonStyle(.bordered)
|
||||
.controlSize(.small)
|
||||
.disabled(!model.canRedo)
|
||||
.keyboardShortcut("z", modifiers: [.command, .shift])
|
||||
.help("Redo stroke (⌘⇧Z)")
|
||||
|
||||
Button {
|
||||
model.clearAnnotations()
|
||||
} label: {
|
||||
Label("Clear", systemImage: "eraser")
|
||||
}
|
||||
.buttonStyle(.bordered)
|
||||
.controlSize(.small)
|
||||
.disabled(model.strokes.isEmpty)
|
||||
.help("Remove all drawn annotations")
|
||||
}
|
||||
.padding(.horizontal, 14)
|
||||
.padding(.vertical, 8)
|
||||
.background(.bar)
|
||||
|
||||
Divider()
|
||||
}
|
||||
|
||||
// Content
|
||||
if model.mode == .screenshot {
|
||||
ScreenshotPane(model: model)
|
||||
.padding(12)
|
||||
} else {
|
||||
ZStack(alignment: .bottomTrailing) {
|
||||
ZStack {
|
||||
ScrollView {
|
||||
Text(model.displayText)
|
||||
.font(.system(.body))
|
||||
.font(.system(.body, design: .monospaced))
|
||||
.textSelection(.enabled)
|
||||
.frame(maxWidth: .infinity, alignment: .leading)
|
||||
.padding(8)
|
||||
.padding(12)
|
||||
}
|
||||
.overlay(
|
||||
RoundedRectangle(cornerRadius: 6)
|
||||
.stroke(Color.secondary.opacity(0.4))
|
||||
)
|
||||
}
|
||||
.frame(maxWidth: .infinity, maxHeight: .infinity)
|
||||
}
|
||||
|
||||
HStack(spacing: 8) {
|
||||
if let msg = model.statusMessage {
|
||||
Text(msg)
|
||||
.font(.caption)
|
||||
.foregroundColor(.secondary)
|
||||
}
|
||||
Button("Copy Text") { model.copyText() }
|
||||
// Status bar
|
||||
HStack(spacing: 8) {
|
||||
if model.isOCRRunning {
|
||||
ProgressView()
|
||||
.controlSize(.small)
|
||||
}
|
||||
Text(model.statusMessage ?? (model.mode == .screenshot
|
||||
? (model.hasSelection ? "Region selected — click OCR Selection." : "Drag to select a region, or Get All Text.")
|
||||
: "Selectable text"))
|
||||
.font(.caption)
|
||||
.foregroundStyle(.secondary)
|
||||
.lineLimit(1)
|
||||
|
||||
Spacer()
|
||||
|
||||
if model.mode == .text, model.extractedText != nil {
|
||||
Button {
|
||||
model.copyText()
|
||||
} label: {
|
||||
Label("Copy Text", systemImage: "doc.on.doc")
|
||||
}
|
||||
.padding(8)
|
||||
.buttonStyle(.bordered)
|
||||
.controlSize(.small)
|
||||
}
|
||||
}
|
||||
.padding(.horizontal, 14)
|
||||
.padding(.vertical, 8)
|
||||
.background(.bar)
|
||||
}
|
||||
.frame(minWidth: 520, minHeight: 420)
|
||||
}
|
||||
}
|
||||
|
||||
if model.statusMessage != nil && model.mode == .screenshot {
|
||||
Text(model.statusMessage!)
|
||||
.font(.caption)
|
||||
.foregroundColor(.secondary)
|
||||
.frame(maxWidth: .infinity, alignment: .leading)
|
||||
.padding(.horizontal, 12)
|
||||
struct ScreenshotPane: View {
|
||||
@ObservedObject var model: CaptureModel
|
||||
|
||||
private var imageSize: CGSize {
|
||||
CGSize(width: model.image.width, height: model.image.height)
|
||||
}
|
||||
|
||||
private func geometry(for geo: GeometryProxy) -> (scale: CGFloat, drawSize: CGSize, origin: CGPoint) {
|
||||
let size = imageSize
|
||||
let scale = min(geo.size.width / size.width, geo.size.height / size.height)
|
||||
let drawSize = CGSize(width: size.width * scale, height: size.height * scale)
|
||||
let origin = CGPoint(
|
||||
x: (geo.size.width - drawSize.width) / 2,
|
||||
y: (geo.size.height - drawSize.height) / 2
|
||||
)
|
||||
return (scale, drawSize, origin)
|
||||
}
|
||||
|
||||
private func toImagePoint(_ viewPoint: CGPoint, scale: CGFloat, origin: CGPoint) -> CGPoint {
|
||||
CGPoint(
|
||||
x: (viewPoint.x - origin.x) / scale,
|
||||
y: (viewPoint.y - origin.y) / scale
|
||||
)
|
||||
}
|
||||
|
||||
private func clampSelection(_ rect: CGRect) -> CGRect {
|
||||
rect.intersection(CGRect(origin: .zero, size: imageSize))
|
||||
}
|
||||
|
||||
@State private var currentStroke: [CGPoint] = []
|
||||
|
||||
var body: some View {
|
||||
GeometryReader { geo in
|
||||
let (scale, drawSize, origin) = geometry(for: geo)
|
||||
|
||||
ZStack {
|
||||
Image(nsImage: NSImage(cgImage: model.image, size: .zero))
|
||||
.resizable()
|
||||
.interpolation(imageSize.width >= geo.size.width ? .high : .none)
|
||||
.aspectRatio(contentMode: .fit)
|
||||
.frame(width: drawSize.width, height: drawSize.height)
|
||||
.position(x: origin.x + drawSize.width / 2, y: origin.y + drawSize.height / 2)
|
||||
|
||||
// Completed strokes
|
||||
Canvas { context, _ in
|
||||
for stroke in model.strokes {
|
||||
guard stroke.points.count > 1 else { continue }
|
||||
if stroke.isRedaction {
|
||||
let xs = stroke.points.map { $0.x }
|
||||
let ys = stroke.points.map { $0.y }
|
||||
let bar = CGRect(
|
||||
x: origin.x + (xs.min()! - 2) * scale,
|
||||
y: origin.y + (ys.min()! - 2) * scale,
|
||||
width: ((xs.max()! - xs.min()!) + 4) * scale,
|
||||
height: (max(stroke.width, ys.max()! - ys.min()!) + 4) * scale
|
||||
)
|
||||
context.fill(Path(bar), with: .color(.black))
|
||||
} else {
|
||||
var path = Path()
|
||||
path.move(to: CGPoint(x: origin.x + stroke.points[0].x * scale, y: origin.y + stroke.points[0].y * scale))
|
||||
for point in stroke.points.dropFirst() {
|
||||
path.addLine(to: CGPoint(x: origin.x + point.x * scale, y: origin.y + point.y * scale))
|
||||
}
|
||||
context.stroke(path, with: .color(Color(nsColor: stroke.color)), lineWidth: stroke.width)
|
||||
}
|
||||
}
|
||||
}
|
||||
.allowsHitTesting(false)
|
||||
|
||||
// In-progress stroke
|
||||
if !currentStroke.isEmpty {
|
||||
Canvas { context, _ in
|
||||
var path = Path()
|
||||
if currentStroke.count > 1 {
|
||||
path.move(to: CGPoint(x: origin.x + currentStroke[0].x * scale, y: origin.y + currentStroke[0].y * scale))
|
||||
for point in currentStroke.dropFirst() {
|
||||
path.addLine(to: CGPoint(x: origin.x + point.x * scale, y: origin.y + point.y * scale))
|
||||
}
|
||||
}
|
||||
context.stroke(path, with: .color(Color(nsColor: model.drawColor)), lineWidth: model.strokeWidth)
|
||||
}
|
||||
.allowsHitTesting(false)
|
||||
}
|
||||
|
||||
if model.tool == .select, let sel = model.selection {
|
||||
let viewRect = CGRect(
|
||||
x: origin.x + sel.minX * scale,
|
||||
y: origin.y + sel.minY * scale,
|
||||
width: sel.width * scale,
|
||||
height: sel.height * scale
|
||||
)
|
||||
Rectangle()
|
||||
.fill(Color.accentColor.opacity(0.15))
|
||||
.border(Color.accentColor, width: 1.5)
|
||||
.frame(width: viewRect.width, height: viewRect.height)
|
||||
.position(x: viewRect.midX, y: viewRect.midY)
|
||||
.allowsHitTesting(false)
|
||||
}
|
||||
}
|
||||
.contentShape(Rectangle())
|
||||
.gesture(
|
||||
DragGesture(minimumDistance: 0)
|
||||
.onChanged { value in
|
||||
if model.tool != .select {
|
||||
let p = toImagePoint(value.location, scale: scale, origin: origin)
|
||||
let imgRect = CGRect(origin: .zero, size: imageSize)
|
||||
if imgRect.contains(p) {
|
||||
currentStroke.append(p)
|
||||
}
|
||||
} else {
|
||||
let start = toImagePoint(value.startLocation, scale: scale, origin: origin)
|
||||
let current = toImagePoint(value.location, scale: scale, origin: origin)
|
||||
let rect = CGRect(
|
||||
x: min(start.x, current.x),
|
||||
y: min(start.y, current.y),
|
||||
width: abs(current.x - start.x),
|
||||
height: abs(current.y - start.y)
|
||||
)
|
||||
model.selection = clampSelection(rect)
|
||||
}
|
||||
}
|
||||
.onEnded { _ in
|
||||
if model.tool != .select {
|
||||
if currentStroke.count > 1 {
|
||||
model.addStroke(currentStroke)
|
||||
}
|
||||
currentStroke = []
|
||||
} else {
|
||||
if let s = model.selection, (s.width < 2 || s.height < 2) {
|
||||
model.selection = nil
|
||||
}
|
||||
}
|
||||
}
|
||||
)
|
||||
}
|
||||
.padding(.bottom, 10)
|
||||
.frame(minWidth: 480, minHeight: 380)
|
||||
}
|
||||
}
|
||||
|
||||
+2
-2
@@ -108,9 +108,9 @@ cat > "$APP/Contents/Info.plist" <<'PLIST'
|
||||
<key>CFBundleIconFile</key>
|
||||
<string>Icon</string>
|
||||
<key>CFBundleShortVersionString</key>
|
||||
<string>1.1.4</string>
|
||||
<string>1.7.4</string>
|
||||
<key>CFBundleVersion</key>
|
||||
<string>14</string>
|
||||
<string>74</string>
|
||||
<key>LSUIElement</key>
|
||||
<true/>
|
||||
<key>NSHighResolutionCapable</key>
|
||||
|
||||
Reference in New Issue
Block a user