Disco/Disco.Services/Plugins/UninstallPluginTask.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using System.Web;
using Disco.Services.Tasks;
using Quartz;

namespace Disco.Services.Plugins
{
    public class UninstallPluginTask : ScheduledTask
    {
        public override string TaskName { get { return "Uninstalling Plugin"; } }

        protected override void ExecuteTask()
        {
            PluginManifest manifest = (PluginManifest)this.ExecutionContext.JobDetail.JobDataMap["PluginManifest"];
            bool UninstallData = (bool)this.ExecutionContext.JobDetail.JobDataMap["UninstallData"];

            this.Status.UpdateStatus(25, string.Format("Uninstalling Plugin: {0} [{1}]", manifest.Name, manifest.Id), "Queuing plugin for uninstall");

            PluginsLog.LogUninstalling(manifest, UninstallData);

            string manifestFileLocation = Path.Combine(manifest.PluginLocation, "manifest.json");
            if (!File.Exists(manifestFileLocation))
                throw new FileNotFoundException("Plugin Manifest File Not Found", manifestFileLocation);

            string manifestUninstallFileLocation = Path.Combine(manifest.PluginLocation, "manifest.uninstall.json");

            if (File.Exists(manifestUninstallFileLocation))
                File.Delete(manifestUninstallFileLocation);

            File.Move(manifestFileLocation, manifestUninstallFileLocation);

            if (UninstallData && Directory.Exists(manifest.StorageLocation))
            {
                var manifestDataUninstallFileLocation = Path.Combine(manifest.StorageLocation, "manifest.uninstall.json");
                if (File.Exists(manifestDataUninstallFileLocation))
                    File.Delete(manifestDataUninstallFileLocation);

                File.Copy(manifestUninstallFileLocation, manifestDataUninstallFileLocation);
            }

            this.Status.Finished("Restarting Disco, please wait...", "/Config/Plugins");
            RestartApp(1500);
        }

        public static ScheduledTaskStatus UninstallPlugin(PluginManifest Manifest, bool UninstallData)
        {
            if (ScheduledTasks.GetTaskStatuses(typeof(InstallPluginTask)).Where(s => s.IsRunning).Count() > 0)
                throw new InvalidOperationException("A plugin is already being Uninstalled");

            JobDataMap taskData = new JobDataMap() { { "PluginManifest", Manifest }, { "UninstallData", UninstallData } };

            var instance = new UninstallPluginTask();

            return instance.ScheduleTask(taskData);
        }

        #region Restart App
        private static object _restartTimerLock = new object();
        private static Timer _restartTimer;
        private void RestartApp(int DelayMilliseconds)
        {
            lock (_restartTimerLock)
            {
                if (_restartTimer != null)
                {
                    _restartTimer.Dispose();
                }

                _restartTimer = new Timer((state) =>
                {
                    HttpRuntime.UnloadAppDomain();
                    //AppDomain.Unload(AppDomain.CurrentDomain);
                }, null, DelayMilliseconds, Timeout.Infinite);
            }
        }
        #endregion
    }
}