Changed project names

Common/AmagnoVirtualPrinter.Utils/RegistryRepository.cs

@@ -0,0 +1,171 @@
+using System;
+using System.IO;
+using System.Linq;
+using AmagnoVirtualPrinter.Agent.Core.Enums;
+using AmagnoVirtualPrinter.Agent.Core.Interfaces;
+using AmagnoVirtualPrinter.Agent.Core.Model;
+using JetBrains.Annotations;
+
+using Microsoft.Win32;
+
+using AmagnoVirtualPrinter.Agent.Core;
+
+namespace AmagnoVirtualPrinter.Utils
+{
+    public class RegistryRepository : IRegistryRepository
+    {
+        private const short DefaultServerPort = 9101;
+
+        public bool TryGetGhostscriptPath(out string path)
+        {
+            path = null;
+
+            try
+            {
+                var regView = GetRegistryView();
+
+                using(var baseKey = RegistryKey.OpenBaseKey(RegistryHive.LocalMachine, regView))
+                {
+                    const string gsSubKey = @"SOFTWARE\GPL Ghostscript";
+                    using(var ghostscript = baseKey.OpenSubKey(gsSubKey))
+                    {
+                        CheckForNull(ghostscript, gsSubKey);
+                        var subKeys = ghostscript.GetSubKeyNames();
+
+                        var lastSubKey = subKeys.Last();
+                        using(var subKey = ghostscript.OpenSubKey(lastSubKey))
+                        {
+                            CheckForNull(subKey, lastSubKey);
+                            path = subKey.GetValue("GS_LIB").ToString().Split(';').FirstOrDefault();
+
+                            if (string.IsNullOrWhiteSpace(path))
+                            {
+                                return false;
+                            }
+
+                            path = Directory.GetParent(path).FullName;
+                            return true;
+                        }
+                    }
+                }
+            }
+            catch (Exception exception)
+            {
+                Console.WriteLine(exception);
+            }
+
+            return false;
+        }
+
+        public IExConfig GetRegistryConfig()
+        {
+            var regView = GetRegistryView();
+            var subKey = GetSubKey();
+            var registryConfig = new RegistryConfig();
+
+            using (var baseKey = RegistryKey.OpenBaseKey(RegistryHive.LocalMachine, regView))
+            {
+                using (var driver = baseKey.OpenSubKey(subKey))
+                {
+                    CheckForNull(driver, subKey);
+
+                    using(var key = driver.OpenSubKey(Keys.POSTCONVERTER_KEY))
+                    {
+                        CheckForNull(key, Keys.POSTCONVERTER_KEY);
+                        registryConfig.Postconverter = key.GetValue(KeyNames.EXECUTABLE_FILE).ToString();
+                    }
+
+                    using(var key = driver.OpenSubKey(Keys.PRECONVERTER_KEY))
+                    {
+                        CheckForNull(key, Keys.PRECONVERTER_KEY);
+                        registryConfig.Preconverter = key.GetValue(KeyNames.EXECUTABLE_FILE).ToString();
+                    }
+
+                    using (var key = driver.OpenSubKey(Keys.CONVERTER_KEY))
+                    {
+                        CheckForNull(key, Keys.CONVERTER_KEY);
+                        registryConfig.OutputDirectory = key.GetValue(KeyNames.OUTPUT_DIR).ToString();
+                        registryConfig.FileNameMask = key.GetValue(KeyNames.FILE_NAME_MASK).ToString();
+                        var portStr = key.GetValue(KeyNames.SERVER_PORT).ToString();
+                        registryConfig.PrinterPort = short.TryParse(portStr, out var portVal) ? portVal : DefaultServerPort;
+                        registryConfig.IntermediateFormat = key.GetValue(KeyNames.FORMAT)?.ToString().ToLower() == "ps" ? IntermediateFormat.Ps : IntermediateFormat.Xps;
+                    }
+                }
+            }
+
+            return registryConfig;
+        }
+
+        [ContractAnnotation("key:null => void")]
+        private void CheckForNull(RegistryKey key, string keyName)
+        {
+            if (key == null)
+            {
+                throw new NullReferenceException(keyName);
+            }
+        }
+
+        public IUserConfig GetUserRegistryConfig(string sid)
+        {
+            var regView = GetRegistryView();
+            var userConfig = new UserRegistryConfig();
+
+            using (var users = RegistryKey.OpenBaseKey(RegistryHive.Users, regView))
+            {
+                var subKey = $@"{sid}\{Keys.PRINTER_DRIVER_KEY32}";
+                using (var driver = users.OpenSubKey(subKey))
+                {
+                    CheckForNull(driver, subKey);
+
+                    using (var converter = driver.OpenSubKey(Keys.CONVERTER_KEY))
+                    {
+                        CheckForNull(converter, Keys.CONVERTER_KEY);
+
+                        subKey = "Redirect";
+                        using (var redirect = converter.OpenSubKey(subKey))
+                        {
+                            CheckForNull(redirect, subKey);
+
+                            userConfig.RedirectEnabled = (int?) redirect.GetValue("Enabled") == 1;
+                            userConfig.RedirectPrinter = redirect.GetValue("Printer").ToString();
+                            userConfig.Format = converter.GetValue("Format").ToString();
+
+                            var dpiStr = (string)driver.GetValue("RENDER: DPI");
+                            if (dpiStr == null)
+                            {
+                                userConfig.UserRenderDpi = null;
+                            }
+                            else
+                            {
+                                userConfig.UserRenderDpi = double.TryParse(dpiStr, out var dpiVal) ? dpiVal : (double?) null;
+                            }
+                        }
+                    }
+                }
+            }
+
+            return userConfig;
+        }
+
+        private RegistryView GetRegistryView()
+        {
+            if (Environment.Is64BitOperatingSystem)
+            {
+                return RegistryView.Registry64;
+            }
+
+            return RegistryView.Registry32;
+        }
+
+        [NotNull]
+        private string GetSubKey()
+        {
+            if (Environment.Is64BitOperatingSystem)
+            {
+                return Keys.PRINTER_DRIVER_KEY64;
+            }
+
+            return Keys.PRINTER_DRIVER_KEY32;
+        }
+    }
+}