inital commit

main.go

@@ -0,0 +1,69 @@
+package main
+
+import (
+	"encoding/json"
+	"flag"
+	"fmt"
+	log "github.com/sirupsen/logrus"
+	"os/exec"
+	"strings"
+)
+
+type TF2Stat struct {
+	Name       string  `json:"name"`
+	Map        string  `json:"map"`
+	Password   bool    `json:"password"`
+	MaxPlayers int     `json:"maxplayers"`
+	NumPlayers int     `json:"numplayers"`
+	QueryPort  int     `json:"queryPort"`
+	Ping       int     `json:"ping"`
+	Error      *string `json:"error,omitempty"`
+}
+
+type TF2Server string
+
+var (
+	server = flag.String("server", "", "comma seperated list of addresses to query")
+)
+
+func main() {
+	flag.Parse()
+
+	servers := strings.Split(*server, ",")
+	var tServer []TF2Server
+	for _, srv := range servers {
+		tServer = append(tServer, TF2Server(strings.TrimSpace(srv)))
+	}
+
+	for _, ip := range tServer {
+		tStat, err := ip.Stats()
+		if err != nil {
+			log.Warnf("error retrieving stats for %s: %v", ip, err)
+			continue
+		}
+
+		escapedName := strings.ReplaceAll(tStat.Name, " ", `\ `)
+		escapedName = strings.ReplaceAll(escapedName, ",", `\,`)
+		escapedName = strings.ReplaceAll(escapedName, "=", `\=`)
+		// myMeasurement,tag1=value1,tag2=value2 fieldKey="fieldValue" 1556813561098000000
+		fmt.Printf("tf2server,address=%s,name=%s player=%di maxplayer=%di map=%q ping=%di\n", string(ip), escapedName, tStat.NumPlayers, tStat.MaxPlayers, tStat.Map, tStat.Ping)
+	}
+}
+
+func (ip TF2Server) Stats() (*TF2Stat, error) {
+	cmd := exec.Command("gamedig", "--type", "teamfortress2", string(ip))
+	bOut, err := cmd.CombinedOutput()
+	if err != nil {
+		return nil, err
+	}
+
+	nStat := new(TF2Stat)
+	if err = json.Unmarshal(bOut, &nStat); err != nil {
+		return nil, err
+	}
+	if nStat.Error != nil {
+		return nil, fmt.Errorf("gamedig returned error: %v", *nStat)
+	}
+
+	return nStat, nil
+}