StartExport.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.IO.Compression;
using System.Security.Cryptography;
using System.Text;
using System.Threading.Tasks;
using System.Xml;
using Waher.Content;
using Waher.Content.Binary;
using Waher.Content.Markdown;
using Waher.Content.Text;
using Waher.Content.Xml;
using Waher.Events;
using Waher.IoTGateway.WebResources.ExportFormats;
using Waher.Networking.HTTP;
using Waher.Networking.XMPP;
using Waher.Networking.XMPP.HttpFileUpload;
using Waher.Persistence;
using Waher.Persistence.XmlLedger;
using Waher.Runtime.Profiling;
 
namespace Waher.IoTGateway.WebResources
{
    public class StartExport : HttpSynchronousResource, IHttpPostMethod
    {
        internal static Aes aes = GetCryptoProvider();
 
        public StartExport()
            : base("/StartExport")
        {
        }
 
        private static Aes GetCryptoProvider()
        {
            Aes Result = Aes.Create();
 
            Result.BlockSize = 128;
            Result.KeySize = 256;
            Result.Mode = CipherMode.CBC;
            Result.Padding = PaddingMode.None;
 
            return Result;
        }
 
        public override bool HandlesSubPaths => false;
 
        public override bool UserSessions => true;
 
        public bool AllowsPOST => true;
 
        public async Task POST(HttpRequest Request, HttpResponse Response)
        {
            try
            {
                Gateway.AssertUserAuthenticated(Request, "Admin.Data.Backup");
 
                if (!Request.HasData || !(await Request.DecodeDataAsync() is Dictionary<string, object> RequestObj))
                    throw new UnsupportedMediaTypeException("Invalid request.");
 
                if (!RequestObj.TryGetValue("TypeOfFile", out object Obj) || !(Obj is string TypeOfFile))
                    throw new BadRequestException("Missing: TypeOfFile");
 
                if (!RequestObj.TryGetValue("Database", out Obj) || !(Obj is bool Database))
                    throw new BadRequestException("Missing: Database");
 
                if (!RequestObj.TryGetValue("Ledger", out Obj) || !(Obj is bool Ledger))
                    Ledger = false;
 
                if (!RequestObj.TryGetValue("WebContent", out Obj) || !(Obj is bool WebContent))
                    throw new BadRequestException("Missing: WebContent");
 
                if (!RequestObj.TryGetValue("OnlySelectedCollections", out Obj) || !(Obj is bool OnlySelectedCollections))
                    throw new BadRequestException("Missing: OnlySelectedCollections");
 
                if (!RequestObj.TryGetValue("selectedCollections", out Obj) || !(Obj is Array SelectedCollections))
                    throw new BadRequestException("Missing: selectedCollections");
 
                if (!RequestObj.TryGetValue("exportOnly", out Obj) || !(Obj is bool ExportOnly))
                    throw new BadRequestException("Missing: exportOnly");
 
                ExportInfo ExportInfo = await GetExporter(TypeOfFile, OnlySelectedCollections, SelectedCollections);
                Task T;
 
                lock (synchObject)
                {
                    if (exporting)
                    {
                        Response.StatusCode = 409;
                        Response.StatusMessage = "Conflict";
                        Response.ContentType = PlainTextCodec.DefaultContentType;
                        T = Response.Write("Export is underway.");
                    }
                    else
                    {
                        exporting = true;
                        T = null;
                    }
                }
 
                if (!(T is null))
                {
                    await T;
                    return;
                }
 
                if (!ExportOnly)
                {
                    Export.ExportType = TypeOfFile;
                    Export.ExportDatabase = Database;
                    Export.ExportLedger = Ledger;
                    Export.ExportWebContent = WebContent;
                }
 
                List<string> Folders = new List<string>();
 
                foreach (Export.FolderCategory FolderCategory in Export.GetRegisteredFolders())
                {
                    if (RequestObj.TryGetValue(FolderCategory.CategoryId, out Obj) && Obj is bool b)
                    {
                        if (!ExportOnly)
                            await Export.SetExportFolderAsync(FolderCategory.CategoryId, b);
 
                        if (b)
                            Folders.AddRange(FolderCategory.Folders);
                    }
                }
 
                Task _ = DoExport(ExportInfo, Database, Ledger, WebContent, Folders.ToArray());
 
                Response.StatusCode = 200;
                Response.ContentType = PlainTextCodec.DefaultContentType;
                await Response.Write(ExportInfo.LocalBackupFileName);
            }
            catch (Exception ex)
            {
                await Response.SendResponse(ex);
            }
        }
 
        private static bool exporting = false;
        private static readonly object synchObject = new object();
 
        internal class ExportInfo
        {
            public string LocalBackupFileName;
            public string LocalKeyFileName;
            public string FullBackupFileName;
            public string FullKeyFileName;
            public IExportFormat Exporter;
        }
 
        internal static async Task<ExportInfo> GetExporter(string TypeOfFile, bool OnlySelectedCollections, Array SelectedCollections)
        {
            ExportInfo Result = new ExportInfo();
            string BasePath = await Export.GetFullExportFolderAsync();
 
            if (!Directory.Exists(BasePath))
                Directory.CreateDirectory(BasePath);
 
            BasePath += Path.DirectorySeparatorChar;
 
            Result.FullBackupFileName = BasePath + DateTime.Now.ToString("yyyy-MM-dd HH_mm_ss");
 
            switch (TypeOfFile)
            {
                case "XML":
                    Result.FullBackupFileName = GetUniqueFileName(Result.FullBackupFileName, ".xml");
                    FileStream fs = new FileStream(Result.FullBackupFileName, FileMode.Create, FileAccess.Write);
                    DateTime Created = File.GetCreationTime(Result.FullBackupFileName);
                    XmlWriterSettings Settings = XML.WriterSettings(true, false);
                    Settings.Async = true;
                    XmlWriter XmlOutput = XmlWriter.Create(fs, Settings);
                    Result.LocalBackupFileName = Result.FullBackupFileName.Substring(BasePath.Length);
                    Result.Exporter = new XmlExportFormat(Result.LocalBackupFileName, Created, XmlOutput, fs, OnlySelectedCollections, SelectedCollections);
                    break;
 
                case "Binary":
                    Result.FullBackupFileName = GetUniqueFileName(Result.FullBackupFileName, ".bin");
                    fs = new FileStream(Result.FullBackupFileName, FileMode.Create, FileAccess.Write);
                    Created = File.GetCreationTime(Result.FullBackupFileName);
                    Result.LocalBackupFileName = Result.FullBackupFileName.Substring(BasePath.Length);
                    Result.Exporter = new BinaryExportFormat(Result.LocalBackupFileName, Created, fs, fs, OnlySelectedCollections, SelectedCollections);
                    break;
 
                case "Compressed":
                    Result.FullBackupFileName = GetUniqueFileName(Result.FullBackupFileName, ".gz");
                    fs = new FileStream(Result.FullBackupFileName, FileMode.Create, FileAccess.Write);
                    Created = File.GetCreationTime(Result.FullBackupFileName);
                    Result.LocalBackupFileName = Result.FullBackupFileName.Substring(BasePath.Length);
                    GZipStream gz = new GZipStream(fs, CompressionLevel.Optimal, false);
                    Result.Exporter = new BinaryExportFormat(Result.LocalBackupFileName, Created, gz, fs, OnlySelectedCollections, SelectedCollections);
                    break;
 
                case "Encrypted":
                    Result.FullBackupFileName = GetUniqueFileName(Result.FullBackupFileName, ".bak");
                    fs = new FileStream(Result.FullBackupFileName, FileMode.Create, FileAccess.Write);
                    Created = File.GetCreationTime(Result.FullBackupFileName);
                    Result.LocalBackupFileName = Result.FullBackupFileName.Substring(BasePath.Length);
 
                    byte[] Key = Gateway.NextBytes(32);
                    byte[] IV = Gateway.NextBytes(16);
 
                    ICryptoTransform AesTransform = aes.CreateEncryptor(Key, IV);
                    CryptoStream cs = new CryptoStream(fs, AesTransform, CryptoStreamMode.Write);
 
                    gz = new GZipStream(cs, CompressionLevel.Optimal, false);
                    Result.Exporter = new BinaryExportFormat(Result.LocalBackupFileName, Created, gz, fs, cs, 32, OnlySelectedCollections, SelectedCollections);
 
                    string BasePath2 = await Export.GetFullKeyExportFolderAsync();
 
                    if (!Directory.Exists(BasePath2))
                        Directory.CreateDirectory(BasePath2);
 
                    BasePath2 += Path.DirectorySeparatorChar;
                    Result.LocalKeyFileName = Result.LocalBackupFileName.Replace(".bak", ".key");
                    Result.FullKeyFileName = BasePath2 + Result.LocalKeyFileName;
 
                    using (XmlOutput = XmlWriter.Create(Result.FullKeyFileName, XML.WriterSettings(true, false)))
                    {
                        XmlOutput.WriteStartDocument();
                        XmlOutput.WriteStartElement("KeyAes256", XmlFileLedger.Namespace);
                        XmlOutput.WriteAttributeString("key", Convert.ToBase64String(Key));
                        XmlOutput.WriteAttributeString("iv", Convert.ToBase64String(IV));
                        XmlOutput.WriteEndElement();
                        XmlOutput.WriteEndDocument();
                    }
 
                    long Size;
 
                    try
                    {
                        using (fs = File.OpenRead(Result.FullKeyFileName))
                        {
                            Size = fs.Length;
                        }
                    }
                    catch (Exception ex)
                    {
                        Log.Exception(ex);
                        Size = 0;
                    }
 
                    Created = File.GetCreationTime(Result.FullKeyFileName);
 
                    ExportFormat.UpdateClientsFileUpdated(Result.LocalKeyFileName, Size, Created);
                    break;
 
                default:
                    throw new NotSupportedException("Unsupported file type.");
            }
 
            return Result;
        }
 
        public static string GetUniqueFileName(string Base, string Extension)
        {
            string Suffix = string.Empty;
            string s;
            int i = 1;
 
            while (true)
            {
                s = Base + Suffix + Extension;
                if (!File.Exists(s))
                    return s;
 
                i++;
                Suffix = " (" + i.ToString() + ")";
            }
        }
 
        internal static async Task DoExport(ExportInfo ExportInfo, bool Database, bool Ledger, bool WebContent, string[] Folders)
        {
            Profiler Profiler = new Profiler("Export", ProfilerThreadType.Sequential);
            Profiler.Start();
            Profiler.NewState("Start");
 
            try
            {
                List<KeyValuePair<string, object>> Tags = new List<KeyValuePair<string, object>>()
                {
                    new KeyValuePair<string, object>("Database", Database),
                    new KeyValuePair<string, object>("Ledger", Ledger)
                };
 
                foreach (string Folder in Folders)
                    Tags.Add(new KeyValuePair<string, object>(Folder, true));
 
                Log.Informational("Starting export.", ExportInfo.Exporter.FileName, Tags.ToArray());
 
                await ExportInfo.Exporter.Start();
 
                if (Database)
                {
                    Profiler.NewState("Database");
 
                    StringBuilder Temp = new StringBuilder();
                    string[] RepairedCollections;
 
                    using (XmlWriter w = XmlWriter.Create(Temp, XML.WriterSettings(true, true)))
                    {
                        RepairedCollections = await Persistence.Database.Analyze(w, Path.Combine(Gateway.AppDataFolder,
                            "Transforms", "DbStatXmlToHtml.xslt"), Path.Combine(Gateway.RootFolder, "Data"), false, true,
                            Profiler.CreateThread("Analyze", ProfilerThreadType.Sequential));
                    }
 
                    if (RepairedCollections.Length > 0)
                    {
                        string Xml = Temp.ToString();
                        string ReportFileName = Path.Combine(Path.GetDirectoryName(ExportInfo.FullBackupFileName),
                            "AutoRepair " + DateTime.Now.ToString("yyyy-MM-ddTHH.mm.ss.ffffff") + ".xml");
                        await Resources.WriteAllTextAsync(ReportFileName, Xml);
                    }
 
                    SortedDictionary<string, bool> CollectionsToExport = new SortedDictionary<string, bool>();
 
                    if (ExportInfo.Exporter.CollectionNames is null)
                    {
                        foreach (string Collection in await Persistence.Database.GetCollections())
                            CollectionsToExport[Collection] = true;
                    }
                    else
                    {
                        foreach (string Collection in ExportInfo.Exporter.CollectionNames)
                            CollectionsToExport[Collection] = true;
                    }
 
                    foreach (string Collection in Persistence.Database.GetExcludedCollections())
                        CollectionsToExport.Remove(Collection);
 
                    string[] ToExport = new string[CollectionsToExport.Count];
                    CollectionsToExport.Keys.CopyTo(ToExport, 0);
 
                    await Persistence.Database.Export(ExportInfo.Exporter, ToExport,
                        Profiler.CreateThread("Database", ProfilerThreadType.Sequential));
                }
 
                if (Ledger && Persistence.Ledger.HasProvider)
                {
                    Profiler.NewState("Ledger");
 
                    LedgerExportRestriction Restricion = new LedgerExportRestriction()
                    {
                        CollectionNames = ExportInfo.Exporter.CollectionNames
                    };
 
                    await Persistence.Ledger.Export(ExportInfo.Exporter, Restricion,
                        Profiler.CreateThread("Ledger", ProfilerThreadType.Sequential));
                }
 
                if (WebContent || Folders.Length > 0)
                {
                    Profiler.NewState("Files");
 
                    await ExportInfo.Exporter.StartFiles();
                    try
                    {
                        string[] FileNames;
                        string Folder2;
 
                        if (WebContent)
                        {
                            string ConfigFileName = Gateway.ConfigFilePath;
                            if (File.Exists(ConfigFileName))
                                await ExportFile(ConfigFileName, ExportInfo.Exporter);
 
                            FileNames = Directory.GetFiles(Gateway.RootFolder, "*.*", SearchOption.TopDirectoryOnly);
 
                            foreach (string FileName in FileNames)
                                await ExportFile(FileName, ExportInfo.Exporter);
 
                            Export.FolderCategory[] ExportFolders = Export.GetRegisteredFolders();
 
                            foreach (string Folder in Directory.GetDirectories(Gateway.RootFolder, "*.*", SearchOption.TopDirectoryOnly))
                            {
                                bool IsWebContent = true;
 
                                foreach (Export.FolderCategory FolderCategory in ExportFolders)
                                {
                                    foreach (string Folder3 in FolderCategory.Folders)
                                    {
                                        if (string.Compare(Folder3, Folder, true) == 0)
                                        {
                                            IsWebContent = false;
                                            break;
                                        }
                                    }
 
                                    if (!IsWebContent)
                                        break;
                                }
 
                                if (IsWebContent)
                                {
                                    FileNames = Directory.GetFiles(Folder, "*.*", SearchOption.AllDirectories);
 
                                    foreach (string FileName in FileNames)
                                        await ExportFile(FileName, ExportInfo.Exporter);
                                }
                            }
                        }
 
                        foreach (string Folder in Folders)
                        {
                            if (Directory.Exists(Folder2 = Path.Combine(Gateway.RootFolder, Folder)))
                            {
                                FileNames = Directory.GetFiles(Folder2, "*.*", SearchOption.AllDirectories);
 
                                foreach (string FileName in FileNames)
                                    await ExportFile(FileName, ExportInfo.Exporter);
                            }
                        }
                    }
                    finally
                    {
                        await ExportInfo.Exporter.EndFiles();
                    }
                }
 
                Log.Informational("Export successfully completed.", ExportInfo.Exporter.FileName);
            }
            catch (Exception ex)
            {
                Profiler.Exception(ex);
 
                Log.Exception(ex);
 
                string[] Tabs = ClientEvents.GetTabIDsForLocation("/Settings/Backup.md");
                await ClientEvents.PushEvent(Tabs, "BackupFailed", "{\"fileName\":\"" + CommonTypes.JsonStringEncode(ExportInfo.Exporter.FileName) +
                    "\", \"message\": \"" + CommonTypes.JsonStringEncode(ex.Message) + "\"}", true, "User");
            }
            finally
            {
                Profiler.NewState("End");
 
                try
                {
                    await ExportInfo.Exporter.End();
                    ExportInfo.Exporter.Dispose();
                }
                catch (Exception ex)
                {
                    Profiler.Exception(ex);
                    Log.Exception(ex);
                }
 
                lock (synchObject)
                {
                    exporting = false;
                }
 
                Profiler.NewState("Upload");
 
                ProfilerThread Thread = Profiler.CreateThread("Upload", ProfilerThreadType.StateMachine);
 
                await UploadBackupFile(ExportInfo.LocalBackupFileName, ExportInfo.FullBackupFileName, false, Thread);
                if (!string.IsNullOrEmpty(ExportInfo.FullKeyFileName))
                    await UploadBackupFile(ExportInfo.LocalKeyFileName, ExportInfo.FullKeyFileName, true, Thread);
 
                Profiler.Stop();
 
                //string Uml = Profiler.ExportPlantUml(TimeUnit.DynamicPerProfiling);
                //string UmlFileName = Path.ChangeExtension(ExportInfo.FullBackupFileName, "uml");
                //long UmlFileSize;
                //
                //await Resources.WriteAllTextAsync(UmlFileName, Uml);
                //
                //using (FileStream fs = File.OpenRead(UmlFileName))
                //{
                //  UmlFileSize = fs.Length;
                //}
                //
                //ExportFormat.UpdateClientsFileUpdated(Path.ChangeExtension(ExportInfo.LocalBackupFileName, "uml"), UmlFileSize, DateTime.Now);
            }
        }
 
        private static async Task<bool> ExportFile(string FileName, IExportFormat Output)
        {
            using (FileStream fs = File.OpenRead(FileName))
            {
                if (FileName.StartsWith(Gateway.AppDataFolder))
                    FileName = FileName.Substring(Gateway.AppDataFolder.Length);
 
                if (!await Output.ExportFile(FileName, fs))
                    return false;
            }
 
            return await Output.UpdateClient(false);
        }
 
        private static Task UploadBackupFile(string LocalFileName, string FullFileName, bool IsKey, ProfilerThread Thread)
        {
            return UploadBackupFile(new BackupInfo()
            {
                LocalFileName = LocalFileName,
                FullFileName = FullFileName,
                IsKey = IsKey,
                Thread = Thread,
                Rescheduled = false
            });
        }
 
        private class BackupInfo
        {
            public string LocalFileName;
            public string FullFileName;
            public bool IsKey;
            public bool Rescheduled;
            public Dictionary<string, bool> Recipients = null;
            public ProfilerThread Thread;
        }
 
        private static async Task UploadBackupFile(object State)
        {
            BackupInfo BackupInfo = (BackupInfo)State;
            bool Reschedule = false;
            string Msg;
 
            try
            {
                if (!File.Exists(BackupInfo.FullFileName))
                {
                    Log.Warning(Msg = "Backup file has been removed. Upload cancelled.", BackupInfo.LocalFileName);
 
                    if (BackupInfo.Rescheduled)
                        await Gateway.SendNotification(Msg);
 
                    return;
                }
                else if (Gateway.XmppClient.State != XmppState.Connected)
                    Reschedule = true;
                else
                {
                    if (BackupInfo.Recipients is null)
                    {
                        BackupInfo.Recipients = new Dictionary<string, bool>();
 
                        string[] Hosts = await Export.GetKeyHostsAsync();
 
                        if (BackupInfo.IsKey && !(Hosts is null))
                        {
                            foreach (string Host in Hosts)
                                BackupInfo.Recipients[Host] = false;
                        }
 
                        Hosts = await Export.GetBackupHostsAsync();
 
                        if (!BackupInfo.IsKey && !(Hosts is null))
                        {
                            foreach (string Host in Hosts)
                                BackupInfo.Recipients[Host] = false;
                        }
                    }
 
                    LinkedList<string> Recipients = new LinkedList<string>();
                    string Recipient;
 
                    foreach (KeyValuePair<string, bool> P in BackupInfo.Recipients)
                    {
                        if (!P.Value)
                            Recipients.AddLast(P.Key);
                    }
 
                    while (!((Recipient = Recipients.First?.Value) is null))
                    {
                        Recipients.RemoveFirst();
 
                        if (Recipient.IndexOf('@') >= 0)
                        {
                            RosterItem Item = Gateway.XmppClient[Recipient];
                            if (Item is null)
                                continue;
 
                            if (!Item.HasLastPresence || !Item.LastPresence.IsOnline)
                            {
                                Reschedule = true;
                                continue;
                            }
                        }
 
                        try
                        {
                            BackupInfo.Thread?.NewState("Discover_" + Recipient);
 
                            using (HttpFileUploadClient UploadClient = new HttpFileUploadClient(Gateway.XmppClient, Recipient, null))
                            {
                                await UploadClient.DiscoverAsync(Recipient);
 
                                if (UploadClient.HasSupport)
                                {
                                    using (FileStream fs = File.OpenRead(BackupInfo.FullFileName))
                                    {
                                        BackupInfo.Thread?.NewState("Prepare_" + Recipient);
 
                                        StringBuilder Xml = new StringBuilder();
                                        long FileSize = fs.Length;
 
                                        Xml.Append("<prepare xmlns='http://waher.se/Schema/Backups.xsd' filename='");
                                        Xml.Append(XML.Encode(BackupInfo.LocalFileName));
                                        Xml.Append("' size='");
                                        Xml.Append(FileSize.ToString());
                                        Xml.Append("' content-type='application/octet-stream'/>");
 
                                        BackupInfo.Thread?.NewState("Get_Slot_" + Recipient);
 
                                        await UploadClient.Client.IqSetAsync(UploadClient.FileUploadJid, Xml.ToString());
                                        // Empty response expected. Errors cause an exception to be raised.
 
                                        HttpFileUploadEventArgs e2 = await UploadClient.RequestUploadSlotAsync(BackupInfo.LocalFileName,
                                            BinaryCodec.DefaultContentType, FileSize, false);
 
                                        if (!e2.Ok)
                                            throw (e2.StanzaError ?? new XmppException("Unable to get HTTP upload slot for backup file."));
 
                                        BackupInfo.Thread?.NewState("Upload_" + Recipient);
 
                                        if (BackupInfo.IsKey)
                                            Log.Informational("Uploading key file to " + Recipient + ".", BackupInfo.LocalFileName);
                                        else
                                            Log.Informational("Uploading backup file to " + Recipient + ".", BackupInfo.LocalFileName);
 
                                        await e2.PUT(fs, BinaryCodec.DefaultContentType, 60 * 60 * 1000);   // 1h timeout
 
                                        if (BackupInfo.IsKey)
                                            Log.Informational("Key file uploaded to " + Recipient + ".", BackupInfo.LocalFileName);
                                        else
                                            Log.Informational("Backup file uploaded to " + Recipient + ".", BackupInfo.LocalFileName);
                                    }
                                }
                            }
                        }
                        catch (Exception ex)
                        {
                            BackupInfo.Thread?.Exception(ex, BackupInfo.LocalFileName);
                            Log.Exception(ex);
                            Reschedule = true;
 
                            await Gateway.SendNotification("Unable to upload backup to " + MarkdownDocument.Encode(Recipient) + 
                                ".\r\n\r\n" + MarkdownDocument.Encode(ex.Message));
                        }
                    }
                }
            }
            catch (Exception ex)
            {
                BackupInfo.Thread?.Exception(ex);
                Log.Exception(ex);
                Reschedule = true;
            }
            finally
            {
                if (Reschedule)
                {
                    BackupInfo.Thread?.NewState("Reschedule");
                    BackupInfo.Rescheduled = true;
 
                    Gateway.ScheduleEvent(UploadBackupFile, DateTime.Now.AddMinutes(15), BackupInfo);
                }
                else if (BackupInfo.Rescheduled)
                {
                    await Gateway.SendNotification("Backup file has been successfully been uploaded. The initial attempt to upload the backup file failed, but a sequent rescheduled upload succeeded.");
                    BackupInfo.Rescheduled = false;
                }
 
                BackupInfo.Thread = null;
            }
        }
 
    }
}