74 lines
1.4 KiB
Go
74 lines
1.4 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"log"
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
type HostInfo struct {
|
||
|
|
ClientID string `json:"clientId"`
|
||
|
|
OS string `json:"os"`
|
||
|
|
Arch string `json:"arch"`
|
||
|
|
Version string `json:"version"`
|
||
|
|
}
|
||
|
|
|
||
|
|
var (
|
||
|
|
hostInfo HostInfo
|
||
|
|
sendFn func(event string, payload []byte)
|
||
|
|
mu sync.Mutex
|
||
|
|
)
|
||
|
|
|
||
|
|
func setSend(fn func(event string, payload []byte)) {
|
||
|
|
mu.Lock()
|
||
|
|
sendFn = fn
|
||
|
|
mu.Unlock()
|
||
|
|
}
|
||
|
|
|
||
|
|
func sendEvent(event string, payload interface{}) {
|
||
|
|
mu.Lock()
|
||
|
|
fn := sendFn
|
||
|
|
mu.Unlock()
|
||
|
|
if fn == nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
data, err := json.Marshal(payload)
|
||
|
|
if err != nil {
|
||
|
|
log.Printf("[sample] marshal error: %v", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
fn(event, data)
|
||
|
|
}
|
||
|
|
|
||
|
|
func handleInit(hostJSON []byte) error {
|
||
|
|
if err := json.Unmarshal(hostJSON, &hostInfo); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
log.Printf("[sample] init: clientId=%s os=%s arch=%s", hostInfo.ClientID, hostInfo.OS, hostInfo.Arch)
|
||
|
|
sendEvent("ready", map[string]string{"message": "sample plugin ready"})
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func handleEvent(event string, payload []byte) error {
|
||
|
|
switch event {
|
||
|
|
case "ui_message":
|
||
|
|
var msg struct {
|
||
|
|
Message string `json:"message"`
|
||
|
|
}
|
||
|
|
_ = json.Unmarshal(payload, &msg)
|
||
|
|
log.Printf("[sample] got ui_message: %s", msg.Message)
|
||
|
|
sendEvent("echo", map[string]string{"message": msg.Message})
|
||
|
|
|
||
|
|
case "ping":
|
||
|
|
sendEvent("pong", nil)
|
||
|
|
|
||
|
|
default:
|
||
|
|
log.Printf("[sample] unhandled event: %s", event)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func handleUnload() {
|
||
|
|
log.Printf("[sample] unloading")
|
||
|
|
}
|