myocr.go

package main

import (
	"fmt"
	"io/ioutil"
	"os"
	"os/exec"
	"runtime"

	"github.com/go-vgo/robotgo"
	"github.com/go-vgo/robotgo/clipboard"
	"github.com/martinlindhe/notify"
	"github.com/otiai10/gosseract/v2"
	hook "github.com/robotn/gohook"
)

var ScreenshotBin = "/usr/sbin/screencapture"

func checkOS() {
	if runtime.GOOS != "darwin" {
		panic("Only support osx!")
	}
	if _, err := os.Stat(ScreenshotBin); os.IsNotExist(err) {
		panic("Not found screencapture")
	}
}

func ocr(path string) string {
	client := gosseract.NewClient()
	defer client.Close()
	client.SetImage(path)
	text, _ := client.Text()
	return text
}

func screenshot() string {
	f, err := ioutil.TempFile(os.TempDir(), "*.png")
	if err != nil {
		panic(err)
	}
	defer os.Remove(f.Name())
	c := exec.Command(ScreenshotBin, "-s", f.Name())
	err = c.Run()
	if err != nil {
		panic(err)
	}
	fi, err := os.Stat(f.Name())
	if err != nil {
		panic(err)
	}
	if fi.Size() > 1024 {
		return ocr(f.Name())
	}
	return ""
}

// https://github.com/otiai10/gosseract/issues/200#issuecomment-817511288
// pkg/mod/github.com/otiai10/gosseract/[email protected]/tessbridge.cpp
// based on the feedback,I modified the original file,otherwise the compilation would report a header file error
func main() {
	checkOS()

	fmt.Println("--- Please press command + option + i to stop hook ---")
	robotgo.EventHook(hook.KeyDown, []string{"command", "option", "i"}, func(e hook.Event) {
		fmt.Println("Start screenshot")
		ocr := screenshot()
		if len(ocr) > 0 {
			message := ""
			clipboard.WriteAll(ocr)
			if len(ocr) > 10 {
				message = ocr[:10]
			} else {
				message = ocr
			}
			notify.Notify("MYOCR", "MYOCR", message, "")
		}
	})

	s := robotgo.EventStart()
	<-robotgo.EventProcess(s)
}