XmlExportFormat.cs

using System;
using System.IO;
using System.Threading.Tasks;
using System.Xml;
using Waher.Content;
using Waher.Content.Xml;
using Waher.Events;
using Waher.Persistence;
using Waher.Persistence.XmlLedger;
using Waher.Runtime.Inventory;
 
namespace Waher.IoTGateway.WebResources.ExportFormats
{
    public class XmlExportFormat : ExportFormat
    {
        private XmlWriter output;
        private bool exportCollection = false;
        private bool inCollection = false;
        private bool inMetaData = false;
 
        public XmlExportFormat(string FileName, DateTime Created, XmlWriter Output, FileStream File,
            bool OnlySelectedCollections, Array SelectedCollections)
            : base(FileName, Created, File, OnlySelectedCollections, SelectedCollections)
        {
            this.output = Output;
        }
 
        public override void Dispose()
        {
            if (!(this.output is null))
            {
                this.output.Flush();
                this.output.Close();
                this.output.Dispose();
                this.output = null;
            }
 
            base.Dispose();
        }
 
        public override async Task<bool> Start()
        {
            await this.output.WriteStartDocumentAsync();
            await this.output.WriteStartElementAsync(string.Empty, "Export", XmlFileLedger.Namespace);
 
            return true;
        }
 
        public override async Task<bool> End()
        {
            await this.output.WriteEndElementAsync();
            await this.output.WriteEndDocumentAsync();
            
            return await this.UpdateClient(true);
        }
 
        public override async Task<bool> StartDatabase()
        {
            await this.output.WriteStartElementAsync(string.Empty, "Database", XmlFileLedger.Namespace);
            return true;
        }
 
        public override async Task<bool> EndDatabase()
        {
            await this.output.WriteEndElementAsync();
            
            return await this.UpdateClient(false);
        }
 
        public override async Task<bool> StartLedger()
        {
            await this.output.WriteStartElementAsync(string.Empty, "Ledger", XmlFileLedger.Namespace);
            return true;
        }
 
        public override async Task<bool> EndLedger()
        {
            await this.output.WriteEndElementAsync();
            return await this.UpdateClient(false);
        }
 
        public override async Task<bool> StartCollection(string CollectionName)
        {
            this.inCollection = true;
            if (this.exportCollection = this.ExportCollection(CollectionName))
            {
                await this.output.WriteStartElementAsync(string.Empty, "Collection", XmlFileLedger.Namespace);
                await this.output.WriteAttributeStringAsync(string.Empty, "name", string.Empty, CollectionName);
            }
 
            return true;
        }
 
        public override async Task<bool> EndCollection()
        {
            this.inCollection = false;
            if (this.exportCollection)
                await this.output.WriteEndElementAsync();
            
            return true;
        }
 
        public override async Task<bool> StartIndex()
        {
            if (this.exportCollection)
                await this.output.WriteStartElementAsync(string.Empty, "Index", XmlFileLedger.Namespace);
            
            return true;
        }
 
        public override async Task<bool> EndIndex()
        {
            if (this.exportCollection)
                await this.output.WriteEndElementAsync();
 
            return true;
        }
 
        public override async Task<bool> ReportIndexField(string FieldName, bool Ascending)
        {
            if (this.exportCollection)
            {
                await this.output.WriteStartElementAsync(string.Empty, "Field", XmlFileLedger.Namespace);
                await this.output.WriteAttributeStringAsync(string.Empty, "name", string.Empty, FieldName);
                await this.output.WriteAttributeStringAsync(string.Empty, "ascending", string.Empty, CommonTypes.Encode(Ascending));
                await this.output.WriteEndElementAsync();
            }
 
            return true;
        }
 
        public override async Task<bool> StartBlock(string BlockID)
        {
            if (this.exportCollection)
            {
                await this.output.WriteStartElementAsync(string.Empty, "Block", XmlFileLedger.Namespace);
                await this.output.WriteAttributeStringAsync(string.Empty, "id", string.Empty, BlockID);
            }
 
            return true;
        }
 
        public override async Task<bool> EndBlock()
        {
            if (this.exportCollection)
            {
                if (this.inMetaData)
                {
                    await this.output.WriteEndElementAsync();
                    this.inMetaData = false;
                }
 
                await this.output.WriteEndElementAsync();
            }
 
            return true;
        }
 
        public override async Task<bool> BlockMetaData(string Key, object Value)
        {
            if (this.exportCollection)
            {
                if (!this.inMetaData)
                {
                    await this.output.WriteStartElementAsync(string.Empty, "MetaData", XmlFileLedger.Namespace);
                    this.inMetaData = true;
                }
 
                await this.ReportProperty(Key, Value);
            }
 
            return true;
        }
 
        public override async Task<string> StartObject(string ObjectId, string TypeName)
        {
            if (this.exportCollection)
            {
                await this.output.WriteStartElementAsync(string.Empty, "Obj", XmlFileLedger.Namespace);
                await this.output.WriteAttributeStringAsync(string.Empty, "id", string.Empty, ObjectId);
                await this.output.WriteAttributeStringAsync(string.Empty, "type", string.Empty, TypeName);
            }
 
            return ObjectId;
        }
 
        public override async Task<bool> EndObject()
        {
            if (this.exportCollection)
            {
                await this.output.WriteEndElementAsync();
                return await this.UpdateClient(false);
            }
            else
                return true;
        }
 
        public override async Task<bool> StartEntry(string ObjectId, string TypeName, EntryType EntryType, DateTimeOffset EntryTimestamp)
        {
            if (this.exportCollection)
            {
                if (this.inMetaData)
                {
                    await this.output.WriteEndElementAsync();
                    this.inMetaData = false;
                }
 
                await this.output.WriteStartElementAsync(string.Empty, EntryType.ToString(), XmlFileLedger.Namespace);
                await this.output.WriteAttributeStringAsync(string.Empty, "id", string.Empty, ObjectId);
                await this.output.WriteAttributeStringAsync(string.Empty, "type", string.Empty, TypeName);
                await this.output.WriteAttributeStringAsync(string.Empty, "ts", string.Empty, XML.Encode(EntryTimestamp));
            }
 
            return true;
        }
 
        public override async Task<bool> EndEntry()
        {
            if (this.exportCollection)
            {
                await this.output.WriteEndElementAsync();
                return await this.UpdateClient(false);
            }
            else
                return true;
        }
 
        public override async Task<bool> CollectionCleared(DateTimeOffset EntryTimestamp)
        {
            if (this.exportCollection)
            {
                if (this.inMetaData)
                {
                    await this.output.WriteEndElementAsync();
                    this.inMetaData = false;
                }
 
                await this.output.WriteStartElementAsync(string.Empty, nameof(EntryType.Clear), XmlFileLedger.Namespace);
                await this.output.WriteAttributeStringAsync(string.Empty, "ts", string.Empty, XML.Encode(EntryTimestamp));
            }
 
            return true;
        }
 
        public override async Task<bool> ReportProperty(string PropertyName, object PropertyValue)
        {
            if (this.exportCollection)
                await XmlFileLedger.ReportProperty(this.output, PropertyName, PropertyValue, XmlFileLedger.Namespace);
            
            return true;
        }
 
        public override async Task<bool> ReportError(string Message)
        {
            if (!this.inCollection || this.exportCollection)
                await this.output.WriteElementStringAsync(string.Empty, "Error", XmlFileLedger.Namespace, Message);
            
            return true;
        }
 
        public override async Task<bool> ReportException(Exception Exception)
        {
            if (!this.inCollection || this.exportCollection)
            {
                await this.output.WriteStartElementAsync(string.Empty, "Exception", XmlFileLedger.Namespace);
                await this.output.WriteAttributeStringAsync(string.Empty, "message", string.Empty, Exception.Message);
                this.output.WriteElementString("StackTrace", Log.CleanStackTrace(Exception.StackTrace));
 
                if (Exception is AggregateException AggregateException)
                {
                    foreach (Exception ex in AggregateException.InnerExceptions)
                        await this.ReportException(ex);
                }
                else if (!(Exception.InnerException is null))
                    await this.ReportException(Exception.InnerException);
 
                await this.output.WriteEndElementAsync();
            }
 
            return true;
        }
 
        public override async Task<bool> StartFiles()
        {
            await this.output.WriteStartElementAsync(string.Empty, "Files", XmlFileLedger.Namespace);
            return true;
        }
 
        public override async Task<bool> EndFiles()
        {
            await this.output.WriteEndElementAsync();
            return true;
        }
 
        public override async Task<bool> ExportFile(string FileName, Stream File)
        {
            await this.output.WriteStartElementAsync(string.Empty, "File", XmlFileLedger.Namespace);
 
            await this.output.WriteAttributeStringAsync(string.Empty, "fileName", string.Empty, FileName);
 
            byte[] Buf = null;
            long c = File.Length;
            long i = 0;
            long d;
            int j;
 
            while (i < c)
            {
                d = c - i;
                if (d > 49152)
                    j = 49152;
                else
                    j = (int)d;
 
                if (Buf is null)
                    Buf = new byte[j];
 
                await File.ReadAllAsync(Buf, 0, j);
 
                this.output.WriteElementString("Chunk", Convert.ToBase64String(Buf, 0, j, Base64FormattingOptions.None));
                i += j;
            }
 
            await this.output.WriteEndElementAsync();
 
            return true;
        }
 
    }
}