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();
}
}
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;
}
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);
}
}
