From e75f0cfaf03c3e6061972f82d10de80442386408 Mon Sep 17 00:00:00 2001
From: Manuel Thalmann <m@nuth.ch>
Date: Tue, 3 Sep 2024 12:02:09 +0200
Subject: [PATCH] Allow backing up TrackMania Forever

---
 .../TrackMania Nations Forever/Manage.ps1     | 49 ++++++++++++++-----
 .../TrackMania United Forever/Manage.ps1      | 30 ++++++++++--
 2 files changed, 61 insertions(+), 18 deletions(-)

diff --git a/scripts/Windows/Software/TrackMania Nations Forever/Manage.ps1 b/scripts/Windows/Software/TrackMania Nations Forever/Manage.ps1
index e9b1fd2c..868a94de 100644
--- a/scripts/Windows/Software/TrackMania Nations Forever/Manage.ps1	
+++ b/scripts/Windows/Software/TrackMania Nations Forever/Manage.ps1	
@@ -3,19 +3,42 @@ param(
     [hashtable] $Arguments
 )
 
-. "$PSScriptRoot/../../../Common/Scripts/Software.ps1";
 
-Start-SoftwareInstaller @PSBoundParameters `
-    -Installer {
-        foreach ($feature in @("DirectPlay", "NetFx3")) {
-            if ((Get-WindowsOptionalFeature -Online -FeatureName $feature).State -ne "Enabled") {
-                Write-Information "Enabling the ``$feature`` feature…";
-                choco install --source windowsFeatures  -y $feature;
+& {
+    param($Parameters)
+    . "$PSScriptRoot/../../Scripts/Restoration.ps1";
+    . "$PSScriptRoot/../../../Common/Scripts/Software.ps1";
+    $path = "$HOME/Documents/TmForever";
+
+    Start-SoftwareInstaller @Parameters `
+        -Installer {
+            foreach ($feature in @("DirectPlay", "NetFx3")) {
+                if ((Get-WindowsOptionalFeature -Online -FeatureName $feature).State -ne "Enabled") {
+                    Write-Information "Enabling the ``$feature`` feature…";
+                    choco install --source windowsFeatures  -y $feature;
+                }
             }
-        }
 
-        Install-WingetPackage Nadeo.TrackManiaNationsForever;
-        Remove-DesktopIcon "*TmNationsForever*";
-    };
-
-# ToDo: Add restoration
+            Install-WingetPackage Nadeo.TrackManiaNationsForever;
+            Remove-DesktopIcon "*TmNationsForever*";
+        } `
+        -UserBackup {
+            param([hashtable] $Arguments)
+            $name = $Arguments.Name;
+            Add-BackupArtifacts -User $name -Source $path -Path "TmNationsForever" `
+                -Include @(
+                    "ChallengeMusics",
+                    "MediaTracker",
+                    "MenuMusics",
+                    "Painter",
+                    "Scores",
+                    "Skins",
+                    "Tracks"
+                );
+        } `
+        -UserConfigurator {
+            param([hashtable] $Arguments)
+            $name = $Arguments.Name;
+            Expand-BackupArtifacts -User $name -Path "TmNationsForever" -Target $path;
+        };
+} $PSBoundParameters;
diff --git a/scripts/Windows/Software/TrackMania United Forever/Manage.ps1 b/scripts/Windows/Software/TrackMania United Forever/Manage.ps1
index 3b97395c..748cac2e 100644
--- a/scripts/Windows/Software/TrackMania United Forever/Manage.ps1	
+++ b/scripts/Windows/Software/TrackMania United Forever/Manage.ps1	
@@ -3,10 +3,15 @@ param(
     [hashtable] $Arguments
 )
 
-. "$PSScriptRoot/../../../Common/Scripts/Software.ps1";
-. "$PSScriptRoot/../../../Common/Scripts/System.ps1";
 
-Start-SoftwareInstaller @PSBoundParameters `
+& {
+    param($Parameters)
+    . "$PSScriptRoot/../../Scripts/Restoration.ps1";
+    . "$PSScriptRoot/../../../Common/Scripts/Software.ps1";
+    . "$PSScriptRoot/../../../Common/Scripts/System.ps1";
+    $path = "$HOME/Documents/TrackMania";
+
+    Start-SoftwareInstaller @Parameters `
     -Installer {
         $file = "TmUnitedForever.exe";
         $dir = New-TemporaryDirectory;
@@ -28,6 +33,21 @@ Start-SoftwareInstaller @PSBoundParameters `
         $null = Pop-Location;
 
         Remove-Item -Recurse $dir;
+    } `
+    -UserBackup {
+        param([hashtable] $Arguments)
+        Add-BackupArtifacts -User $Arguments.Name -Source $path -Path "TmUnitedForever" `
+            -Include @(
+                "ChallengeMusics",
+                "MediaTracker",
+                "MenuMusics",
+                "Scores",
+                "Skins",
+                "Tracks"
+            );
+    } `
+    -UserConfigurator {
+        param([hashtable] $Arguments)
+        Expand-BackupArtifacts -User $Arguments.Name -Path "TmUnitedForever" -Target $path;
     };
-
-# ToDo: Add restoration
+} $PSBoundParameters;