Compare commits
3 Commits
Author | SHA1 | Date | |
---|---|---|---|
7185d84eac | |||
d3245fd7c0 | |||
1cca2bc8a5 |
4
go.mod
4
go.mod
@ -1,5 +1,5 @@
|
||||
module go.balki.me/tglistbot
|
||||
|
||||
go 1.20
|
||||
go 1.21
|
||||
|
||||
require go.balki.me/anyhttp v0.2.0
|
||||
require go.balki.me/anyhttp v0.3.0
|
||||
|
4
go.sum
4
go.sum
@ -1,2 +1,2 @@
|
||||
go.balki.me/anyhttp v0.2.0 h1:W6aGcmjF5CMJvJYtbYCywxnYoErFhFc76vwaqUG5FAQ=
|
||||
go.balki.me/anyhttp v0.2.0/go.mod h1:JhfekOIjgVODoVqUCficjpIgmB3wwlB7jhN0eN2EZ/s=
|
||||
go.balki.me/anyhttp v0.3.0 h1:WtBQ0rnkg567sX/O4ij/+qBbdCIUt5VURSe718sITBY=
|
||||
go.balki.me/anyhttp v0.3.0/go.mod h1:JhfekOIjgVODoVqUCficjpIgmB3wwlB7jhN0eN2EZ/s=
|
||||
|
54
main.go
54
main.go
@ -3,6 +3,7 @@ package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
@ -16,6 +17,7 @@ import (
|
||||
"time"
|
||||
|
||||
"go.balki.me/anyhttp"
|
||||
"go.balki.me/anyhttp/idle"
|
||||
"go.balki.me/tglistbot/glist"
|
||||
)
|
||||
|
||||
@ -25,13 +27,15 @@ var apiToken string
|
||||
|
||||
var usage string = `Telegram List bot
|
||||
Environment variables:
|
||||
TGLB_API_TOKEN (required): See https://core.telegram.org/bots#how-do-i-create-a-bot
|
||||
TGLB_PORT (default 28923): Set numerical port or unix//run/path.sock for unix socket
|
||||
TGLB_DATA_PATH (default .): Directory path where list data is stored
|
||||
TGLB_API_TOKEN (required) : See https://core.telegram.org/bots#how-do-i-create-a-bot
|
||||
TGLB_ADDR (default 28923) : See https://pkg.go.dev/go.balki.me/anyhttp#readme-address-syntax
|
||||
TGLB_DATA_PATH (default .) : Directory path where list data is stored
|
||||
TGLB_TIMEOUT (default 30m) : Timeout to auto shutdown if using systemd-fd, See https://pkg.go.dev/time#ParseDuration
|
||||
`
|
||||
|
||||
func main() {
|
||||
|
||||
log.SetFlags(log.Flags() | log.Lshortfile)
|
||||
apiToken = os.Getenv("TGLB_API_TOKEN")
|
||||
|
||||
if apiToken == "" {
|
||||
@ -40,7 +44,7 @@ func main() {
|
||||
}
|
||||
|
||||
addr := func() string {
|
||||
addr := os.Getenv("TGLB_PORT")
|
||||
addr := os.Getenv("TGLB_ADDR")
|
||||
if addr == "" {
|
||||
return "28923"
|
||||
}
|
||||
@ -59,6 +63,18 @@ func main() {
|
||||
return dataPath
|
||||
}()
|
||||
|
||||
timeout := func() time.Duration {
|
||||
timeoutStr := os.Getenv("TGLB_TIMEOUT")
|
||||
if timeoutStr == "" {
|
||||
return 30 * time.Minute
|
||||
}
|
||||
timeout, err := time.ParseDuration(timeoutStr)
|
||||
if err != nil {
|
||||
log.Panicf("Invalid timeout: %q\n", timeoutStr)
|
||||
}
|
||||
return timeout
|
||||
}()
|
||||
|
||||
glist.DataPath = dataPath
|
||||
|
||||
version := func() string {
|
||||
@ -124,6 +140,16 @@ func main() {
|
||||
if update.Message != nil && update.Message.Text != "" && update.Message.Text[0] != '/' {
|
||||
|
||||
chatID := update.Message.Chat.ID
|
||||
if len(update.Message.Text) > 60 {
|
||||
replyURL := fmt.Sprintf("https://api.telegram.org/bot%s/sendMessage?chat_id=%d&reply_to_message_id=%d&text=ItemTooLong-Max60", apiToken, chatID, update.Message.ID)
|
||||
resp, err := http.Get(replyURL)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
return
|
||||
}
|
||||
logBody(resp.Body)
|
||||
return
|
||||
}
|
||||
g, _ := chats.LoadOrStore(chatID, glist.NewGList(chatID))
|
||||
gl := g.(*glist.GList)
|
||||
go handleTextAdded(gl, update.Message.Text)
|
||||
@ -143,7 +169,24 @@ func main() {
|
||||
|
||||
}
|
||||
})
|
||||
log.Panicln(anyhttp.ListenAndServe(addr, nil))
|
||||
addrType, server, done, err := anyhttp.Serve(addr, idle.WrapHandler(nil))
|
||||
if err != nil {
|
||||
log.Panicln(err)
|
||||
}
|
||||
if addrType == anyhttp.SystemdFD {
|
||||
log.Println("server started")
|
||||
if err := idle.Wait(timeout); err != nil {
|
||||
log.Panicln(err)
|
||||
}
|
||||
log.Printf("server idle for %v, shutting down\n", timeout)
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 1*time.Minute) // Don't want any stuck connections
|
||||
defer cancel()
|
||||
if err := server.Shutdown(ctx); err != nil {
|
||||
log.Panicln(err)
|
||||
}
|
||||
} else {
|
||||
<-done
|
||||
}
|
||||
}
|
||||
|
||||
func handleTextAdded(gl *glist.GList, text string) {
|
||||
@ -209,6 +252,7 @@ func sendList(gl *glist.GList, method glist.SendMethod) []byte {
|
||||
log.Println(err)
|
||||
return nil
|
||||
}
|
||||
log.Println(string(sendMsgReq))
|
||||
resp, err := http.Post(url, "application/json", bytes.NewReader(sendMsgReq))
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
|
Reference in New Issue
Block a user