labctl playground lifetime

Author: iximiuz

api/plays.go

@@ -64,6 +64,8 @@ type Play struct {
 
 	ExpiresIn int `json:"expiresIn" yaml:"expiresIn"`
 
+	MaxPlayTime string `json:"maxPlayTime,omitempty" yaml:"maxPlayTime,omitempty"`
+
 	Status *PlayStatus `json:"status,omitempty" yaml:"status,omitempty"`
 
 	Playground Playground `json:"playground" yaml:"playground"`
@@ -370,6 +372,19 @@ func (c *Client) PersistPlay(ctx context.Context, id string) error {
 	return c.PostInto(ctx, "/plays/"+id+"/actions", nil, nil, body, &p)
 }
 
+func (c *Client) SetPlayMaxPlayTime(ctx context.Context, id string, maxPlayTimeMinutes int) (*Play, error) {
+	body, err := toJSONBody(map[string]any{
+		"action":             "set_max_play_time",
+		"maxPlayTimeMinutes": maxPlayTimeMinutes,
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	var p Play
+	return &p, c.PostInto(ctx, "/plays/"+id+"/actions", nil, nil, body, &p)
+}
+
 func (c *Client) RebootPlayMachine(ctx context.Context, id, machine string) (*Play, error) {
 	body, err := toJSONBody(map[string]any{"action": "machine.reboot", "machine": machine})
 	if err != nil {

cmd/playground/lifetime.go

@@ -0,0 +1,92 @@
+package playground
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"github.com/spf13/cobra"
+
+	"github.com/iximiuz/labctl/internal/completion"
+	"github.com/iximiuz/labctl/internal/labcli"
+)
+
+type lifetimeOptions struct {
+	playID   string
+	lifetime time.Duration
+}
+
+func newLifetimeCommand(cli labcli.CLI) *cobra.Command {
+	var opts lifetimeOptions
+
+	cmd := &cobra.Command{
+		Use:               "lifetime <playground-id> [new-lifetime]",
+		Short:             `Show or set a playground's lifetime - the total session duration counted from the playground's start (e.g., 90m, 3h, 2h30m)`,
+		Args:              cobra.RangeArgs(1, 2),
+		ValidArgsFunction: completion.ActivePlays(cli),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			opts.playID = args[0]
+
+			if len(args) == 1 {
+				return labcli.WrapStatusError(runShowPlaygroundLifetime(cmd.Context(), cli, &opts))
+			}
+
+			lifetime, err := time.ParseDuration(args[1])
+			if err != nil {
+				return labcli.WrapStatusError(fmt.Errorf("invalid lifetime %q: %w", args[1], err))
+			}
+			opts.lifetime = lifetime
+
+			return labcli.WrapStatusError(runSetPlaygroundLifetime(cmd.Context(), cli, &opts))
+		},
+	}
+
+	return cmd
+}
+
+func runShowPlaygroundLifetime(ctx context.Context, cli labcli.CLI, opts *lifetimeOptions) error {
+	play, err := cli.Client().GetPlay(ctx, opts.playID)
+	if err != nil {
+		return fmt.Errorf("couldn't get playground: %w", err)
+	}
+
+	if play.MaxPlayTime == "" {
+		return labcli.NewStatusError(1, "playground %s has no lifetime information", opts.playID)
+	}
+
+	lifetime, err := time.ParseDuration(play.MaxPlayTime)
+	if err != nil {
+		cli.PrintOut("%s\n", play.MaxPlayTime)
+		return nil
+	}
+
+	if play.ExpiresIn > 0 {
+		cli.PrintOut("%s (expires in %s)\n", lifetime, (time.Duration(play.ExpiresIn) * time.Millisecond).Round(time.Second))
+	} else {
+		cli.PrintOut("%s\n (expired)", lifetime)
+	}
+
+	return nil
+}
+
+func runSetPlaygroundLifetime(ctx context.Context, cli labcli.CLI, opts *lifetimeOptions) error {
+	minutes := int(opts.lifetime.Minutes())
+	if minutes < 1 {
+		return fmt.Errorf("lifetime must be at least 1 minute")
+	}
+
+	cli.PrintAux("Setting lifetime of playground %s to %s...\n", opts.playID, opts.lifetime)
+
+	play, err := cli.Client().SetPlayMaxPlayTime(ctx, opts.playID, minutes)
+	if err != nil {
+		return fmt.Errorf("couldn't set the playground lifetime: %w", err)
+	}
+
+	if play.ExpiresIn > 0 {
+		cli.PrintAux("Playground will now expire in %s.\n", (time.Duration(play.ExpiresIn) * time.Millisecond).Round(time.Second))
+	} else {
+		cli.PrintAux("Playground lifetime is set.\n")
+	}
+
+	return nil
+}

cmd/playground/playground.go

@@ -27,6 +27,7 @@ func NewCommand(cli labcli.CLI) *cobra.Command {
 		newRestartCommand(cli),
 		newDestroyCommand(cli),
 		newPersistCommand(cli),
+		newLifetimeCommand(cli),
 		newMachinesCommand(cli),
 		newMachineCommand(cli),
 		newCreateCommand(cli),