add playground mgmt commands

cmd/auth/whoami.go

@@ -26,7 +26,7 @@ func runWhoAmI(ctx context.Context, cli labcli.CLI) error {
  return nil
  }
 
- me, err := cli.Client().GetMe(ctx)
+ me, err := cli.Client().GetAccount(ctx)
  if err != nil {
  return err
  }

cmd/playground/list.go

@@ -0,0 +1,149 @@
+package playground
+
+import (
+ "context"
+ "fmt"
+ "io"
+ "strings"
+ "text/tabwriter"
+ "time"
+
+ "github.com/dustin/go-humanize"
+ "github.com/spf13/cobra"
+
+ "github.com/iximiuz/labctl/internal/api"
+ "github.com/iximiuz/labctl/internal/labcli"
+)
+
+type listOptions struct {
+ all bool
+ quiet bool
+}
+
+func newListCommand(cli labcli.CLI) *cobra.Command {
+ var opts listOptions
+
+ cmd := &cobra.Command{
+ Use: "list [flags]",
+ Aliases: []string{"ls"},
+ Short: `List recent playgrounds (up to 50)`,
+ RunE: func(cmd *cobra.Command, args []string) error {
+ return labcli.WrapStatusError(runListPlaygrounds(cmd.Context(), cli, &opts))
+ },
+ }
+
+ flags := cmd.Flags()
+
+ flags.BoolVarP(
+ &opts.all,
+ "all",
+ "a",
+ false,
+ `List all playgrounds (including terminated)`,
+ )
+ flags.BoolVarP(
+ &opts.quiet,
+ "quiet",
+ "q",
+ false,
+ `Only print playground IDs`,
+ )
+
+ return cmd
+}
+
+func runListPlaygrounds(ctx context.Context, cli labcli.CLI, opts *listOptions) error {
+ printer := newListPrinter(cli.OutputStream(), opts.quiet)
+ defer printer.flush()
+
+ printer.printHeader()
+
+ plays, err := cli.Client().ListPlays(ctx)
+ if err != nil {
+ return fmt.Errorf("couldn't list playgrounds: %w", err)
+ }
+
+ for _, play := range plays {
+ if opts.all || play.Active {
+ printer.printOne(play)
+ }
+ }
+
+ return nil
+}
+
+type listPrinter struct {
+ quiet bool
+ header []string
+ writer *tabwriter.Writer
+}
+
+func newListPrinter(outStream io.Writer, quiet bool) *listPrinter {
+ header := []string{
+ "PLAYGROUND ID",
+ "NAME",
+ "CREATED",
+ "STATUS",
+ "LINK",
+ }
+
+ return &listPrinter{
+ quiet: quiet,
+ header: header,
+ writer: tabwriter.NewWriter(outStream, 0, 4, 2, ' ', 0),
+ }
+}
+
+func (p *listPrinter) printHeader() {
+ if !p.quiet {
+ fmt.Fprintln(p.writer, strings.Join(p.header, "\t"))
+ }
+}
+
+func (p *listPrinter) printOne(play *api.Play) {
+ if p.quiet {
+ fmt.Fprintln(p.writer, play.ID)
+ return
+ }
+
+ var link string
+ if play.Active {
+ link = play.PageURL
+ }
+
+ fields := []string{
+ play.ID,
+ play.Playground.Name,
+ humanize.Time(safeParseTime(play.CreatedAt)),
+ playStatus(play),
+ link,
+ }
+
+ fmt.Fprintln(p.writer, strings.Join(fields, "\t"))
+}
+
+func (p *listPrinter) flush() {
+ p.writer.Flush()
+}
+
+func playStatus(play *api.Play) string {
+ if play.Running {
+ return fmt.Sprintf("running (expires in %s)",
+ humanize.Time(time.Now().Add(time.Duration(play.ExpiresIn)*time.Millisecond)))
+ }
+ if play.Destroyed {
+ return fmt.Sprintf("terminated %s",
+ humanize.Time(safeParseTime(play.LastStateAt)))
+ }
+ if play.Failed {
+ return fmt.Sprintf("failed %s",
+ humanize.Time(safeParseTime(play.LastStateAt)))
+ }
+
+ return "unknown"
+}
+
+func safeParseTime(s string) time.Time {
+ t, _ := time.Parse(time.RFC3339, s)
+ return t
+}

cmd/playground/playground.go

@@ -0,0 +1,23 @@
+package playground
+
+import (
+ "github.com/spf13/cobra"
+
+ "github.com/iximiuz/labctl/internal/labcli"
+)
+
+func NewCommand(cli labcli.CLI) *cobra.Command {
+ cmd := &cobra.Command{
+ Use: "playground <list|start|stop> [playground-name]",
+ Aliases: []string{"p", "playgrounds"},
+ Short: "List, start and stop playgrounds",
+ }
+
+ cmd.AddCommand(
+ newListCommand(cli),
+ newStartCommand(cli),
+ newStopCommand(cli),
+ )
+
+ return cmd
+}

cmd/playground/start.go

@@ -0,0 +1,73 @@
+package playground
+
+import (
+ "context"
+ "fmt"
+
+ "github.com/skratchdot/open-golang/open"
+ "github.com/spf13/cobra"
+
+ "github.com/iximiuz/labctl/internal/api"
+ "github.com/iximiuz/labctl/internal/labcli"
+)
+
+type startOptions struct {
+ playground string
+
+ open bool
+ quiet bool
+}
+
+func newStartCommand(cli labcli.CLI) *cobra.Command {
+ var opts startOptions
+
+ cmd := &cobra.Command{
+ Use: "start [flags] <playground-name>",
+ Short: `Start a new playground, possibly open it in a browser`,
+ Args: cobra.ExactArgs(1),
+ RunE: func(cmd *cobra.Command, args []string) error {
+ cli.SetQuiet(opts.quiet)
+
+ opts.playground = args[0]
+
+ return labcli.WrapStatusError(runStartPlayground(cmd.Context(), cli, &opts))
+ },
+ }
+
+ flags := cmd.Flags()
+
+ flags.BoolVar(
+ &opts.open,
+ "open",
+ false,
+ `Open the playground page in a browser`,
+ )
+ flags.BoolVarP(
+ &opts.quiet,
+ "quiet",
+ "q",
+ false,
+ `Only print playground's ID`,
+ )
+
+ return cmd
+}
+
+func runStartPlayground(ctx context.Context, cli labcli.CLI, opts *startOptions) error {
+ play, err := cli.Client().CreatePlay(ctx, api.CreatePlayRequest{
+ Playground: opts.playground,
+ })
+ if err != nil {
+ return fmt.Errorf("couldn't create a new playground: %w", err)
+ }
+
+ cli.PrintAux("Opening %s in your browser...\n", play.PageURL)
+
+ if err := open.Run(play.PageURL); err != nil {
+ cli.PrintAux("Couldn't open the browser. Copy the above URL into a browser manually to access the playground.\n")
+ }
+
+ cli.PrintOut("%s\n", play.ID)
+
+ return nil
+}

cmd/playground/stop.go

@@ -0,0 +1,80 @@
+package playground
+
+import (
+ "context"
+ "fmt"
+ "time"
+
+ "github.com/briandowns/spinner"
+ "github.com/spf13/cobra"
+
+ "github.com/iximiuz/labctl/internal/labcli"
+)
+
+const stopCommandTimeout = 5 * time.Minute
+
+type stopOptions struct {
+ playID string
+
+ quiet bool
+}
+
+func newStopCommand(cli labcli.CLI) *cobra.Command {
+ var opts stopOptions
+
+ cmd := &cobra.Command{
+ Use: "stop [flags] <playground-id>",
+ Short: `Start a new playground, possibly open it in a browser`,
+ Args: cobra.ExactArgs(1),
+ RunE: func(cmd *cobra.Command, args []string) error {
+ cli.SetQuiet(opts.quiet)
+
+ opts.playID = args[0]
+
+ return labcli.WrapStatusError(runStopPlayground(cmd.Context(), cli, &opts))
+ },
+ }
+
+ flags := cmd.Flags()
+
+ flags.BoolVarP(
+ &opts.quiet,
+ "quiet",
+ "q",
+ false,
+ `Do not print any diagnostic messages`,
+ )
+
+ return cmd
+}
+
+func runStopPlayground(ctx context.Context, cli labcli.CLI, opts *stopOptions) error {
+ cli.PrintAux("Stopping playground %s...\n", opts.playID)
+
+ if err := cli.Client().DeletePlay(ctx, opts.playID); err != nil {
+ return fmt.Errorf("couldn't delete the playground: %w", err)
+ }
+
+ s := spinner.New(spinner.CharSets[38], 300*time.Millisecond)
+ s.Writer = cli.AuxStream()
+ s.Prefix = "Waiting for playground to terminate... "
+ s.Start()
+
+ ctx, cancel := context.WithTimeout(ctx, stopCommandTimeout)
+ defer cancel()
+
+ for ctx.Err() == nil {
+ if play, err := cli.Client().GetPlay(ctx, opts.playID); err == nil && !play.Active {
+ s.FinalMSG = "Waiting for playground to terminate... Done.\n"
+ s.Stop()
+
+ return nil
+ }
+
+ time.Sleep(2 * time.Second)
+ }
+
+ cli.PrintAux("Playground has been stopped.\n")
+
+ return nil
+}

go.mod

@@ -5,6 +5,7 @@ go 1.22.0
 require (
  github.com/briandowns/spinner v1.23.0
  github.com/docker/cli v25.0.3+incompatible
+ github.com/dustin/go-humanize v1.0.1
  github.com/google/uuid v1.6.0
  github.com/iximiuz/wsmux v0.0.2
  github.com/mikesmitty/edkey v0.0.0-20170222072505-3356ea4e686a

go.sum

@@ -10,6 +10,8 @@ github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/docker/cli v25.0.3+incompatible h1:KLeNs7zws74oFuVhgZQ5ONGZiXUUdgsdy6/EsX/6284=
 github.com/docker/cli v25.0.3+incompatible/go.mod h1:JLrzqnKDaYBop7H2jaqPtU4hHvMKP+vjCwu2uszcLI8=
+github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY=
+github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto=
 github.com/fatih/color v1.16.0 h1:zmkK9Ngbjj+K0yRhTVONQh1p/HknKYSlNT+vZCzyokM=
 github.com/fatih/color v1.16.0/go.mod h1:fL2Sau1YI5c0pdGEVCbKQbLXB6edEj1ZgiY4NijnWvE=
 github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=

internal/api/account.go

@@ -9,7 +9,7 @@ type Me struct {
  GithubProfileId string `json:"githubProfileId" yaml:"githubProfileId"`
 }
 
-func (c *Client) GetMe(ctx context.Context) (*Me, error) {
+func (c *Client) GetAccount(ctx context.Context) (*Me, error) {
  var me Me
  return &me, c.GetInto(ctx, "/account", nil, nil, &me)
 }