使用PowerShell從文件中提取函數體

Question

如何提取PowerShell函數定義的內容？ 假設代碼是這樣，

Function fun1($choice){ 
    switch($choice) 
    { 
     1{ 
     "within 1" 
     } 
     2{ 
     "within 2" 
     } 
     default{ 
     "within default" 
     } 

    } 

} 

fun1 1 

我只想要函數的定義，並沒有其他的文本內容。

Answer 1

使用PowerShell 3.0+ Language namespace AST解析器：

$code = Get-Content -literal 'R:\source.ps1' -raw 
$name = 'fun1' 

$body = [Management.Automation.Language.Parser]::ParseInput($code, [ref]$null, [ref]$null). 
    Find([Func[Management.Automation.Language.Ast,bool]]{ 
     param ($ast) 
     $ast.name -eq $name -and $ast.body 
    }, $true) | ForEach { 
     $_.body.extent.text 
    } 

輸出單一的多行字符串在$體：

{ 
    switch($choice) 
    { 
     1{ 
     "within 1" 
     } 
     2{ 
     "within 2" 
     } 
     default{ 
     "within default" 
     } 

    } 

} 

要提取的第一個函數定義體，無論名稱：

$body = [Management.Automation.Language.Parser]::ParseInput($code, [ref]$null, [ref]$null). 
    Find([Func[Management.Automation.Language.Ast,bool]]{$args[0].body}, $true) | ForEach { 
     $_.body.extent.text 
    } 

提取從開始的整個函數定義關鍵字，使用$_.extent.text：

$fun = [Management.Automation.Language.Parser]::ParseInput($code, [ref]$null, [ref]$null). 
    Find([Func[Management.Automation.Language.Ast,bool]]{$args[0].body}, $true) | ForEach { 
     $_.extent.text 
    }