Software/Visual_Studio/PPC/Tango.PPC.Common/UpdatePackages/DefaultPackageRunner.cs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
pre { line-height: 125%; }
td.linenos .normal { color: inherit; background-color: transparent; padding-left: 5px; padding-right: 5px; }
span.linenos { color: inherit; background-color: transparent; padding-left: 5px; padding-right: 5px; }
td.linenos .special { color: #000000; background-color: #ffffc0; padding-left: 5px; padding-right: 5px; }
span.linenos.special { color: #000000; background-color: #ffffc0; padding-left: 5px; padding-right: 5px; }
.highlight .hll { background-color: #ffffcc }
.highlight .c { color: #888888 } /* Comment */
.highlight .err { color: #a61717; background-color: #e3d2d2 } /* Error */
.highlight .k { color: #008800; font-weight: bold } /* Keyword */
.highlight .ch { color: #888888 } /* Comment.Hashbang */
.highlight .cm { color: #888888 } /* Comment.Multiline */
.highlight .cp { color: #cc0000; font-weight: bold } /* Comment.Preproc */
.highlight .cpf { color: #888888 } /* Comment.PreprocFile */
.highlight .c1 { color: #888888 } /* Comment.Single */
.highlight .cs { color: #cc0000; font-weight: bold; background-color: #fff0f0 } /* Comment.Special */
.highlight .gd { color: #000000; background-color: #ffdddd } /* Generic.Deleted */
.highlight .ge { font-style: italic } /* Generic.Emph */
.highlight .ges { font-weight: bold; font-style: italic } /* Generic.EmphStrong */
.highlight .gr { color: #aa0000 } /* Generic.Error */
.highlight .gh { color: #333333 } /* Generic.Heading */
.highlight .gi { color: #000000; background-color: #ddffdd } /* Generic.Inserted */
.highlight .go { color: #888888 } /* Generic.Output */
.highlight .gp { color: #555555 } /* Generic.Prompt */
.highlight .gs { font-weight: bold } /* Generic.Strong */
.highlight .gu { color: #666666 } /* Generic.Subheading */
.highlight .gt { color: #aa0000 } /* Generic.Traceback */
.highlight .kc { color: #008800; font-weight: bold } /* Keyword.Constant */
.highlight .kd { color: #008800; font-weight: bold } /* Keyword.Declaration */
.highlight .kn { color: #008800; font-weight: bold } /* Keyword.Namespace */
.highlight .kp { color: #008800 } /* Keyword.Pseudo */
.highlight .kr { color: #008800; font-weight: bold } /* Keyword.Reserved */
.highlight .kt { color: #888888; font-weight: bold } /* Keyword.Type */
.highlight .m { color: #0000DD; font-weight: bold } /* Literal.Number */
.highlight .s { color: #dd2200; background-color: #fff0f0 } /* Literal.String */
.highlight .na { color: #336699 } /* Name.Attribute */
.highlight .nb { color: #003388 } /* Name.Builtin */
.highlight .nc { color: #bb0066; font-weight: bold } /* Name.Class */
.highlight .no { color: #003366; font-weight: bold } /* Name.Constant */
.highlight .nd { color: #555555 } /* Name.Decorator */
.highlight .ne { color: #bb0066; font-weight: bold } /* Name.Exception */
.highlight .nf { color: #0066bb; font-weight: bold } /* Name.Function */
.highlight .nl { color: #336699; font-style: italic } /* Name.Label */
.highlight .nn { color: #bb0066; font-weight: bold } /* Name.Namespace */
.highlight .py { color: #336699; font-weight: bold } /* Name.Property */
.highlight .nt { color: #bb0066; font-weight: bold } /* Name.Tag */
.highlight .nv { color: #336699 } /* Name.Variable */
.highlight .ow { color: #008800 } /* Operator.Word */
.highlight .w { color: #bbbbbb } /* Text.Whitespace */
.highlight .mb { color: #0000DD; font-weight: bold } /* Literal.Number.Bin */
.highlight .mf { color: #0000DD; font-weight: bold } /* Literal.Number.Float */
.highlight .mh { color: #0000DD; font-weight: bold } /* Literal.Number.Hex */
.highlight .mi { color: #0000DD; font-weight: bold } /* Literal.Number.Integer */
.highlight .mo { color: #0000DD; font-weight: bold } /* Literal.Number.Oct */
.highlight .sa { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Affix */
.highlight .sb { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Backtick */
.highlight .sc { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Char */
.highlight .dl { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Delimiter */
.highlight .sd { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Doc */
.highlight .s2 { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Double */
.highlight .se { color: #0044dd; background-color: #fff0f0 } /* Literal.String.Escape */
.highlight .sh { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Heredoc */
.highlight .si { color: #3333bb; background-color: #fff0f0 } /* Literal.String.Interpol */
.highlight .sx { color: #22bb22; background-color: #f0fff0 } /* Literal.String.Other */
.highlight .sr { color: #008800; background-color: #fff0ff } /* Literal.String.Regex */
.highlight .s1 { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Single */
.highlight .ss { color: #aa6600; background-color: #fff0f0 } /* Literal.String.Symbol */
.highlight .bp { color: #003388 } /* Name.Builtin.Pseudo */
.highlight .fm { color: #0066bb; font-weight: bold } /* Name.Function.Magic */
.highlight .vc { color: #336699 } /* Name.Variable.Class */
.highlight .vg { color: #dd7700 } /* Name.Variable.Global */
.highlight .vi { color: #3333bb } /* Name.Variable.Instance */
.highlight .vm { color: #336699 } /* Name.Variable.Magic */
.highlight .il { color: #0000DD; font-weight: bold } /* Literal.Number.Integer.Long */using System;
using System.Collections.Generic;
using System.Linq;
using System.Runtime.Serialization;
using System.Text;
using System.Threading.Tasks;

namespace Tango.PPC.Common.Update
{
    [DataContract]
    public class UploadCompletedRequest
    {
        [DataMember]
        public String Token { get; set; }
    }
}
using Newtonsoft.Json;
using Newtonsoft.Json.Converters;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using Tango.Core;
using Tango.Core.DI;
using Tango.Core.Helpers;
using Tango.PPC.Common.Application;
using Tango.PPC.Common.Connection;
using Tango.PPC.Common.Notifications;
using Tango.PPC.Shared.Updates;

namespace Tango.PPC.Common.UpdatePackages
{
    public class DefaultPackageRunner : ExtendedObject, IPackageRunner
    {
        private JsonSerializerSettings _jsonSettings;
        private String _configFile;
        private PackagesFile _packagesFile;

        public event EventHandler<PackageStateChangedEventArgs> PackageStateChanged;
        public event EventHandler<PackageProgressEventArgs> PackageProgress;

        public DefaultPackageRunner()
        {
            _jsonSettings = new JsonSerializerSettings
            {
                Formatting = Formatting.Indented,
                Error = (sender, args) =>
                {
                    args.ErrorContext.Handled = true;
                    LogManager.Log(args.ErrorContext.Error.Message);
                }
            };

            _jsonSettings.Converters.Add(new StringEnumConverter(false));

            _configFile = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData), "Twine", "Tango", "Packages", "packages.json");
        }

        public Task<PackagesFile> GetPackagesFile()
        {
            return Task.Factory.StartNew<PackagesFile>(() =>
            {
                if (_packagesFile != null)
                {
                    return _packagesFile;
                }
                else
                {
                    Directory.CreateDirectory(Path.GetDirectoryName(_configFile));

                    _packagesFile = new PackagesFile();

                    try
                    {
                        if (File.Exists(_configFile))
                        {
                            LogManager.Log("Loading packages config from " + _configFile + "...");
                            _packagesFile = JsonConvert.DeserializeObject<PackagesFile>(File.ReadAllText(_configFile), _jsonSettings);
                        }
                    }
                    catch (Exception ex)
                    {
                        LogManager.Log(ex, "Error loading packages file.");
                    }

                    return _packagesFile;
                }
            });
        }

        private void SavePackagesConfig()
        {
            Directory.CreateDirectory(Path.GetDirectoryName(_configFile));

            String json = String.Empty;

            if (_packagesFile != null)
            {
                json = JsonConvert.SerializeObject(_packagesFile, _jsonSettings);
            }
            else
            {
                json = JsonConvert.SerializeObject(new PackagesFile(), _jsonSettings);
            }

            File.WriteAllText(_configFile, json);
        }

        public Task<PackageRunnerResult> Run(PackageType type, Version deltaVersion, String packagesFolder)
        {
            return Task.Factory.StartNew<PackageRunnerResult>(() =>
            {
                PackageRunnerResult result = new PackageRunnerResult();

                PackageContext context = new PackageContext();
                context.ApplicationManager = TangoIOC.Default.GetInstance<IPPCApplicationManager>();
                context.MachineProvider = TangoIOC.Default.GetInstance<IMachineProvider>();
                context.NotificationProvider = TangoIOC.Default.GetInstance<INotificationProvider>();
                context.InstalledVersion = context.ApplicationManager.Version;
                context.DeltaVersion = deltaVersion;

                LogManager.Log($"Running {type}-update packages...");

                //Get installed packages.
                _packagesFile = GetPackagesFile().Result;

                LogManager.Log($"Installed packages file:\n{_packagesFile}");

                if (Debugger.IsAttached)
                {
                    LogManager.Log("Debugger attached detected. switching packages folder to main application path...");

                    //TO DEBUG PACKAGES --
                    //
                    //On DEBUG build the packages assemblies are copied to the main application output folder using post-build events.
                    //Then, if a debugger is attached, we change the packages folder to the main output folder so they can be debugged easily.
                    packagesFolder = AssemblyHelper.GetCurrentAssemblyFolder();
                }

                LogManager.Log($"Scanning for packages on '{packagesFolder}'...");

                //Get all packages in folder.
                foreach (var packageFile in Directory.GetFiles(packagesFolder, "*.dll"))
                {
                    LogManager.Log($"Loading assembly '{Path.GetFileName(packageFile)}'...");

                    Assembly asm;

                    //Load assembly and investigate for types based on package type.
                    try
                    {
                        asm = Assembly.LoadFile(packageFile);
                    }
                    catch (Exception ex)
                    {
                        LogManager.Log(ex, "Error loading assembly!");
                        continue;
                    }

                    try
                    {
                        foreach (var packageType in asm.GetTypes().Where(
                            x => typeof(IPPCPackage).IsAssignableFrom(x) &&
                            x.GetCustomAttribute<PPCPackageAttribute>() != null &&
                            x.GetCustomAttribute<PPCPackageAttribute>().Type == type))
                        {
                            LogManager.Log($"Checking package '{packageType.FullName}'...");

                            try
                            {
                                //Getting installed package from file.
                                var installedPackage = _packagesFile.PackageInstallations.SingleOrDefault(x => x.PackageName == packageType.FullName);

                                //Check if requires installation.
                                if (installedPackage == null || installedPackage.State != PackageInstallationState.Installed)
                                {
                                    if (installedPackage == null)
                                    {
                                        LogManager.Log("Package was never installed.");

                                        installedPackage = new PackageInstallation();
                                        installedPackage.State = PackageInstallationState.NotInstalled;
                                        installedPackage.PackageName = packageType.FullName;
                                        installedPackage.Type = type;
                                        _packagesFile.PackageInstallations.Add(installedPackage);
                                    }
                                    else
                                    {
                                        LogManager.Log($"Package installation state is '{installedPackage.State}' due to {installedPackage.FailedReason}");
                                    }

                                    LogManager.Log("Installing package...");

                                    //Install package...
                                    var att = packageType.GetCustomAttribute<PPCPackageAttribute>();

                                    var packageInstance = Activator.CreateInstance(packageType) as IPPCPackage;

                                    if (packageInstance != null)
                                    {
                                        try
                                        {
                                            OnPackageRuns(att.Name, installedPackage.State, installedPackage.Type);
                                            installedPackage.InstallationDate = DateTime.Now;
                                            context.ProgressAction = (message, isIntermediate, progress, total) =>
                                            {
                                                PackageProgress?.Invoke(this, new PackageProgressEventArgs()
                                                {
                                                    PackageName = att.Name,
                                                    Message = message,
                                                    IsIntermediate = isIntermediate,
                                                    Progress = progress,
                                                    Total = total
                                                });
                                            };

                                            PackageProgress?.Invoke(this, new PackageProgressEventArgs()
                                            {
                                                PackageName = type == PackageType.Pre ? "Preparing" : "Finalizing",
                                                Message = att.Name,
                                                IsIntermediate = true,
                                                Progress = 0,
                                                Total = 100
                                            });

                                            packageInstance.Run(context).GetAwaiter().GetResult();
                                            installedPackage.State = PackageInstallationState.Installed;
                                            installedPackage.FailedReason = null;
                                            OnPackageRuns(att.Name, installedPackage.State, installedPackage.Type);
                                            LogManager.Log("Package installed successfully.");

                                            if (att.RequiresRestart)
                                            {
                                                result.RestartRequired = true;
                                            }

                                            Thread.Sleep(2000);
                                        }
                                        catch (Exception ex)
                                        {
                                            LogManager.Log(ex, "Package installation failed.");
                                            installedPackage.State = PackageInstallationState.Failed;
                                            installedPackage.FailedReason = ex.FlattenMessage();
                                            OnPackageRuns(att.Name, installedPackage.State, installedPackage.Type);
                                            continue;
                                        }
                                    }
                                }
                                else
                                {
                                    LogManager.Log("Package is already installed.");
                                }
                            }
                            catch (Exception ex)
                            {
                                LogManager.Log(ex, "Error in handling the package!");
                                continue;
                            }
                        }
                    }
                    catch (Exception ex)
                    {
                        LogManager.Log(ex, "Error investigating assembly!");
                        continue;
                    }
                }

                //Save package file.
                LogManager.Log("Running packages has completed. Saving packages config file.");

                try
                {
                    SavePackagesConfig();
                }
                catch (Exception ex)
                {
                    LogManager.Log(ex, "Error saving packages file!");
                }

                return result;
            });
        }

        public Task<bool> IsPackageInstallationRequired(PackageType type, String packagesFolder)
        {
            return Task.Factory.StartNew<bool>(() =>
            {
                LogManager.Log("Checking if any package installation is required...");

                _packagesFile = GetPackagesFile().Result;

                //Get all packages in folder.
                foreach (var packageFile in Directory.GetFiles(packagesFolder, "*.dll"))
                {
                    LogManager.Log($"Loading assembly '{Path.GetFileName(packageFile)}'...");

                    Assembly asm;

                    //Load assembly and investigate for types based on package type.
                    try
                    {
                        asm = Assembly.LoadFile(packageFile);
                    }
                    catch (Exception ex)
                    {
                        LogManager.Log(ex, "Error loading assembly!");
                        continue;
                    }

                    try
                    {
                        foreach (var packageType in asm.GetTypes().Where(
                            x => typeof(IPPCPackage).IsAssignableFrom(x) &&
                            x.GetCustomAttribute<PPCPackageAttribute>() != null &&
                            x.GetCustomAttribute<PPCPackageAttribute>().Type == type))
                        {
                            LogManager.Log($"Checking package '{packageType.FullName}'...");

                            try
                            {
                                //Getting installed package from file.
                                var installedPackage = _packagesFile.PackageInstallations.SingleOrDefault(x => x.PackageName == packageType.FullName);

                                //Check if requires installation.
                                if (installedPackage == null || installedPackage.State != PackageInstallationState.Installed)
                                {
                                    if (installedPackage == null)
                                    {
                                        LogManager.Log("Package was never installed.");
                                        return true;
                                    }
                                    else
                                    {
                                        LogManager.Log($"Package installation state is '{installedPackage.State}' due to {installedPackage.FailedReason}");
                                        return true;
                                    }
                                }
                                else
                                {
                                    LogManager.Log("Package is already installed.");
                                }
                            }
                            catch (Exception ex)
                            {
                                LogManager.Log(ex, "Error in handling the package!");
                                continue;
                            }
                        }
                    }
                    catch (Exception ex)
                    {
                        LogManager.Log(ex, "Error investigating assembly!");
                        continue;
                    }
                }

                LogManager.Log("No packages to install.");

                return false;
            });
        }

        protected virtual void OnPackageRuns(String packageName, PackageInstallationState state, PackageType type)
        {
            PackageStateChanged?.Invoke(this, new PackageStateChangedEventArgs()
            {
                PackageName = packageName,
                State = state,
                PackageType = type,
            });
        }
    }
}