Add function/api to enable ssh (for network mode) (#649)

Co-authored-by: Giulio <giulio@Giulios-MacBook-Pro.local>
Co-authored-by: Luca Rinaldi <lucarin@protonmail.com>

Author: 3 people

cmd/arduino-app-cli/board/board.go

@@ -66,6 +66,9 @@ func NewBoardCmd() *cobra.Command {
 	fsCmd.AddCommand(newBoardListCmd())
 	fsCmd.AddCommand(newBoardSetName())
 	fsCmd.AddCommand(newSetPasswordCmd())
+	fsCmd.AddCommand(newEnableNetworkModeCmd())
+	fsCmd.AddCommand(newDisableNetworkModeCmd())
+	fsCmd.AddCommand(newNetworkModeStatusCmd())
 
 	return fsCmd
 }
@@ -189,3 +192,58 @@ func newSetPasswordCmd() *cobra.Command {
 		},
 	}
 }
+
+func newEnableNetworkModeCmd() *cobra.Command {
+	return &cobra.Command{
+		Use:   "enable-ssh",
+		Short: "Enable and start the SSH service on the board",
+		Args:  cobra.ExactArgs(0),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			conn := cmd.Context().Value(remoteConnKey).(remote.RemoteConn)
+
+			if err := board.EnableNetworkMode(cmd.Context(), conn); err != nil {
+				return fmt.Errorf("failed to enable SSH: %w", err)
+			}
+
+			feedback.Printf("SSH service enabled and started\n")
+			return nil
+		},
+	}
+}
+
+func newDisableNetworkModeCmd() *cobra.Command {
+	return &cobra.Command{
+		Use:   "disable-ssh",
+		Short: "Disable and stop the SSH service on the board",
+		Args:  cobra.ExactArgs(0),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			conn := cmd.Context().Value(remoteConnKey).(remote.RemoteConn)
+
+			if err := board.DisableNetworkMode(cmd.Context(), conn); err != nil {
+				return fmt.Errorf("failed to disable SSH: %w", err)
+			}
+
+			feedback.Printf("SSH service disabled and stopped\n")
+			return nil
+		},
+	}
+}
+
+func newNetworkModeStatusCmd() *cobra.Command {
+	return &cobra.Command{
+		Use:   "status-ssh",
+		Short: "Check the status of the network mode on the board",
+		Args:  cobra.ExactArgs(0),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			conn := cmd.Context().Value(remoteConnKey).(remote.RemoteConn)
+
+			isEnabled, err := board.NetworkModeStatus(cmd.Context(), conn)
+			if err != nil {
+				return fmt.Errorf("failed to check network mode status: %w", err)
+			}
+
+			feedback.Printf("Network mode is %s\n", map[bool]string{true: "enabled", false: "disabled"}[isEnabled])
+			return nil
+		},
+	}
+}

debian/arduino-app-cli/etc/sudoers.d/20-arduino-ssh

@@ -0,0 +1 @@
+arduino ALL=(ALL) NOPASSWD: /usr/sbin/dpkg-reconfigure openssh-server, /usr/bin/systemctl enable ssh, /usr/bin/systemctl start ssh, /usr/bin/systemctl disable ssh, /usr/bin/systemctl stop ssh

pkg/board/board.go

@@ -3,9 +3,11 @@ package board
 import (
 	"bytes"
 	"context"
+	"errors"
 	"fmt"
 	"io"
 	"log/slog"
+	"os/exec"
 	"regexp"
 	"slices"
 	"strings"
@@ -300,6 +302,60 @@ func SetUserPassword(ctx context.Context, conn remote.RemoteConn, newPass string
 	return nil
 }
 
+func EnableNetworkMode(ctx context.Context, conn remote.RemoteConn) error {
+	if err := conn.GetCmd("sudo", "dpkg-reconfigure", "openssh-server").Run(ctx); err != nil {
+		return fmt.Errorf("failed to reconfigure openssh-server: %w", err)
+	}
+
+	if err := conn.GetCmd("sudo", "systemctl", "enable", "ssh").Run(ctx); err != nil {
+		return fmt.Errorf("failed to enable ssh service: %w", err)
+	}
+
+	if err := conn.GetCmd("sudo", "systemctl", "start", "ssh").Run(ctx); err != nil {
+		return fmt.Errorf("failed to start ssh service: %w", err)
+	}
+
+	return nil
+}
+
+func NetworkModeStatus(ctx context.Context, conn remote.RemoteConn) (bool, error) {
+	err := conn.GetCmd("systemctl", "is-enabled", "ssh").Run(ctx)
+	if err != nil {
+		var exitErr *exec.ExitError
+		if errors.As(err, &exitErr) {
+			if exitErr.ExitCode() != 0 {
+				return false, nil
+			}
+		}
+		return false, fmt.Errorf("failed to check ssh service status: %w", err)
+	}
+
+	err = conn.GetCmd("systemctl", "is-active", "ssh").Run(ctx)
+	if err != nil {
+		var exitErr *exec.ExitError
+		if errors.As(err, &exitErr) {
+			if exitErr.ExitCode() != 0 {
+				return false, nil
+			}
+		}
+		return false, fmt.Errorf("failed to check ssh service status: %w", err)
+	}
+
+	return true, nil
+}
+
+func DisableNetworkMode(ctx context.Context, conn remote.RemoteConn) error {
+	if err := conn.GetCmd("sudo", "systemctl", "disable", "ssh").Run(ctx); err != nil {
+		return fmt.Errorf("failed to disable ssh service: %w", err)
+	}
+
+	if err := conn.GetCmd("sudo", "systemctl", "stop", "ssh").Run(ctx); err != nil {
+		return fmt.Errorf("failed to stop ssh service: %w", err)
+	}
+
+	return nil
+}
+
 func getSerial(conn remote.RemoteConn) (string, error) {
 	f, err := conn.ReadFile(SerialPath)
 	if err != nil {