X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/0bc8db71aa5d2c55b215a7a15b073bb85a1172eb..12bd9e96f3b3e67290484e3829461e054110db27:/terminal.go?ds=inline diff --git a/terminal.go b/terminal.go index 203cb5e..da0d78a 100644 --- a/terminal.go +++ b/terminal.go @@ -23,9 +23,10 @@ var errTerminalDaemonClosed = errors.New("the daemon closed the connection") // terminal runs "onechat" and "onechat chat": the terminal client of the // daemon. Each line of input is a user message, the line /stop interrupts the // running turn, the line /stop NAME stops the subagent NAME, and a line that -// starts with /whatsapp is a WhatsApp command. The client shows the replies, -// the tool calls, and their results as they happen. It ends at the end of the -// input or when the daemon closes the connection. +// starts with /whatsapp is a WhatsApp command. /image PATH [caption] sends an +// image from the client working directory. The client shows the replies, the +// tool calls, and their results as they happen. It ends at the end of the input +// or when the daemon closes the connection. func terminal(ctx context.Context, in io.Reader, out io.Writer) error { dir, err := dataDir() if err != nil { @@ -99,6 +100,7 @@ func terminalSendLines(ctx context.Context, conn net.Conn, in io.Reader) error { name, stop := stopSubagentName(line) args, whatsapp := whatsappCommandArgs(line) + image, isImage := terminalImageRequest(line) switch { case strings.TrimSpace(line) == "": continue @@ -111,6 +113,16 @@ func terminalSendLines(ctx context.Context, conn net.Conn, in io.Reader) error { Type: "stop", Text: name, } + case isImage: + request = image + if request.Path != "" { + var err error + + request.Path, err = filepath.Abs(request.Path) + if err != nil { + return fmt.Errorf("find terminal image path: %w", err) + } + } case whatsapp: request = SocketRequest{ Type: "whatsapp", @@ -134,8 +146,10 @@ func terminalSendLines(ctx context.Context, conn net.Conn, in io.Reader) error { // terminalShowEvents writes each event from the daemon to out until the // connection closes. Only replies and work messages use Markdown styles, and -// only when out is a terminal. +// only when out is a terminal. Terminal output ends each shown idle event with +// an input prompt and clears that prompt before the next shown event. func terminalShowEvents(conn net.Conn, out io.Writer) error { + busy, prompt := false, false scanner := bufio.NewScanner(conn) scanner.Buffer(make([]byte, 64*1024), socketRequestBytes) @@ -157,7 +171,24 @@ func terminalShowEvents(conn net.Conn, out io.Writer) error { text = strings.Join(terminalMarkdown(text, columns), "\n") } - _, err = io.WriteString(out, text+"\n") + switch event.Kind { + case "busy": + busy = true + case "idle": + busy = false + } + + text += "\n" + if prompt { + text = "\r\x1b[2K" + text + } + + prompt = columns > 0 && !busy + if prompt { + text += "> " + } + + _, err = io.WriteString(out, text) if err != nil { return fmt.Errorf("write terminal event: %w", err) }