76 lines
1.6 KiB
Go
76 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"html/template"
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/james4k/rcon"
|
|
)
|
|
|
|
type serverData struct {
|
|
IPAddr string
|
|
Port int
|
|
Title string
|
|
Players string
|
|
}
|
|
|
|
func main() {
|
|
|
|
serverAddr := flag.String("serverAddr", "localhost", "Server to check status of (optional; defaults to localhost")
|
|
serverPort := flag.Int("serverPort", 34196, "RCON port on the Factorio server (optional; defaults to 34196)")
|
|
password := flag.String("password", "", "RCON password of the server (required)")
|
|
flag.Parse()
|
|
|
|
if *serverPort < 1 || *serverPort > 65535 {
|
|
fmt.Printf("Invalid server port %v\n", *serverPort)
|
|
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,
|
|
34197,
|
|
"Server with Bob's Mod, est. Feb 2020",
|
|
playersOnline,
|
|
}
|
|
|
|
t.Execute(w, data)
|
|
})
|
|
|
|
fmt.Println("Serving...")
|
|
http.ListenAndServe(":http", nil)
|
|
|
|
}
|