1 files changed, 106 insertions, 0 deletions
diff --git a/MediaBrowser.ClickOnce/ApplicationUpdateCheck.cs b/MediaBrowser.ClickOnce/ApplicationUpdateCheck.cs
new file mode 100644
index 000000000..72c42f586
--- /dev/null
+++ b/MediaBrowser.ClickOnce/ApplicationUpdateCheck.cs
@@ -0,0 +1,106 @@
+using MediaBrowser.Model.Updates;
+using System;
+using System.Deployment.Application;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace MediaBrowser.ClickOnce
+{
+    /// <summary>
+    /// Class ApplicationUpdateCheck
+    /// </summary>
+    public class ApplicationUpdateCheck
+    {
+        /// <summary>
+        /// The _task completion source
+        /// </summary>
+        private TaskCompletionSource<CheckForUpdateResult> _taskCompletionSource;
+
+        /// <summary>
+        /// The _progress
+        /// </summary>
+        private IProgress<double> _progress;
+
+        /// <summary>
+        /// Checks for application update.
+        /// </summary>
+        /// <param name="cancellationToken">The cancellation token.</param>
+        /// <param name="progress">The progress.</param>
+        /// <returns>Task{CheckForUpdateCompletedEventArgs}.</returns>
+        /// <exception cref="System.InvalidOperationException">Current deployment is not a ClickOnce deployment</exception>
+        public Task<CheckForUpdateResult> CheckForApplicationUpdate(CancellationToken cancellationToken, IProgress<double> progress)
+        {
+            if (!ApplicationDeployment.IsNetworkDeployed)
+            {
+                throw new InvalidOperationException("Current deployment is not network deployed.");
+            }
+
+            _progress = progress;
+
+            _taskCompletionSource = new TaskCompletionSource<CheckForUpdateResult>();
+
+            var deployment = ApplicationDeployment.CurrentDeployment;
+
+            cancellationToken.Register(deployment.CheckForUpdateAsyncCancel);
+
+            cancellationToken.ThrowIfCancellationRequested();
+
+            deployment.CheckForUpdateCompleted += deployment_CheckForUpdateCompleted;
+            deployment.CheckForUpdateProgressChanged += deployment_CheckForUpdateProgressChanged;
+
+            deployment.CheckForUpdateAsync();
+
+            return _taskCompletionSource.Task;
+        }
+
+        /// <summary>
+        /// To the result.
+        /// </summary>
+        /// <param name="args">The <see cref="CheckForUpdateCompletedEventArgs" /> instance containing the event data.</param>
+        /// <returns>CheckForUpdateResult.</returns>
+        private CheckForUpdateResult ToResult(CheckForUpdateCompletedEventArgs args)
+        {
+            return new CheckForUpdateResult
+            {
+                AvailableVersion = args.AvailableVersion,
+                IsUpdateAvailable = args.UpdateAvailable
+            };
+        }
+
+        /// <summary>
+        /// Handles the CheckForUpdateCompleted event of the deployment control.
+        /// </summary>
+        /// <param name="sender">The source of the event.</param>
+        /// <param name="e">The <see cref="CheckForUpdateCompletedEventArgs" /> instance containing the event data.</param>
+        void deployment_CheckForUpdateCompleted(object sender, CheckForUpdateCompletedEventArgs e)
+        {
+            var deployment = ApplicationDeployment.CurrentDeployment;
+
+            deployment.CheckForUpdateCompleted -= deployment_CheckForUpdateCompleted;
+            deployment.CheckForUpdateProgressChanged -= deployment_CheckForUpdateProgressChanged;
+
+            if (e.Error != null)
+            {
+                _taskCompletionSource.SetException(e.Error);
+            }
+            else if (e.Cancelled)
+            {
+                _taskCompletionSource.SetCanceled();
+            }
+            else
+            {
+                _taskCompletionSource.SetResult(ToResult(e));
+            }
+        }
+
+        /// <summary>
+        /// Handles the CheckForUpdateProgressChanged event of the deployment control.
+        /// </summary>
+        /// <param name="sender">The source of the event.</param>
+        /// <param name="e">The <see cref="DeploymentProgressChangedEventArgs" /> instance containing the event data.</param>
+        void deployment_CheckForUpdateProgressChanged(object sender, DeploymentProgressChangedEventArgs e)
+        {
+            _progress.Report(e.ProgressPercentage);
+        }
+    }
+}