[Setup] Implement modulesRegistry API

microsoft · Oct 25, 2021 · c324cd5 · c324cd5
1 parent 5cfbd72
commit c324cd5
Show file tree

Hide file tree

Showing 4 changed files with 428 additions and 0 deletions.
diff --git a/PowerToys.sln b/PowerToys.sln
@@ -276,10 +276,12 @@ Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "utils", "utils", "{B39DC643
 		src\common\utils\HttpClient.h = src\common\utils\HttpClient.h
 		src\common\utils\json.h = src\common\utils\json.h
 		src\common\utils\logger_helper.h = src\common\utils\logger_helper.h
+		src\common\utils\modulesRegistry.h = src\common\utils\modulesRegistry.h
 		src\common\utils\MsiUtils.h = src\common\utils\MsiUtils.h
 		src\common\utils\os-detect.h = src\common\utils\os-detect.h
 		src\common\utils\process_path.h = src\common\utils\process_path.h
 		src\common\utils\ProcessWaiter.h = src\common\utils\ProcessWaiter.h
+		src\common\utils\registry.h = src\common\utils\registry.h
 		src\common\utils\resources.h = src\common\utils\resources.h
 		src\common\utils\string_utils.h = src\common\utils\string_utils.h
 		src\common\utils\timeutil.h = src\common\utils\timeutil.h

diff --git a/src/common/utils/modulesRegistry.h b/src/common/utils/modulesRegistry.h
@@ -0,0 +1,88 @@
+#pragma once
+
+#include "registry.h"
+
+#include <filesystem>
+
+namespace fs = std::filesystem;
+
+inline registry::Changeset getSvgPreviewHandlerChangset(const std::wstring installationDir, const bool perUser)
+{
+    using namespace registry::shellex;
+    return generatePreviewHandler(PreviewHandlerType::preview,
+                                  perUser,
+                                  L"{ddee2b8a-6807-48a6-bb20-2338174ff779}",
+                                  get_std_product_version(),
+                                  (fs::path{ installationDir } /
+                                   LR"d(modules\FileExplorerPreview\SvgPreviewHandler.comhost.dll)d")
+                                      .wstring(),
+                                  registry::DOTNET_COMPONENT_CATEGORY_CLSID,
+                                  L"Microsoft.PowerToys.PreviewHandler.Svg.SvgPreviewHandler",
+                                  L"Svg Preview Handler",
+                                  L".svg");
+}
+
+inline registry::Changeset getMdPreviewHandlerChangset(const std::wstring installationDir, const bool perUser)
+{
+    using namespace registry::shellex;
+    return generatePreviewHandler(PreviewHandlerType::preview,
+                                  perUser,
+                                  L"{45769bcc-e8fd-42d0-947e-02beef77a1f5}",
+                                  get_std_product_version(),
+                                  (fs::path{ installationDir } / LR"d(modules\FileExplorerPreview\MarkdownPreviewHandler.comhost.dll)d").wstring(),
+                                  registry::DOTNET_COMPONENT_CATEGORY_CLSID,
+                                  L"Microsoft.PowerToys.PreviewHandler.Markdown.MarkdownPreviewHandler",
+                                  L"Markdown Preview Handler",
+                                  L".md");
+}
+
+inline registry::Changeset getPdfPreviewHandlerChangset(const std::wstring installationDir, const bool perUser)
+{
+    using namespace registry::shellex;
+    return generatePreviewHandler(PreviewHandlerType::preview,
+                                  perUser,
+                                  L"{07665729-6243-4746-95b7-79579308d1b2}",
+                                  get_std_product_version(),
+                                  (fs::path{ installationDir } / LR"d(modules\FileExplorerPreview\PdfPreviewHandler.comhost.dll)d").wstring(),
+                                  registry::DOTNET_COMPONENT_CATEGORY_CLSID,
+                                  L"Microsoft.PowerToys.PreviewHandler.Pdf.PdfPreviewHandler",
+                                  L"Pdf Preview Handler",
+                                  L".pdf");
+}
+
+inline registry::Changeset getSvgThumbnailHandlerChangset(const std::wstring installationDir, const bool perUser)
+{
+    using namespace registry::shellex;
+    return generatePreviewHandler(PreviewHandlerType::thumbnail,
+                                  perUser,
+                                  L"{36B27788-A8BB-4698-A756-DF9F11F64F84}",
+                                  get_std_product_version(),
+                                  (fs::path{ installationDir } / LR"d(modules\FileExplorerPreview\SvgThumbnailProvider.comhost.dll)d").wstring(),
+                                  registry::DOTNET_COMPONENT_CATEGORY_CLSID,
+                                  L"Microsoft.PowerToys.ThumbnailHandler.Svg.SvgThumbnailProvider",
+                                  L"Svg Thumbnail Provider",
+                                  L".svg");
+}
+
+inline registry::Changeset getPdfThumbnailHandlerChangset(const std::wstring installationDir, const bool perUser)
+{
+    using namespace registry::shellex;
+    return generatePreviewHandler(PreviewHandlerType::thumbnail,
+                                  perUser,
+                                  L"{BCC13D15-9720-4CC4-8371-EA74A274741E}",
+                                  get_std_product_version(),
+                                  (fs::path{ installationDir } / LR"d(modules\FileExplorerPreview\PdfThumbnailProvider.comhost.dll)d").wstring(),
+                                  registry::DOTNET_COMPONENT_CATEGORY_CLSID,
+                                  L"Microsoft.PowerToys.ThumbnailHandler.Pdf.PdfThumbnailProvider",
+                                  L"Pdf Thumbnail Provider",
+                                  L".pdf");
+}
+
+inline std::vector<registry::Changeset> getAllModulesChangesets(const std::wstring installationDir, const bool perUser)
+{
+    return { getSvgPreviewHandlerChangset(installationDir, perUser),
+             getMdPreviewHandlerChangset(installationDir, perUser),
+             getPdfPreviewHandlerChangset(installationDir, perUser),
+             getSvgThumbnailHandlerChangset(installationDir, perUser),
+             getPdfThumbnailHandlerChangset(installationDir, perUser) };
+}