]> git.giacomolli.uk - onechat/blobdiff - telegram.go
Retry the first WhatsApp connection
[onechat] / telegram.go
index 16eff80c473eaba73d6cd5291b6aee7cdfc63a57..af1b91e8a899aa8ef1f54f6344844574693eed91 100644 (file)
@@ -145,7 +145,9 @@ func (c TelegramConfig) validate() error {
 // makes Telegram send the update again, and the message repeats once.
 //
 // A failed Bot API call waits and tries again, from minWait up to maxWait, or
-// as long as the Bot API asks. It never stops the agent.
+// as long as the Bot API asks. It never stops the agent. A TelegramOutage logs
+// the first failure and the end of each run of failures, and the retries
+// between them log nothing.
 type TelegramBot struct {
        api        *TelegramAPI
        owner      int64
@@ -160,10 +162,15 @@ type TelegramBot struct {
 }
 
 // Notify queues event for the chat. It never blocks, so a slow network does not
-// stop the agent.
+// stop the agent. The WhatsApp batches and the runs of the schedules do not go
+// to the chat: the reply of the agent tells the user what needs the user.
 func (b *TelegramBot) Notify(event Event) {
        switch event.Kind {
-       case "busy", "idle", "reply", "error", "notice", "subagent", "work":
+       case "busy", "idle", "reply", "error", "notice", "subagent":
+       case "work":
+               if event.Source == whatsappSource || event.Source == scheduleSource {
+                       return
+               }
        case "user":
                if event.Source == telegramSource {
                        return
@@ -253,6 +260,12 @@ func (b *TelegramBot) poll(ctx context.Context, l *TurnLoop, whatsapp *WhatsApp)
                return err
        }
 
+       outage := &TelegramOutage{
+               logger: b.logger,
+               loss:   "receive failed",
+               back:   "receive back",
+       }
+
        wait := b.minWait
        for ctx.Err() == nil {
                updates, err := b.api.GetUpdates(ctx, offset)
@@ -262,11 +275,12 @@ func (b *TelegramBot) poll(ctx context.Context, l *TurnLoop, whatsapp *WhatsApp)
 
                if err != nil {
                        _, asked := telegramRetryable(err)
-                       wait = b.pause(ctx, err, wait, asked)
+                       wait = b.pause(ctx, outage, err, wait, asked)
 
                        continue
                }
 
+               outage.End()
                wait = b.minWait
                for _, update := range updates {
                        if ctx.Err() != nil {
@@ -330,11 +344,11 @@ func (b *TelegramBot) post(ctx context.Context, event Event) {
        }
 }
 
-// pause logs err and waits the longer of wait and asked, or until ctx ends. It
-// returns the wait of the next failure: twice wait, up to maxWait.
-func (b *TelegramBot) pause(ctx context.Context, err error, wait, asked time.Duration) time.Duration {
+// pause gives err to outage and waits the longer of wait and asked, or until
+// ctx ends. It returns the wait of the next failure: twice wait, up to maxWait.
+func (b *TelegramBot) pause(ctx context.Context, outage *TelegramOutage, err error, wait, asked time.Duration) time.Duration {
        delay := max(wait, asked)
-       b.logger.Printf("%v (next try in %s)", err, delay)
+       outage.Fail(err, delay)
 
        timer := time.NewTimer(delay)
        defer timer.Stop()
@@ -357,7 +371,7 @@ func (b *TelegramBot) pause(ctx context.Context, err error, wait, asked time.Dur
 func (b *TelegramBot) handle(ctx context.Context, l *TurnLoop, whatsapp *WhatsApp, update TelegramUpdate) error {
        message := update.Message
        if message == nil || message.Chat.Type != "private" || message.Chat.ID != b.owner || message.From == nil || message.From.ID != b.owner {
-               b.logger.Print("update ignored: not a message of the owner in the private chat")
+               b.logger.Print("message ignored: not from the owner")
                return nil
        }
 
@@ -401,28 +415,33 @@ func (b *TelegramBot) handle(ctx context.Context, l *TurnLoop, whatsapp *WhatsAp
 // reply is not lost to a bad entity. It tries again after each failure that a
 // later try can fix, until ctx ends. It drops the part after any other failure.
 func (b *TelegramBot) deliver(ctx context.Context, part TelegramPart) {
+       outage := b.sendOutage()
        wait := b.minWait
        for {
                err := b.api.SendMessage(ctx, b.owner, part.text, part.entities)
+               retry, asked := telegramRetryable(err)
+               if !retry {
+                       outage.End()
+               }
+
                if err == nil || ctx.Err() != nil {
                        return
                }
 
                var answer *TelegramError
                if len(part.entities) > 0 && errors.As(err, &answer) && answer.Status == http.StatusBadRequest {
-                       b.logger.Printf("entities refused, the message goes again as plain text: %v", err)
+                       b.logger.Printf("reply sent as plain text: %v", err)
                        part.entities = nil
 
                        continue
                }
 
-               retry, asked := telegramRetryable(err)
                if !retry {
-                       b.logger.Printf("message dropped: %v", err)
+                       b.logger.Printf("reply lost: %v", err)
                        return
                }
 
-               wait = b.pause(ctx, err, wait, asked)
+               wait = b.pause(ctx, outage, err, wait, asked)
        }
 }
 
@@ -434,15 +453,21 @@ func (b *TelegramBot) deliver(ctx context.Context, part TelegramPart) {
 // the model wrote it. It tries again after each failure that a later try can
 // fix, until ctx ends. It drops the message after any other failure.
 func (b *TelegramBot) deliverRich(ctx context.Context, markdown string) {
+       outage := b.sendOutage()
        wait := b.minWait
        for {
                err := b.api.SendRichMessage(ctx, b.owner, markdown)
+               retry, asked := telegramRetryable(err)
+               if !retry {
+                       outage.End()
+               }
+
                if err == nil || ctx.Err() != nil {
                        return
                }
 
                if telegramRejected(err) {
-                       b.logger.Printf("rich message refused, the reply goes with entities: %v", err)
+                       b.logger.Printf("reply sent with simple formatting: %v", err)
                        for _, part := range telegramReplyParts(telegramUnescapeTags(markdown)) {
                                b.deliver(ctx, part)
                        }
@@ -450,13 +475,21 @@ func (b *TelegramBot) deliverRich(ctx context.Context, markdown string) {
                        return
                }
 
-               retry, asked := telegramRetryable(err)
                if !retry {
-                       b.logger.Printf("message dropped: %v", err)
+                       b.logger.Printf("reply lost: %v", err)
                        return
                }
 
-               wait = b.pause(ctx, err, wait, asked)
+               wait = b.pause(ctx, outage, err, wait, asked)
+       }
+}
+
+// sendOutage returns a new outage of the messages to the chat.
+func (b *TelegramBot) sendOutage() *TelegramOutage {
+       return &TelegramOutage{
+               logger: b.logger,
+               loss:   "send failed",
+               back:   "send back",
        }
 }
 
@@ -468,6 +501,37 @@ func (b *TelegramBot) notice(text string) {
        })
 }
 
+// TelegramOutage is one run of failures of a kind of Bot API call. The first
+// failure logs loss, its cause, and the wait of the next try. The retries log
+// nothing. The next answer of Telegram logs back and the length of the outage.
+// loss and back say what the user loses and gets again.
+type TelegramOutage struct {
+       logger *log.Logger
+       loss   string
+       back   string
+       start  time.Time
+}
+
+// Fail logs the start of an outage with err and delay, unless an outage runs.
+func (o *TelegramOutage) Fail(err error, delay time.Duration) {
+       if !o.start.IsZero() {
+               return
+       }
+
+       o.start = time.Now()
+       o.logger.Printf("%s: %v (retry in %s)", o.loss, err, durationText(delay))
+}
+
+// End logs the end of the outage that runs, with its length.
+func (o *TelegramOutage) End() {
+       if o.start.IsZero() {
+               return
+       }
+
+       o.logger.Printf("%s after %s", o.back, durationText(time.Since(o.start)))
+       o.start = time.Time{}
+}
+
 // TelegramPart is the text of one message to the chat, at most
 // telegramMessageUnits long, and its entities.
 type TelegramPart struct {