XmlFileEventSink.cs

using System;
using System.IO;
using System.Text;
using System.Xml;
using System.Threading.Tasks;
using Waher.Runtime.IO;
 
namespace Waher.Events.Files
{
    public class XmlFileEventSink : XmlWriterEventSink
    {
        private readonly XmlWriterSettings settings;
        private readonly FileNameTimeSequence fileSequence;
        private StreamWriter file;
        private readonly string transform = null;
        private readonly int deleteAfterDays;
 
        public XmlFileEventSink(string ObjectID, string FileName)
            : this(ObjectID, FileName, string.Empty, 7)
        {
        }
 
        public XmlFileEventSink(string ObjectID, string FileName, int DeleteAfterDays)
            : this(ObjectID, FileName, string.Empty, DeleteAfterDays)
        {
        }
 
        public XmlFileEventSink(string ObjectID, string FileName, string Transform)
            : this(ObjectID, FileName, Transform, 7)
        {
        }
 
        public XmlFileEventSink(string ObjectID, string FileName, string Transform, int DeleteAfterDays)
            : base(ObjectID, null)
        {
            this.file = null;
            this.output = null;
            this.fileSequence = new FileNameTimeSequence(FileName, true);
            this.transform = Transform;
            this.deleteAfterDays = DeleteAfterDays;
 
            this.settings = new XmlWriterSettings
            {
                CloseOutput = true,
                ConformanceLevel = ConformanceLevel.Document,
                Encoding = Encoding.UTF8,
                Indent = true,
                IndentChars = "\t",
                NewLineChars = "\r\n",
                NewLineHandling = NewLineHandling.Replace,
                NewLineOnAttributes = false,
                OmitXmlDeclaration = false,
                WriteEndDocumentOnClose = true
            };
 
            string FolderName = Path.GetDirectoryName(FileName);
 
            if (!Directory.Exists(FolderName))
            {
                Log.Informational("Creating folder.", FolderName);
                Directory.CreateDirectory(FolderName);
            }
        }
 
        protected override async Task BeforeWrite()
        {
            if (!this.fileSequence.TryGetNewFileName(out string s))
                return;
 
            try
            {
                this.output?.WriteEndElement();
                this.output?.WriteEndDocument();
                this.output?.Flush();
                this.file?.Dispose();
            }
            catch (Exception)
            {
                try
                {
                    this.DisposeOutput();
                }
                catch (Exception)
                {
                    // Ignore
                }
            }
 
            this.file = null;
            this.output = null;
 
            this.file = File.CreateText(s);
 
            this.output = XmlWriter.Create(this.file, this.settings);
            this.output.WriteStartDocument();
 
            if (!string.IsNullOrEmpty(this.transform))
            {
                if (File.Exists(this.transform))
                {
                    try
                    {
                        byte[] XsltBin = await Runtime.IO.Files.ReadAllBytesAsync(this.transform);
 
                        this.output.WriteProcessingInstruction("xml-stylesheet", "type=\"text/xsl\" href=\"data:text/xsl;base64," +
                            Convert.ToBase64String(XsltBin) + "\"");
                    }
                    catch (Exception)
                    {
                        this.output.WriteProcessingInstruction("xml-stylesheet", "type=\"text/xsl\" href=\"" + this.transform + "\"");
                    }
                }
                else
                    this.output.WriteProcessingInstruction("xml-stylesheet", "type=\"text/xsl\" href=\"" + this.transform + "\"");
            }
 
            this.output.WriteStartElement("EventOutput", EventExtensions.LogNamespace);
            this.output.Flush();
 
            if (this.deleteAfterDays < int.MaxValue)
            {
                string FolderName = Path.GetDirectoryName(s);
 
                Runtime.IO.Files.DeleteOldFiles(FolderName, TimeSpan.FromDays(this.deleteAfterDays), SearchOption.AllDirectories, true);
            }
        }
 
        public override void DisposeOutput()
        {
            base.DisposeOutput();
 
            this.file?.Dispose();
            this.file = null;
        }
    }
}