XpandPosh

1.0.40

Cmdlets/src/XpandPosh.Cmdlets/CmdletExtensions.cs

using System;
using System.Collections.ObjectModel;
using System.Linq;
using System.Management.Automation;
 
namespace XpandPosh.CmdLets{
    static class CmdletExtensions{
        public static string GetCmdletName<T>() where T : Cmdlet{
            var cmdletAttribute = typeof(T).GetCustomAttributes(false).OfType<CmdletAttribute>().First();
            return $"{cmdletAttribute.VerbName}-{cmdletAttribute.NounName}";
        }
 
        public static T GetVariableValue<T>(this PSCmdlet cmdlet, string name){
            var psVariable = cmdlet.Invoke<PSVariable>($"Get-Variable|where{{$_.Name -eq '{name}'}}").FirstOrDefault();
            if (psVariable==null)
                throw new NullReferenceException(name);
            return (T)psVariable.Value;
        }
 
        public static ActionPreference ErrorAction(this PSCmdlet cmdlet){
            if (cmdlet.MyInvocation.BoundParameters.ContainsKey("ErrorAction")){
                return (ActionPreference) cmdlet.MyInvocation.BoundParameters["ErrorAction"];
            }
            return cmdlet.GetVariableValue<ActionPreference>("ErrorActionPreference");
        }
 
        public static void GetCallerPreference(this Cmdlet cmdlet){
            try{
                cmdlet.Invoke("Get-CallerPreference -Cmdlet $PSCmdlet -SessionState $ExecutionContext.SessionState");
            }
            catch{
                // ignored
            }
        }
 
        public static Collection<PSObject> Invoke(this Cmdlet cmdlet, string script){
            return cmdlet.Invoke<PSObject>(script);
        }
 
        public static Collection<T> Invoke<T>(this Cmdlet cmdlet,string script){
            using (var powerShell = PowerShell.Create(RunspaceMode.CurrentRunspace)){
                powerShell.Commands.AddScript(script);
                return powerShell.Invoke<T>();
            }
        }
    }
}