// Copyright (c) 2014 AlphaSierraPapa for the SharpDevelop Team
// 
// Permission is hereby granted, free of charge, to any person obtaining a copy of this
// software and associated documentation files (the "Software"), to deal in the Software
// without restriction, including without limitation the rights to use, copy, modify, merge,
// publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons
// to whom the Software is furnished to do so, subject to the following conditions:
// 
// The above copyright notice and this permission notice shall be included in all copies or
// substantial portions of the Software.
// 
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
// INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
// PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE
// FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.

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.Scripting;

namespace ICSharpCode.PackageManagement.Scripting
{
	public class PowerShellHost : PSHost, IPowerShellHost
	{
		public static readonly string EnvironmentPathVariableName = "env:path";
		
		IScriptingConsole scriptingConsole;
		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;
		object dte;
		Version version;
		
		public PowerShellHost(
			IScriptingConsole scriptingConsole,
			Version version,
			object privateData,
			object dte)
		{
			this.scriptingConsole = scriptingConsole;
			this.version = version;
			this.privateData = new PSObject(privateData);
			this.dte = dte;
			userInterface = new PowerShellHostUserInterface(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) {
				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 options = ScopedItemOptions.AllScope | ScopedItemOptions.Constant;
			return new SessionStateVariableEntry("DTE", dte, "SharpDevelop DTE object", options);
		}
		
		public override Version Version {
			get { return 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 "Package Manager 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) {
				scriptingConsole.WriteLine(ex.Message, ScriptingStyle.Error);
			}
		}
		
		public void SetDefaultRunspace()
		{
			Runspace.DefaultRunspace = runspace;
		}
	}
}