Merge pull request #935 from mandiant/package-magika

Package proposal: magika
mandiant · Apr 10, 2024 · 5dcdd11 · 5dcdd11
2 parents 9fe5557 + 3ec1e77
commit 5dcdd11
Show file tree

Hide file tree

Showing 5 changed files with 49 additions and 1 deletion.
diff --git a/packages/common.vm/common.vm.nuspec b/packages/common.vm/common.vm.nuspec
@@ -2,7 +2,7 @@
 <package xmlns="http://schemas.microsoft.com/packaging/2015/06/nuspec.xsd">
   <metadata>
     <id>common.vm</id>
-    <version>0.0.0.20240404</version>
+    <version>0.0.0.20240409</version>
     <description>Common libraries for VM-packages</description>
     <authors>Mandiant</authors>
   </metadata>

diff --git a/packages/common.vm/tools/vm.common/vm.common.psm1 b/packages/common.vm/tools/vm.common/vm.common.psm1
@@ -1550,3 +1550,13 @@ function VM-Get-MSIInstallerPathByProductName {
         VM-Write-Log-Exception $_ "An error occurred: $_"
     }
 }
+
+function VM-Pip-Install {
+    param (
+        [string]$package
+    )
+    # Create output file to log python module installation details
+    $outputFile = VM-New-Install-Log ${Env:VM_COMMON_DIR}
+
+    Invoke-Expression "py -3.10 -m pip install $package --disable-pip-version-check 2>&1 >> $outputFile"
+}
diff --git a/packages/magika.vm/magika.vm.nuspec b/packages/magika.vm/magika.vm.nuspec
@@ -0,0 +1,13 @@
+<?xml version="1.0" encoding="utf-8"?>
+<package xmlns="http://schemas.microsoft.com/packaging/2015/06/nuspec.xsd">
+  <metadata>
+    <id>magika.vm</id>
+    <version>0.5.1</version>
+    <authors>Yanick Fratantonio, Luca Invernizzi, Marina Zhang, Giancarlo Metitieri, Thomas Kurt, Francois Galilee, Alexandre Petit-Bianco, Loua Farah, Ange Albertini, Elie Bursztein</authors>
+    <description>Magika is an AI powered file type detection tool that uses deep learning to provide accurate detection.</description>
+    <dependencies>
+      <dependency id="common.vm" />
+      <dependency id="python3.vm" />
+    </dependencies>
+  </metadata>
+</package>
diff --git a/packages/magika.vm/tools/chocolateyinstall.ps1 b/packages/magika.vm/tools/chocolateyinstall.ps1
@@ -0,0 +1,16 @@
+$ErrorActionPreference = 'Stop'
+Import-Module vm.common -Force -DisableNameChecking
+
+try {
+    $toolName = 'magika'
+    $category = 'File Information'
+
+    VM-Pip-Install $toolName
+
+    $executablePath = "$(where.exe $toolName)"
+    $arguments = "--help"
+
+    VM-Install-Shortcut $toolName $category $executablePath -consoleApp $true -arguments $arguments -iconLocation $iconLocation
+} catch {
+    VM-Write-Log-Exception $_
+}
diff --git a/packages/magika.vm/tools/chocolateyuninstall.ps1 b/packages/magika.vm/tools/chocolateyuninstall.ps1
@@ -0,0 +1,9 @@
+$ErrorActionPreference = 'Continue'
+Import-Module vm.common -Force -DisableNameChecking
+
+$toolName = 'magika'
+$category = 'File Information'
+
+Invoke-Expression "py -3.10 -m pip uninstall $toolName -y"
+
+VM-Uninstall $toolName $category