Add support for INSTALLER_EXE templates

This commit adds support for creating package templates of type INSTALLER_EXE to
deal with the installers these are distributed as EXE files.

Author: d35ha

categories.txt

@@ -31,3 +31,4 @@ Utilities
 Visual Basic
 Web Application
 Wordlists
+VB

scripts/utils/create_package_template.py

@@ -118,6 +118,20 @@ def package_version(dependency_version):
 VM-Install-From-Zip $toolName $category $zipUrl -zipSha256 $zipSha256 -consoleApp ${console_app} -innerFolder ${inner_folder} -arguments $arguments
 """
 
+INSTALLER_EXE_TEMPLATE = r"""$ErrorActionPreference = 'Stop'
+Import-Module vm.common -Force -DisableNameChecking
+
+$toolName = '{tool_name}'
+$category = VM-Get-Category($MyInvocation.MyCommand.Definition)
+
+$exeUrl = '{target_url}'
+$exeSha256 = '{target_hash}'
+
+$toolDir = Join-Path ${{Env:RAW_TOOLS_DIR}} $toolName
+$executablePath = (Join-Path $toolDir '{target_file}')
+VM-Install-With-Installer $toolName $category "EXE" "/VERYSILENT /SUPPRESSMSGBOXES /NORESTART /SP- /Dir=`"$($toolDir)`"" $executablePath $exeUrl -sha256 $exeSha256
+"""
+
 """
 Needs the following format strings:
     tool_name="...", category="..."
@@ -228,6 +242,17 @@ def package_version(dependency_version):
 VM-Uninstall $toolName $category
 """
 
+UNINSTALLER_EXE_TEMPLATE = r"""$ErrorActionPreference = 'Continue'
+Import-Module vm.common -Force -DisableNameChecking
+
+$toolName = '{tool_name}'
+$category = VM-Get-Category($MyInvocation.MyCommand.Definition)
+
+VM-Uninstall-With-Uninstaller $toolName $category "EXE" "/VERYSILENT /SUPPRESSMSGBOXES /NORESTART /SP-"
+
+VM-Uninstall $toolName $category
+"""
+
 """
 Needs the following format strings:
     tool_name="...", category="..."
@@ -280,6 +305,23 @@ def create_zip_exe_template(packages_path, **kwargs):
     )
 
 
+def create_installer_exe_template(packages_path, **kwargs):
+    create_template(
+        INSTALLER_EXE_TEMPLATE,
+        uninstall_template=UNINSTALLER_EXE_TEMPLATE,
+        packages_path=packages_path,
+        pkg_name=kwargs.get("pkg_name"),
+        version=kwargs.get("version"),
+        authors=kwargs.get("authors"),
+        description=kwargs.get("description"),
+        tool_name=kwargs.get("tool_name"),
+        category=kwargs.get("category"),
+        target_url=kwargs.get("target_url"),
+        target_hash=kwargs.get("target_hash"),
+        target_file=kwargs.get("target_file"),
+    )
+
+
 def create_node_template(packages_path, **kwargs):
     create_template(
         NODE_TEMPLATE,
@@ -388,6 +430,7 @@ def create_template(
     category="",
     target_url="",
     target_hash="",
+    target_file="",
     shim_path="",
     dependency="",
     console_app="",
@@ -427,6 +470,7 @@ def create_template(
                 arguments=arguments,
                 target_url=target_url,
                 target_hash=target_hash,
+                target_file=target_file,
                 shim_path=shim_path,
                 console_app=console_app,
                 inner_folder=inner_folder,
@@ -476,6 +520,22 @@ def get_script_directory():
             "target_hash",
         ],
     },
+    "INSTALLER_EXE": {
+        "cb": create_installer_exe_template,
+        "doc": "An installer distributed as an EXE file",
+        "example": "<url>/tool.exe",
+        "arguments": [
+            "pkg_name",
+            "version",
+            "authors",
+            "description",
+            "tool_name",
+            "category",
+            "target_url",
+            "target_hash",
+            "target_file",
+        ],
+    },
     "NODE": {
         "cb": create_node_template,
         "doc": "An tool from the JavaScript Package Registry installed with npm",
@@ -627,6 +687,7 @@ def main(argv=None):
     parser.add_argument("--dependency", type=str, default="", help="Metapackage dependency")
     parser.add_argument("--target_url", type=str, default="", help="URL to target file (zip or executable)")
     parser.add_argument("--target_hash", type=str, default="", help="SHA256 hash of target file (zip or executable)")
+    parser.add_argument("--target_file", type=str, default="", help="EXE name/path under the installed tool folder")
     parser.add_argument("--shim_path", type=str, default="", help="Metapackage shim path")
     parser.add_argument(
         "--console_app",