Rename mcstatus to gamestatus

2026-04-05 13:43:11 +03:00
parent 39a3ceadbb
commit cdf375ba80
15 changed files with 319 additions and 55 deletions
--- a/cmd/gamestatus/main.go
+++ b/cmd/gamestatus/main.go
@@ -0,0 +1,301 @@
+package main
+
+import (
+	"encoding/json"
+	"flag"
+	"fmt"
+	"os"
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/VexoraDevelopment/gamestatus"
+	"github.com/VexoraDevelopment/gamestatus/text"
+)
+
+func main() {
+	cfg, target, err := parseCLI(os.Args[1:])
+	if err != nil {
+		fatal(err)
+	}
+	if target == "" {
+		printUsage()
+		os.Exit(2)
+	}
+
+	host, port, err := parseTarget(target)
+	if err != nil {
+		fatal(err)
+	}
+
+	client := gamestatus.NewAll(cfg.timeout)
+
+	info, err := connect(client, cfg.edition, host, port, cfg.useQuery)
+	if err != nil {
+		fatal(err)
+	}
+
+	if cfg.asJSON {
+		enc := json.NewEncoder(os.Stdout)
+		enc.SetIndent("", "  ")
+		if err := enc.Encode(info); err != nil {
+			fatal(err)
+		}
+		return
+	}
+
+	printText(info, cfg.pretty, cfg.motdRaw)
+}
+
+type cliConfig struct {
+	edition  string
+	timeout  time.Duration
+	asJSON   bool
+	useQuery bool
+	pretty   bool
+	motdRaw  bool
+}
+
+func parseCLI(args []string) (cliConfig, string, error) {
+	cfg := cliConfig{
+		edition: "auto",
+		timeout: 3 * time.Second,
+	}
+
+	fs := flag.NewFlagSet("gamestatus", flag.ContinueOnError)
+	fs.SetOutput(os.Stderr)
+	edition := fs.String("edition", "auto", "server edition: auto, java, bedrock, source, rust, cs2, valve")
+	timeout := fs.Duration("timeout", 3*time.Second, "request timeout")
+	asJSON := fs.Bool("json", false, "print JSON output")
+	useQuery := fs.Bool("query", false, "use full query when supported")
+	pretty := fs.Bool("pretty", false, "use compact human-friendly output")
+	motdRaw := fs.Bool("motd-raw", false, "print raw MOTD without cleanup")
+
+	if len(args) > 0 {
+		switch strings.ToLower(strings.TrimSpace(args[0])) {
+		case "auto", "java", "bedrock", "source", "rust", "cs2", "valve":
+			cfg.edition = strings.ToLower(strings.TrimSpace(args[0]))
+			args = args[1:]
+		}
+	}
+
+	if err := fs.Parse(args); err != nil {
+		return cfg, "", err
+	}
+
+	cfg.edition = strings.ToLower(strings.TrimSpace(*edition))
+	if cfg.edition == "auto" && len(args) > 0 {
+		// Preserve subcommand semantics while still allowing -edition overrides.
+		switch strings.ToLower(strings.TrimSpace(os.Args[1])) {
+		case "auto", "java", "bedrock", "source", "rust", "cs2", "valve":
+			cfg.edition = strings.ToLower(strings.TrimSpace(os.Args[1]))
+		}
+	}
+	cfg.timeout = *timeout
+	cfg.asJSON = *asJSON
+	cfg.useQuery = *useQuery
+	cfg.pretty = *pretty
+	cfg.motdRaw = *motdRaw
+
+	if fs.NArg() != 1 {
+		return cfg, "", nil
+	}
+	return cfg, fs.Arg(0), nil
+}
+
+func printUsage() {
+	fmt.Fprintf(os.Stderr, "Usage: %s [auto|java|bedrock|source|rust|cs2|valve] [flags] <host[:port]>\n\n", os.Args[0])
+	fmt.Fprintln(os.Stderr, "Examples:")
+	fmt.Fprintf(os.Stderr, "  %s mc.hypixel.net\n", os.Args[0])
+	fmt.Fprintf(os.Stderr, "  %s java mc.hypixel.net\n", os.Args[0])
+	fmt.Fprintf(os.Stderr, "  %s bedrock -query play.example.com:19132\n\n", os.Args[0])
+	fmt.Fprintf(os.Stderr, "  %s source 127.0.0.1:27015\n\n", os.Args[0])
+	fmt.Fprintf(os.Stderr, "  %s rust 127.0.0.1:28015\n\n", os.Args[0])
+	fmt.Fprintf(os.Stderr, "  %s cs2 127.0.0.1:27015\n\n", os.Args[0])
+	fmt.Fprintf(os.Stderr, "  %s valve 127.0.0.1:27015\n\n", os.Args[0])
+	fmt.Fprintln(os.Stderr, "Flags:")
+	fmt.Fprintln(os.Stderr, "  -edition string")
+	fmt.Fprintln(os.Stderr, "        server edition: auto, java, bedrock, source, rust, cs2, valve")
+	fmt.Fprintln(os.Stderr, "  -timeout duration")
+	fmt.Fprintln(os.Stderr, "        request timeout (default 3s)")
+	fmt.Fprintln(os.Stderr, "  -json")
+	fmt.Fprintln(os.Stderr, "        print JSON output")
+	fmt.Fprintln(os.Stderr, "  -query")
+	fmt.Fprintln(os.Stderr, "        use full query when supported")
+	fmt.Fprintln(os.Stderr, "  -pretty")
+	fmt.Fprintln(os.Stderr, "        use compact human-friendly output")
+	fmt.Fprintln(os.Stderr, "  -motd-raw")
+	fmt.Fprintln(os.Stderr, "        print raw MOTD without cleanup")
+}
+
+func connect(client *gamestatus.Client, edition, host string, port int, useQuery bool) (*gamestatus.ServerInfo, error) {
+	switch edition {
+	case "auto":
+		if info, err := connectJava(client, host, port, useQuery); err == nil {
+			return info, nil
+		}
+		if info, err := connectBedrock(client, host, port, useQuery); err == nil {
+			return info, nil
+		}
+		return connectSource(client, host, port, useQuery)
+	case "java":
+		return connectJava(client, host, port, useQuery)
+	case "bedrock":
+		return connectBedrock(client, host, port, useQuery)
+	case "source":
+		return connectSource(client, host, port, useQuery)
+	case "rust":
+		return connectRust(client, host, port, useQuery)
+	case "cs2":
+		return connectCS2(client, host, port, useQuery)
+	case "valve":
+		return connectValve(client, host, port, useQuery)
+	default:
+		return nil, fmt.Errorf("unsupported edition %q", edition)
+	}
+}
+
+func connectJava(client *gamestatus.Client, host string, port int, useQuery bool) (*gamestatus.ServerInfo, error) {
+	if useQuery {
+		return client.Java.QueryFull(host, port)
+	}
+	return client.Java.Connect(host, port)
+}
+
+func connectBedrock(client *gamestatus.Client, host string, port int, useQuery bool) (*gamestatus.ServerInfo, error) {
+	if useQuery {
+		return client.Bedrock.QueryFull(host, port)
+	}
+	return client.Bedrock.Connect(host, port)
+}
+
+func connectSource(client *gamestatus.Client, host string, port int, useQuery bool) (*gamestatus.ServerInfo, error) {
+	if useQuery {
+		return client.Source.QueryFull(host, port)
+	}
+	return client.Source.Connect(host, port)
+}
+
+func connectRust(client *gamestatus.Client, host string, port int, useQuery bool) (*gamestatus.ServerInfo, error) {
+	if useQuery {
+		return client.Source.QueryFullRust(host, port)
+	}
+	return client.Source.ConnectRust(host, port)
+}
+
+func connectCS2(client *gamestatus.Client, host string, port int, useQuery bool) (*gamestatus.ServerInfo, error) {
+	if useQuery {
+		return client.Source.QueryFullCS2(host, port)
+	}
+	return client.Source.ConnectCS2(host, port)
+}
+
+func connectValve(client *gamestatus.Client, host string, port int, useQuery bool) (*gamestatus.ServerInfo, error) {
+	if useQuery {
+		return client.Source.QueryFullValve(host, port)
+	}
+	return client.Source.ConnectValve(host, port)
+}
+
+func parseTarget(target string) (string, int, error) {
+	host := strings.TrimSpace(target)
+	if host == "" {
+		return "", 0, fmt.Errorf("empty target")
+	}
+
+	if strings.Count(host, ":") == 1 && !strings.HasSuffix(host, "]") {
+		parts := strings.SplitN(host, ":", 2)
+		p, err := strconv.Atoi(parts[1])
+		if err != nil {
+			return "", 0, fmt.Errorf("invalid port %q", parts[1])
+		}
+		return parts[0], p, nil
+	}
+
+	if strings.HasPrefix(host, "[") {
+		end := strings.LastIndex(host, "]")
+		if end == -1 {
+			return "", 0, fmt.Errorf("invalid IPv6 target %q", target)
+		}
+		ipv6Host := host[1:end]
+		if len(host) == end+1 {
+			return ipv6Host, 0, nil
+		}
+		if !strings.HasPrefix(host[end+1:], ":") {
+			return "", 0, fmt.Errorf("invalid IPv6 target %q", target)
+		}
+		p, err := strconv.Atoi(host[end+2:])
+		if err != nil {
+			return "", 0, fmt.Errorf("invalid port %q", host[end+2:])
+		}
+		return ipv6Host, p, nil
+	}
+
+	return host, 0, nil
+}
+
+func printText(info *gamestatus.ServerInfo, pretty bool, motdRaw bool) {
+	motd := info.MOTD
+	if !motdRaw {
+		motd = text.CleanMOTD(motd)
+	}
+	if pretty {
+		printPretty(info, motd)
+		return
+	}
+
+	fmt.Printf("Edition: %s\n", info.Edition)
+	fmt.Printf("Address: %s:%d\n", info.Address, info.Port)
+	if info.IP != "" {
+		fmt.Printf("IP: %s\n", info.IP)
+	}
+	if info.Version != "" {
+		fmt.Printf("Version: %s\n", info.Version)
+	}
+	if motd != "" {
+		fmt.Printf("MOTD: %s\n", motd)
+	}
+	fmt.Printf("Players: %d/%d\n", info.Online, info.Slots)
+	if info.GameMode != "" {
+		fmt.Printf("GameMode: %s\n", info.GameMode)
+	}
+	if info.Map != "" {
+		fmt.Printf("Map: %s\n", info.Map)
+	}
+	if info.Software != "" {
+		fmt.Printf("Software: %s\n", info.Software)
+	}
+	if len(info.Plugins) > 0 {
+		fmt.Printf("Plugins: %s\n", strings.Join(info.Plugins, ", "))
+	}
+	if len(info.Players) > 0 {
+		fmt.Printf("Sample Players: %s\n", strings.Join(info.Players, ", "))
+	}
+	if info.Latency > 0 {
+		fmt.Printf("Latency: %s\n", info.Latency)
+	}
+	fmt.Printf("Query: %t\n", info.Query)
+}
+
+func printPretty(info *gamestatus.ServerInfo, motd string) {
+	fmt.Printf("%s %s:%d\n", strings.ToUpper(string(info.Edition)), info.Address, info.Port)
+	if motd != "" {
+		fmt.Printf("%s\n", motd)
+	}
+	fmt.Printf("Players %d/%d", info.Online, info.Slots)
+	if info.Version != "" {
+		fmt.Printf(" | Version %s", info.Version)
+	}
+	if info.Software != "" {
+		fmt.Printf(" | %s", info.Software)
+	}
+	if info.Latency > 0 {
+		fmt.Printf(" | %s", info.Latency)
+	}
+	fmt.Println()
+}
+
+func fatal(err error) {
+	fmt.Fprintln(os.Stderr, "gamestatus:", err)
+	os.Exit(1)
+}