// Set the text that will display.
public static void SetText(string text)
{
if (statusText != null)
{
// Show the text if it exists.
StatusText.Show();
statusText.text = text;
}
}
private async void FetchAddressBtn_Click(object sender, EventArgs e)
{
FetchAddressBtn.Enabled = false;
StatusText.Show();
StatusPic.Show();
IPAddress addy = null;
// WebClient can sometimes cause the GUI to lock being all slow and such, so task it up
await Task.Run(() =>
{
// Use a web client to fetch our IP Address
using (WebClient Web = new WebClient())
{
// Disable Proxy to prevent slowness
Web.Proxy = null;
// Set headers (Copied from my current Chrome headers)
Web.Headers["User-Agent"] = "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/43.0.2357.81 Safari/537.36";
Web.Headers["Accept"] = "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8";
Web.Headers["Accept-Language"] = "en-US,en;q=0.8";
// Loop through each service and check for our IP
for (int i = 0; i < IpServices.Length; i++)
{
try
{
// Attempt to Fetch the IP address from this service
string result = Web.DownloadString(IpServices[i]);
Match match = Regex.Match(result, @"(?<IpAddress>\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})");
if (match.Success && IPAddress.TryParse(match.Groups["IpAddress"].Value, out addy))
{
break; // Success, we have an IP so stop here
}
}
catch
{
continue; // Error, Skip to next service
}
}
}
});
// If we were unable to fetch the IP address, then alert the user
if (addy == null)
{
StatusText.Text = "Unable to fetch external address!";
StatusPic.Image = BF2Statistics.Properties.Resources.error;
}
else
{
StatusText.Text = "Address Fetched Successfully!";
StatusPic.Image = BF2Statistics.Properties.Resources.check;
AddressTextBox.Text = addy.ToString();
}
}
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);
}
}
// Wait for the object to load. Adds a layer of abstraction.
private IEnumerator WaitForLoad()
{
// Don't try to load two different elements at once. Could get messy.
while (SiteManager.loading)
{
yield return(null);
}
// Lock user input so they can't interact while an object is loading.
GamepadInput.LockInput(true);
SiteManager.loading = true;
// Show the loading status.
StatusText.Show();
// If there's no specified scene name in the JSON, then we just load the data normally.
if (string.IsNullOrEmpty(siteData.sceneName))
{
yield return(StartCoroutine(LoadCoroutine()));
yield return(StartCoroutine(LoadCustomData()));
}
// Otherwise, we just load the scene specified in JSON.
else
{
// Variable to keep track if we actually found the scene or not.
bool foundScene = false;
// Iterate through all existing scenes.
for (int i = 0; i < SceneManager.sceneCount; i++)
{
// Get the scene at specified index.
Scene scene = SceneManager.GetSceneAt(i);
// Check if the names match.
if (scene.name.Equals(siteData.sceneName))
{
// If they match, scene was found. Load the scene.
foundScene = true;
AsyncOperation sceneLoad = SceneManager.LoadSceneAsync(siteData.sceneName);
// Load the scene specified in JSON and wait for it to finish loading.
while (!sceneLoad.isDone)
{
yield return(null);
}
// Break, since we found the scene.
break;
}
}
// If we didn't find the scene, log an error and try to load normally... which probably won't work.
if (!foundScene)
{
Debug.LogErrorFormat("Could not find scene name '{0}', which was provided in JSON data file. Ensure this scene exists and is added to build settings.");
yield return(StartCoroutine(LoadCoroutine()));
}
}
// Everything is now loaded.
loaded = true;
// Hide the status text.
StatusText.Hide();
// Unlock input and mark scene as not loading anymore.
GamepadInput.LockInput(false);
SiteManager.loading = false;
}