feat(as): add ssh verb on the server resource to connect to a server

Author: remyleone

cmd/scw/testdata/test-all-usage-apple-silicon-server-ssh-usage.golden

@@ -0,0 +1,22 @@
+🎲🎲🎲 EXIT CODE: 0 🎲🎲🎲
+🟥🟥🟥 STDERR️️ 🟥🟥🟥️
+Connect to distant server via the SSH protocol.
+
+USAGE:
+  scw apple-silicon server ssh <server-id ...> [arg=value ...]
+
+ARGS:
+  server-id         Server ID to SSH into
+  [username=m1]     Username used for the SSH connection
+  [port=22]         Port used for the SSH connection
+  [command]         Command to execute on the remote server
+  [zone=fr-par-1]   Zone to target. If none is passed will use default zone from the config
+
+FLAGS:
+  -h, --help   help for ssh
+
+GLOBAL FLAGS:
+  -c, --config string    The path to the config file
+  -D, --debug            Enable debug mode
+  -o, --output string    Output format: json or human, see 'scw help output' for more info (default "human")
+  -p, --profile string   The config profile to use

cmd/scw/testdata/test-all-usage-apple-silicon-server-usage.golden

@@ -12,6 +12,7 @@ AVAILABLE COMMANDS:
   list        List all servers
   reboot      Reboot a server
   reinstall   Reinstall a server
+  ssh         SSH into a server
   update      Update a server
 
 FLAGS:

internal/namespaces/applesilicon/v1alpha1/custom.go

@@ -9,6 +9,10 @@ import (
 func GetCommands() *core.Commands {
 	cmds := GetGeneratedCommands()
 
+	cmds.Merge(core.NewCommands(
+		serverSSHCommand(),
+	))
+
 	human.RegisterMarshalerFunc(applesilicon.ServerStatus(""), human.EnumMarshalFunc(serverStatusMarshalSpecs))
 
 	return cmds

internal/namespaces/applesilicon/v1alpha1/custom_server_ssh.go

@@ -0,0 +1,98 @@
+package applesilicon
+
+import (
+	"context"
+	"fmt"
+	"os/exec"
+	"reflect"
+
+	"github.com/scaleway/scaleway-cli/internal/core"
+	applesilicon "github.com/scaleway/scaleway-sdk-go/api/applesilicon/v1alpha1"
+	"github.com/scaleway/scaleway-sdk-go/scw"
+)
+
+type serverSSHConnectRequest struct {
+	Zone     scw.Zone
+	ServerID string
+	Username string
+	Port     uint
+	Command  string
+}
+
+func serverSSHCommand() *core.Command {
+	return &core.Command{
+		Short:     `SSH into a server`,
+		Long:      `Connect to distant server via the SSH protocol.`,
+		Namespace: "apple-silicon",
+		Verb:      "ssh",
+		Resource:  "server",
+		ArgsType:  reflect.TypeOf(serverSSHConnectRequest{}),
+		ArgSpecs: core.ArgSpecs{
+			{
+				Name:       "server-id",
+				Short:      "Server ID to SSH into",
+				Required:   true,
+				Positional: true,
+			},
+			{
+				Name:    "username",
+				Short:   "Username used for the SSH connection",
+				Default: core.DefaultValueSetter("m1"),
+			},
+			{
+				Name:    "port",
+				Short:   "Port used for the SSH connection",
+				Default: core.DefaultValueSetter("22"),
+			},
+			{
+				Name:  "command",
+				Short: "Command to execute on the remote server",
+			},
+			core.ZoneArgSpec(),
+		},
+		Run: serverSSHRun,
+	}
+}
+
+func serverSSHRun(ctx context.Context, argsI interface{}) (i interface{}, e error) {
+	args := argsI.(*serverSSHConnectRequest)
+
+	client := core.ExtractClient(ctx)
+	asAPI := applesilicon.NewAPI(client)
+	serverResp, err := asAPI.GetServer(&applesilicon.GetServerRequest{
+		Zone:     args.Zone,
+		ServerID: args.ServerID,
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	if serverResp.Status != applesilicon.ServerStatusReady {
+		return nil, &core.CliError{
+			Err:     fmt.Errorf("server is not ready"),
+			Details: fmt.Sprintf("Server %s currently in %s", serverResp.Name, serverResp.Status),
+		}
+	}
+
+	sshArgs := []string{
+		serverResp.IP.String(),
+		"-p", fmt.Sprintf("%d", args.Port),
+		"-l", args.Username,
+		"-t",
+	}
+	if args.Command != "" {
+		sshArgs = append(sshArgs, args.Command)
+	}
+
+	sshCmd := exec.Command("ssh", sshArgs...)
+
+	exitCode, err := core.ExecCmd(ctx, sshCmd)
+	if err != nil {
+		return nil, err
+	}
+	if exitCode != 0 {
+		return nil, &core.CliError{Empty: true, Code: exitCode}
+	}
+
+	return &core.SuccessResult{Empty: true}, nil
+}