112 lines
4.1 KiB
C#
112 lines
4.1 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.ComponentModel;
|
|
using System.IO;
|
|
using System.Net;
|
|
using System.Threading.Tasks;
|
|
using System.Windows.Forms;
|
|
using Giants.WebApi.Clients;
|
|
|
|
namespace Giants.Launcher
|
|
{
|
|
public class Updater
|
|
{
|
|
private readonly IDictionary<ApplicationType, Version> appVersions;
|
|
private readonly AsyncCompletedEventHandler updateCompletedCallback;
|
|
private readonly DownloadProgressChangedEventHandler updateProgressCallback;
|
|
|
|
public Updater(
|
|
IDictionary<ApplicationType, Version> appVersions,
|
|
AsyncCompletedEventHandler updateCompletedCallback,
|
|
DownloadProgressChangedEventHandler updateProgressCallback)
|
|
{
|
|
this.appVersions = appVersions;
|
|
this.updateCompletedCallback = updateCompletedCallback;
|
|
this.updateProgressCallback = updateProgressCallback;
|
|
}
|
|
|
|
public Task UpdateApplication(ApplicationType applicationType, VersionInfo versionInfo)
|
|
{
|
|
try
|
|
{
|
|
if (this.ToVersion(versionInfo.Version) > this.appVersions[applicationType])
|
|
{
|
|
this.StartApplicationUpdate(applicationType, versionInfo);
|
|
}
|
|
}
|
|
catch (Exception)
|
|
{
|
|
}
|
|
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
private int GetHttpFileSize(Uri uri)
|
|
{
|
|
HttpWebRequest req = (HttpWebRequest)HttpWebRequest.Create(uri);
|
|
req.Proxy = null;
|
|
req.Method = "HEAD";
|
|
HttpWebResponse resp = (HttpWebResponse)req.GetResponse();
|
|
|
|
if (resp.StatusCode == HttpStatusCode.OK && int.TryParse(resp.Headers.Get("Content-Length"), out int contentLength))
|
|
{
|
|
resp.Close();
|
|
return contentLength;
|
|
}
|
|
|
|
resp.Close();
|
|
return -1;
|
|
|
|
}
|
|
|
|
private void StartApplicationUpdate(ApplicationType applicationType, VersionInfo versionInfo)
|
|
{
|
|
// Display update prompt
|
|
string updateMsg = applicationType == ApplicationType.Game ?
|
|
string.Format(Resources.UpdateAvailableText, this.ToVersion(versionInfo.Version).ToString()) :
|
|
string.Format(Resources.LauncherUpdateAvailableText, this.ToVersion(versionInfo.Version).ToString());
|
|
|
|
if (MessageBox.Show(updateMsg, Resources.UpdateAvailableTitle, MessageBoxButtons.YesNo) == DialogResult.No)
|
|
{
|
|
return; // User declined update
|
|
}
|
|
|
|
string patchFileName = Path.GetFileName(versionInfo.InstallerUri.AbsoluteUri);
|
|
string localPath = Path.Combine(Path.GetTempPath(), patchFileName);
|
|
|
|
// Delete the file locally if it already exists, just to be safe
|
|
if (File.Exists(localPath))
|
|
{
|
|
File.Delete(localPath);
|
|
}
|
|
|
|
int fileSize = this.GetHttpFileSize(versionInfo.InstallerUri);
|
|
if (fileSize == -1)
|
|
{
|
|
string errorMsg = string.Format(Resources.UpdateDownloadFailedText, "File not found on server.");
|
|
MessageBox.Show(errorMsg, Resources.UpdateDownloadFailedTitle, MessageBoxButtons.OK, MessageBoxIcon.Error);
|
|
return;
|
|
}
|
|
|
|
var updateInfo = new UpdateInfo()
|
|
{
|
|
FilePath = localPath,
|
|
FileSize = fileSize,
|
|
ApplicationType = applicationType
|
|
};
|
|
|
|
// Download the update
|
|
// TODO: Super old code, replace this with async HttpClient
|
|
WebClient client = new WebClient();
|
|
client.DownloadFileAsync(versionInfo.InstallerUri, localPath, updateInfo);
|
|
client.DownloadFileCompleted += this.updateCompletedCallback;
|
|
client.DownloadProgressChanged += this.updateProgressCallback;
|
|
}
|
|
|
|
private Version ToVersion(AppVersion version)
|
|
{
|
|
return new Version(version.Major, version.Minor, version.Build, version.Revision);
|
|
}
|
|
}
|
|
}
|