]> git.giacomolli.uk - onechat/blobdiff - telegram.go
Give the rebuild its own call limit
[onechat] / telegram.go
index af1b91e8a899aa8ef1f54f6344844574693eed91..2a9345a3a4b6823c0a17039c207dcf437f12d640 100644 (file)
@@ -73,8 +73,10 @@ func loadTelegramConfig(dir string) (TelegramConfig, bool, error) {
        }
 
        var config TelegramConfig
+
        decoder := json.NewDecoder(bytes.NewReader(data))
        decoder.DisallowUnknownFields()
+
        err = decoder.Decode(&config)
        if err == nil && !errors.Is(decoder.Decode(&struct{}{}), io.EOF) {
                err = errors.New("data after the JSON object")
@@ -180,6 +182,7 @@ func (b *TelegramBot) Notify(event Event) {
        }
 
        b.mutex.Lock()
+
        b.queue = append(b.queue, event)
        b.mutex.Unlock()
 
@@ -197,7 +200,9 @@ func (b *TelegramBot) Run(ctx context.Context, l *TurnLoop, whatsapp *WhatsApp)
        defer cancel()
 
        var sender sync.WaitGroup
+
        sender.Add(1)
+
        go func() {
                defer sender.Done()
 
@@ -205,6 +210,7 @@ func (b *TelegramBot) Run(ctx context.Context, l *TurnLoop, whatsapp *WhatsApp)
        }()
 
        err := b.poll(ctx, l, whatsapp)
+
        cancel()
        sender.Wait()
 
@@ -237,6 +243,7 @@ func (b *TelegramBot) send(ctx context.Context) {
                switch event.Kind {
                case "busy":
                        busy = true
+
                        b.typing(ctx)
                        ticker.Reset(telegramTypingInterval)
                case "idle":
@@ -281,6 +288,7 @@ func (b *TelegramBot) poll(ctx context.Context, l *TurnLoop, whatsapp *WhatsApp)
                }
 
                outage.End()
+
                wait = b.minWait
                for _, update := range updates {
                        if ctx.Err() != nil {
@@ -372,6 +380,7 @@ func (b *TelegramBot) handle(ctx context.Context, l *TurnLoop, whatsapp *WhatsAp
        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("message ignored: not from the owner")
+
                return nil
        }
 
@@ -429,8 +438,10 @@ func (b *TelegramBot) deliver(ctx context.Context, part TelegramPart) {
                }
 
                var answer *TelegramError
+
                if len(part.entities) > 0 && errors.As(err, &answer) && answer.Status == http.StatusBadRequest {
                        b.logger.Printf("reply sent as plain text: %v", err)
+
                        part.entities = nil
 
                        continue
@@ -438,6 +449,7 @@ func (b *TelegramBot) deliver(ctx context.Context, part TelegramPart) {
 
                if !retry {
                        b.logger.Printf("reply lost: %v", err)
+
                        return
                }
 
@@ -468,6 +480,7 @@ func (b *TelegramBot) deliverRich(ctx context.Context, markdown string) {
 
                if telegramRejected(err) {
                        b.logger.Printf("reply sent with simple formatting: %v", err)
+
                        for _, part := range telegramReplyParts(telegramUnescapeTags(markdown)) {
                                b.deliver(ctx, part)
                        }
@@ -477,6 +490,7 @@ func (b *TelegramBot) deliverRich(ctx context.Context, markdown string) {
 
                if !retry {
                        b.logger.Printf("reply lost: %v", err)
+
                        return
                }
 
@@ -529,6 +543,7 @@ func (o *TelegramOutage) End() {
        }
 
        o.logger.Printf("%s after %s", o.back, durationText(time.Since(o.start)))
+
        o.start = time.Time{}
 }
 
@@ -557,6 +572,7 @@ func telegramLoadOffset(path string) (int64, error) {
        }
 
        var file TelegramOffsetFile
+
        err = json.Unmarshal(data, &file)
        if err != nil {
                return 0, fmt.Errorf("%s: %w", path, err)
@@ -588,6 +604,7 @@ func telegramSaveOffset(path string, offset int64) error {
 // never cuts a character.
 func telegramSplitMessage(text string, limit int) []string {
        var parts []string
+
        for {
                end := telegramPrefixEnd(text, limit)
                if end == len(text) {
@@ -637,6 +654,7 @@ func telegramText(event Event) string {
 // not cut an entity of telegramEscapeTags. It never cuts a character.
 func telegramSplitRich(markdown string, limit int) []string {
        var parts []string
+
        for len(markdown) > limit {
                cut, fence, opening := telegramRichCut(markdown, limit)
                if fence == "" {
@@ -692,7 +710,9 @@ func telegramPrefixEnd(text string, limit int) int {
 // so each cut makes the rest shorter.
 func telegramRichCut(markdown string, limit int) (int, string, string) {
        blank, line, code := 0, 0, 0
+
        var fence, opening, inFence, inOpening string
+
        for start := 0; start < limit; {
                end := strings.IndexByte(markdown[start:], '\n')
                if end < 0 || start+end+1 > limit {
@@ -751,10 +771,13 @@ func telegramRichCut(markdown string, limit int) (int, string, string) {
 // An entity that crosses a cut becomes one entity in each part.
 func telegramSplitEntities(text string, entities []TelegramEntity, limit int) []TelegramPart {
        var parts []TelegramPart
+
        start := 0
        for _, part := range telegramSplitMessage(text, limit) {
                end := start + telegramUTF16Units(part)
+
                var kept []TelegramEntity
+
                for _, entity := range entities {
                        from := max(entity.Offset, start)
                        to := min(entity.Offset+entity.Length, end)