我完全不会 Lua，但我用嘴做了一个 OCR

zh-Hans
zh-Hant
en-US

local hs = hs
local hotkey = hs.hotkey
local task = hs.task
local alert = hs.alert
local pasteboard = hs.pasteboard
local timer = hs.timer
local screenRecordingState = hs.screenRecordingState

local ocr = {}

local screenshotCommand = "/usr/sbin/screencapture"
local osascriptCommand = "/usr/bin/osascript"

local function trim(text)
  return (text or ""):gsub("^%s+", ""):gsub("%s+$", "")
end

local function stripProcessTrailingNewline(text)
  return (text or ""):gsub("\r?\n$", "")
end

local function removeFile(path)
  if path and #path > 0 then
    os.remove(path)
  end
end

local function tempPath(suffix)
  return os.tmpname() .. suffix
end

local function copyOCRResult(text)
  pasteboard.setContents(text)
end

local function appleScriptQuote(value)
  return '"' .. tostring(value):gsub("\\", "\\\\"):gsub('"', '\\"') .. '"'
end

local function writeFile(path, content)
  local file, err = io.open(path, "w")
  if not file then
    return nil, err
  end

  file:write(content)
  file:close()
  return true
end

local function buildOCRAppleScript(imagePath)
  local quotedPath = appleScriptQuote(imagePath)
  return table.concat({
    'use framework "Foundation"',
    'use framework "Vision"',
    'use scripting additions',
    "",
    "set theFile to current application's |NSURL|'s fileURLWithPath:" .. quotedPath,
    "set requestHandler to current application's VNImageRequestHandler's alloc()'s initWithURL:theFile options:(missing value)",
    "set theRequest to current application's VNRecognizeTextRequest's alloc()'s init()",
    "theRequest's setRecognitionLevel:(current application's VNRequestTextRecognitionLevelAccurate)",
    'theRequest\'s setRecognitionLanguages:{"zh-Hans", "zh-Hant", "en-US"}',
    "theRequest's setUsesLanguageCorrection:false",
    "requestHandler's performRequests:(current application's NSArray's arrayWithObject:(theRequest)) |error|:(missing value)",
    "set theResults to theRequest's results()",
    'if theResults is missing value then return ""',
    "set theArray to current application's NSMutableArray's new()",
    "repeat with aResult in theResults",
    "    set theCandidates to aResult's topCandidates:1",
    "    if (theCandidates's |count|()) > 0 then",
    "        (theArray's addObject:(((theCandidates's objectAtIndex:0)'s |string|())))",
    "    end if",
    "end repeat",
    "return (theArray's componentsJoinedByString:linefeed) as text",
  }, "\n")
end

local function runAppleScriptOCR(imagePath, callback)
  local scriptPath = tempPath(".applescript")
  local ok, err = writeFile(scriptPath, buildOCRAppleScript(imagePath))
  if not ok then
    callback(nil, "无法创建临时 AppleScript: " .. tostring(err))
    return
  end

  local ocrTask = task.new(osascriptCommand, function(exitCode, stdout, stderr)
    removeFile(scriptPath)

    if exitCode ~= 0 then
      local message = trim(stderr)
      if message == "" then
        message = "系统 OCR 执行失败"
      end
      callback(nil, message)
      return
    end

    callback(stripProcessTrailingNewline(stdout), nil)
  end, {scriptPath})

  if not ocrTask then
    removeFile(scriptPath)
    callback(nil, "无法启动 osascript")
    return
  end

  ocrTask:start()
end

function ocr.runOnImage(imagePath)
  runAppleScriptOCR(imagePath, function(text, err)
    removeFile(imagePath)

    if not text then
      alert.show(err, 3)
      return
    end

    if text == "" then
      alert.show("没有识别到文本", 2)
      return
    end

    copyOCRResult(text)
  end)
end

function ocr.captureSelectionAndRecognize()
  if not screenRecordingState(false) then
    screenRecordingState(true)
    alert.show("请先给 Hammerspoon 开启屏幕录制权限，然后重新尝试 OCR", 3)
    return
  end

  local imagePath = tempPath(".png")
  task.new(screenshotCommand, function(exitCode, _, stderr)
    if exitCode ~= 0 then
      removeFile(imagePath)

      local message = trim(stderr)
      if not screenRecordingState(false) then
        message = "缺少屏幕录制权限，请在系统设置里允许 Hammerspoon"
      elseif message == "" then
        message = "截图已取消"
      end

      alert.show(message, 1.8)
      return
    end

    timer.doAfter(0.05, function()
      ocr.runOnImage(imagePath)
    end)
  end, {"-i", "-x", imagePath}):start()
end

local modifiers = {"ctrl"}
local triggerKey = "a"

if hotkey.assignable(modifiers, triggerKey) then
  hotkey.bind(modifiers, triggerKey, nil, function()
    ocr.captureSelectionAndRecognize()
  end)
else
  alert.show("Ctrl+" .. triggerKey .. " 已被系统占用，OCR 热键未绑定")
end

return ocr

require("ocr")