75 lines
1.6 KiB
Go
75 lines
1.6 KiB
Go
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)
|
|
|
|
}
|