Autoload Module 1.2 by Joel Bennett 4 years ago (modification of post by Joel Bennett view diff)
View followups from Travis | diff | embed code: <script type="text/javascript" src="/img/spacer.gif"> download | new post
Autoload function like the Korn shell — can inject functions to modules.
In order to make this easier to use, I’ve added the ability to run AutoLoad against script files (that don’t have a function in them). It will ultimately create a function out of the script.
The main purpose of this module is to allow defining functions in a script file without needing to dot-source the script in order to use them. You simply autoload that function, and there’s little overhead until the function is called (at which time it’s parsed and loaded, once). This way, autoload helps you avoid loading functions into memory until you need them — but allows you to write code as though the functions were already loaded.
Note: The Functions/Scripts are only actually parsed the first time you call the function (or when you call Get-AutoloadHelp on the function). You should be aware of that when making changes to the script.
- #Requires -Version 2.0
- ## Automatically load functions from scripts on-demand, instead of having to dot-source them ahead of time, or reparse them from the script every time.
- ## Provides significant memory benefits over pre-loading all your functions, and significant performance benefits over using plain scripts. Can also *inject* functions into Modules so they inherit the module scope instead of the current local scope.
- ## Please see the use example in the script below
- ## Version History
- ## v 1.2 - 2011.05.02
- ## - Exposed the LoadNow alias and the Resolve-Autoloaded function
- ## v 1.1 - 2011.02.09
- ## Added support for autoloading scripts (files that don't have a "function" in them)
- ## v 1.0 - 2010.10.20
- ## Officially out of beta -- this is working for me without problems on a daily basis.
- ## Renamed functions to respect the Verb-Noun expectations, and added Export-ModuleMember
- ## beta 8 - 2010.09.20
- ## Finally fixed the problem with aliases that point at Invoke-Autoloaded functions!
- ## beta 7 - 2010.06.03
- ## Added some help, and a function to force loading "now"
- ## Added a function to load AND show the help...
- ## beta 6 - 2010.05.18
- ## Fixed a bug in output when multiple outputs happen in the END block
- ## beta 5 - 2010.05.10
- ## Fixed non-pipeline use using $MyInvocation.ExpectingInput
- ## beta 4 - 2010.05.10
- ## I made a few tweaks and bug fixes while testing it's use with PowerBoots.
- ## beta 3 - 2010.05.10
- ## fix for signed scripts (strip signature)
- ## beta 2 - 2010.05.09
- ## implement module support
- ## beta 1 - 2010.04.14
- ## Initial Release
- ## To use:
- ## 1) Create a function. To be 100% compatible, it should specify pipeline arguments
- ## For example:
- <#
- function Skip-Object {
- param(
- [int]$First = 0, [int]$Last = 0, [int]$Every = 0, [int]$UpTo = 0,
- [Parameter(Mandatory=$true,ValueFromPipeline=$true)]
- $InputObject
- )
- begin {
- if($Last) {
- $Queue = new-object System.Collections.Queue $Last
- }
- $e = $every; $UpTo++; $u = 0
- }
- process {
- $InputObject | where { --$First -lt 0 } |
- foreach {
- if($Last) {
- $Queue.EnQueue($_)
- if($Queue.Count -gt $Last) { $Queue.DeQueue() }
- } else { $_ }
- } |
- foreach {
- if(!$UpTo) { $_ } elseif( --$u -le 0) { $_; $U = $UpTo }
- } |
- foreach {
- if($every -and (--$e -le 0)) { $e = $every } else { $_ }
- }
- }
- }
- #>
- ## 2) Put the function into a script (for our example: C:\Users\${Env:UserName}\Documents\WindowsPowerShell\Scripts\SkipObject.ps1 )
- ## 3) Import the Autoload Module
- ## 5) Run this command (or add it to your profile):
- <#
- New-Autoload C:\Users\${Env:UserName}\Documents\WindowsPowerShell\Scripts\SkipObject.ps1 Skip-Object
- #>
- ## This tells us that you want to have that function loaded for you out of the script file if you ever try to use it.
- ## Now, you can just use the function:
- ## 1..10 | Skip-Object -first 2 -upto 2
- function Invoke-Autoloaded {
- #.Synopsis
- # This function was autoloaded, but it has not been parsed yet.
- # Use Get-AutoloadHelp to force parsing and get the correct help (or just invoke the function once).
- #.Description
- # You are seeing this help because the command you typed was imported via the New-Autoload command from the Autoload module. The script file containing the function has not been loaded nor parsed yet. In order to see the correct help for your function we will need to parse the full script file, to force that at this time you may use the Get-AutoloadHelp function.
- #
- # For example, if your command was Get-PerformanceHistory, you can force loading the help for it by running the command: Get-AutoloadHelp Get-PerformanceHistory
- [CmdletBinding()]Param()
- DYNAMICPARAM {
- $CommandName = $MyInvocation.InvocationName
- return Resolve-Autoloaded $CommandName
- }#DynamicParam
- begin {
- Write-Verbose "Command: $CommandName"
- if(!$Script:AutoloadHash[$CommandName]) {
- do {
- $Alias = $CommandName
- $CommandName = Get-Alias $CommandName -ErrorAction SilentlyContinue | Select -Expand Definition
- Write-Verbose "Invoke-Autoloaded Begin: $Alias -> $CommandName"
- } while(!$Script:AutoloadHash[$CommandName] -and (Get-Alias $CommandName -ErrorAction SilentlyContinue))
- } else {
- Write-Verbose "CommandHash: $($Script:AutoloadHash[$CommandName])"
- }
- if(!$Script:AutoloadHash[$CommandName]) { throw "Unable to determine command!" }
- $ScriptName, $ModuleName, $FunctionName = $Script:AutoloadHash[$CommandName]
- Write-Verbose "Invoke-Autoloaded Begin: $Alias -> $CommandName -> $FunctionName"
- #Write-Host "Parameters: $($PSBoundParameters | ft | out-string)" -Fore Magenta
- $global:command = $ExecutionContext.InvokeCommand.GetCommand( $FunctionName, [System.Management.Automation.CommandTypes]::Function )
- Write-Verbose "Autoloaded Command: $($Command|Out-String)"
- $scriptCmd = {& $command @PSBoundParameters | Write-Output }
- $steppablePipeline = $scriptCmd.GetSteppablePipeline($myInvocation.CommandOrigin)
- $steppablePipeline.Begin($myInvocation.ExpectingInput)
- }
- process
- {
- Write-Verbose "Invoke-Autoloaded Process: $CommandName ($_)"
- try {
- if($_) {
- $steppablePipeline.Process($_)
- } else {
- $steppablePipeline.Process()
- }
- } catch {
- throw
- }
- }
- end
- {
- try {
- $steppablePipeline.End()
- } catch {
- throw
- }
- Write-Verbose "Invoke-Autoloaded End: $CommandName"
- }
- }#Invoke-Autoloaded
- function Get-AutoloadHelp {
- [CmdletBinding()]
- Param([Parameter(Mandatory=$true)][String]$CommandName)
- $null = Resolve-Autoloaded $CommandName
- Get-Help $CommandName
- }
- function Resolve-Autoloaded {
- [CmdletBinding()]
- param(
- [Parameter(Mandatory=$true,ValueFromPipelineByPropertyName=$true)]
- [Alias("Name")]
- [String]$CommandName
- )
- $OriginalCommandName = "($CommandName)"
- Write-Verbose "Command: $CommandName"
- if(!$Script:AutoloadHash[$CommandName]) {
- do {
- $Alias = $CommandName
- $CommandName = Get-Alias $CommandName -ErrorAction SilentlyContinue | Select -Expand Definition
- $OriginalCommandName += "($CommandName)"
- Write-Verbose "Resolve-Autoloaded Begin: $Alias -> $CommandName"
- } while(!$Script:AutoloadHash[$CommandName] -and (Get-Alias $CommandName -ErrorAction SilentlyContinue))
- } else {
- Write-Verbose "CommandHash: $($Script:AutoloadHash[$CommandName])"
- }
- if(!$Script:AutoloadHash[$CommandName]) { throw "Unable to determine command $OriginalCommandName!" }
- Write-Verbose "Resolve-Autoloaded DynamicParam: $CommandName from $($Script:AutoloadHash[$CommandName])"
- $ScriptName, $ModuleName, $FunctionName = $Script:AutoloadHash[$CommandName]
- Write-Verbose "Autoloading:`nScriptName: $ScriptName `nModuleName: $ModuleName `nFunctionName: $FunctionName"
- if(!$ScriptName){ $ScriptName = $CommandName }
- if(!$FunctionName){ $FunctionName = $CommandName }
- if($ModuleName) {
- $Module = Get-Module $ModuleName
- } else { $Module = $null }
- ## Determine the command name based on the alias used to invoke us
- ## Store the parameter set for use in the function later...
- $paramDictionary = new-object System.Management.Automation.RuntimeDefinedParameterDictionary
- #$externalScript = $ExecutionContext.InvokeCommand.GetCommand( $CommandName, [System.Management.Automation.CommandTypes]::ExternalScript )
- $externalScript = Get-Command $ScriptName -Type ExternalScript | Select -First 1
- Write-Verbose "Processing Script: $($externalScript |Out-String)"
- $parserrors = $null
- $prev = $null
- $script = $externalScript.ScriptContents
- [System.Management.Automation.PSToken[]]$tokens = [PSParser]::Tokenize( $script, [ref]$parserrors )
- [Array]::Reverse($tokens)
- $function = $false
- ForEach($token in $tokens) {
- if($prev -and $token.Content -eq "# SIG # Begin signature block") {
- $script = $script.SubString(0, $token.Start )
- }
- if($prev -and $token.Type -eq "Keyword" -and $token.Content -ieq "function" -and $prev.Content -ieq $FunctionName ) {
- $script = $script.Insert( $prev.Start, "global:" )
- $function = $true
- Write-Verbose "Globalized: $($script[(($prev.Start+7)..($prev.Start + 7 +$prev.Content.Length))] -join '')"
- }
- $prev = $token
- }
- if(!$function) {
- $script = "function global:$Functionname { $script }"
- }
- if($Module) {
- $script = Invoke-Expression "{ $Script }"
- Write-Verbose "Importing Function into $($Module) module."
- &$Module $Script | Out-Null
- $command = Get-Command $FunctionName -Type Function
- Write-Verbose "Loaded Module Function: $($command | ft CommandType, Name, ModuleName, Visibility|Out-String)"
- } else {
- Write-Verbose "Importing Function without module."
- Invoke-Expression $script | out-null
- $command = Get-Command $FunctionName -Type Function
- Write-Verbose "Loaded Local Function: $($command | ft CommandType, Name, ModuleName, Visibility|Out-String)"
- }
- if(!$command) {
- throw "Something went wrong autoloading the $($FunctionName) function. Function definition doesn't exist in script: $($externalScript.Path)"
- }
- if($CommandName -eq $FunctionName) {
- Remove-Item Alias::$($CommandName)
- Write-Verbose "Defined the function $($FunctionName) and removed the alias $($CommandName)"
- } else {
- Set-Alias $CommandName $FunctionName -Scope Global
- Write-Verbose "Defined the function $($FunctionName) and redefined the alias $($CommandName)"
- }
- foreach( $pkv in $command.Parameters.GetEnumerator() ){
- $parameter = $pkv.Value
- if( $parameter.Aliases -match "vb|db|ea|wa|ev|wv|ov|ob" ) { continue }
- $param = new-object System.Management.Automation.RuntimeDefinedParameter( $parameter.Name, $parameter.ParameterType, $parameter.Attributes)
- $paramdictionary.Add($pkv.Key, $param)
- }
- return $paramdictionary
- }
- function New-Autoload {
- [CmdletBinding()]
- param(
- [Parameter(Position