laily
021fc3a828
Route hook notifications through /dev/tty and client_tty so OSC reaches Ghostty when Cursor captures stdout. Add afterAgentResponse hook, hook logging, debounce, split test modes, and use directory name for titles. Co-authored-by: Cursor <cursoragent@cursor.com>
112 lines
2.9 KiB
Go
112 lines
2.9 KiB
Go
package notify
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/longbin/agent-notify/internal/tmux"
|
|
)
|
|
|
|
const defaultTestTitle = "agent-notify"
|
|
const defaultTestBody = "测试通知 — 如果你看到这条，说明配置正确"
|
|
|
|
type claudeTestResponse struct {
|
|
TerminalSequence string `json:"terminalSequence"`
|
|
}
|
|
|
|
func TestCursor(title, body string, tryAll bool) (SendResult, error) {
|
|
if title == "" {
|
|
title = defaultTestTitle
|
|
}
|
|
if body == "" {
|
|
body = defaultTestBody + " [cursor]"
|
|
}
|
|
|
|
if tryAll {
|
|
return testCursorAll(title, body)
|
|
}
|
|
|
|
result, err := SendAutoWithResult("osc777", title, body)
|
|
printTestStatus("cursor", result, err)
|
|
return result, err
|
|
}
|
|
|
|
func testCursorAll(title, body string) (SendResult, error) {
|
|
inTmux := tmux.InTmux()
|
|
methods := []DeliveryMethod{MethodDirectStdout, MethodPassthroughStdout, MethodClientTTYRaw, MethodClientTTYPassthrough}
|
|
if !inTmux {
|
|
methods = []DeliveryMethod{MethodDirectStdout}
|
|
}
|
|
clientTTY, _ := tmux.ClientTTY()
|
|
|
|
var lastResult SendResult
|
|
var lastErr error
|
|
for i, method := range methods {
|
|
result, err := SendWithResult(SendOptions{
|
|
Protocol: "osc777",
|
|
Title: title,
|
|
Body: fmt.Sprintf("%s [%s]", body, method),
|
|
InTmux: inTmux,
|
|
Layers: 1,
|
|
ClientTTY: clientTTY,
|
|
Method: method,
|
|
})
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "try %d/%d %s: FAIL %v\n", i+1, len(methods), method, err)
|
|
lastErr = err
|
|
continue
|
|
}
|
|
fmt.Fprintf(os.Stderr, "try %d/%d %s: OK (check Ghostty notification)\n", i+1, len(methods), method)
|
|
lastResult = result
|
|
}
|
|
if lastResult.Method == "" && lastErr != nil {
|
|
return lastResult, lastErr
|
|
}
|
|
fmt.Fprintf(os.Stderr, "mode=cursor try-all done ssh=%v tmux=%v\n", tmux.IsSSHSession(), inTmux)
|
|
return lastResult, nil
|
|
}
|
|
|
|
func printTestStatus(mode string, result SendResult, err error) {
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "agent-notify: %s test FAILED: %v\n", mode, err)
|
|
return
|
|
}
|
|
fmt.Fprintf(os.Stderr,
|
|
"agent-notify: %s test sent via %s (ssh=%v tmux=%v) — check Ghostty desktop notification\n",
|
|
mode, result.Method, tmux.IsSSHSession(), tmux.InTmux(),
|
|
)
|
|
}
|
|
|
|
func TestClaude(title, body string, apply bool, out io.Writer) (SendResult, error) {
|
|
if title == "" {
|
|
title = defaultTestTitle
|
|
}
|
|
if body == "" {
|
|
body = defaultTestBody + " [claude]"
|
|
}
|
|
seq := BuildSequence("osc777", title, body)
|
|
if out == nil {
|
|
out = os.Stdout
|
|
}
|
|
if !apply {
|
|
resp := claudeTestResponse{TerminalSequence: seq}
|
|
enc := json.NewEncoder(out)
|
|
enc.SetEscapeHTML(false)
|
|
if err := enc.Encode(resp); err != nil {
|
|
return SendResult{}, err
|
|
}
|
|
fmt.Fprintln(os.Stderr, "agent-notify: claude test JSON printed (use --apply to emit)")
|
|
return SendResult{Method: "terminal-sequence-json"}, nil
|
|
}
|
|
result, err := EmitSequence(seq)
|
|
printTestStatus("claude", result, err)
|
|
return result, err
|
|
}
|
|
|
|
func TestNotification(title, body string) error {
|
|
_, err := TestCursor(title, body, false)
|
|
return err
|
|
}
|