/Raven.Studio/Features/Input/AskUser.cs

using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using System.Windows;
using System.Windows.Controls;

namespace Raven.Studio.Features.Input
{
	public static class AskUser
	{
		public static Task<T> ShowAsync<T>(this T window) where T : ChildWindow
		{
			var tcs = new TaskCompletionSource<T>();

			window.Closed += (sender, args) =>
			{
				if (window.DialogResult == true)
					tcs.SetResult(window);
				else
					tcs.SetCanceled();
			};

			window.Show();

			return tcs.Task;
		}


		public static Task<string> QuestionAsync(string title, string question)
		{
			var dataContext = new InputModel
			{
				Title = title,
				Question = question
			};

			var inputWindow = new InputWindow
			{
				DataContext = dataContext
			};

			var tcs = new TaskCompletionSource<string>();

			inputWindow.Closed += (sender, args) =>
			{
				if (inputWindow.DialogResult == true)
					tcs.SetResult(dataContext.Answer);
				else
					tcs.SetCanceled();
			};

			inputWindow.Show();

			return tcs.Task;
		}

		public static Task<string> QuestionWithSuggestionAsync(string title, string question, Func<string, Task<IList<object>>> provideSuggestions)
		{
			var dataContext = new InputModelWithSuggestion(provideSuggestions)
			{
				Title = title,
				Question = question
			};

			var inputWindow = new InputWindowWithSuggestion
			{
				DataContext = dataContext
			};

			var tcs = new TaskCompletionSource<string>();

			inputWindow.Closed += (sender, args) =>
			{
				if (inputWindow.DialogResult == true)
					tcs.SetResult(dataContext.Answer);
				else
					tcs.SetCanceled();
			};

			inputWindow.Show();

			return tcs.Task;
		}

		public static Task<string> SelectItem(string title, string question, Func<Task<IList<string>>> provideList)
		{
			var dataContext = new SelectModel(provideList)
			{
				Title = title,
				Question = question
			};

			var inputWindow = new SelectWindow
			{
				DataContext = dataContext
			};

			var tcs = new TaskCompletionSource<string>();

			inputWindow.Closed += (sender, args) =>
			{
				if (inputWindow.DialogResult == true)
					tcs.SetResult(dataContext.Answer);
				else
					tcs.SetCanceled();
			};

			inputWindow.Show();

			return tcs.Task;
		}

		public static Task<bool> ConfirmationAsync(string title, string question)
		{
			var dataContext = new ConfirmModel
			{
				Title = title,
				Question = question
			};

			var inputWindow = new ConfirmWindow
			{
				DataContext = dataContext
			};

			var tcs = new TaskCompletionSource<bool>();

			inputWindow.Closed += (sender, args) =>
			{
				if (inputWindow.DialogResult != null)
					tcs.SetResult(inputWindow.DialogResult.Value);
				else
					tcs.SetCanceled();
			};

			inputWindow.Show();

			return tcs.Task;
		}

		public static void ConfirmationWithEvent(string title, string question, Action onOkay)
		{
			var dataContext = new ConfirmModel
			{
				Title = title,
				Question = question
			};

			var inputWindow = new ConfirmWindow
			{
				DataContext = dataContext
			};


			inputWindow.Closed += (sender, args) =>
			{
				if (inputWindow.DialogResult == true)
					onOkay();
			};

			inputWindow.Show();

		}

        public static Task<T> ConfirmationWithContinuation<T>(string title, string question, Func<Task<T>> onOkay, Func<Task<T>> onCancelled)
        {
            var dataContext = new ConfirmModel
            {
                Title = title,
                Question = question
            };

            var inputWindow = new ConfirmWindow
            {
                DataContext = dataContext
            };

            var tcs = new TaskCompletionSource<T>();

            inputWindow.Closed += async (sender, args) =>
            {
                try
                {
                    var result = inputWindow.DialogResult == true ? await onOkay() : await onCancelled();
                    tcs.SetResult(result);
                }
                catch (Exception ex)
                {
                    tcs.SetException(ex);
                }
            };

            inputWindow.Show();

            return tcs.Task;
        }

        public static bool Confirmation(string title, string question)
        {
            return MessageBox.Show(question, title, MessageBoxButton.OKCancel) == MessageBoxResult.OK;
        }
	}
}