boxstarter choco extension

Boxstarter.Bootstrapper/BoxStarter.Bootstrapper.psm1

@@ -11,4 +11,5 @@ Export-ModuleMember Invoke-BoxStarter, `
                     Stop-TimedSection, `
                     Out-Boxstarter, `
                     Enter-BoxstarterLogable, `
-                    Get-BoxstarterTempDir
+                    Get-BoxstarterTempDir, `
+                    Setup-BoxstarterExtension

Boxstarter.Bootstrapper/Cleanup-Boxstarter.ps1

@@ -7,6 +7,22 @@ function Cleanup-Boxstarter {
     }
     Start-UpdateServices
 
+    $extensionBaseDir = "$env:ChocolateyInstall\extensions\boxstarter-choco"
+    if (Test-Path $extensionBaseDir) {
+        Remove-Item $extensionBaseDir -Recurse -Force -ErrorAction SilentlyContinue
+
+        Import-Module "$env:ChocolateyInstall/helpers/chocolateyProfile.psm1" -DisableNameChecking:$true
+        . "$env:ChocolateyInstall/helpers/functions/Write-FunctionCallLogMessage.ps1"
+        . "$env:ChocolateyInstall/helpers/functions/Test-ProcessAdminRights.ps1"
+        . "$env:ChocolateyInstall/helpers/functions/Set-EnvironmentVariable.ps1"
+        . "$env:ChocolateyInstall/helpers/functions/Uninstall-ChocolateyEnvironmentVariable.ps1"
+        $uninstallEnvVarParam = @{
+            variableName = 'BoxstarterChocoExtension'
+            variableType = [System.EnvironmentVariableTarget]::Machine
+        }
+        Uninstall-ChocolateyEnvironmentVariable @uninstallEnvVarParam
+    }
+
     if(Test-Path "$(Get-BoxstarterTempDir)\BoxstarterReEnableUAC") {
         del "$(Get-BoxstarterTempDir)\BoxstarterReEnableUAC"
         Enable-UAC

Boxstarter.Bootstrapper/Setup-BoxstarterExtension.ps1

@@ -0,0 +1,25 @@
+
+function Setup-BoxstarterExtension {
+    $extensionBaseDir = "$env:ChocolateyInstall\extensions\boxstarter-choco"
+    New-Item -ItemType Directory -Path $extensionBaseDir -Force | Out-Null
+
+    @"
+if (`$env:BoxstarterChocoExtension) {
+Write-Host "Lets.GetBoxstarter() :-)"        
+Import-Module '$($boxstarter.BaseDir)\Boxstarter.Chocolatey\Boxstarter.Chocolatey.psd1' -DisableNameChecking -ArgumentList `$true
+Write-Host "Boxstarter superpowers initiated..."
+}
+"@ | Out-File "$extensionBaseDir/boxstarter-choco.psm1" -Encoding utf8
+
+    Import-Module "$env:ChocolateyInstall/helpers/chocolateyProfile.psm1" -DisableNameChecking:$true
+    . "$env:ChocolateyInstall/helpers/functions/Write-FunctionCallLogMessage.ps1"
+    . "$env:ChocolateyInstall/helpers/functions/Test-ProcessAdminRights.ps1"
+    . "$env:ChocolateyInstall/helpers/functions/Set-EnvironmentVariable.ps1"
+    . "$env:ChocolateyInstall/helpers/functions/Install-ChocolateyEnvironmentVariable.ps1"
+    $installEnvVarParam = @{
+        variableName  = 'BoxstarterChocoExtension'
+        variableValue = 'absolutely'
+        variableType  = [System.EnvironmentVariableTarget]::Machine
+    }
+    Install-ChocolateyEnvironmentVariable @installEnvVarParam
+}