50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
|
package client
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"os"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"forge.cadoles.com/arcad/arcast/internal/command/client/flag"
|
||
|
|
"forge.cadoles.com/arcad/arcast/pkg/client"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
"github.com/urfave/cli/v2"
|
||
|
|
"gitlab.com/wpetit/goweb/cli/format"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Scan() *cli.Command {
|
||
|
|
return &cli.Command{
|
||
|
|
Name: "scan",
|
||
|
|
Flags: flag.ComposeFlags(
|
||
|
|
&cli.DurationFlag{
|
||
|
|
Name: "timeout",
|
||
|
|
EnvVars: []string{"ARCAST_CLIENT_SCAN_TIMEOUT"},
|
||
|
|
Usage: "Use `TIMEOUT` as maximum scan duration",
|
||
|
|
Value: 5 * time.Second,
|
||
|
|
},
|
||
|
|
),
|
||
|
|
Action: func(ctx *cli.Context) error {
|
||
|
|
baseFlags := flag.GetBaseFlags(ctx)
|
||
|
|
timeout := ctx.Duration("timeout")
|
||
|
|
|
||
|
|
client := client.New()
|
||
|
|
|
||
|
|
scanCtx, cancel := context.WithTimeout(ctx.Context, timeout)
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
players, err := client.Scan(scanCtx)
|
||
|
|
if err != nil && (!errors.Is(err, context.Canceled) && !errors.Is(err, context.DeadlineExceeded)) {
|
||
|
|
return errors.Wrap(err, "could not scan for players")
|
||
|
|
}
|
||
|
|
|
||
|
|
hints := playerHints(baseFlags.OutputMode)
|
||
|
|
|
||
|
|
if err := format.Write(baseFlags.Format, os.Stdout, hints, AsAnySlice(players...)...); err != nil {
|
||
|
|
return errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
},
|
||
|
|
}
|
||
|
|
}
|