add support for set

Author: Steve Lee (POWERSHELL HE/HIM) (from Dev Box)

resources/registry/locales/en-us.toml

@@ -3,6 +3,8 @@ _version = 1
 [adapter]
 getProcessingKey = "Processing key: %{key}"
 getNoAdaptedRegistryValueFound = "No adapted registry value found for key: %{key}"
+setProcessingKey = "Setting key: %{key}"
+setNoAdaptedRegistryValueFound = "No adapted registry value found for key: %{key}"
 registryValueNotFound = "Registry value not found for key path: %{key_path} and value name: %{value_name}"
 couldNotConvertRegistryValue = "Could not convert registry value for key path '%{key_path}' and value name '%{value_name}' to JSON type: %{json_type}"
 unsupportedConversionToJsonType = "Unsupported conversion from %{registry_value_data} to JSON type: %{json_type}"

resources/registry/src/adapter.rs

@@ -298,15 +298,37 @@ fn get_registry_value_data(value_name: &str, value: &Value, map: &Map<String, Va
     Ok(registry_value_data)
 }
 
-pub fn adapter_set(input: &str, adapted_resource: &str) -> Result<String, RegistryResourceError> {
-    trace!("Adapter Set with input: {input}");
+pub fn adapter_set(input: &str, adapted_resource: &str) -> Result<(), RegistryResourceError> {
     let adapted_resource: AdaptedRegistryResource = serde_json::from_str(adapted_resource)
         .map_err(|e| RegistryResourceError::AdaptedResource(e.to_string()))?;
-    
+    let mut resource_map = HashMap::new();
+
     for (key, value) in adapted_resource.properties.iter() {
-        trace!("Property: {key} = {value}");
+        let adapted_registry_value: AdaptedRegistryValue = serde_json::from_value(value.clone())
+            .map_err(|e| RegistryResourceError::AdaptedResource(e.to_string()))?;
+        resource_map.insert(key.clone(), adapted_registry_value);
     }
-    Ok("{}".to_string())
+
+    let input_map: Map<String, Value> = serde_json::from_str(input)
+        .map_err(|e| RegistryResourceError::AdaptedResource(e.to_string()))?;
+    for (key, value) in input_map.iter() {
+        if let Some(adapted_registry_value) = resource_map.get(key) {
+            debug!("{}", t!("adapter.setProcessingKey", key = key));
+            if let Some(json_map) = adapted_registry_value.map_json_to_registry.as_object() {
+                let registry_data = get_registry_value_data(&adapted_registry_value.value_name, value, json_map, &adapted_registry_value.value_type)?;
+                let registry_helper = RegistryHelper::new(&adapted_registry_value.key_path, Some(adapted_registry_value.value_name.clone()), Some(registry_data))?;
+                if let Err(e) = registry_helper.set() {
+                    return Err(RegistryResourceError::RegistryError(e));
+                }
+            } else {
+                warn!("No mapping found for key {}", key);
+            }
+        } else {
+            debug!("{}", t!("adapter.setNoAdaptedRegistryValueFound", key = key));
+        }
+    }
+
+    Ok(())
 }
 
 pub fn adapter_export(input: &str, adapted_resource: &str) -> Result<String, RegistryResourceError> {

resources/registry/src/main.rs

@@ -43,7 +43,11 @@ fn main() {
                     adapter_get(&input, &adapted_resource)
                 },
                 AdapterSubCommand::Set { input, adapted_resource } => {
-                    adapter_set(&input, &adapted_resource)
+                    if let Err(e) = adapter_set(&input, &adapted_resource) {
+                        error!("{e}");
+                        exit(EXIT_REGISTRY_ERROR);
+                    }
+                    exit(EXIT_SUCCESS);
                 },
                 AdapterSubCommand::Export { input, adapted_resource } => {
                     adapter_export(&input, &adapted_resource)

resources/windows_personalization/personalization_set.tests.ps1

@@ -0,0 +1,45 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
+Describe 'Personalization resource set tests' -Skip:(!$IsWindows) {
+    BeforeAll {
+        $currentSettings = dsc resource export -r Microsoft.Windows/Personalization 2>$TestDrive/error.log
+        $LASTEXITCODE | Should -Be 0 -Because "Failed to export current personalization settings with exit code $LASTEXITCODE. Error log: $(Get-Content $TestDrive/error.log -Raw)"
+    }
+
+    AfterAll {
+        dsc resource set -r Microsoft.Windows/Personalization -i $currentSettings 2>$TestDrive/error.log
+        $LASTEXITCODE | Should -Be 0 -Because (Get-Content $TestDrive/error.log -Raw)
+    }
+
+    It 'Can set personalization settings' {
+        $newSettings = @{
+            appsUseLightTheme = $true
+            systemUsesLightTheme = $true
+            autoColorization = $true
+            colorPrevalence = $true
+            transparencyEffects = $true
+            startMenuVisiblePlaces = @('Documents', 'Downloads','Music', 'Pictures', 'Videos', 'Network', 'UserProfile', 'Explorer', 'Settings')
+            startMenuShowRecentList = $true
+            showRecommendedApps = $true
+            taskbarShowBadges = $true
+            desktopTaskbarShowBadges = $true
+            multimonitorTaskbarGroupingMode = 'CombineWhenTaskbarIsFull'
+            multimonitorTaskbar = $true
+            multimonitorDesktopTaskbar = $true
+            multimonitorTaskbarMode = 'PrimaryAndMonitorWindowIsOn'
+            multimonitorDesktopTaskbarMode = 'MonitorWindowIsOn'
+        }
+
+        dsc resource set -r Microsoft.Windows/Personalization -i ($newSettings | ConvertTo-Json) 2>$TestDrive/error.log
+        $LASTEXITCODE | Should -Be 0 -Because (Get-Content $TestDrive/error.log -Raw)
+
+        $out = dsc resource export -r Microsoft.Windows/Personalization 2>$TestDrive/error.log | ConvertFrom-Json
+        $LASTEXITCODE | Should -Be 0 -Because (Get-Content $TestDrive/error.log -Raw)
+        foreach ($key in $newSettings.Keys) {
+            $keyValue = $out.resources[0].properties.$key
+            $expectedValue = $newSettings.$key
+            $keyValue | Should -Be $expectedValue -Because "Property $key has value $keyValue which is not equal to expected value $expectedValue"
+        }
+    }
+}