basic flag support

cmd/main.go

@@ -3,26 +3,16 @@ package main
 import (
     "log"
     "net/http"
-    "time"
-    "io/ioutil"
-    "encoding/json"
 
     "git.snrd.de/Spaenny/steamServer"
     "git.snrd.de/Spaenny/steamServer/postgres"
     "git.snrd.de/Spaenny/steamServer/web"
+    "github.com/rumblefrog/go-a2s"
     "github.com/robfig/cron/v3"
 )
 
-type Server struct {
-    AppID       int     `json:"AppID"`
-    Name        string  `json:"Name"`
-    Map         string  `json:"Map"`
-    Players     int     `json:"Players"`
-    MaxPlayers  int     `json:"MaxPlayers"`
-}
-
 func main() {
-    dsn := "postgres://postgres:secret@postgres:5432/postgres?sslmode=disable"
+    dsn := "postgres://postgres:secret@localhost:5432/postgres?sslmode=disable"
 
     store, err := postgres.NewStore(dsn)
     if err != nil {
@@ -47,35 +37,32 @@ func main() {
     http.ListenAndServe(":3000", h)
 }
 
-func updateServers(serverString string, store steamServer.Store) {
-    url := "http://steam-server-api:8080/server/" + serverString
+func updateServers(host string, store steamServer.Store) {
+    client, err := a2s.NewClient(host)
 
-    reqClient := http.Client {
-        Timeout: time.Second * 10,
-    }
-
-    req, err := http.NewRequest(http.MethodGet, url, nil)
     if err != nil {
-        log.Fatal(err)
+        log.Fatal("Creation of client failed: %w", err)
+        return
     }
 
-    res, err := reqClient.Do(req)
+    defer client.Close()
+
+    info, err := client.QueryInfo()
+
     if err != nil {
-        log.Fatal(err)
+        log.Printf("Querying failed: %s", err)
+        return
     }
 
-    if res.Body != nil {
-        defer res.Body.Close()
+    log.Printf("%s - %s\n", host, info.Name)
+    
+    server := steamServer.Server {
+        AppID: info.ID,
+        Name: info.Name,
+        Map: info.Map,
+        Players: info.Players,
+        MaxPlayers: info.MaxPlayers,
     }
 
-    body, err := ioutil.ReadAll(res.Body)
-    if err != nil {
-        log.Fatal(err)
-    }
-
-
-    server := steamServer.Server{}
-    json.Unmarshal([]byte(body), &server)
-    store.UpdateServerByQuery(&server, serverString)
+    store.UpdateServerByQuery(&server, host)
 }
-