示例#1
0
 private void endInstallation(string reason, bool failed)
 {
     InstallProgress.Value = 100;
     BetaRadio.Hide();
     InstallButton.Hide();
     UninstallButton.Hide();
     UpdateButton.Hide();
     ReinjectButton.Hide();
     StatusText.Hide();
     StatusLabel.Show();
     StatusLabel.Text      = operation == "UPDATE" ? "Update " + (failed ? "failed" : "complete") : (operation == "UNINSTALL" ? "Unin" : "In") + "stallation " + (failed ? " failed." : "completed!");
     StatusLabel.ForeColor = failed ? Color.Red : Color.Lime;
     StatusLabel2.Show();
     StatusLabel2.Text = reason;
     StatusCloseButton.Show();
     if (platform != "Linux")
     {
         OpenFolderButton.Show();
     }
 }
示例#2
0
        async private void continueUpdate(Process proc)
        {
            string path = proc.MainModule.FileName;

            operation = "UPDATE";
            BetaRadio.Hide();
            InstallButton.Hide();
            UninstallButton.Hide();
            UpdateButton.Hide();
            ReinjectButton.Hide();
            StatusText.Show();
            InstallProgress.Show();
            InstallProgress.Value = 0;

            string tempPath = Path.Combine(Path.GetTempPath(), "EnhancedDiscord");

            if (Directory.Exists(tempPath))
            {
                try
                {
                    Directory.Delete(tempPath, true);
                }
                catch (Exception e)
                {
                    StatusText.Text = "Error deleting temp folders.";
                    Logger.Log(StatusText.Text + " " + e.Message);
                }
            }
            Directory.CreateDirectory(tempPath);

            StatusText.Text = "Downloading package...";
            Logger.Log(StatusText.Text);
            string zipPath = Path.Combine(tempPath, "EnhancedDiscord.zip");
            string zipLink = Properties.Resources.zipLink + branch;

            ServicePointManager.SecurityProtocol = SecurityProtocolType.Tls | SecurityProtocolType.Tls11 | SecurityProtocolType.Tls12 | SecurityProtocolType.Ssl3;
            WebClient wc = new WebClient();

            try
            {
                await wc.DownloadFileTaskAsync(new Uri(zipLink), zipPath);
            }
            catch (Exception e)
            {
                Logger.Error("Failed to download ED files. " + e.Message);
                endInstallation("Failed to download ED files.", true); return;
            }
            InstallProgress.Value = 40;
            StatusText.Text       = "Successfully downloaded. Extracting...";
            Logger.Log(StatusText.Text);

            try
            {
                ZipFile.ExtractToDirectory(zipPath, tempPath);
            }
            catch (Exception e)
            {
                Logger.Error("Failed to extract zip file. " + e.Message);
                endInstallation("Failed to extract zip file.", true); return;
            }
            InstallProgress.Value = 50;
            StatusText.Text       = "Finished extracting zip. Checking core...";
            Logger.Log(StatusText.Text);

            string extractedPath = Path.Combine(tempPath, "EnhancedDiscord-" + branch);
            string enhancedPath  = "./EnhancedDiscord";

            if (!File.Exists(Path.Combine(enhancedPath, "config.json")))
            {
                try
                {
                    File.WriteAllText(Path.Combine(enhancedPath, "config.json"), "{}");
                }
                catch (Exception e)
                {
                    Logger.Error("Failed to write config.json. " + e.Message);
                }
            }

            string[] garbage = new string[] { "README.md", "plugins.md", ".gitignore", "advanced_installation.md" };
            foreach (string file in Directory.GetFiles(extractedPath))
            {
                string filename = Path.GetFileName(file);
                if (Array.Exists(garbage, f => f == filename))
                {
                    continue;
                }
                string equiv      = Path.Combine(enhancedPath, filename);
                bool   filesEqual = false;
                bool   fileExists = File.Exists(equiv);
                if (fileExists)
                {
                    filesEqual = FilesEqual(file, equiv);
                }
                try
                {
                    if (fileExists && !filesEqual)
                    {
                        File.Delete(equiv);
                    }
                    if (!fileExists || !filesEqual)
                    {
                        File.Copy(file, equiv);
                    }
                }
                catch (Exception e)
                {
                    StatusText.Text = "Could not update plugin: " + filename;
                    Logger.Log(StatusText.Text + " " + e.Message);
                }
            }
            InstallProgress.Value = 70;
            StatusText.Text       = "Core finished. Checking plugins...";
            Logger.Log(StatusText.Text);

            string pluginPath = Path.Combine(enhancedPath, "plugins");

            if (!Directory.Exists(pluginPath))
            {
                Directory.CreateDirectory(pluginPath);
            }
            foreach (string file in Directory.GetFiles(Path.Combine(extractedPath, "plugins")))
            {
                string filename = Path.GetFileName(file);
                if (filename == "style.css")
                {
                    continue;
                }
                string equiv      = Path.Combine(pluginPath, filename);
                bool   filesEqual = false;
                bool   fileExists = File.Exists(equiv);
                if (fileExists)
                {
                    filesEqual = FilesEqual(file, equiv);
                }
                try
                {
                    if (fileExists && !filesEqual)
                    {
                        File.Delete(equiv);
                    }
                    if (!fileExists || !filesEqual)
                    {
                        File.Copy(file, equiv);
                    }
                }
                catch (Exception e)
                {
                    StatusText.Text = "Could not update plugin: " + filename;
                    Logger.Log(StatusText.Text + " " + e.Message);
                }
            }

            StatusText.Text = "Cleaning up...";
            Logger.Log(StatusText.Text);
            if (Directory.Exists(tempPath))
            {
                try
                {
                    Directory.Delete(tempPath, true);
                }
                catch (Exception e)
                {
                    StatusText.Text = "Error deleting temp folders.";
                    Logger.Log(StatusText.Text + " " + e.Message);
                }
            }
            InstallProgress.Value = 90;
            endInstallation("ED files updated.", false); return;
        }
示例#3
0
        private void InstallButton_Click(object sender, EventArgs e)
        {
            if (BetaRadio.Checked)
            {
                branch = "beta";
            }
            BetaRadio.Hide();
            InstallButton.Hide();
            UninstallButton.Hide();
            UpdateButton.Hide();
            ReinjectButton.Hide();
            StatusText.Show();
            InstallProgress.Show();
            StatusText.Text = "Finding Discord processes...";

            Process[] stable = Process.GetProcessesByName("Discord");
            Process[] canary = Process.GetProcessesByName("DiscordCanary");
            Process[] ptb    = Process.GetProcessesByName("DiscordPtb");
            Process[] dev    = Process.GetProcessesByName("DiscordDevelopment");

            List <Process> discordProcesses = new List <Process>();

            discordProcesses.AddRange(stable);
            discordProcesses.AddRange(canary);
            discordProcesses.AddRange(ptb);
            discordProcesses.AddRange(dev);

            if (discordProcesses.Count == 0)
            {
                endInstallation("No Discord processes found. Please open Discord and try again.", true); return;
            }
            List <Process> uniqueProcesses = new List <Process>();

            // First look for processes with unique filenames that have a title
            for (int i = 0; i < discordProcesses.Count; i++)
            {
                bool isUnique = true;
                for (int j = 0; j < uniqueProcesses.Count; j++)
                {
                    if (uniqueProcesses[j].MainModule.FileName.Equals(discordProcesses[i].MainModule.FileName))
                    {
                        isUnique = false; break;
                    }
                }
                if (!isUnique || discordProcesses[i].MainWindowTitle == "" || discordProcesses[i].MainWindowTitle.StartsWith("Developer Tools"))
                {
                    continue;
                }

                uniqueProcesses.Add(discordProcesses[i]);
            }
            // Then look for all processes with unique filenames
            for (int i = 0; i < discordProcesses.Count; i++)
            {
                bool isUnique = true;
                for (int j = 0; j < uniqueProcesses.Count; j++)
                {
                    if (uniqueProcesses[j].MainModule.FileName.Equals(discordProcesses[i].MainModule.FileName))
                    {
                        isUnique = false; break;
                    }
                }
                if (!isUnique)
                {
                    continue;
                }
                uniqueProcesses.Add(discordProcesses[i]);
            }
            StatusText.Text       = "Found " + uniqueProcesses.Count + " Discord process" + (uniqueProcesses.Count == 1 ? "" : "es") + ".";
            InstallProgress.Value = 10;
            Process finalProcess = uniqueProcesses[0];

            if (uniqueProcesses.Count > 1)
            {
                // Enable selection buttons
                List <Button> clients = new List <Button>();
                for (int i = 0; i < uniqueProcesses.Count; i++)
                {
                    if (canary.Contains(uniqueProcesses[i]))
                    {
                        CanaryButton.Show();
                        clients.Add(CanaryButton);
                        canaryProcess = uniqueProcesses[i];
                    }
                    else if (ptb.Contains(uniqueProcesses[i]))
                    {
                        PTBButton.Show();
                        clients.Add(PTBButton);
                        ptbProcess = uniqueProcesses[i];
                    }
                    else if (dev.Contains(uniqueProcesses[i]))
                    {
                        DevButton.Show();
                        clients.Add(DevButton);
                        devProcess = uniqueProcesses[i];
                    }
                    else if (stable.Contains(uniqueProcesses[i]))
                    {
                        StableButton.Show();
                        clients.Add(StableButton);
                        stableProcess = uniqueProcesses[i];
                    }
                }
                // position buttons correctly
                if (clients.Count == 3)
                {
                    clients[0].Left = 55;
                    clients[1].Left = 131;
                    clients[2].Left = 207;
                }
                else if (clients.Count == 2)
                {
                    clients[0].Left = 88;
                    clients[1].Left = 164;
                }
                return; // stuff continues w/ button events
            }
            if (operation == "UPDATE")
            {
                continueUpdate(finalProcess);
            }
            else
            {
                continueInstall(finalProcess);
            }
        }