BetterDiscordApp-v2/WindowsInstaller/BetterDiscordWI/panels/Panel2.cs

using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.IO.Compression;
using System.Net;
using System.Threading;
using System.Windows.Forms;
using asardotnet;

namespace BetterDiscordWI.panels
{
    public partial class Panel2 : UserControl, IPanel
    {

        private String _dataPath, _tempPath;
        private Utils _utils;

        public Panel2()
        {
            InitializeComponent();
        }

        public void SetVisible()
        {
            GetParent().btnBack.Enabled = false;
            GetParent().btnNext.Enabled = false;
            GetParent().btnBack.Visible = false;
            GetParent().btnNext.Visible = false;
            GetParent().btnCancel.Enabled = false;

            _utils = new Utils();

            AppendLog("Killing Discord");
            foreach (var process in Process.GetProcessesByName("Discord"))
            {
                process.Kill();
            }
            
            CreateDirectories();
        }

        private void CreateDirectories()
        {
            Thread t = new Thread(() =>
            {
                _dataPath = Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData) + "\\BetterDiscord";
                _tempPath = _dataPath + "\\temp";


                if (Directory.Exists(_tempPath))
                {
                    AppendLog("Deleting temp path");
                    Directory.Delete(_tempPath, true);
                }

                while (Directory.Exists(_tempPath))
                {
                    Debug.Print("Waiting for dirdel");
                    Thread.Sleep(100);
                }

                Directory.CreateDirectory(_tempPath);

                DownloadResource("BetterDiscord.zip", "https://github.com/Jiiks/BetterDiscordApp/archive/stable.zip");

                while (!File.Exists(_tempPath + "\\BetterDiscord.zip"))
                {
                    Debug.Print("Waiting for download");
                    Thread.Sleep(100);
                }

                AppendLog("Extracting BetterDiscord");

                ZipArchive zar =
                    ZipFile.OpenRead(Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData) +
                                     "\\BetterDiscord\\temp\\BetterDiscord.zip");
                zar.ExtractToDirectory(Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData) +
                                       "\\BetterDiscord\\temp\\");

                DeleteDirs();
            });
            t.Start();

        }


        private void DeleteDirs()
        {
            Thread t = new Thread(() =>
            {
                String dir = GetParent().DiscordPath + "\\resources\\app";

                if (Directory.Exists(dir))
                {
                    AppendLog("Deleting " + dir);
                    Directory.Delete(dir, true);
                }

                while (Directory.Exists(dir))
                {
                    Debug.Print("Waiting for direl");
                    Thread.Sleep(100);
                }

                dir = GetParent().DiscordPath + "\\resources\\node_modules\\BetterDiscord";


                if (Directory.Exists(dir))
                {
                    AppendLog("Deleting " + dir);
                    Directory.Delete(dir, true);
                }

                while (Directory.Exists(dir))
                {
                    Debug.Print("Waiting for direl");
                    Thread.Sleep(100);
                }

                AppendLog("Moving BetterDiscord to resources\\node_modules\\");

                Directory.Move(Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData) + "\\BetterDiscord\\temp\\BetterDiscordApp-stable", GetParent().DiscordPath + "\\resources\\node_modules\\BetterDiscord");

                AppendLog("Extracting app.asar");

                AsarArchive archive = new AsarArchive(GetParent().DiscordPath + "\\resources\\app.asar");

                AsarExtractor extractor = new AsarExtractor();
                extractor.ExtractAll(archive, GetParent().DiscordPath + "\\resources\\app\\");

                Splice();
            });
            
            
            t.Start();
        }


        private void DownloadResource(String resource, String url)
        {
            AppendLog("Downloading Resource: " + resource);

            WebClient webClient = new WebClient();
            webClient.Headers["User-Agent"] = "Mozilla/5.0";
            
            webClient.DownloadFile(new Uri(url), Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData) + "\\BetterDiscord\\temp\\" + resource);
        }

        private void Splice()
        {
            String indexloc = GetParent().DiscordPath + "\\resources\\app\\app\\index.js";

            Thread t = new Thread(() =>
            {

                List<String> lines = new List<string>();

                AppendLog("Spicing index");

                using (FileStream fs = new FileStream(indexloc, FileMode.Open))
                {
                    using (StreamReader reader = new StreamReader(fs))
                    {
                        String line = "";
                        while((line = reader.ReadLine()) != null)
                        {
                            if (line.Contains("var _overlay2"))
                            {
                                AppendLog("Splicing require");
                                lines.Add(line);
                                lines.Add("var _betterDiscord = require('betterdiscord');");
                            }else if (line.Contains("mainWindow = new _BrowserWindow2"))
                            {
                                AppendLog("Splicing function call");
                                lines.Add(line);
                                lines.Add("betterDiscord(mainWindow);");
                                AppendLog("Splicing plugin&theme loaders");

                                lines.Add("    mainWindow.webContents.on('dom-ready', function() {");
                                lines.Add("        var fs = require('fs');");
                                lines.Add("        fs.readdir('G:/BetterDiscord/plugins/',function(err,files) {");
                                lines.Add("            if (err) { console.log(err); return; } ");
                                lines.Add("            mainWindow.webContents.executeJavaScript('var bdplugins = {};');");
                                lines.Add("            files.forEach(function(fileName){");
                                lines.Add("                var plugin = fs.readFileSync('G:/BetterDiscord/plugins/' + fileName,'utf8');");
                                lines.Add("                var meta = plugin.split('\\n')[0];");
                                lines.Add("                if(meta.indexOf('META') < 0) {");
                                lines.Add("                    console.log('BetterDiscord: ERROR[Plugin META not found in file: '+fileName+']');");
                                lines.Add("                    return;");
                                lines.Add("                }");
                                lines.Add("                var pluginVar = meta.substring(meta.lastIndexOf('//META')+6, meta.lastIndexOf('*//'));");
                                lines.Add("                var parse = JSON.parse(pluginVar);");
                                lines.Add("                var pluginName = parse['name'];");
                                lines.Add("                console.log('BetterDiscord: Loading Plugin: ' + pluginName);");
                                lines.Add("                mainWindow.webContents.executeJavaScript(plugin);");
                                lines.Add("                mainWindow.webContents.executeJavaScript('(function() { var plugin = new '+pluginName+'(); bdplugins[plugin.getName()] = { \"plugin\": plugin, \"enabled\": false } })();')");
                                lines.Add("            });");
                                lines.Add("        });");
                                lines.Add("\n");
                                lines.Add("        fs.readdir('G:/BetterDiscord/themes/', function(err, files) {");
                                lines.Add("            if (err) { console.log(err); return; }");
                                lines.Add("            mainWindow.webContents.executeJavaScript('var bdthemes = {};');");
                                lines.Add("            files.forEach(function(fileName) {");
                                lines.Add("                var theme = fs.readFileSync('G:/BetterDiscord/themes/' + fileName, 'utf8');");
                                lines.Add("                var split = theme.split('\\n');");
                                lines.Add("                var meta = split[0];");
                                lines.Add("                if(meta.indexOf('META') < 0) {");
                                lines.Add("                    console.log('BetterDiscord: ERROR[Theme META not found in file: '+fileName+']');");
                                lines.Add("                    return;");
                                lines.Add("                }");
                                lines.Add("                var themeVar = meta.substring(meta.lastIndexOf('//META')+6, meta.lastIndexOf('*//'));");
                                lines.Add("                var parse = JSON.parse(themeVar);");
                                lines.Add("                var themeName = parse['name'];");
                                lines.Add("                console.log('BetterDiscord: Loading Theme: ' + themeName);");
                                lines.Add("                split.splice(0, 1);");
                                lines.Add("                theme = split.join('\\n');");
                                lines.Add("                theme = theme.replace(/(\\r\\n|\\n|\\r)/gm,'');");
                                lines.Add("                theme = theme.replace(/\\s/g, '');");
                                lines.Add("                mainWindow.webContents.executeJavaScript('var theme' + themeName + ' = \"' + escape(theme) + '\";');");
                                lines.Add("                mainWindow.webContents.executeJavaScript('(function() { bdthemes[theme'+themeName+'] = false })();');");
                                lines.Add("            });");
                                lines.Add("        });");
                                lines.Add("    });\n");

                            }
                            else if (line.Contains("main();"))
                            {
                                AppendLog("Splicing function");
                                lines.Add("function betterDiscord(mw) { _betterDiscord = new _betterDiscord.BetterDiscord(mw); _betterDiscord.init(); }");
                                lines.Add(line);
                            }
                            else
                            {
                                lines.Add(line);
                            }
                        }
                    }
                }

                AppendLog("Writing index");

                File.WriteAllLines(indexloc, lines.ToArray());

                Finalize();
            });
            t.Start();
        }

        private void Finalize()
        {
            AppendLog("Finished installing BetterDiscord");

            Invoke((MethodInvoker) delegate
            {
                GetParent().finished = true;
                GetParent().btnCancel.Text = "OK";
                GetParent().btnCancel.Enabled = true;
            });
        }

        public FormMain GetParent()
        {
            return (FormMain)ParentForm;
        }

        public void BtnNext()
        {
            throw new NotImplementedException();
        }

        public void BtnPrev()
        {
            throw new NotImplementedException();
        }

        private void AppendLog(String message)
        {
            Invoke((MethodInvoker) delegate
            {
                rtLog.AppendText(message + "\n");
                rtLog.SelectionStart = rtLog.Text.Length;
                rtLog.ScrollToCaret();
            });
            
        }
    }
}