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;
using Disco.Data.Repository;

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);

            using (DiscoDataContext dbContext = new DiscoDataContext())
            {
                manifest.UninstallPlugin(dbContext, UninstallData, this.Status);
            }

            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");
            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");
            if (ScheduledTasks.GetTaskStatuses(typeof(UpdatePluginTask)).Where(s => s.IsRunning).Count() > 0)
                throw new InvalidOperationException("A plugin is being Updated");
            if (ScheduledTasks.GetTaskStatuses(typeof(InstallPluginTask)).Where(s => s.IsRunning).Count() > 0)
                throw new InvalidOperationException("A plugin is being Installed");

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

            var instance = new UninstallPluginTask();

            return instance.ScheduleTask(taskData);
        }
    }
}