92 lines
2.2 KiB
Go
92 lines
2.2 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"forge.cadoles.com/arcad/arcast/pkg/client"
|
|
"forge.cadoles.com/arcad/arcast/pkg/server"
|
|
"github.com/pkg/errors"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
var (
|
|
playerScanTimeoutFlag = "scan-timeout"
|
|
flagPlayerAddr = "player-address"
|
|
flagPlayerID = "player-id"
|
|
playerFlags = []cli.Flag{
|
|
&cli.DurationFlag{
|
|
Name: playerScanTimeoutFlag,
|
|
EnvVars: []string{"ARCAST_CLIENT_CAST_TIMEOUT"},
|
|
Usage: "Use `TIMEOUT` as maximum cast duration",
|
|
Value: 5 * time.Second,
|
|
},
|
|
&cli.StringSliceFlag{
|
|
Name: flagPlayerID,
|
|
Aliases: []string{"i"},
|
|
EnvVars: []string{"ARCAST_CLIENT_PLAYER_ID"},
|
|
Usage: "Use `ID` as player id",
|
|
Value: cli.NewStringSlice(),
|
|
},
|
|
&cli.StringSliceFlag{
|
|
Name: flagPlayerAddr,
|
|
Aliases: []string{"a"},
|
|
EnvVars: []string{"ARCAST_CLIENT_PLAYER_ADDR"},
|
|
Usage: "Use `ADDR` as player address",
|
|
Value: cli.NewStringSlice(),
|
|
},
|
|
}
|
|
)
|
|
|
|
func forEachPlayer(ctx *cli.Context, fn func(cl *client.Client, playerAddr string) error) error {
|
|
playerAddrs := ctx.StringSlice(flagPlayerAddr)
|
|
|
|
cl := client.New()
|
|
|
|
if ctx.IsSet(flagPlayerID) || !ctx.IsSet(flagPlayerAddr) {
|
|
scanTimeout := ctx.Duration(playerScanTimeoutFlag)
|
|
scanCtx, cancel := context.WithTimeout(ctx.Context, scanTimeout)
|
|
defer cancel()
|
|
|
|
playerIDs := ctx.StringSlice(flagPlayerID)
|
|
|
|
players, err := cl.Scan(scanCtx, client.WithPlayerIDs(playerIDs...))
|
|
if err != nil && (!errors.Is(err, context.Canceled) && !errors.Is(err, context.DeadlineExceeded)) {
|
|
return errors.Wrap(err, "could not scan for players")
|
|
}
|
|
|
|
for _, p := range players {
|
|
preferredIP, err := server.FindPreferredLocalAddress(p.IPs...)
|
|
if err != nil {
|
|
return errors.Errorf("could not retrieve player '%s' preferred address", p.ID)
|
|
}
|
|
|
|
playerAddr := fmt.Sprintf("%s:%d", preferredIP, p.Port)
|
|
playerAddrs = append(playerAddrs, playerAddr)
|
|
}
|
|
}
|
|
|
|
if len(playerAddrs) == 0 {
|
|
return errors.New("no players found")
|
|
}
|
|
|
|
for _, addr := range playerAddrs {
|
|
if err := fn(cl, addr); err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func AsAnySlice[T any](src ...T) []any {
|
|
dst := make([]any, len(src))
|
|
|
|
for i, s := range src {
|
|
dst[i] = s
|
|
}
|
|
|
|
return dst
|
|
}
|