Cosmos/source/Cosmos.Build.Builder/MainWindow.xaml.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Documents;
using System.Windows.Input;
using System.Windows.Media;
using System.Windows.Threading;
using Microsoft.VisualBasic;
using TaskScheduler;

namespace Cosmos.Build.Builder
{
  public partial class MainWindow : Window
  {
    protected int mTailLineCount = 5;
    protected int mTailCurrent = 0;
    protected List<TextBlock> mTailLines = new List<TextBlock>();
    protected string mCosmosDir;
    protected string mSetupPath;
    // Needs updating with each new release.
    protected int mReleaseNo = 106027;

    public MainWindow()
    {
      InitializeComponent();
      mApp = (App)Application.Current;

      for (int i = 0; i < mTailLineCount; i++)
      {
        var xTextBlock = new TextBlock();
        xTextBlock.FontSize = 16;
        mTailLines.Add(xTextBlock);
        spnlTail.Children.Add(xTextBlock);
      }
    }

    bool mPreventAutoClose = false;
    App mApp;
    TextBlock mSection;
    TextBlock mContent;
    StringBuilder mClipboard = new StringBuilder();
    DispatcherTimer mCloseTimer;

    const string InstallScheduledTaskName = "CosmosSetup";

    // Install the UAC bypass CosmosTask. This must be performed while the program is running under
    // administrator credentials with elevation.
    void InstallScheduledTask()
    {
      ITaskService xService = new TaskScheduler.TaskScheduler();
      xService.Connect();
      ITaskFolder xFolder = xService.GetFolder(@"\");
      IRegisteredTask xTask = TryGetInstallScheduledTask(xFolder);

      if (null != xTask)
      {
        // The first parameter MUST NOT be prefixed with the folder path.
        xFolder.DeleteTask(InstallScheduledTaskName, 0);
      }

      ITaskDefinition xTaskDef = xService.NewTask(0);
      xTaskDef.RegistrationInfo.Description = "Cosmos DevKit UAC Bypass";
      xTaskDef.RegistrationInfo.Author = "Cosmos Group";
      xTaskDef.Settings.Compatibility = _TASK_COMPATIBILITY.TASK_COMPATIBILITY_V2_1;
      xTaskDef.Settings.DisallowStartIfOnBatteries = false;
      xTaskDef.Principal.RunLevel = _TASK_RUNLEVEL.TASK_RUNLEVEL_HIGHEST;

      IActionCollection xActions = xTaskDef.Actions;
      IAction xAction = xActions.Create(_TASK_ACTION_TYPE.TASK_ACTION_EXEC);
      IExecAction xExecAction = xAction as IExecAction;
      xExecAction.Path = mSetupPath;
      xExecAction.Arguments = @"/SILENT";

      // 6 = task can be created or updated ["CreateOrUpdate" flag]
      // if Name id empty or null, System will create a task with name as GUID
      xTask = xFolder.RegisterTaskDefinition(InstallScheduledTaskName, xTaskDef, 6, null, null, _TASK_LOGON_TYPE.TASK_LOGON_NONE, null);
    }

    // Check for task path change. This is invoked in normal user mode (not administrator) and is intended
    // to fix issue #15528
    bool IsInstallScheduledTaskPathFixRequired(IRegisteredTask existingTask, string expectedPath)
    {
      // This is a defensive programming test. This should never happen unless someone modifies
      // the task creation code and forget to fix the task update part.
      if (1 != existingTask.Definition.Actions.Count) { return true; }
      IExecAction xExistingExecAction = null;
      IActionCollection xActions;

      try
      {
        xActions = existingTask.Definition.Actions;
        IEnumerator xActionsEnumerator = xActions.GetEnumerator();
        if (!xActionsEnumerator.MoveNext()) { return true; }
        xExistingExecAction = xActionsEnumerator.Current as IExecAction;
      }
      catch { return true; }

      if (null == xExistingExecAction) { return true; }
      if (0 != string.Compare(xExistingExecAction.Path, expectedPath, true)) { return true; }
      return false;
    }

    // http://yoursandmyideas.wordpress.com/2012/01/07/task-scheduler-in-c-net/
    bool ScheduledTaskIsInstalled()
    {
      ITaskService xService = new TaskScheduler.TaskScheduler();
      xService.Connect();

      ITaskFolder xFolder = xService.GetFolder(@"\");
      IRegisteredTask xExistingTask = TryGetInstallScheduledTask(xFolder);

      if (null == xExistingTask) { return false; }
      return (!IsInstallScheduledTaskPathFixRequired(xExistingTask, mSetupPath));
    }

    // Attempt to retrieve the CosmosSetup UAC bypass task. If the task can't be retrieved either
    // because it doesn't exist or because access is denied, return a null reference.
    IRegisteredTask TryGetInstallScheduledTask(ITaskFolder folder)
    {
      try { return folder.GetTask(InstallScheduledTaskName); }
      catch { return null; }
    }

    void InstallTaskAsAdmin()
    {
      // Restart with UAC and just install scheduled task
      using (var xProcess = new Process())
      {
        var xPSI = xProcess.StartInfo;
        xPSI.UseShellExecute = true;
        xPSI.FileName = Assembly.GetEntryAssembly().GetName().CodeBase.Replace("file:///", "");
        xPSI.Arguments = "-InstallTask";
        xPSI.Arguments += " -VS2017";

        xPSI.Verb = "runas";
        try
        {
          xProcess.Start();
        }
        catch (System.ComponentModel.Win32Exception)
        {
          // happens if user press "cancel" on UAC dialog
          Log_LogSection("Error");
          Log_LogLine("User pressed \"Cancel\" on UAC dialog for install task!");
          Log_LogError();
          return;
        }
        xProcess.WaitForExit();
      }
    }

    public bool Build()
    {
      if (App.IsUserKit)
      {
        mReleaseNo = Int32.Parse(DateTime.Now.ToString("yyyyMMdd"));
      }
      else
      {
        if (App.UseTask)
        {
          if (!ScheduledTaskIsInstalled())
          {
            InstallTaskAsAdmin();
          }
        }
      }
      if (mPreventAutoClose)
        return true;

      var xTask = new CosmosTask(mCosmosDir, mReleaseNo);
      xTask.Log.LogLine += new Installer.Log.LogLineHandler(Log_LogLine);
      xTask.Log.LogSection += new Installer.Log.LogSectionHandler(Log_LogSection);
      xTask.Log.LogError += new Installer.Log.LogErrorHandler(Log_LogError);

      var xThread = new System.Threading.Thread(delegate ()
      {
        xTask.Run();
        ThreadDone();
      });
      xThread.Start();

      return true;
    }

    void ThreadDone()
    {
      Dispatcher.Invoke(DispatcherPriority.Normal, (Action)delegate ()
      {
        if (App.StayOpen == false)
        {
          mCloseTimer = new DispatcherTimer();
          mCloseTimer.Interval = TimeSpan.FromSeconds(5);
          mCloseTimer.Tick += delegate
          {
            mCloseTimer.Stop();
            if (mPreventAutoClose)
            {
              if (WindowState == WindowState.Minimized)
              {
                WindowState = WindowState.Normal;
              }
            }
            else
            {
              Close();
            }
          };
          mCloseTimer.Start();
        }
      });
    }

    void ClearTail()
    {
      mTailCurrent = 0;
      foreach (var x in mTailLines)
      {
        x.Text = "";
      }
    }

    void Log_LogError()
    {
      Dispatcher.Invoke(DispatcherPriority.Normal, (Action)delegate ()
      {
        ClearTail();

        mSection.Foreground = Brushes.Red;
        mContent.Visibility = Visibility.Visible;
        mPreventAutoClose = true;
      });
    }

    void Log_LogLine(string aLine)
    {
      Dispatcher.Invoke(DispatcherPriority.Normal, (Action)delegate ()
      {
        WriteTail(aLine);

        mClipboard.AppendLine(aLine);

        mContent.Inlines.Add(aLine);
        mContent.Inlines.Add(new LineBreak());
      });
    }

    void Log_LogSection(string aLine)
    {
      Dispatcher.Invoke(DispatcherPriority.Normal, (Action)delegate ()
      {
        Title = aLine;

        ClearTail();

        mClipboard.AppendLine();
        mClipboard.AppendLine(new string('=', aLine.Length));
        mClipboard.AppendLine(aLine);
        mClipboard.AppendLine(new string('=', aLine.Length));
        mClipboard.AppendLine();

        mSection = new TextBlock();
        mSection.Text = aLine;
        mSection.Background = Brushes.LightGray;
        mSection.Foreground = Brushes.Green;
        mSection.FontSize = 18;
        mSection.FontWeight = FontWeights.Bold;
        mSection.MouseUp += new MouseButtonEventHandler(mSection_MouseUp);
        spnlLog.Children.Add(mSection);

        mContent = new TextBlock();
        mContent.Visibility = Visibility.Collapsed;
        spnlLog.Children.Add(mContent);
        mSection.Tag = mContent;
      });
    }

    void mSection_MouseUp(object sender, MouseButtonEventArgs e)
    {
      var xSection = (TextBlock)sender;
      var xContent = (TextBlock)xSection.Tag;
      xContent.Visibility = xContent.Visibility == Visibility.Visible ? Visibility.Collapsed : Visibility.Visible;
      mPreventAutoClose = true;
    }

    void ScrollTail()
    {
      for (int i = 0; i < mTailLineCount - 1; i++)
      {
        mTailLines[i].Text = mTailLines[i + 1].Text;
      }
    }

    void WriteTail(string aText)
    {
      if (mTailCurrent == mTailLineCount - 1)
      {
        ScrollTail();
      }
      mTailLines[mTailCurrent].Text = aText;
      if (mTailCurrent < mTailLineCount - 1)
      {
        mTailCurrent++;
      }
    }

    protected bool mLoaded = false;
    void Window_Loaded(object sender, RoutedEventArgs e)
    {
      if (!App.mArgs.Any())
      {
        MessageBox.Show("Builder not meant to be called directly. Use install.bat instead.", "Error", MessageBoxButton.OK, MessageBoxImage.Error);
        Close();
        return;
      }

      LoadPosition();
      mLoaded = true;
      
      string xAppPath = AppContext.BaseDirectory;
      mCosmosDir = Path.GetFullPath(xAppPath + @"..\..\..\..\");
      mSetupPath = Path.Combine(mCosmosDir, @"Setup\Output\" + CosmosTask.GetSetupName(mReleaseNo) + ".exe");
      if (App.InstallTask)
      {
        InstallScheduledTask();
        Close();
      }
      else if (!Build())
      {
        Close();
      }
    }

    void butnCopy_Click(object sender, RoutedEventArgs e)
    {
      mPreventAutoClose = true;
      Clipboard.SetText(mClipboard.ToString());
    }

    void LoadPosition()
    {
      Left = Properties.Settings.Default.Location.X;
      Top = Properties.Settings.Default.Location.Y;
      Width = Properties.Settings.Default.Size.Width;
      Height = Properties.Settings.Default.Size.Height;
    }

    protected void SavePosition()
    {
      Properties.Settings.Default.Location = new System.Drawing.Point((int)Left, (int)Top);
      Properties.Settings.Default.Size = new System.Drawing.Size((int)Width, (int)Height);
      Properties.Settings.Default.Save();
    }

    private void Window_SizeChanged(object sender, SizeChangedEventArgs e)
    {
      // User had non minimized window, or maximized it, or otherwise manually intervened.
      // Even if starting minimized, this event gets called with Normal before load.
      // This is why we have mLoaded.
      if (mLoaded && WindowState != System.Windows.WindowState.Minimized)
      {
        mPreventAutoClose = true;
      }
    }

    private void Window_Closing(object sender, System.ComponentModel.CancelEventArgs e)
    {
      SavePosition();
    }
  }
}