telebot: add username field to bot

yfzhe · Jan 27, 2024 · c4a53a2 · c4a53a2
1 parent 42c39f7
commit c4a53a2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/telebot/main.rkt b/telebot/main.rkt
@@ -22,7 +22,12 @@
   ((webhook-url string? "url"))
   -> true?)
 
+(define (bot-init! bot)
+  (define me (bot-get-me bot))
+  (set-bot-username! bot (ref (me : user) .username #f)))
+
 (define (bot-start/poll bot handle-update)
+  (bot-init! bot)
   (let loop ([offset 0] [updates '()])
     (cond
       [(null? updates)
@@ -34,6 +39,7 @@
        (loop (add1 (ref (update : update) .id)) (cdr updates))])))
 
 (define (bot-start/webhook bot handle-update webhook-base port)
+  (bot-init! bot)
   (bot-set-webhook bot
                    #:webhook-url (string-append webhook-base "/webhook"))
 

diff --git a/telebot/private/bot.rkt b/telebot/private/bot.rkt
@@ -3,12 +3,13 @@
          "error.rkt")
 
 (provide bot? make-bot
+         set-bot-username!
          bot-post)
 
-(struct bot (token))
+(struct bot (token [username #:mutable]))
 
 (define (make-bot token)
-  (bot token))
+  (bot token #f))
 
 (define *tg-api-base* "https://api.telegram.org/bot")