From ade414284d6e489f5a7a2d858d5abc2b2ebe0f5a Mon Sep 17 00:00:00 2001
From: Saksham Singh <sakshamsingh.93502@gmail.com>
Date: Sun, 15 Sep 2024 20:07:08 +0530
Subject: [PATCH] Added option to enable ssh server under config tab

---
 config/feature.json                           | 10 ++-
 functions/private/Invoke-WinUtilSSHServer.ps1 | 80 +++++++++++++++++++
 functions/public/Invoke-WPFButton.ps1         |  1 +
 3 files changed, 90 insertions(+), 1 deletion(-)
 create mode 100644 functions/private/Invoke-WinUtilSSHServer.ps1

diff --git a/config/feature.json b/config/feature.json
index 10835451..137cd9ee 100644
--- a/config/feature.json
+++ b/config/feature.json
@@ -313,5 +313,13 @@
     "Order": "a083_",
     "Type": "Button",
     "ButtonWidth": "300"
-  }
+  },
+  "WPFWinUtilSSHServer": {
+    "Content": "Enable OpenSSH Server",
+    "category": "Remote Access",
+    "panel": "2",
+    "Order": "a084_",
+    "Type": "Button",
+    "ButtonWidth": "300"
+  },
 }
diff --git a/functions/private/Invoke-WinUtilSSHServer.ps1 b/functions/private/Invoke-WinUtilSSHServer.ps1
new file mode 100644
index 00000000..9124353a
--- /dev/null
+++ b/functions/private/Invoke-WinUtilSSHServer.ps1
@@ -0,0 +1,80 @@
+function Invoke-WinUtilSSHServer {
+    <#
+    .SYNOPSIS
+        Enables OpenSSH server to remote into your windows device
+    #>
+
+    # Get the latest version of OpenSSH Server
+    $FeatureName = Get-WindowsCapability -Online | Where-Object { $_.Name -like "OpenSSH.Server*" }
+
+    # Install the OpenSSH Server feature if not already installed
+    if ($FeatureName.State -ne "Installed") {
+        Write-Host "Enabling OpenSSH Server"
+        Add-WindowsCapability -Online -Name $FeatureName.Name
+    }
+
+    # Sets up the OpenSSH Server service
+    Write-Host "Starting the services"
+    Start-Service -Name sshd
+    Set-Service -Name sshd -StartupType Automatic
+
+    # Sets up the ssh-agent service
+    Start-Service 'ssh-agent'
+    Set-Service -Name 'ssh-agent' -StartupType 'Automatic'
+
+    # Confirm the required services are running
+    $service = Get-Service -Name sshd
+    $service1 = Get-Service -Name 'ssh-agent'
+    if ($service.Status -eq 'Running') {
+        Write-Host "OpenSSH Server is running."
+    } else {
+        try {
+            Write-Host "OpenSSH Server is not running. Attempting to restart..."
+            Restart-Service -Name sshd -Force
+            Write-Host "OpenSSH Server has been restarted successfully."
+        } catch {
+            Write-Host "Failed to restart OpenSSH Server: $_"
+        }
+    }
+    if ($service1.Status -eq 'Running') {
+        Write-Host "ssh-agent is running."
+    } else {
+        try {
+            Write-Host "ssh-agent is not running. Attempting to restart..."
+            Restart-Service -Name sshd -Force
+            Write-Host "ssh-agent has been restarted successfully."
+        } catch {
+            Write-Host "Failed to restart ssh-agent : $_"
+        }
+    }
+
+    #Adding Firewall rule for port 22
+    Write-Host "Setting up firewall rules"
+    $firewallRule = (Get-NetFirewallRule -Name 'sshd').Enabled
+    if ($firewallRule) {
+        Write-Host "Firewall rule for OpenSSH Server (sshd) already exists."
+    } else {
+        New-NetFirewallRule -Name sshd -DisplayName 'OpenSSH Server (sshd)' -Enabled True -Direction Inbound -Protocol TCP -Action Allow -LocalPort 22
+        Write-Host "Firewall rule for OpenSSH Server created and enabled."
+    }
+
+    # Check for the authorized_keys file
+    $sshFolderPath = "$env:HOMEPATH\ssh"
+    $authorizedKeysPath = "$sshFolderPath\authorized_keys"
+
+    if (-not (Test-Path -Path $sshFolderPath)) {
+        Write-Host "Creating ssh directory..."
+        New-Item -Path $sshFolderPath -ItemType Directory -Force
+    }
+
+    if (-not (Test-Path -Path $authorizedKeysPath)) {
+        Write-Host "Creating authorized_keys file..."
+        New-Item -Path $authorizedKeysPath -ItemType File -Force
+        Write-Host "authorized_keys file created at $authorizedKeysPath."
+    } else {
+        Write-Host "authorized_keys file already exists at $authorizedKeysPath."
+    }
+    Write-Host "OpenSSH server was successfully enabled."
+    Write-Host "The config file can be located at C:\ProgramData\ssh\sshd_config "
+    Write-Host "Add your public keys to this file -> %HOMEPATH%\ssh\authorized_keys"
+}
diff --git a/functions/public/Invoke-WPFButton.ps1 b/functions/public/Invoke-WPFButton.ps1
index 9f768fd8..2d6ab93c 100644
--- a/functions/public/Invoke-WPFButton.ps1
+++ b/functions/public/Invoke-WPFButton.ps1
@@ -59,5 +59,6 @@ function Invoke-WPFButton {
         "WPFCloseButton" {Invoke-WPFCloseButton}
         "MicrowinScratchDirBT" {Invoke-ScratchDialog}
         "WPFWinUtilPSProfile" {Invoke-WinUtilpsProfile}
+        "WPFWinUtilSSHServer" {Invoke-WinUtilSSHServer}
     }
 }