From f30eb1b3b962ef64386fdb3b9127f5eb920fd6a0 Mon Sep 17 00:00:00 2001 From: Bob Da Ross <52760019+BobDaRoss@users.noreply.github.com> Date: Sat, 19 Jun 2021 22:19:44 -0500 Subject: [PATCH] Revert "Removed updater again. Forgot I was in the wrong branch" This reverts commit 845bb0ecdd9699afde874912013a923a73f29aa7. --- SeamlessClient.cs | 6 + SeamlessClientPlugin.csproj | 1 + Utilities/UpdateChecker.cs | 315 ++++++++++++++++++++++++++++++++++++ 3 files changed, 322 insertions(+) create mode 100644 Utilities/UpdateChecker.cs diff --git a/SeamlessClient.cs b/SeamlessClient.cs index 5f53c77..c650e20 100644 --- a/SeamlessClient.cs +++ b/SeamlessClient.cs @@ -122,6 +122,12 @@ namespace SeamlessClientPlugin public void Init(object gameInstance) { + Utilities.UpdateChecker Checker = new Utilities.UpdateChecker(false); + Task UpdateChecker = new Task(() => Checker.PingUpdateServer()); + UpdateChecker.Start(); + + + Patches.GetPatches(); TryShow("Running Seamless Client Plugin v[" + Version + "]"); PingTimer.Elapsed += PingTimer_Elapsed; diff --git a/SeamlessClientPlugin.csproj b/SeamlessClientPlugin.csproj index 943e1bf..936fa49 100644 --- a/SeamlessClientPlugin.csproj +++ b/SeamlessClientPlugin.csproj @@ -167,6 +167,7 @@ + diff --git a/Utilities/UpdateChecker.cs b/Utilities/UpdateChecker.cs new file mode 100644 index 0000000..7cdc29f --- /dev/null +++ b/Utilities/UpdateChecker.cs @@ -0,0 +1,315 @@ +using ProtoBuf; +using Sandbox.Graphics.GUI; +using System; +using System.Collections.Generic; +using System.IO; +using System.IO.Compression; +using System.Linq; +using System.Net; +using System.Net.Sockets; +using System.Reflection; +using System.Runtime.Serialization; +using System.Runtime.Serialization.Json; +using System.Text; +using System.Threading.Tasks; +using System.Windows.Forms; +using System.Xml.Serialization; +namespace SeamlessClientPlugin.Utilities +{ + public class UpdateChecker + { + public string PluginFolder; + public string CurrentVersion; + public bool DownloadUpdate; + private string GitHubAPILink = "https://api.github.com/repos/Casimir255/SeamlessClientPlugin/releases/latest"; + + private WebClient Client; + + + + + public UpdateChecker(bool AutoUpdate) + { + PluginFolder = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location); + Client = new WebClient(); + DeleteOLDFiles(); + } + + + + + public void PingUpdateServer() + { + try + { + //Create new webclient and insert a user-agent + Client.Headers["User-Agent"] = "SeamlessClientUpdater"; + + //Grap API data for latest seamless client release + string data = Client.DownloadString(GitHubAPILink); + + //SeamlessClient.TryShow(data); + + + DataContractJsonSerializer s = new DataContractJsonSerializer(typeof(GithubRelease)); + + GithubRelease Release; + using (MemoryStream stream = new MemoryStream(Encoding.UTF8.GetBytes(data))) + { + Release = (GithubRelease)s.ReadObject(stream); + } + + + if (Release == null || !TryGetMainRelease(Release.Content, out GitZipFile MainReleaseFile)) + return; + + + //Check if the client needs an update based off of github latest release version + + if (!NeedsUpdate(SeamlessClient.Version, Release.LatestVersion)) + return; + + + //Ask client if they want to update! + ShowDialog(Release, MainReleaseFile); + } + catch (Exception Ex) + { + SeamlessClient.TryShow(Ex.ToString()); + } + } + + + private bool TryGetMainRelease(GitZipFile[] Files, out GitZipFile Release) + { + Release = null; + + //Sanity saftey checks + if (Files == null || Files.Length <= 0) + return false; + + foreach (GitZipFile File in Files) + { + if (File.Name == "SeamlessClientPlugin.zip") + { + Release = File; + return true; + } + + } + + return false; + } + + + private void ShowDialog(GithubRelease Release, GitZipFile MainReleaseFile) + { + StringBuilder Response = new StringBuilder(); + Response.AppendLine($"Current version: {SeamlessClient.Version} Latest: {Release.LatestVersion}"); + Response.AppendLine($"Update: {Release.Name}"); + Response.AppendLine($"Description: {Release.Description}"); + Response.AppendLine($"Size: {MainReleaseFile.Size / 1000}kb"); + Response.AppendLine(); + Response.AppendLine("Warning: If you have a version less than latest seamless will be disabled to prevent crashes!"); + Response.AppendLine("(Clicking yes should restart your game)"); + + DialogResult Result = MessageBox.Show(Response.ToString(), $"Download Seamless Client Plugin Update v{ Release.LatestVersion}?", MessageBoxButtons.YesNo, MessageBoxIcon.None, MessageBoxDefaultButton.Button1, MessageBoxOptions.DefaultDesktopOnly); + SeamlessClient.TryShow(Response.ToString()); + + if (Result == DialogResult.Yes) + { + SeamlessClient.TryShow("Client wants to update!"); + string DownloadPath = Path.Combine(PluginFolder, MainReleaseFile.Name); + Client.DownloadFile(new Uri(MainReleaseFile.ZipURL), DownloadPath); + + if (!File.Exists(DownloadPath)) + { + SeamlessClient.TryShow("Failed to download zip!"); + return; + } + + if (ExtractAndReplace(DownloadPath)) + { + StringBuilder ErrorResponse = new StringBuilder(); + ErrorResponse.AppendLine("There was an error during the extraction proccess! Check your logs for more information!"); + ErrorResponse.AppendLine(); + ErrorResponse.AppendLine("You can download manually here:"); + ErrorResponse.AppendLine(Release.GitHubPage); + SeamlessClient.TryShow(ErrorResponse.ToString()); + MessageBox.Show(ErrorResponse.ToString(), $"Failed to update plugin to v{ Release.LatestVersion}!", MessageBoxButtons.OK, MessageBoxIcon.Error, MessageBoxDefaultButton.Button1, MessageBoxOptions.DefaultDesktopOnly); + return; + } + + } + else + { + SeamlessClient.TryShow("Client skipped Update!"); + return; + } + + + + + } + + + + + + private void DeleteOLDFiles() + { + foreach (var OLDFile in Directory.GetFiles(PluginFolder, "*.old")) + { + File.Delete(OLDFile); + } + + SeamlessClient.TryShow("Deleted all OLD update files"); + + } + + + private bool ExtractAndReplace(string ZipPath) + { + try + { + + //Start extractor + using (ZipArchive archive = ZipFile.OpenRead(ZipPath)) + { + foreach (ZipArchiveEntry entry in archive.Entries) + { + string ExsistingFilePath = Path.Combine(PluginFolder, entry.Name); + string OldFilePath = Path.Combine(PluginFolder, entry.Name + ".old"); + + //No need to extract to files that dont exsist + if (!File.Exists(ExsistingFilePath)) + continue; + + SeamlessClient.TryShow(ExsistingFilePath + "=>" + OldFilePath); + + if (File.Exists(OldFilePath)) + File.Delete(OldFilePath); + + File.Move(ExsistingFilePath, OldFilePath); + entry.ExtractToFile(ExsistingFilePath, false); + //File.Delete(OldFilePath); + } + } + + //Delete latest zip + File.Delete(ZipPath); + + //Restart client + SeamlessClient.TryShow("UpdateComplete!"); + SeamlessClient.RestartClientAfterUpdate(); + return true; + + } + catch (Exception ex) + { + SeamlessClient.TryShow(ex.ToString()); + return false; + } + } + + private bool NeedsUpdate(string ClientVersion, string ServerVersion) + { + + + Version Client = new Version(ClientVersion); + Version Latest = new Version(ServerVersion); + + var result = Client.CompareTo(Latest); + if (result > 0) + { + //Console.WriteLine("Client is greater"); + SeamlessClient.TryShow("Client version is greater than latest! Wow!"); + return false; + } + else if (result < 0) + { + //Console.WriteLine("Latest is greater"); + SeamlessClient.TryShow("Client version is out-of-date!"); + return true; + } + else + { + //Console.WriteLine("versions are equal"); + SeamlessClient.TryShow("Client is up-to-date!"); + return false; + } + } + + } + + + [ProtoContract] + public class UpdateMessage + { + [ProtoMember(10)] + public string ClientVersion = ""; + + [ProtoMember(20)] + public bool UpToDate = false; + + [ProtoMember(30)] + public bool DownloadNewUpdate = false; + + [ProtoMember(40)] + public string ServerVersion = ""; + + [ProtoMember(50)] + public string UpdateNotes = ""; + + + + /* Misc Stuff incase I need it */ + [ProtoMember(60)] + public byte[] XmlCharactersAsBytes; + + + public UpdateMessage() { } + + + } + + + [DataContract] + public class GithubRelease + { + + [DataMember(Name = "url")] + public string GitHubPage { get; set; } + + [DataMember(Name = "name")] + public string Name { get; set; } + + + [DataMember(Name = "tag_name")] + public string LatestVersion { get; set; } + + [DataMember(Name = "prerelease")] + public bool Beta { get; set; } + + [DataMember(Name = "body")] + public string Description { get; set; } + + [DataMember(Name = "assets")] + public GitZipFile[] Content { get; set; } + + } + + [DataContract] + public class GitZipFile + { + [DataMember(Name = "name")] + public string Name { get; set; } + + [DataMember(Name = "browser_download_url")] + public string ZipURL { get; set; } + + [DataMember(Name = "size")] + public int Size { get; set; } + + } +}