From 25a43268c93d15958978f5f5c0f3439ed4e520f2 Mon Sep 17 00:00:00 2001
From: Manuel Thalmann <m@nuth.ch>
Date: Wed, 28 Aug 2024 05:05:55 +0200
Subject: [PATCH] Add scripts for backing up and restoring LGHub

---
 scripts/Windows/Software/LGHub/Manage.ps1 | 84 ++++++++++++++++++++---
 1 file changed, 76 insertions(+), 8 deletions(-)

diff --git a/scripts/Windows/Software/LGHub/Manage.ps1 b/scripts/Windows/Software/LGHub/Manage.ps1
index d7fca148..ee840bed 100644
--- a/scripts/Windows/Software/LGHub/Manage.ps1
+++ b/scripts/Windows/Software/LGHub/Manage.ps1
@@ -3,13 +3,81 @@ param(
     [hashtable] $Arguments
 )
 
-. "$PSScriptRoot/../../Scripts/AppAssociations.ps1";
-. "$PSScriptRoot/../../../Common/Scripts/Software.ps1";
+& {
+    param ($Parameters)
+    . "$PSScriptRoot/../../Scripts/Restoration.ps1";
+    . "$PSScriptRoot/../../../Common/Scripts/Software.ps1";
+    $configPath = "$env:LocalAppData/LGHUB";
 
-Start-SoftwareInstaller @PSBoundParameters `
-    -Installer {
-        Install-ChocoPackage lghub;
-        Remove-DesktopIcon "*G HUB*";
-    };
+    <#
+        .SYNOPSIS
+        Stops the running Logitech G Hub instance.
+    #>
+    function Stop-LGHub {
+        [OutputType([string])]
+        param()
 
-# ToDo: Add restoration
+        $hubName = "lghub_system_tray.exe";
+
+        $candidates = Get-Process | Where-Object { $_.Path -and ((Split-Path -Leaf $_.Path) -eq $hubName) };
+
+        if ($candidates.Count -gt 0) {
+            $lghubPath = $candidates[0].Path;
+        } else {
+            $lghubPath = $null;
+        }
+
+        $mainProcesses = Get-Process | Where-Object {
+            $_.Path -and
+            (@("lghub.exe", "lghub_agent.exe", "lghub_updater.exe", $hubName) -contains (Split-Path -Leaf $_.Path)) -and
+            (($_.Parent.ProcessName -eq "explorer") -or ($null -eq $_.Parent))
+        };
+
+        $null = $mainProcesses | ForEach-Object { $_.Kill($true) };
+        return $lghubPath;
+    }
+
+    <#
+        .SYNOPSIS
+        Edits the Logitech G Hub configuration.
+    #>
+    function Edit-LGHubConfig {
+        param(
+            [scriptblock] $Action
+        )
+
+        Write-Host "Stopping Logitech G Hub";
+        $path = Stop-LGHub;
+        & $Action;
+
+        if ($path) {
+            Write-Host "Restarting Logitech G Hub";
+            Start-Process $path;
+        }
+    }
+
+    Start-SoftwareInstaller @Parameters `
+        -Installer {
+            Install-ChocoPackage lghub;
+            Remove-DesktopIcon "*G HUB*";
+        } `
+        -UserBackup {
+            param(
+                [hashtable] $Arguments
+            )
+
+            Edit-LGHubConfig {
+                Add-BackupArtifacts -User $Arguments.Name -Source $configPath -Path "LGHub" `
+                    -Include @("settings.db", "icon_cache")
+            };
+        } `
+        -UserConfigurator {
+            param(
+                [hashtable] $Arguments
+            )
+
+            Edit-LGHubConfig {
+                Expand-BackupArtifacts -User $Arguments.Name -Path "LGHub" -Target $configPath;
+            };
+        };
+} $PSBoundParameters;