SharpDevelop/src/AddIns/Misc/PackageManagement/Project/Src/Scripting/PowerShellHost.cs

// Copyright (c) AlphaSierraPapa for the SharpDevelop Team (for details please see \doc\copyright.txt)
// This code is distributed under the GNU LGPL (for details please see \doc\license.txt)

using System;
using System.Collections;
using System.Collections.Generic;
using System.Globalization;
using System.Management.Automation;
using System.Management.Automation.Host;
using System.Management.Automation.Runspaces;
using System.Threading;

using ICSharpCode.PackageManagement.EnvDTE;
using ICSharpCode.Scripting;

namespace ICSharpCode.PackageManagement.Scripting
{
	public class PowerShellHost : PSHost, IPowerShellHost
	{
		public static readonly string EnvironmentPathVariableName = "env:path";
		
		IPackageManagementConsoleHost consoleHost;
		CultureInfo currentUICulture = Thread.CurrentThread.CurrentUICulture;
		CultureInfo currentCulture = Thread.CurrentThread.CurrentCulture;
		Guid instanceId = Guid.NewGuid();
		Runspace runspace;
		PowerShellHostUserInterface userInterface;
		List<string> modulesToImport = new List<string>();
		PSObject privateData;
		
		public PowerShellHost(
			IPackageManagementConsoleHost consoleHost,
			object privateData)
		{
			this.consoleHost = consoleHost;
			this.privateData = new PSObject(privateData);
			userInterface = new PowerShellHostUserInterface(consoleHost.ScriptingConsole);
		}
		
		public override PSObject PrivateData {
			get { return privateData; }
		}
		
		public IList<string> ModulesToImport {
			get { return modulesToImport; }
		}
		
		public void SetRemoteSignedExecutionPolicy()
		{
			ExecuteCommand("Set-ExecutionPolicy RemoteSigned -Scope 0 -Force");	
		}
		
		public void UpdateFormatting(IEnumerable<string> formattingFiles)
		{
			foreach (string file in formattingFiles) {
				string command = String.Format("Update-FormatData '{0}'", file);
				ExecuteCommand(command);
			}
		}
		
		public void ExecuteCommand(string command)
		{
			try {
				CreateRunspace();
				
				using (Pipeline pipeline = CreatePipeline(command)) {
					pipeline.Invoke();
				}
				
			} catch (Exception ex) {
				consoleHost.ScriptingConsole.WriteLine(ex.Message, ScriptingStyle.Error);
			}
		}
		
		Pipeline CreatePipeline(string command)
		{
			Pipeline pipeline = runspace.CreatePipeline();
			pipeline.Commands.AddScript(command);
			pipeline.Commands.Add("out-default");
			pipeline.Commands[0].MergeMyResults(PipelineResultTypes.Error, PipelineResultTypes.Output);
			return pipeline;
		}
		
		void CreateRunspace()
		{
			if (runspace == null) {
				InitialSessionState initialSessionState = CreateInitialSessionState();
				runspace = RunspaceFactory.CreateRunspace(this, initialSessionState);
				runspace.Open();
			}
		}
		
		InitialSessionState CreateInitialSessionState()
		{
			var initialSessionState = InitialSessionState.CreateDefault();
			initialSessionState.ImportPSModule(modulesToImport.ToArray());
			SessionStateVariableEntry variable = CreateDTESessionVariable();
			initialSessionState.Variables.Add(variable);
			return initialSessionState;
		}
		
		SessionStateVariableEntry CreateDTESessionVariable()
		{
			var dte = new DTE();
			var options = ScopedItemOptions.AllScope | ScopedItemOptions.Constant;
			return new SessionStateVariableEntry("DTE", dte, "SharpDevelop DTE object", options);
		}
		
		public override Version Version {
			get { return NuGetVersion.Version; }
		}
		
		public override PSHostUserInterface UI {
			get { return userInterface; }
		}
		
		public override void SetShouldExit(int exitCode)
		{
		}
		
		public override void NotifyEndApplication()
		{
		}
		
		public override void NotifyBeginApplication()
		{
		}
		
		public override string Name {
			get { return "PowerShell Host"; }
		}
		
		public override Guid InstanceId {
			get { return instanceId; }
		}
		
		public override void ExitNestedPrompt()
		{
			throw new NotImplementedException();
		}
		
		public override void EnterNestedPrompt()
		{
			throw new NotImplementedException();
		}
		
		public override CultureInfo CurrentUICulture {
			get { return currentUICulture; }
		}
		
		public override CultureInfo CurrentCulture {
			get { return currentCulture; }
		}
		
		public void RunScript(string fileName, IEnumerable<object> input)
		{
			try {
				CreateRunspace();
				
				string command = 
					"$__args = @(); " +
					"$input | ForEach-Object {$__args += $_}; " +
					"& '" + fileName + "' $__args[0] $__args[1] $__args[2] $__args[3]" +
					"Remove-Variable __args -Scope 0";
				using (Pipeline pipeline = CreatePipeline(command)) {
					pipeline.Invoke(input);
				}
				
			} catch (Exception ex) {
				consoleHost.ScriptingConsole.WriteLine(ex.Message, ScriptingStyle.Error);
			}
		}
		
		public void SetEnvironmentPath(string path)
		{
			Environment.SetEnvironmentVariable(EnvironmentPathVariableName, path);
		}
		
		public string GetEnvironmentPath()
		{
			return Environment.GetEnvironmentVariable(EnvironmentPathVariableName);
		}
		
		public void AddVariable(string name, object value)
		{
			runspace.SessionStateProxy.SetVariable(name, value);
		}
		
		public void RemoveVariable(string name)
		{
			runspace.SessionStateProxy.PSVariable.Remove(name);
		}
		
		public void InvokeScript(string script)
		{
			consoleHost.ScriptingConsole.SendLine(script);
		}
	}
}