Add support for restarting a instance

Signed-off-by: Ansuman Sahoo <[email protected]>

Author: unsuman

cmd/limactl/main.go

@@ -159,6 +159,7 @@ func newApp() *cobra.Command {
 		newUnprotectCommand(),
 		newTunnelCommand(),
 		newTemplateCommand(),
+		newRestartCommand(),
 	)
 	if runtime.GOOS == "darwin" || runtime.GOOS == "linux" {
 		rootCmd.AddCommand(startAtLoginCommand())

cmd/limactl/restart.go

@@ -0,0 +1,52 @@
+// SPDX-FileCopyrightText: Copyright The Lima Authors
+// SPDX-License-Identifier: Apache-2.0
+
+package main
+
+import (
+	"github.com/lima-vm/lima/pkg/instance"
+	"github.com/lima-vm/lima/pkg/store"
+	"github.com/spf13/cobra"
+)
+
+func newRestartCommand() *cobra.Command {
+	restartCmd := &cobra.Command{
+		Use:               "restart INSTANCE",
+		Short:             "Restart a running instance",
+		RunE:              restartAction,
+		ValidArgsFunction: restartBashComplete,
+		GroupID:           basicCommand,
+	}
+
+	restartCmd.Flags().BoolP("force", "f", false, "force stop and restart the instance")
+	return restartCmd
+}
+
+func restartAction(cmd *cobra.Command, args []string) error {
+	instName := DefaultInstanceName
+	if len(args) > 0 {
+		instName = args[0]
+	}
+
+	inst, err := store.Inspect(instName)
+	if err != nil {
+		return err
+	}
+
+	force, err := cmd.Flags().GetBool("force")
+	if err != nil {
+		return err
+	}
+
+	ctx := cmd.Context()
+	if force {
+		err = instance.RestartForcibly(ctx, inst)
+	} else {
+		err = instance.Restart(ctx, inst)
+	}
+	return err
+}
+
+func restartBashComplete(cmd *cobra.Command, _ []string, _ string) ([]string, cobra.ShellCompDirective) {
+	return bashCompleteInstanceNames(cmd)
+}

pkg/instance/restart.go

@@ -0,0 +1,73 @@
+// SPDX-FileCopyrightText: Copyright The Lima Authors
+// SPDX-License-Identifier: Apache-2.0
+
+package instance
+
+import (
+	"context"
+	"errors"
+	"time"
+
+	networks "github.com/lima-vm/lima/pkg/networks/reconcile"
+	"github.com/lima-vm/lima/pkg/store"
+)
+
+const launchHostAgentForeground = false
+
+func Restart(ctx context.Context, inst *store.Instance) error {
+	if err := StopGracefully(inst); err != nil {
+		return err
+	}
+
+	if err := waitForInstanceShutdown(ctx, inst); err != nil {
+		return err
+	}
+
+	if err := networks.Reconcile(ctx, inst.Name); err != nil {
+		return err
+	}
+
+	if err := Start(ctx, inst, "", launchHostAgentForeground); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func RestartForcibly(ctx context.Context, inst *store.Instance) error {
+	StopForcibly(inst)
+
+	if err := networks.Reconcile(ctx, inst.Name); err != nil {
+		return err
+	}
+
+	if err := Start(ctx, inst, "", launchHostAgentForeground); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func waitForInstanceShutdown(ctx context.Context, inst *store.Instance) error {
+	ctx2, cancel := context.WithTimeout(ctx, 3*time.Minute)
+	defer cancel()
+
+	ticker := time.NewTicker(500 * time.Millisecond)
+	defer ticker.Stop()
+
+	for {
+		select {
+		case <-ticker.C:
+			updatedInst, err := store.Inspect(inst.Name)
+			if err != nil {
+				return errors.New("failed to inspect instance status: " + err.Error())
+			}
+
+			if updatedInst.Status == store.StatusStopped {
+				return nil
+			}
+		case <-ctx2.Done():
+			return errors.New("timed out waiting for instance to stop")
+		}
+	}
+}