Better screenshots and test VBA Module1 execution

Author: DecimalTurn

scripts/Build-VBA.ps1

@@ -4,6 +4,11 @@
 # and imports all .bas files from a specified folder into the document.
 # It then saves and closes the document, and cleans up the COM objects.
 
+# Load utiliies
+$scriptPath = Split-Path -Parent $MyInvocation.MyCommand.Path
+. "$scriptPath/utils/Screenshot.ps1"
+
+# Args
 $folderName = $args[0]
 $officeAppName = $args[1]
 
@@ -37,6 +42,12 @@ if (-not (Test-Path $outputFilePath)) {
     exit 1
 }
 
+$screenshotDir = (DirUp $outputDir) + "screenshots/"
+if (-not (Test-Path $screenshotDir)) {
+    New-Item -ItemType Directory -Path $screenshotDir -Force | Out-Null
+    Write-Host "Created screenshot directory: $screenshotDir"
+}
+
 # Allows to double-ckeck if the VBOM is enabled
 # HKCU:\Software\Microsoft\Office\16.0\Common\TrustCenter
 # Just check the registry entry
@@ -105,6 +116,7 @@ if ($officeAppName -eq "Excel") {
 # Check if the document was opened successfully
 if ($null -eq $doc) {
     Write-Host "Error: Failed to open the document: $outputFilePath"
+    Take-Screenshot -OutputPath "${screenshotDir}Screenshot_${fileNameNoExt}_{{timestamp}}.png"
     exit 1
 }
 
@@ -156,15 +168,7 @@ $basFiles | ForEach-Object {
     }
 }
 
-# Take a screenshot of the Office application
-$scriptPath = Split-Path -Parent $MyInvocation.MyCommand.Path
-. "$scriptPath/utils/Screenshot.ps1"
-$screenshotDir = (DirUp $outputDir) + "screenshots/"
-if (-not (Test-Path $screenshotDir)) {
-    New-Item -ItemType Directory -Path $screenshotDir -Force | Out-Null
-    Write-Host "Created screenshot directory: $screenshotDir"
-}
-Take-Screenshot -OutputPath "${screenshotDir}Screenshot_${fileNameNoExt}.png"
+Take-Screenshot -OutputPath "${screenshotDir}Screenshot_${fileNameNoExt}_{{timestamp}}.png"
 
 # Save the document
 Write-Host "Saving document..."
@@ -179,6 +183,7 @@ try {
     }
 } catch {
     Write-Host "Warning: Could not save document: $($_.Exception.Message)"
+    Take-Screenshot -OutputPath "${screenshotDir}Screenshot_${fileNameNoExt}_{{timestamp}}.png"
 
     # Alternative approach for PowerPoint if SaveAs fails
     if ($officeAppName -eq "PowerPoint") {
@@ -210,10 +215,39 @@ try {
             exit 0
         } catch {
             Write-Host "Error: Alternative save method also failed: $($_.Exception.Message)"
+            Take-Screenshot -OutputPath "${screenshotDir}Screenshot_${fileNameNoExt}_{{timestamp}}.png"
         }
     }
 }
 
+# Call the WriteFile macro to check if the module was imported correctly
+try {
+    $macroName = "WriteFile"
+    $officeApp.Run($macroName)
+    
+    # Check if the file was created successfully with the correct content
+    $outputFile = "$outputDir/$fileNameNoExt.txt"
+    if (Test-Path $outputFile) {
+        $fileContent = Get-Content -Path $outputFile
+        if ($fileContent -eq "Hello, World") {
+            Write-Host "🟢 Macro executed successfully and file content is correct."
+        } else {
+            Write-Host "🟡 Warning: Macro executed, but file content is incorrect."
+        }
+
+        # Delete the output file after checking
+        Remove-Item -Path $outputFile -Force
+        Write-Host "Output test file deleted successfully."
+
+    } else {
+        Write-Host "🟡 Warning: Macro executed, but output file was not created."
+    }
+
+} catch {
+    Write-Host "🟡 Warning: Could not execute macro ${macroName}: $($_.Exception.Message)"
+    Take-Screenshot -OutputPath "${screenshotDir}Screenshot_${fileNameNoExt}_{{timestamp}}.png"
+}
+
 # Close the document
 try {
     $doc.Close()

scripts/utils/Screenshot.ps1

@@ -20,6 +20,10 @@ function Take-Screenshot {
     # Copy the screen to the bitmap
     $graphic.CopyFromScreen($bounds.X, $bounds.Y, 0, 0, $bounds.Size)
 
+    # Replace {{timestamp}} with the current date and time
+    $timestamp = (Get-Date).ToString("yyyyMMdd_HHmmss")
+    $OutputPath = $OutputPath -replace "{{timestamp}}", $timestamp
+
     # Save the bitmap as a file
     $bitmap.Save($OutputPath)
 

tests/AccessDatabase.accdb/Modules/Module1.bas

@@ -5,4 +5,25 @@ Option Explicit
 
 Sub Demo()
     MsgBox "Hello, World!"
+End Sub
+
+'This code will be called via COM to test if the VBA import was successful
+Sub WriteToFile()
+    Dim filePath As String
+    Dim fileNum As Integer
+    
+    ' Specify the path to the text file
+    filePath = CurrentProject.Path & "\AccessDatabase.txt"
+    
+    ' Get a free file number
+    fileNum = FreeFile
+    
+    ' Open the file for output
+    Open filePath For Output As #fileNum
+    
+    ' Write some text to the file
+    Print #fileNum, "Hello, World!"
+    
+    ' Close the file
+    Close #fileNum
 End Sub

tests/ExcelAddin.xlam/Modules/Module1.bas

@@ -5,4 +5,25 @@ Option Explicit
 
 Sub Demo()
     MsgBox "Hello, World!"
+End Sub
+
+'This code will be called via COM to test if the VBA import was successful
+Sub WriteToFile()
+    Dim filePath As String
+    Dim fileNum As Integer
+    
+    ' Specify the path to the text file
+    filePath = ThisWorkbook.Path & "\ExcelAddin.txt"
+    
+    ' Get a free file number
+    fileNum = FreeFile
+    
+    ' Open the file for output
+    Open filePath For Output As #fileNum
+    
+    ' Write some text to the file
+    Print #fileNum, "Hello, World!"
+    
+    ' Close the file
+    Close #fileNum
 End Sub

tests/ExcelWorkbook.xlsm/Modules/Module1.bas

@@ -5,4 +5,25 @@ Option Explicit
 
 Sub Demo()
     MsgBox "Hello, World!"
+End Sub
+
+'This code will be called via COM to test if the VBA import was successful
+Sub WriteToFile()
+    Dim filePath As String
+    Dim fileNum As Integer
+    
+    ' Specify the path to the text file
+    filePath = ThisWorkbook.Path & "\ExcelWorkbook.txt"
+    
+    ' Get a free file number
+    fileNum = FreeFile
+    
+    ' Open the file for output
+    Open filePath For Output As #fileNum
+    
+    ' Write some text to the file
+    Print #fileNum, "Hello, World!"
+    
+    ' Close the file
+    Close #fileNum
 End Sub

tests/PowerPointPresentation.pptm/Modules/Module1.bas

@@ -5,4 +5,25 @@ Option Explicit
 
 Sub Demo()
     MsgBox "Hello, World!"
+End Sub
+
+'This code will be called via COM to test if the VBA import was successful
+Sub WriteToFile()
+    Dim filePath As String
+    Dim fileNum As Integer
+    
+    ' Specify the path to the text file
+    filePath = ThisPresentation.Path & "\PowerPointPresentation.txt"
+    
+    ' Get a free file number
+    fileNum = FreeFile
+    
+    ' Open the file for output
+    Open filePath For Output As #fileNum
+    
+    ' Write some text to the file
+    Print #fileNum, "Hello, World!"
+    
+    ' Close the file
+    Close #fileNum
 End Sub

tests/WordDocument.docm/Modules/Module1.bas

@@ -5,4 +5,25 @@ Option Explicit
 
 Sub Demo()
     MsgBox "Hello, World!"
+End Sub
+
+'This code will be called via COM to test if the VBA import was successful
+Sub WriteToFile()
+    Dim filePath As String
+    Dim fileNum As Integer
+    
+    ' Specify the path to the text file
+    filePath = ThisDocument.Path & "\WordDocument.txt"
+    
+    ' Get a free file number
+    fileNum = FreeFile
+    
+    ' Open the file for output
+    Open filePath For Output As #fileNum
+    
+    ' Write some text to the file
+    Print #fileNum, "Hello, World!"
+    
+    ' Close the file
+    Close #fileNum
 End Sub