import (
"bufio"
+ "bytes"
"context"
"encoding/json"
"errors"
once sync.Once
}
-// Accept serves each client of listener until the listener closes. A new
-// client gets a ready event with the name of the agent, and a busy event when a
-// turn runs.
+// Accept serves each client of listener until the listener closes. A new client
+// gets a ready event with the name of the agent, and a busy event when a turn
+// runs.
func (h *DaemonHub) Accept(ctx context.Context, listener net.Listener, l *TurnLoop) {
for {
conn, err := listener.Accept()
err := json.Unmarshal(scanner.Bytes(), &request)
if err == nil {
- if request.Type == "backup" {
+ switch request.Type {
+ case "notify":
+ var receipt Event
+
+ decoder := json.NewDecoder(bytes.NewReader(scanner.Bytes()))
+ decoder.DisallowUnknownFields()
+
+ err = decoder.Decode(&request)
+ if err == nil {
+ receipt, err = l.messages.Notify(request.Notify)
+ }
+
+ if err == nil {
+ h.Send(client, receipt)
+ }
+ case "backup":
var size int64
size, err = backupData(ctx, filepath.Dir(l.agent.dir), request.Text, h.whatsapp, h.snapshot)
Text: fmt.Sprintf("%s (%d bytes)", request.Text, size),
})
}
- } else {
+ default:
err = daemonRunRequest(l, h.whatsapp, request, func(text string) {
h.Send(client, Event{
Kind: "notice",
}
}
-// remove removes client, closes its connection, and ends its writer. The
-// caller holds the mutex.
+// remove removes client, closes its connection, and ends its writer. The caller
+// holds the mutex.
func (h *DaemonHub) remove(client *DaemonClient) {
delete(h.clients, client)
client.once.Do(func() {
close(client.events)
- _ = client.conn.Close()
+ // The client is removed already, so a close failure cannot
+ // restore it.
+ client.conn.Close() // nolint: errcheck
})
}
h.workers.Wait()
}
+// daemonRunRequest runs one request of a client. The answers of a WhatsApp
+// command go to answer.
+func daemonRunRequest(l *TurnLoop, whatsapp *WhatsApp, request SocketRequest, answer func(text string)) error {
+ switch request.Type {
+ case "send":
+ if request.Text == "" {
+ return errors.New("empty message")
+ }
+
+ _, err := l.Submit(terminalSource, request.Text)
+ if err != nil {
+ return fmt.Errorf("submit terminal message: %w", err)
+ }
+
+ return nil
+ case "image":
+ if !filepath.IsAbs(request.Path) {
+ return errors.New("use /image PATH [caption]")
+ }
+
+ data, err := imageReadFile(request.Path)
+ if err != nil {
+ return fmt.Errorf("read terminal image: %w", err)
+ }
+
+ name, err := l.agent.Images().Save(data)
+ if err != nil {
+ return fmt.Errorf("store terminal image: %w", err)
+ }
+
+ _, err = l.Submit(terminalSource, imageText([]string{name}, request.Text))
+ if err != nil {
+ return fmt.Errorf("submit terminal image: %w", err)
+ }
+
+ return nil
+ case "interrupt":
+ if !l.Interrupt() {
+ return errors.New("no turn runs")
+ }
+
+ return nil
+ case "whatsapp":
+ whatsapp.Command(request.Text, answer)
+
+ return nil
+ case "stop":
+ found, err := l.StopSubagent(request.Text)
+ if err == nil && !found {
+ err = fmt.Errorf("no subagent %s runs", request.Text)
+ }
+
+ if err != nil {
+ return fmt.Errorf("stop subagent: %w", err)
+ }
+
+ return nil
+ default:
+ return fmt.Errorf("unknown request %q", request.Type)
+ }
+}
+
// daemon runs "onechat daemon": it owns the root, its children, and their
// files, and serves the clients on the Unix socket in the data directory until
// ctx ends. When telegram.json exists, it also runs the Telegram bot. It also
func daemon(ctx context.Context, out io.Writer) (err error) {
dir, err := dataDir()
if err != nil {
- return err
+ return fmt.Errorf("find data directory: %w", err)
}
// Check the socket path before the daemon writes a file, so that a data
socket := filepath.Join(dir, socketFileName)
err = checkSocketPath(socket)
if err != nil {
- return err
+ return fmt.Errorf("check daemon socket path: %w", err)
}
config, err := loadConfig(dir)
if err != nil {
- return err
+ return fmt.Errorf("load daemon config: %w", err)
}
botSettings, hasBot, err := loadTelegramConfig(dir)
if err != nil {
- return err
+ return fmt.Errorf("load Telegram config: %w", err)
}
tokens, err := lockTokens(dir)
if err != nil {
- return err
+ return fmt.Errorf("lock login tokens: %w", err)
}
defer func() {
logger := daemonLogger(out, "onechat: ")
whatsapp, err := openWhatsApp(ctx, dir, time.Now, daemonLogger(out, "whatsapp: "), whatsappNewClient)
if err != nil {
- return err
+ return fmt.Errorf("open WhatsApp: %w", err)
}
defer func() {
address, auth, err := modelEndpoint(config, dir)
if err != nil {
- return err
+ return fmt.Errorf("configure model endpoint: %w", err)
}
agents, err := openDaemonAgents(ctx, dir, address, auth, config.Compactor, time.Now, logger)
if err != nil {
- return err
+ return fmt.Errorf("open daemon agents: %w", err)
}
defer func() {
listener, err := daemonListenSocket(socket)
if err != nil {
- return err
+ return fmt.Errorf("listen for daemon clients: %w", err)
}
linked := "not linked"
logger.Info("Telegram bot started")
}
- return daemonServe(ctx, listener, telegram, whatsapp, schedules, func(notify func(Event)) (*TurnLoop, error) {
+ err = daemonServe(ctx, listener, telegram, whatsapp, schedules, func(notify func(Event)) (*TurnLoop, error) {
return root.Loop(whatsapp, notify)
}, agents[1:]...)
+ if err != nil {
+ return fmt.Errorf("serve daemon: %w", err)
+ }
+
+ return nil
}
// daemonListenSocket listens on the Unix socket at path, with mode 0600. The
func daemonListenSocket(path string) (net.Listener, error) {
err := checkSocketPath(path)
if err != nil {
- return nil, err
+ return nil, fmt.Errorf("check daemon socket path: %w", err)
}
err = os.Remove(path)
root: l,
messages: messages,
}
+
l.tools.children = runtime
var workers sync.WaitGroup
return errors.Join(err, <-done)
}
-
-// daemonRunRequest runs one request of a client. The answers of a WhatsApp
-// command go to answer.
-func daemonRunRequest(l *TurnLoop, whatsapp *WhatsApp, request SocketRequest, answer func(text string)) error {
- switch request.Type {
- case "send":
- if request.Text == "" {
- return errors.New("empty message")
- }
-
- _, err := l.Submit(terminalSource, request.Text)
-
- return err
- case "image":
- if !filepath.IsAbs(request.Path) {
- return errors.New("use /image PATH [caption]")
- }
-
- data, err := imageReadFile(request.Path)
- if err != nil {
- return fmt.Errorf("read terminal image: %w", err)
- }
-
- name, err := l.agent.Images().Save(data)
- if err != nil {
- return fmt.Errorf("store terminal image: %w", err)
- }
-
- _, err = l.Submit(terminalSource, imageText([]string{name}, request.Text))
- if err != nil {
- return fmt.Errorf("submit terminal image: %w", err)
- }
-
- return nil
- case "interrupt":
- if !l.Interrupt() {
- return errors.New("no turn runs")
- }
-
- return nil
- case "whatsapp":
- whatsapp.Command(request.Text, answer)
-
- return nil
- case "stop":
- found, err := l.StopSubagent(request.Text)
- if err == nil && !found {
- err = fmt.Errorf("no subagent %s runs", request.Text)
- }
-
- return err
- default:
- return fmt.Errorf("unknown request %q", request.Type)
- }
-}
projects / onechat / blobdiff