Replace unix socket logic with go.balki.me/anyhttp

Now also support systemd socket activation
This commit is contained in:
Balakrishnan Balasubramanian 2023-04-20 17:59:40 -04:00
parent 107d90deb8
commit dd62f237b0
3 changed files with 12 additions and 42 deletions

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=

50
main.go
View File

@ -4,21 +4,18 @@ 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"
) )
@ -42,19 +39,12 @@ func main() {
log.Panicln("TGLB_API_TOKEN is empty") 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 {
@ -85,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", version, dataPath, listeningOn)
var chats sync.Map var chats sync.Map
if err := loadData(dataPath, &chats); err != nil { if err := loadData(dataPath, &chats); err != nil {
@ -160,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) {