v0.3.2 #3

Merged
ScuroNeko merged 4 commits from dev into main 2026-01-29 11:47:30 +03:00
3 changed files with 58 additions and 1 deletions
Showing only changes of commit f1bb4b62c1 - Show all commits

47
api.go
View File

@@ -16,6 +16,53 @@ type ApiResponse struct {
Result any `json:"result,omitempty"` Result any `json:"result,omitempty"`
ErrorCode int `json:"error_code,omitempty"` ErrorCode int `json:"error_code,omitempty"`
} }
type ApiResponseG[R any] struct {
Ok bool `json:"ok"`
Description string `json:"description,omitempty"`
Result R `json:"result,omitempty"`
ErrorCode int `json:"error_code,omitempty"`
}
type TelegramRequest[R, P any] struct {
method string
params P
}
type EmptyParams struct{}
func NewRequest[R, P any](method string, params P) TelegramRequest[R, P] {
return TelegramRequest[R, P]{
method: method,
params: params,
}
}
func (r TelegramRequest[R, P]) Do(bot *Bot) (*R, error) {
var buf bytes.Buffer
err := json.NewEncoder(&buf).Encode(r.params)
if err != nil {
return nil, err
}
req, err := http.Post(fmt.Sprintf("https://api.telegram.org/bot%s/%s", bot.token, r.method), "application/json", &buf)
if err != nil {
return nil, err
}
defer req.Body.Close()
data, err := io.ReadAll(req.Body)
if err != nil {
return nil, err
}
response := new(ApiResponseG[R])
err = json.Unmarshal(data, &response)
if err != nil {
return nil, err
}
if !response.Ok {
return nil, fmt.Errorf("[%d] %s", response.ErrorCode, response.Description)
}
return &response.Result, nil
}
// request is a low-level call to api. // request is a low-level call to api.
func (b *Bot) request(methodName string, params any) (map[string]any, error) { func (b *Bot) request(methodName string, params any) (map[string]any, error) {

View File

@@ -22,7 +22,7 @@ func (b *Bot) Updates() ([]*Update, error) {
if err != nil { if err != nil {
return res, err return res, err
} }
for _, u := range res { for _, u := range res {
b.updateOffset = u.UpdateID + 1 b.updateOffset = u.UpdateID + 1
err = b.updateQueue.Enqueue(u) err = b.updateQueue.Enqueue(u)
@@ -43,6 +43,8 @@ func (b *Bot) Updates() ([]*Update, error) {
} }
func (b *Bot) GetMe() (*User, error) { func (b *Bot) GetMe() (*User, error) {
//req := NewRequest[User, EmptyParams]("getMe", EmptyParams{})
//user, err := req.Do(b)
data, err := b.request("getMe", NoParams) data, err := b.request("getMe", NoParams)
if err != nil { if err != nil {
return nil, err return nil, err
@@ -69,6 +71,9 @@ type SendMessageP struct {
} }
func (b *Bot) SendMessage(params *SendMessageP) (*Message, error) { func (b *Bot) SendMessage(params *SendMessageP) (*Message, error) {
req := NewRequest[Message, SendMessageP]("sendMessage", *params)
return req.Do(b)
data, err := b.request("sendMessage", params) data, err := b.request("sendMessage", params)
if err != nil { if err != nil {
return nil, err return nil, err

View File

@@ -61,10 +61,15 @@ func (b *Bot) ExecRunners() {
} }
}() }()
} else if !runner.Async && runner.Onetime { } else if !runner.Async && runner.Onetime {
t := time.Now()
err := runner.Fn(b) err := runner.Fn(b)
if err != nil { if err != nil {
b.logger.Warnf("Runner %s failed: %s\n", runner.Name, err) b.logger.Warnf("Runner %s failed: %s\n", runner.Name, err)
} }
elapsed := time.Since(t)
if elapsed > time.Second*2 {
b.logger.Warnf("Runner %s too slow. Elapsed time %s>=2s", runner.Name, elapsed)
}
} else if !runner.Onetime { } else if !runner.Onetime {
go func() { go func() {
for { for {