4 Commits

Author SHA1 Message Date
dd62f237b0 Replace unix socket logic with go.balki.me/anyhttp
Now also support systemd socket activation
2023-04-20 17:59:40 -04:00
107d90deb8 fix typo in Makefile 2023-04-05 11:33:02 -04:00
6f111a42e9 Add usage 2023-04-05 11:28:16 -04:00
2aa043b5fc Add main.Version 2023-04-05 11:17:07 -04:00
5 changed files with 40 additions and 44 deletions

1
.gitignore vendored Normal file
View File

@ -0,0 +1 @@
tglistbot*

12
Makefile Normal file
View File

@ -0,0 +1,12 @@
VERSION = $(shell git describe --tags --abbrev=0)
build_release:
CGO_ENABLED=0 go build -buildmode=pie -ldflags "-X main.Version=$(VERSION)" -o tglistbot-$(VERSION)
release_check:
go list -m go.balki.me/tglistbot@$(VERSION)
clean:
rm -i tglistbot* || true

2
go.mod
View File

@ -1,3 +1,5 @@
module go.balki.me/tglistbot module go.balki.me/tglistbot
go 1.20 go 1.20
require go.balki.me/anyhttp v0.1.0

2
go.sum Normal file
View File

@ -0,0 +1,2 @@
go.balki.me/anyhttp v0.1.0 h1:ULzLWS1pRWMEduHHJxXCbvxoTmxNaWSNANV9gQ0Pigw=
go.balki.me/anyhttp v0.1.0/go.mod h1:JhfekOIjgVODoVqUCficjpIgmB3wwlB7jhN0eN2EZ/s=

67
main.go
View File

@ -4,47 +4,47 @@ package main
import ( import (
"bytes" "bytes"
"encoding/json" "encoding/json"
"errors"
"fmt" "fmt"
"io" "io"
"io/fs"
"log" "log"
"net"
"net/http" "net/http"
"os" "os"
"path" "path"
"runtime/debug" "runtime/debug"
"strconv"
"strings" "strings"
"sync" "sync"
"time" "time"
"go.balki.me/anyhttp"
"go.balki.me/tglistbot/glist" "go.balki.me/tglistbot/glist"
) )
// Version will be set from build commandline
var Version string
var apiToken string 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
`
func main() { func main() {
apiToken = os.Getenv("TGLB_API_TOKEN") apiToken = os.Getenv("TGLB_API_TOKEN")
if apiToken == "" { if apiToken == "" {
log.Panicln("TG_API_TOKEN is empty") log.Print(usage)
log.Panicln("TGLB_API_TOKEN is empty")
} }
port, unixSocketPath := func() (int, string) { addr := func() string {
portStr := os.Getenv("TGLB_PORT") addr := os.Getenv("TGLB_PORT")
if addr == "" {
defaultPort := 28923 return "28923"
if strings.HasPrefix(portStr, "unix/") {
return defaultPort, strings.TrimPrefix(portStr, "unix/")
} }
return addr
if port, err := strconv.Atoi(portStr); err == nil {
return port, ""
}
return defaultPort, ""
}() }()
dataPath := func() string { dataPath := func() string {
@ -61,7 +61,10 @@ func main() {
glist.DataPath = dataPath glist.DataPath = dataPath
commit := func() string { version := func() string {
if Version != "" {
return Version
}
if bi, ok := debug.ReadBuildInfo(); ok { if bi, ok := debug.ReadBuildInfo(); ok {
for _, s := range bi.Settings { for _, s := range bi.Settings {
if s.Key == "vcs.revision" { if s.Key == "vcs.revision" {
@ -72,14 +75,7 @@ func main() {
return "unknown" return "unknown"
}() }()
listeningOn := func() string { log.Printf("List bot (%s) starting with datapath: %q, %s\n", version, dataPath, addr)
if unixSocketPath != "" {
return fmt.Sprintf("socket: %q", unixSocketPath)
}
return fmt.Sprintf("port: %v", port)
}()
log.Printf("List bot (%s) starting with datapath: %q, %s\n", commit, dataPath, listeningOn)
var chats sync.Map var chats sync.Map
if err := loadData(dataPath, &chats); err != nil { if err := loadData(dataPath, &chats); err != nil {
@ -147,24 +143,7 @@ func main() {
} }
}) })
if unixSocketPath != "" { log.Panicln(anyhttp.ListenAndServe(addr, nil))
// Remove old one
if err := os.Remove(unixSocketPath); err != nil && !errors.Is(err, fs.ErrNotExist) {
log.Panicf("Failed to remove unix socket : %q err: %v\n", unixSocketPath, err)
}
l, err := net.Listen("unix", unixSocketPath)
if err != nil {
log.Panicf("Unable to listen to unix socket : %q err: %v\n", unixSocketPath, err)
}
if err = os.Chmod(unixSocketPath, 0666); err != nil {
log.Panicf("Failed to set permission of unix socket %q err: %v\n", unixSocketPath, err)
}
log.Panicln(http.Serve(l, nil))
}
log.Panicln(http.ListenAndServe(fmt.Sprintf(":%v", port), nil))
} }
func handleTextAdded(gl *glist.GList, text string) { func handleTextAdded(gl *glist.GList, text string) {