move a few tools to a more central utility cli

This commit is contained in:
adman
2026-03-18 20:45:54 +01:00
parent e0cf1834f1
commit f554e91354
6 changed files with 94 additions and 20 deletions
+2
View File
@@ -8,6 +8,7 @@ import (
) )
// Connect connects to the given address note that it does not resolve the SRV records it's just a raw net.Dial // Connect connects to the given address note that it does not resolve the SRV records it's just a raw net.Dial
// You must close the connection when you're done with it
func Connect(addr string) (ret *proto.Conn, err error) { func Connect(addr string) (ret *proto.Conn, err error) {
s, err := Status(addr) s, err := Status(addr)
if err != nil { if err != nil {
@@ -23,6 +24,7 @@ func Connect(addr string) (ret *proto.Conn, err error) {
// ConnectVersionLess connects to the given address without setting the version you should set it // ConnectVersionLess connects to the given address without setting the version you should set it
// when using this function. // when using this function.
// You must close the connection when you're done with it
func ConnectVersionLess(addr string) (ret *proto.Conn, err error) { func ConnectVersionLess(addr string) (ret *proto.Conn, err error) {
ret = &proto.Conn{} ret = &proto.Conn{}
ret.SetState(proto_base.Handshaking) ret.SetState(proto_base.Handshaking)
+1 -1
View File
@@ -53,6 +53,7 @@ func StatusRaw(addr string) (s *StatusClient, err error) {
if err != nil { if err != nil {
return return
} }
defer c.Close()
err = c.SetVersion("1.21.8") err = c.SetVersion("1.21.8")
if err != nil { if err != nil {
return return
@@ -61,7 +62,6 @@ func StatusRaw(addr string) (s *StatusClient, err error) {
Conn: c, Conn: c,
} }
err = s.Start(s) err = s.Start(s)
defer c.Close()
return return
} }
+52
View File
@@ -0,0 +1,52 @@
package main
import (
"errors"
"fmt"
"log/slog"
"maps"
"os"
"slices"
"github.com/admin-else/strom/tools/cli/offline_uuid"
"github.com/admin-else/strom/tools/cli/print_nbt"
)
var ExpectedASubcommandErr = errors.New("expected a subcommand")
var UnknownSubcommandErr = errors.New("unknown subcommand")
var subcommands = map[string]func(args []string) error{
"offline-uuid": offline_uuid.Run,
"print-nbt": print_nbt.Run,
}
func Help(args []string) (err error) {
fmt.Println("Available subcommands", slices.Collect(maps.Keys(subcommands)))
return
}
func mainE() (err error) {
subcommands["help"] = Help // this is here because if it isn't tools/cli/main.go:17:5: initialization cycle for subcommands
// tools/cli/main.go:17:5: subcommands refers to Help
// tools/cli/main.go:22:6: Help refers to subcommands
if len(os.Args) < 2 {
err = ExpectedASubcommandErr
return
}
f, ok := subcommands[os.Args[1]]
if !ok {
err = UnknownSubcommandErr
return
}
err = f(os.Args[2:])
return
}
func main() {
err := mainE()
if err != nil {
slog.Error("exiting", "err", err)
}
}
+20
View File
@@ -0,0 +1,20 @@
package offline_uuid
import (
"flag"
"fmt"
"github.com/admin-else/strom/crypto"
)
var cmd = flag.NewFlagSet("offline-uuid", flag.ContinueOnError)
var nameFlag = cmd.String("name", "Notch", "name of the player")
func Run(args []string) (err error) {
err = cmd.Parse(args)
if err != nil {
return
}
fmt.Println(crypto.FromOfflinePlayer(*nameFlag))
return
}
@@ -1,4 +1,4 @@
package main package print_nbt
import ( import (
"encoding/json" "encoding/json"
@@ -9,11 +9,15 @@ import (
) )
var ( var (
FileFlag = flag.String("file", "", "The NBT file to print") cmd = flag.NewFlagSet("print-nbt", flag.ContinueOnError)
FileFlag = cmd.String("file", "", "The NBT file to print")
) )
func mainE() (err error) { func Run(args []string) (err error) {
flag.Parse() err = cmd.Parse(args)
if err != nil {
return
}
f, err := os.Open(*FileFlag) f, err := os.Open(*FileFlag)
if err != nil { if err != nil {
return return
@@ -30,10 +34,3 @@ func mainE() (err error) {
_, err = os.Stdout.Write(b) _, err = os.Stdout.Write(b)
return return
} }
func main() {
err := mainE()
if err != nil {
panic(err)
}
}
@@ -1,4 +1,4 @@
package main package status
import ( import (
"flag" "flag"
@@ -7,23 +7,26 @@ import (
"github.com/admin-else/strom/client" "github.com/admin-else/strom/client"
) )
var cmd = flag.NewFlagSet("status", flag.ContinueOnError)
var addrFlag = flag.String("addr", "localhost:25565", "address to connect to") var addrFlag = flag.String("addr", "localhost:25565", "address to connect to")
var srvFlag = flag.Bool("srv", false, "use SRV records to resolve the address") var srvFlag = flag.Bool("srv", false, "use SRV records to resolve the address")
func main() { func Run(args []string) (err error) {
flag.Parse() err = cmd.Parse(args)
var err error if err != nil {
return
}
if *srvFlag { if *srvFlag {
*addrFlag, err = client.DoDNS(*addrFlag) *addrFlag, err = client.DoDNS(*addrFlag)
if err != nil { if err != nil {
panic(err) return
} }
} }
status, err := client.StatusRaw(*addrFlag) status, err := client.StatusRaw(*addrFlag)
if err != nil { if err != nil {
panic(err) return
} }
fmt.Println(status.Status) fmt.Println(status)
fmt.Printf("%v ping", status.PingReceiveTime.Sub(status.PingSendTime)) return
} }