SiteMapping.cs

#nullable disable
namespace AngleSharp.Css.Tests.Mocks
{
    using AngleSharp.Dom;
    using AngleSharp.Xml;
    using AngleSharp.Xml.Dom;
    using AngleSharp.Xml.Parser;
    using System;
    using System.IO;

    sealed class SiteMapping
    {
        private readonly String _fileName;
        private readonly IXmlDocument _xml;

        public SiteMapping(String fileName)
        {
            _fileName = fileName;
            var parser = new XmlParser();
            var content = File.Exists(fileName) ? File.ReadAllText(_fileName) : "<entries></entries>";
            _xml = parser.ParseDocument(content);
        }

        public String this[String url]
        {
            get
            {
                var element = _xml.QuerySelector("entry[url='" + url + "']");
                return element?.TextContent;
            }
        }

        public Boolean Contains(String url) => this[url] != null;

        public void Add(String url, String fileName)
        {
            if (Contains(url) == false)
            {
                var element = _xml.CreateElement("entry");
                element.SetAttribute("url", url);
                element.TextContent = fileName;
                _xml.DocumentElement.AppendChild(element);

                using (var writer = File.CreateText(_fileName))
                {
                    _xml.ToHtml(writer, XmlMarkupFormatter.Instance);
                }
            }
        }
    }
}