main.go RawHistoryFolder

1// Tiny Go program that drives `zut rpc` as a subprocess.
2//
3// For an in-process Go embedding, use github.com/bnema/zut/packages/agent/sdk
4// instead — this example exists to show what consumers in OTHER
5// languages have to do.
6package main
7
8import (
9	"bufio"
10	"encoding/json"
11	"fmt"
12	"io"
13	"os"
14	"os/exec"
15	"strings"
16)
17
18func main() {
19	if len(os.Args) < 2 {
20		fmt.Fprintln(os.Stderr, "usage: rpcdemo <prompt>")
21		os.Exit(2)
22	}
23	prompt := strings.Join(os.Args[1:], " ")
24
25	cmd := exec.Command("zut", "rpc")
26	cmd.Stderr = os.Stderr
27	stdin, err := cmd.StdinPipe()
28	must(err)
29	stdout, err := cmd.StdoutPipe()
30	must(err)
31	must(cmd.Start())
32
33	if tok := os.Getenv("ZUTCORE_RPC_TOKEN"); tok != "" {
34		send(stdin, map[string]any{"id": "0", "type": "hello", "token": tok})
35	}
36	send(stdin, map[string]any{
37		"id":      "1",
38		"type":    "prompt",
39		"message": prompt,
40	})
41
42	sc := bufio.NewScanner(stdout)
43	sc.Buffer(make([]byte, 0, 64*1024), 16*1024*1024)
44	for sc.Scan() {
45		var ev map[string]any
46		if err := json.Unmarshal(sc.Bytes(), &ev); err != nil {
47			continue
48		}
49		switch ev["type"] {
50		case "text_delta":
51			fmt.Print(ev["delta"])
52		case "tool_call":
53			fmt.Fprintf(os.Stderr, "\n[tool] %v\n", ev["name"])
54		case "error":
55			fmt.Fprintf(os.Stderr, "\n[error] %v\n", ev["message"])
56		case "done":
57			fmt.Println()
58			stdin.Close()
59			_ = cmd.Wait()
60			return
61		}
62	}
63}
64
65func send(w io.Writer, m map[string]any) {
66	b, _ := json.Marshal(m)
67	_, _ = w.Write(append(b, '\n'))
68}
69
70func must(err error) {
71	if err != nil {
72		fmt.Fprintln(os.Stderr, err)
73		os.Exit(1)
74	}
75}