Install.cs

using System;
using System.IO;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using Waher.Security;
using Waher.Service.IoTBroker.Provisioning.SoftwareUpdates;
 
namespace Waher.Service.IoTBroker.Admin.Software
{
    public class Install : CommandRegEx
    {
        public Install()
            : base(@"^(upgrade|update|install)(\s+nobackup)?(\s+[^\s]+){0,2}$")
        {
        }
 
        public override string Name => "Install";
 
        public override string[] Aliases => new string[] { "Update", "Upgrade" };
 
        public override async Task Execute(ChatState State, string[] Arguments, string OrgMessage, Match Details,
            ResponseCallbackHandler ResponseCallback)
        {
            int i = 0;
            int c = Arguments.Length;
            bool Backup = true;
 
            if (i < c && string.Compare(Arguments[i], "nobackup", true) == 0)
            {
                i++;
                Backup = false;
            }
 
            if (i >= c)
                await XmppServerModule.Instance.UpdateSoftware(null, State, Backup, ResponseCallback);
            else
            {
                string PackageFileName = Arguments[i++];
                Package Package = await Provisioning.ProvisioningComponent.GetPackage(PackageFileName);
                if (Package is null)
                {
                    await ResponseCallback("Package not found.", string.Empty);
                    return;
                }
 
                string FullFileName = Path.Combine(XmppServerModule.PackagesFolder, PackageFileName);
                if (!File.Exists(FullFileName))
                {
                    await ResponseCallback("Package file not found.", string.Empty);
                    return;
                }
 
                if (Package.FileName == BrokerPackage.FileName)
                {
                    await ResponseCallback("Reserved package.", string.Empty);
                    return;
                }
 
                byte[] PublicKey;
                byte[] AesKey;
 
                if (i < c)
                {
                    string Key = Arguments[i++];
 
                    if (Key.Length != 76 && Key.Length != 76 + 32)
                    {
                        await ResponseCallback("Invalid key.", string.Empty);
                        return;
                    }
 
                    PublicKey = Convert.FromBase64String(Key.Substring(0, 76));
                    AesKey = Key.Length == 76 ? null : Hashes.StringToBinary(Key.Substring(76));
                }
                else
                {
                    PublicKey = Package.PublicKey;
                    AesKey = Package.AesKey;
                }
 
                using (FileStream fs = File.OpenRead(FullFileName))
                {
                    if (!XmppServerModule.ValidatePackage(fs, PublicKey, Package.Signature))
                    {
                        await ResponseCallback("Invalid public key.", string.Empty);
                        return;
                    }
                }
 
                Package.Installed = Package.Published;
                Package.PublicKey = PublicKey;
                Package.AesKey = AesKey;
                Package.ContentOnly = XmppServerModule.IsContentPackage(Package);
 
                await Persistence.Database.Update(Package);
 
                await XmppServerModule.Instance.UpdateSoftware(Package, State, Backup, ResponseCallback);
            }
        }
 
        public override HelpItem[] GetHelp()
        {
            return new HelpItem[]
            {
                new HelpItem("upgrade", "Upgrades the software of the service, including installed packages. Performas a backup first."),
                new HelpItem("upgrade PACKAGE", "Upgrades a specific software package that is already installed. Performas a backup first."),
                new HelpItem("install PACKAGE KEY", "install a specific software package that is downloaded but not installed. Performas a backup first."),
                new HelpItem("upgrade nobackup", "Upgrades the software of the service, including installed packages. Does not perform a backup first."),
                new HelpItem("upgrade nobackup PACKAGE", "Upgrades a specific software package that is already installed. Does not perform a backup first."),
                new HelpItem("install nobackup PACKAGE KEY", "install a specific software package that is downloaded but not installed. Does not perform a backup first."),
            };
        }
    }
}