Move server -> factorio-site

factorio-site.go

@@ -0,0 +1,74 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"html/template"
+	"log"
+	"net/http"
+
+	"github.com/james4k/rcon"
+)
+
+type serverData struct {
+	IPAddr  string
+	Title   string
+	Players string
+}
+
+func main() {
+
+	port := flag.Int("port", 65536, "Port on which the HTTP server should serve (required)")
+	serverAddr := flag.String("serverAddr", "factorio.blackolivepineapple.pizza", "Server to check status of (optional, defaults to factorio.bopp")
+	serverPort := flag.Int("serverport", 34196, "RCON port on the Factorio server")
+	password := flag.String("password", "", "RCON password of the server (required)")
+	flag.Parse()
+
+	if *port < 1 || *port > 65535 {
+		fmt.Printf("Invalid port %v\n", *port)
+		return
+	}
+
+	if *password == "" {
+		fmt.Printf("Password flag is required")
+	}
+
+	fmt.Print("Parsing templates...\n")
+	t, err := template.ParseFiles("templates/index.html")
+	if err != nil {
+		fmt.Printf("Error parsing HTML template: %v\n", err)
+	}
+
+	rconConnection, err := rcon.Dial(fmt.Sprintf("%v:%v", *serverAddr, *serverPort), *password)
+	if err != nil {
+		log.Fatalf("Error making RCON connection: %v", err)
+	}
+	defer rconConnection.Close()
+
+	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+
+		_, err := rconConnection.Write("/players o")
+		if err != nil {
+			fmt.Print(w, "Error connecting to server")
+			return
+		}
+
+		playersOnline, _, err := rconConnection.Read()
+		if err != nil {
+			fmt.Print(w, "Error receiving data from server")
+			return
+		}
+
+		data := serverData{
+			*serverAddr,
+			"Server with Bob's Mod, est. Feb 2020",
+			playersOnline,
+		}
+
+		t.Execute(w, data)
+	})
+
+	fmt.Printf("Serving on :%v...\n", *port)
+	http.ListenAndServe(fmt.Sprintf(":%v", *port), nil)
+
+}