mirror of
https://github.com/lukaszraczylo/go-telegram.git
synced 2026-06-10 23:09:04 +00:00
Initial release of go-telegram
A fully-generated, strongly-typed Go client for the Telegram Bot API. * 176 methods + 301 types generated from Bot API v10.0 * 1408 auto-generated tests (8 scenarios per method) * Typed unions throughout — no 'any' in the public surface * Pluggable HTTP transport and JSON codec (default goccy/go-json) * Built-in retry middleware honouring Telegram's retry_after * Generic dispatcher with filters and conversation handlers * Self-verifying codegen pipeline (regen → audit → emit → run tests) * 14 example bots covering common patterns
This commit is contained in:
@@ -0,0 +1,29 @@
|
||||
# polls
|
||||
|
||||
Create polls and tally answers in real time via `OnPollAnswer`.
|
||||
|
||||
## What it shows
|
||||
|
||||
- `api.SendPoll` with `[]api.InputPollOption` and `IsAnonymous: false`
|
||||
- `router.OnPollAnswer` to receive vote updates (`PollAnswer.OptionIds`, `PollAnswer.User`)
|
||||
- Concurrent-safe in-memory tally with `sync.Mutex`
|
||||
|
||||
## Commands
|
||||
|
||||
| Command | Description |
|
||||
|---|---|
|
||||
| `/poll <question>` | Creates a poll with four preset options (A/B/C/D) |
|
||||
| `/tally <poll_id>` | Shows current vote counts for a poll |
|
||||
|
||||
## Notes
|
||||
|
||||
- `OnPollAnswer` only fires for **non-anonymous** polls. For anonymous polls, Telegram does not send user identifiers.
|
||||
- The poll ID is logged when the poll is created; copy it to use with `/tally`.
|
||||
- Vote tallies are in-memory and reset on restart.
|
||||
|
||||
## Running
|
||||
|
||||
```bash
|
||||
export TELEGRAM_BOT_TOKEN=123456:ABC...
|
||||
go run ./examples/polls
|
||||
```
|
||||
@@ -0,0 +1,146 @@
|
||||
// Package main demonstrates creating polls and tallying answers via OnPollAnswer.
|
||||
//
|
||||
// Usage: send "/poll <question>" in a group or private chat.
|
||||
// The bot creates a non-anonymous poll with four preset options and tallies votes.
|
||||
//
|
||||
// TELEGRAM_BOT_TOKEN=xxx go run ./examples/polls
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"os/signal"
|
||||
"strings"
|
||||
"sync"
|
||||
"syscall"
|
||||
|
||||
"github.com/lukaszraczylo/go-telegram/api"
|
||||
"github.com/lukaszraczylo/go-telegram/client"
|
||||
"github.com/lukaszraczylo/go-telegram/dispatch"
|
||||
"github.com/lukaszraczylo/go-telegram/transport"
|
||||
)
|
||||
|
||||
// pollTally maps pollID → optionIndex → voteCount.
|
||||
type pollTally struct {
|
||||
mu sync.Mutex
|
||||
votes map[string]map[int64]int
|
||||
}
|
||||
|
||||
func newPollTally() *pollTally {
|
||||
return &pollTally{votes: make(map[string]map[int64]int)}
|
||||
}
|
||||
|
||||
func (t *pollTally) record(pollID string, opts []int64) {
|
||||
t.mu.Lock()
|
||||
defer t.mu.Unlock()
|
||||
if t.votes[pollID] == nil {
|
||||
t.votes[pollID] = make(map[int64]int)
|
||||
}
|
||||
for _, opt := range opts {
|
||||
t.votes[pollID][opt]++
|
||||
}
|
||||
}
|
||||
|
||||
func (t *pollTally) summary(pollID string) string {
|
||||
t.mu.Lock()
|
||||
defer t.mu.Unlock()
|
||||
m := t.votes[pollID]
|
||||
if len(m) == 0 {
|
||||
return "No votes yet."
|
||||
}
|
||||
var sb strings.Builder
|
||||
fmt.Fprintf(&sb, "Tally for poll %s:\n", pollID)
|
||||
for opt, count := range m {
|
||||
fmt.Fprintf(&sb, " Option %d: %d vote(s)\n", opt, count)
|
||||
}
|
||||
return sb.String()
|
||||
}
|
||||
|
||||
var pollOptions = []api.InputPollOption{
|
||||
{Text: "Option A"},
|
||||
{Text: "Option B"},
|
||||
{Text: "Option C"},
|
||||
{Text: "Option D"},
|
||||
}
|
||||
|
||||
func main() {
|
||||
token := os.Getenv("TELEGRAM_BOT_TOKEN")
|
||||
if token == "" {
|
||||
log.Fatal("TELEGRAM_BOT_TOKEN required")
|
||||
}
|
||||
|
||||
ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
|
||||
defer stop()
|
||||
|
||||
bot := client.New(token,
|
||||
client.WithHTTPClient(client.NewRetryDoer(client.NewDefaultHTTPDoer())),
|
||||
)
|
||||
|
||||
tally := newPollTally()
|
||||
router := dispatch.New(bot)
|
||||
|
||||
router.OnCommand("/poll", func(c *dispatch.Context, m *api.Message) error {
|
||||
question := strings.TrimSpace(m.Text)
|
||||
// Strip the "/poll" command prefix.
|
||||
if after, ok := strings.CutPrefix(question, "/poll"); ok {
|
||||
question = strings.TrimSpace(after)
|
||||
}
|
||||
if question == "" {
|
||||
_, _ = api.SendMessage(c.Ctx, c.Bot, &api.SendMessageParams{
|
||||
ChatID: api.ChatIDFromInt(m.Chat.ID),
|
||||
Text: "Usage: /poll <question>",
|
||||
})
|
||||
return nil
|
||||
}
|
||||
isAnon := false
|
||||
msg, err := api.SendPoll(c.Ctx, c.Bot, &api.SendPollParams{
|
||||
ChatID: api.ChatIDFromInt(m.Chat.ID),
|
||||
Question: question,
|
||||
Options: pollOptions,
|
||||
IsAnonymous: &isAnon,
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if msg != nil && msg.Poll != nil {
|
||||
log.Printf("poll created: id=%s question=%q", msg.Poll.ID, msg.Poll.Question)
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
||||
router.OnCommand("/tally", func(c *dispatch.Context, m *api.Message) error {
|
||||
pollID := strings.TrimSpace(m.Text)
|
||||
if after, ok := strings.CutPrefix(pollID, "/tally"); ok {
|
||||
pollID = strings.TrimSpace(after)
|
||||
}
|
||||
if pollID == "" {
|
||||
_, _ = api.SendMessage(c.Ctx, c.Bot, &api.SendMessageParams{
|
||||
ChatID: api.ChatIDFromInt(m.Chat.ID),
|
||||
Text: "Usage: /tally <poll_id>",
|
||||
})
|
||||
return nil
|
||||
}
|
||||
_, _ = api.SendMessage(c.Ctx, c.Bot, &api.SendMessageParams{
|
||||
ChatID: api.ChatIDFromInt(m.Chat.ID),
|
||||
Text: tally.summary(pollID),
|
||||
})
|
||||
return nil
|
||||
})
|
||||
|
||||
router.OnPollAnswer(func(c *dispatch.Context, pa *api.PollAnswer) error {
|
||||
userID := int64(0)
|
||||
if pa.User != nil {
|
||||
userID = pa.User.ID
|
||||
}
|
||||
log.Printf("poll answer: poll=%s user=%d options=%v", pa.PollID, userID, pa.OptionIds)
|
||||
tally.record(pa.PollID, pa.OptionIds)
|
||||
return nil
|
||||
})
|
||||
|
||||
poller := transport.NewLongPoller(bot)
|
||||
if err := router.Run(ctx, poller); err != nil && err != context.Canceled {
|
||||
log.Printf("router exited: %v", err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user