From e0cb0fec9b8fba61ffbe0ceb4e5f00a0263a3d2b Mon Sep 17 00:00:00 2001
From: scottr <scottr@amazon.com>
Date: Wed, 19 May 2021 19:39:52 -0700
Subject: [PATCH] [cpack_installer] add desktop and start menu shortcuts

---
 .../Platform/Windows/Packaging/Shortcuts.wxs  | 68 +++++++++++++++++++
 .../Windows/Packaging/Template.wxs.in         |  5 +-
 .../Platform/Windows/Packaging_windows.cmake  |  4 ++
 .../Windows/platform_windows_files.cmake      |  1 +
 4 files changed, 77 insertions(+), 1 deletion(-)
 create mode 100644 cmake/Platform/Windows/Packaging/Shortcuts.wxs

diff --git a/cmake/Platform/Windows/Packaging/Shortcuts.wxs b/cmake/Platform/Windows/Packaging/Shortcuts.wxs
new file mode 100644
index 0000000000..fb9d359b5a
--- /dev/null
+++ b/cmake/Platform/Windows/Packaging/Shortcuts.wxs
@@ -0,0 +1,68 @@
+<?xml version='1.0' encoding='windows-1252'?>
+
+<?include "cpack_variables.wxi"?>
+
+<Wix xmlns='http://schemas.microsoft.com/wix/2006/wi'>
+    <Fragment>
+
+        <DirectoryRef Id="TARGETDIR">
+            <Directory Id="ProgramMenuFolder">
+                <Directory Id="ApplicationProgramsFolder" Name="$(var.CPACK_PACKAGE_NAME)">
+                    <Directory Id="VersionedApplicationProgramsFolder" Name="$(var.CPACK_PACKAGE_VERSION)"/>
+                </Directory>
+            </Directory>
+            <Directory Id="DesktopFolder" SourceName="Desktop"/>
+        </DirectoryRef>
+
+        <!-- the WorkingDirectory attribute for shortcuts uses directory refs -->
+        <DirectoryRef Id="INSTALL_ROOT">
+            <Directory Id="root.bin" Name="bin">
+                <Directory Id="root.bin.Windows" Name="Windows">
+                    <Directory Id="root.bin.Windows.profile" Name="profile" />
+                </Directory>
+            </Directory>
+        </DirectoryRef>
+
+        <DirectoryRef Id="DesktopFolder">
+            <Component Id="DesktopShortcuts" Guid="{2600B54A-65FB-4507-A7CD-3CE4817C7173}">
+
+                <Shortcut Id="DesktopShortcut_Launcher"
+                    Target="[root.bin.Windows.profile]o3de.exe"
+                    WorkingDirectory="root.bin.Windows.profile"
+                    Name="$(var.CPACK_PACKAGE_NAME) $(var.CPACK_PACKAGE_VERSION)" />
+
+                <RemoveFolder Id="DesktopFolder" On="uninstall"/>
+
+                <RegistryValue
+                    Root="HKCU"
+                    Key="Software\$(var.CPACK_PACKAGE_VENDOR)\$(var.CPACK_PACKAGE_NAME)"
+                    Name="installed"
+                    Type="integer"
+                    Value="1"
+                    KeyPath="yes"/>
+            </Component>
+        </DirectoryRef>
+
+        <DirectoryRef Id="VersionedApplicationProgramsFolder">
+            <Component Id="StartMenuShortcuts" Guid="{E6447F0F-A46E-4A72-83D8-600707B590E8}">
+
+                <Shortcut Id="StartMenuShortcut_Launcher"
+                    Target="[root.bin.Windows.profile]o3de.exe"
+                    WorkingDirectory="root.bin.Windows.profile"
+                    Name="$(var.CPACK_PACKAGE_NAME)" />
+
+                <RemoveFolder Id="RemoveVersionedApplicationProgramsFolder" Directory='VersionedApplicationProgramsFolder' On="uninstall"/>
+                <RemoveFolder Id="RemoveApplicationProgramsFolder" Directory="ApplicationProgramsFolder" On="uninstall"/>
+
+                <RegistryValue
+                    Root="HKCU"
+                    Key="Software\$(var.CPACK_PACKAGE_VENDOR)\$(var.CPACK_PACKAGE_NAME)"
+                    Name="installed"
+                    Type="integer"
+                    Value="1"
+                    KeyPath="yes"/>
+            </Component>
+        </DirectoryRef>
+
+    </Fragment>
+</Wix>
diff --git a/cmake/Platform/Windows/Packaging/Template.wxs.in b/cmake/Platform/Windows/Packaging/Template.wxs.in
index 0b3c597ab6..2900b96f41 100644
--- a/cmake/Platform/Windows/Packaging/Template.wxs.in
+++ b/cmake/Platform/Windows/Packaging/Template.wxs.in
@@ -38,7 +38,10 @@
         <WixVariable Id="WixUIDialogBmp" Value="$(var.CPACK_WIX_UI_DIALOG)"/>
         <?endif?>
 
-        <FeatureRef Id="ProductFeature"/>
+        <FeatureRef Id="ProductFeature">
+            <ComponentRef Id="DesktopShortcuts"/>
+            <ComponentRef Id="StartMenuShortcuts"/>
+        </FeatureRef>
 
         <UIRef Id="$(var.CPACK_WIX_UI_REF)" />
 
diff --git a/cmake/Platform/Windows/Packaging_windows.cmake b/cmake/Platform/Windows/Packaging_windows.cmake
index 8504447d4f..204d59852a 100644
--- a/cmake/Platform/Windows/Packaging_windows.cmake
+++ b/cmake/Platform/Windows/Packaging_windows.cmake
@@ -83,6 +83,10 @@ set(CPACK_WIX_PRODUCT_ICON ${CPACK_SOURCE_DIR}/Platform/Windows/Packaging/produc
 
 set(CPACK_WIX_TEMPLATE "${CPACK_SOURCE_DIR}/Platform/Windows/Packaging/Template.wxs.in")
 
+set(CPACK_WIX_EXTRA_SOURCES
+    "${CPACK_SOURCE_DIR}/Platform/Windows/Packaging/Shortcuts.wxs"
+)
+
 set(_embed_artifacts "yes")
 
 if(LY_INSTALLER_DOWNLOAD_URL)
diff --git a/cmake/Platform/Windows/platform_windows_files.cmake b/cmake/Platform/Windows/platform_windows_files.cmake
index b760a8760d..3ce53fbcea 100644
--- a/cmake/Platform/Windows/platform_windows_files.cmake
+++ b/cmake/Platform/Windows/platform_windows_files.cmake
@@ -26,5 +26,6 @@ set(FILES
     Packaging_windows.cmake
     PackagingPostBuild.cmake
     Packaging/Bootstrapper.wxs
+    Packaging/Shortcuts.wxs
     Packaging/Template.wxs.in
 )