UpTool2/Installer/InstallerForm.cs

using System;
using System.Drawing;
using System.Globalization;
using System.IO;
using System.IO.Compression;
using System.Linq;
using System.Net;
using System.Security.Cryptography;
using System.Threading;
using System.Windows.Forms;
using System.Xml.Linq;
using Microsoft.Win32;
using UpToolLib;
using UpToolLib.v1.Tool;
using UpToolLib.v2;

namespace Installer
{
    public partial class InstallerForm : Form
    {
        private const string AppName = "UpTool2";
        private readonly RegistryKey _rkApp;
        private string _log = "";
        private UpToolLibMain _lib;

        public InstallerForm()
        {
            _lib = new UpToolLibMain(new UtLibFunctionsGui(Log));
            InitializeComponent();
            Step(0, "Initialized");
            _log = _log.TrimStart(Environment.NewLine.ToCharArray());
            _rkApp = Registry.CurrentUser.OpenSubKey(@"SOFTWARE\Microsoft\Windows\CurrentVersion\Run", true);
            pathBox.Checked = !File.Exists(_lib.V1.PathTool.InfoXml) ||
                              Path.Content.Contains(Path.GetName(_lib.V1.PathTool.GetRelative("Install")));
            startupBox.Checked = pathBox.Checked && _rkApp.GetValue(AppName) != null;
            updateAppsBox.Checked = pathBox.Checked && startupBox.Checked &&
                                    (string) _rkApp.GetValue(AppName) == "uptool dist-upgrade";
        }

        private void install_Click(object sender, EventArgs e)
        {
            log.Visible = false;
            try
            {
                progress.Visible = true;
                WebClient client = new();
                Step(1, "Downloading metadata");
                XElement meta = XDocument
                    .Load("https://gitlab.com/JFronny/UpTool2/-/jobs/artifacts/master/raw/meta.xml?job=uptool")
                    .Element("meta");
                Step(2, "Downloading binary");
                byte[] dl = client.DownloadData(meta.Element("File").Value);
                Step(3, "Verifying integrity");
                using (SHA256CryptoServiceProvider sha256 = new())
                {
                    string pkgHash = BitConverter.ToString(sha256.ComputeHash(dl)).Replace("-", string.Empty).ToUpper();
                    if (pkgHash != meta.Element("Hash").Value.ToUpper())
                        throw new Exception(
                            $@"The hash is not equal to the one stored in the repo:
Package: {pkgHash}
Online: {meta.Element("Hash").Value.ToUpper()}");
                }
                Step(4, "Extracting");
                if (Directory.Exists(_lib.V1.PathTool.GetRelative("Install")))
                    Directory.Delete(_lib.V1.PathTool.GetRelative("Install"), true);
                Directory.CreateDirectory(_lib.V1.PathTool.GetRelative("Install"));
                using (MemoryStream ms = new(dl))
                {
                    using ZipArchive ar = new(ms);
                    ar.ExtractToDirectory(_lib.V1.PathTool.GetRelative("Install"), true);
                }
                Step(5, "Creating shortcut");
                Shortcut.Make(_lib.V1.PathTool.GetRelative("Install", "UpTool2.exe"),
                    System.IO.Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.Programs),
                        "UpTool2.lnk"));
                Step(6, "Preparing Repos");
                _lib.V1.XmlTool.FixXml();
                _lib.V2.RepoManagement.FetchRepos();
                if (pathBox.Checked)
                {
                    Step(7, startupBox.Checked ? "Creating PATH & Autostart entry" : "Creating PATH entry");
                    if (!Path.Content.Contains(Path.GetName(_lib.V1.PathTool.GetRelative("Install"))))
                        Path.Append(_lib.V1.PathTool.GetRelative("Install"));
                    if (startupBox.Checked)
                        _rkApp.SetValue(AppName, updateAppsBox.Checked ? "uptool dist-upgrade" : "uptool upgrade-self");
                    else if (_rkApp.GetValue(AppName) != null)
                        _rkApp.DeleteValue(AppName, false);
                }
                Step(8, "Done!");
            }
            catch (Exception ex)
            {
                Step(progress.Value, $"Failed!{Environment.NewLine}{ex}");
                BackColor = Color.Red;
                processLabel.Text = "Failed";
                new Thread(() =>
                {
                    Thread.Sleep(1000);
                    Invoke(new Action(() =>
                    {
                        BackColor = SystemColors.Control;
                        progress.Visible = false;
                    }));
                }).Start();
            }
            finally
            {
                log.Visible = true;
            }
        }

        private void Step(int p, string text)
        {
            progress.Value = p;
            processLabel.Text = text;
            Log(text);
        }

        private void Log(string text) => _log +=
            $"{Environment.NewLine}[{DateTime.Now.ToString(CultureInfo.InvariantCulture).Split(' ')[1]}] {text}";

        private void log_Click(object sender, EventArgs e) => new Thread(() => MessageBox.Show(_log)).Start();

        private void pathBox_CheckedChanged(object sender, EventArgs e)
        {
            startupBox.Enabled = pathBox.Checked;
            if (!pathBox.Checked)
            {
                startupBox.Checked = false;
                updateAppsBox.Checked = false;
                updateAppsBox.Enabled = false;
            }
        }

        private void startupBox_CheckedChanged(object sender, EventArgs e)
        {
            updateAppsBox.Enabled = startupBox.Checked;
            if (!startupBox.Checked)
                updateAppsBox.Checked = false;
        }
    }
}