From 4e873cd7929e12083dea3b0f069b495453a24f1c Mon Sep 17 00:00:00 2001
From: Giovani <giovani@boldquest.net>
Date: Fri, 16 Jul 2021 22:26:19 +0000
Subject: [PATCH] feat: improve msi build script

---
 build.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/build.py b/build.py
index 7cb3b05..d36dd78 100644
--- a/build.py
+++ b/build.py
@@ -5,13 +5,31 @@ base = None
 if sys.platform == 'win32':
     base = "Win32GUI"
 
-# TODO add icon to cx freeze set up
+shortcut_table = [
+    ("DesktopShortcut",        # Shortcut
+     "DesktopFolder",          # Directory_
+     "Tetri5",                 # Name
+     "TARGETDIR",              # Component_
+     "[TARGETDIR]main.exe",    # Target
+     None,                     # Arguments
+     None,                     # Description
+     None,                     # Hotkey
+     None,                     # Icon
+     None,                     # IconIndex
+     None,                     # ShowCmd
+     'TARGETDIR'               # WkDir
+     )
+    ]
+
+msi_data = {"Shortcut": shortcut_table}
+bdist_msi_options = {'data': msi_data}
 
 setup(
     name="Tetri5",
     version="1.0.0",
     description="A full featured tetris game using python's pygame library. https://github.com/gio101046/tetri5",
     options={
+        "bdist_msi": bdist_msi_options,
         "build_exe": { 
             "packages": [
                 "pygame",