Main.psm1
|
$ErrorActionPreference = "Stop" # Entry point of the module. Functions listed here can be exported to the user if declared in manifest. function Invoke-DocumentCompiler() { [CmdletBinding()] param( # Path to the Bikeshed file to build. If null, will look for a single .bs file in current directory. [Parameter()] [string]$path, # If provided, will ignore Bikeshed syntax errors and output whatever HTML it can. # This is useful for troubleshooting because Bikeshed errors can sometimes be rather vague. [Parameter()] [switch]$force ) $externals = PrepareExternals $inputFile = ResolveInputFile $path Write-Host "Building Bikeshed document: $($inputFile.FullName)" $workspaceRootPath = Split-Path -Parent $inputFile.FullName # Output goes into "Output/" directory relative to input file. $outputPath = Join-Path $workspaceRootPath "Output" Write-Verbose "Will generate output in $outputPath." # Assets come from "Assets/" directory relative to input file. # Assets are static files that are not generated, e.g. manually created JPG/PNG/XML or other files. $assetsPath = Join-Path $workspaceRootPath "Assets" Write-Verbose "Expending to find static assets in $assetsPath." PrepareOutputDirectory $outputPath $assetsPath # We will build all diagrams that exist in a Diagrams folder relative to the input document. $diagramsPath = Join-Path $workspaceRootPath "Diagrams" BuildDiagrams $outputPath $diagramsPath $externals $htmlFilePath = BuildBikeshedDocument $outputPath $inputFile $force $pdfFilePath = BuildPdf $outputPath $htmlFilePath $externals Write-Host "Build completed." return @{ htmlFilePath = $htmlFilePath pdfFilePath = $pdfFilePath } } ### Below this line is internal logic not exported to user. ### # This will setup all the external tools that we use (if not already done so) # and return a structure with paths to all of them, for easy reference later. function PrepareExternals() { # Ensure that prerequisites exist. if (!(Get-Command -Name "java" -ErrorAction SilentlyContinue)) { Write-Error "To use this command you must first install Java. You can verify installation using 'java -version'." } $externalsPath = Join-Path $PSScriptRoot "Externals" -Resolve $plantuml = Join-Path $externalsPath "plantuml.jar" -Resolve # On Linux we rely on user-installed binaries for the following externals: # graphviz # wkhtmltopdf if ($IsLinux) { $dot = Get-Command "dot" -ErrorAction SilentlyContinue $wk = Get-Command "wkhtmltopdf" -ErrorAction SilentlyContinue if (!$dot) { Write-Error "To use this command on Linux you must first install graphviz. You can verify installation using 'dot -V'." } if (!$wk) { Write-Error "To use this command on Linux you must first install wkhtmltopdf. You can verify installation using 'wkhtmltopdf --version'." } # We want just the path, not the command itself. $graphvizdot = $dot.Path $wkPath = $wk.Path } else { # Unpack Graphviz. It is in a zip for easy packaging - all we need to do is unpack it. $graphvizSource = Join-Path $externalsPath "graphviz-win64.zip" -Resolve $graphvizDestination = Join-Path $externalsPath "graphviz" if (!(Test-Path $graphvizDestination)) { # Got to extract graphviz (only on first run). # On upgrade we get a clean directory (there is never an old graphviz there). Expand-Archive -Path $graphvizSource -DestinationPath $graphvizDestination } $graphvizdot = Join-Path $graphvizDestination "dot.exe" -Resolve $wkPath = Join-Path $externalsPath "wkhtmltopdf.exe" -Resolve } # Validate that PlantUML can use the installed Graphviz dot.exe without issues. $stdout = java -jar "$plantuml" -graphvizdot "$graphvizdot" -testdot if ($stdout -match "Error: *") { $stdout | Out-Host Write-Error "Graphviz setup failed - PlantUML was unable to execute Graphviz!" } return @{ wkhtmltopdf = $wkPath plantuml = $plantuml dot = $graphvizdot } } function ResolveInputFile($path) { if ($path) { # If there is a path, it must be a single Bikeshed file. # We will accept any filename/extension as long as the file exists. if (!(Test-Path $path)) { Write-Error "Input file not found." } $file = Get-Item $path if ($file.Count -ne 1) { Write-Error "Input path resolved to multiple files. You can only build one document at a time." } if ($file -isnot [SYstem.IO.FileInfo]) { Write-Error "Input path did not point to a file." } return $file } else { # If there is no path, we expect a single .bs file in whatever the current directory is. $candidates = Get-ChildItem -File -Path "*.bs" if ($candidates.Count -ne 1) { Write-Error "Expected to find exactly 1 .bs file in current directory to use as input. Instead found $($candidates.Count)." } return $candidates } } function PrepareOutputDirectory($outputPath, $assetsPath) { Write-Verbose "Cleaning output directory." if (Test-Path $outputPath) { # Do not delete the directory itself, just contents. # This avoids some file locking issues. Get-ChildItem -Path $outputPath | Remove-Item -Force -Recurse } [IO.Directory]::CreateDirectory($outputPath) | Out-Null if (!(Test-Path $assetsPath) -or (Get-Item $assetsPath) -isnot [IO.DirectoryInfo]) { Write-Verbose "No Assets directory was found next to the input file. Will not copy any static assets." } else { Write-Host "Copying static assets to output directory." # Recursively copy all the static assets to the output directory. Get-ChildItem $assetsPath | % { Copy-Item -Path $_.FullName $outputPath -Recurse } } } function BuildDiagrams($outputPath, $diagramsPath, $externals) { if (!(Test-Path $diagramsPath) -or (Get-Item $diagramsPath) -isnot [IO.DirectoryInfo]) { Write-Verbose "No Diagrams directory was found next to the input file. Will not generate images from any PlantUML diagrams." } else { Write-Host "Generating images from PlantUML diagrams in $diagramsPath." # TODO: Support deeper paths here (if we chapterize the files, we would want Diagrams/Chapter1/*.wsd). # # This might be easily supported in PlantUML itself but alas, it does not really work that way. # http://forum.plantuml.net/8293/double-star-recursive-input-does-not-appear-to-work-reliably # However, we could work around it by individually processing each directory with PlantUML. java -jar "$($externals.plantuml)" -graphvizdot "$($externals.dot)" -output "$outputPath" -timeout 60 "$(Join-Path $diagramsPath '*.wsd')" if ($LASTEXITCODE -ne 0) { Write-Error "Diagram generation failed! See log above for errors." } } } # Builds the document and returns the path to the resulting HTML file. function BuildBikeshedDocument($outputPath, $inputFile, $force) { $outputFilePath = Join-Path $outputPath ($inputFile.BaseName + ".html") if ($force) { Write-Host "Building Bikeshed document." [Bikeshed]::Compile($inputFile.FullName, $true) | Out-File $outputFilePath } else { # First validate, because build only fails on super critical errors. Write-Host "Validating Bikeshed document." [Bikeshed]::Validate($inputFile.FullName) Write-Host "Building Bikeshed document." [Bikeshed]::Compile($inputFile.FullName, $false) | Out-File $outputFilePath } return $outputFilePath } function BuildPdf($outputPath, $htmlFilePath, $externals) { Write-Host "Generating PDF." $outputFilePath = Join-Path $outputPath ([IO.Path]::GetFileNameWithoutExtension($htmlFilePath) + ".pdf") # TODO: block access to unrelated files once a related defect is fixed # See https://github.com/wkhtmltopdf/wkhtmltopdf/issues/3846 & $externals.wkhtmltopdf --load-media-error-handling abort $htmlFilePath "$outputFilePath" if ($LASTEXITCODE -ne 0) { Write-Error "Conversion to PDF failed. See above log for errors." } return $outputFilePath } |