Support human-readable reclaim sizes

Author: sjmiller609

go.mod

@@ -3,6 +3,7 @@ module github.com/kernel/hypeman-cli
 go 1.25
 
 require (
+	github.com/c2h5oh/datasize v0.0.0-20231215233829-aa82cc1e6500
 	github.com/charmbracelet/bubbles v0.21.0
 	github.com/charmbracelet/bubbletea v1.3.6
 	github.com/charmbracelet/lipgloss v1.1.0

go.sum

@@ -6,6 +6,8 @@ github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiE
 github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8=
 github.com/aymanbagabas/go-udiff v0.2.0 h1:TK0fH4MteXUDspT88n8CKzvK0X9O2xu9yQjWpi6yML8=
 github.com/aymanbagabas/go-udiff v0.2.0/go.mod h1:RE4Ex0qsGkTAJoQdQQCA0uG+nAzJO/pI/QwceO5fgrA=
+github.com/c2h5oh/datasize v0.0.0-20231215233829-aa82cc1e6500 h1:6lhrsTEnloDPXyeZBvSYvQf8u86jbKehZPVDDlkgDl4=
+github.com/c2h5oh/datasize v0.0.0-20231215233829-aa82cc1e6500/go.mod h1:S/7n9copUssQ56c7aAgHqftWO4LTf4xY6CGWt8Bc+3M=
 github.com/cenkalti/backoff/v4 v4.3.0 h1:MyRJ/UdXutAwSAT+s3wNd7MfTIcy71VQueUuFK343L8=
 github.com/cenkalti/backoff/v4 v4.3.0/go.mod h1:Y3VNntkOUPxTVeUxJ/G5vcM//AlwfmyYozVcomhLiZE=
 github.com/charmbracelet/bubbles v0.21.0 h1:9TdC97SdRVg/1aaXNVWfFH3nnLAwOXr8Fn6u6mfQdFs=

pkg/cmd/resourcecmd.go

@@ -3,9 +3,11 @@ package cmd
 import (
 	"context"
 	"fmt"
+	"math"
 	"os"
 	"strings"
 
+	"github.com/c2h5oh/datasize"
 	"github.com/kernel/hypeman-go"
 	"github.com/kernel/hypeman-go/option"
 	"github.com/tidwall/gjson"
@@ -39,10 +41,15 @@ Examples:
 var resourcesReclaimMemoryCmd = cli.Command{
 	Name:  "reclaim-memory",
 	Usage: "Request guest memory reclaim from reclaim-eligible instances",
+	Description: `Request guest memory reclaim across eligible instances.
+
+Examples:
+  hypeman resources reclaim-memory --reclaim-bytes 512MB --dry-run
+  hypeman resources reclaim-memory --reclaim-bytes 1073741824 --hold-for 10m --reason "pack host before launch"`,
 	Flags: []cli.Flag{
-		&cli.Int64Flag{
+		&cli.StringFlag{
 			Name:     "reclaim-bytes",
-			Usage:    "Total bytes of guest memory to reclaim across eligible VMs",
+			Usage:    `Total guest memory to reclaim (e.g., "512MB", "2GB", or "1048576" for raw bytes)`,
 			Required: true,
 		},
 		&cli.BoolFlag{
@@ -93,9 +100,9 @@ func handleResources(ctx context.Context, cmd *cli.Command) error {
 func handleResourcesReclaimMemory(ctx context.Context, cmd *cli.Command) error {
 	client := hypeman.NewClient(getDefaultRequestOptions(cmd)...)
 
-	reclaimBytes := cmd.Int64("reclaim-bytes")
-	if reclaimBytes <= 0 {
-		return fmt.Errorf("reclaim-bytes must be greater than 0")
+	reclaimBytes, err := parseReclaimBytes(cmd.String("reclaim-bytes"))
+	if err != nil {
+		return err
 	}
 
 	request := hypeman.MemoryReclaimRequestParam{
@@ -122,7 +129,7 @@ func handleResourcesReclaimMemory(ctx context.Context, cmd *cli.Command) error {
 
 	var res []byte
 	opts = append(opts, option.WithResponseBodyInto(&res))
-	_, err := client.Resources.ReclaimMemory(ctx, params, opts...)
+	_, err = client.Resources.ReclaimMemory(ctx, params, opts...)
 	if err != nil {
 		return err
 	}
@@ -143,6 +150,25 @@ func handleResourcesReclaimMemory(ctx context.Context, cmd *cli.Command) error {
 	return ShowJSON(os.Stdout, "resources reclaim-memory", obj, format, transform)
 }
 
+func parseReclaimBytes(raw string) (int64, error) {
+	if raw == "" {
+		return 0, fmt.Errorf("reclaim-bytes is required")
+	}
+
+	var size datasize.ByteSize
+	if err := size.UnmarshalText([]byte(raw)); err != nil {
+		return 0, fmt.Errorf("invalid reclaim-bytes %q: %w", raw, err)
+	}
+	if size == 0 {
+		return 0, fmt.Errorf("reclaim-bytes must be greater than 0")
+	}
+	if size.Bytes() > math.MaxInt64 {
+		return 0, fmt.Errorf("reclaim-bytes %q exceeds the maximum supported size", raw)
+	}
+
+	return int64(size.Bytes()), nil
+}
+
 func showResourcesTable(data []byte) error {
 	obj := gjson.ParseBytes(data)
 

pkg/cmd/resourcecmd_test.go

@@ -4,8 +4,39 @@ import (
 	"testing"
 
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )
 
+func TestParseReclaimBytes(t *testing.T) {
+	tests := []struct {
+		name     string
+		input    string
+		expected int64
+		wantErr  string
+	}{
+		{name: "raw bytes", input: "1048576", expected: 1048576},
+		{name: "megabytes", input: "512MB", expected: 512 * 1024 * 1024},
+		{name: "gigabytes with space", input: "2 GB", expected: 2 * 1024 * 1024 * 1024},
+		{name: "empty", input: "", wantErr: "reclaim-bytes is required"},
+		{name: "zero", input: "0", wantErr: "reclaim-bytes must be greater than 0"},
+		{name: "invalid", input: "nope", wantErr: "invalid reclaim-bytes \"nope\""},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := parseReclaimBytes(tt.input)
+			if tt.wantErr != "" {
+				require.Error(t, err)
+				assert.Contains(t, err.Error(), tt.wantErr)
+				return
+			}
+
+			require.NoError(t, err)
+			assert.Equal(t, tt.expected, got)
+		})
+	}
+}
+
 func TestFormatBytes(t *testing.T) {
 	tests := []struct {
 		bytes    int64