TShock/TShockAPI/PluginUpdater/PluginVersionCheck.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Net;
using System.Text;
using JsonLoader;
using Newtonsoft.Json;

namespace TShockAPI.PluginUpdater
{
    public class PluginVersionCheck
    {
        public delegate void PluginUpdateD(UpdateArgs e);
        public static event PluginUpdateD PluginUpdate;
        public static void OnPluginUpdate(UpdateArgs args)
        {
            if (PluginUpdate == null)
            {
                return;
            }

            PluginUpdate(args);
        }

        public static void CheckPlugin(object p)
        {
            TerrariaPlugin plugin = (TerrariaPlugin)p;
            UpdateArgs args = new UpdateArgs {Plugin = plugin, Success = true, Error = ""};
            List<string> files = new List<string>();

            try
            {
                if (!String.IsNullOrEmpty(plugin.UpdateURL))
                {
                    var request = HttpWebRequest.Create(plugin.UpdateURL);
                    VersionInfo vi;

                    request.Timeout = 5000;
                    using (var response = request.GetResponse())
                    {
                        using (var reader = new StreamReader(response.GetResponseStream()))
                        {
                            vi = JsonConvert.DeserializeObject<VersionInfo>(reader.ReadToEnd());
                        }
                    }

                    System.Version v = System.Version.Parse((vi.version.ToString()));

                    if (!v.Equals(plugin.Version))
                    {
                        DownloadPackage pkg;
                        request = HttpWebRequest.Create(vi.url);

                        request.Timeout = 5000;
                        using (var response = request.GetResponse())
                        {
                            using (var reader = new StreamReader(response.GetResponseStream()))
                            {
                                pkg = JsonConvert.DeserializeObject<DownloadPackage>(reader.ReadToEnd());
                            }
                        }

                        foreach (PluginFile f in pkg.files)
                        {
                            using (WebClient Client = new WebClient())
                            {
                                string dir = Path.Combine(TShock.SavePath, "UpdatedPlugins");
                                if (!Directory.Exists(dir))
                                    Directory.CreateDirectory(dir);

                                Client.DownloadFile(f.url,
                                                    Path.Combine(dir, f.destination));

                                files.Add(Path.Combine(dir, f.destination));
                            }                            
                        }
                    }
                    else
                    {
                        args.Error = "Plugin is up to date.";
                    }
                }
                else
                {
                    args.Error = "Plugin has no updater recorded.";
                }
            }
            catch(Exception e)
            {
                args.Success = false;
                args.Error = e.Message;
                if(files.Count > 0)
                {
                    foreach(string s in files)
                    {
                        File.Delete(s);
                    }
                }
            }

            OnPluginUpdate(args);
        }
    }

    public class UpdateArgs
    {
        public TerrariaPlugin Plugin { get; set; }
        public bool Success { get; set; }
        public string Error { get; set; }
    }
}