mirror-processhacker/trunk/ProcessHacker.Common/Settings/XmlFileSettingsStore.cs

/*
 * Process Hacker - 
 *   XML file settings store
 * 
 * Copyright (C) 2009 wj32
 * 
 * This file is part of Process Hacker.
 * 
 * Process Hacker is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Process Hacker is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Process Hacker.  If not, see <http://www.gnu.org/licenses/>.
 */

using System;
using System.Collections.Generic;
using System.IO;
using System.Text;
using System.Xml;

namespace ProcessHacker.Common.Settings
{
    public sealed class XmlFileSettingsStore : ISettingsStore
    {
        private const string _rootElementName = "settings";
        private const string _settingElementName = "setting";
        private const string _nameAttributeName = "name";

        private string _fileName;
        private XmlDocument _doc;
        private XmlNode _rootNode;

        public XmlFileSettingsStore(string fileName)
        {
            _fileName = fileName;
            _doc = new XmlDocument();

            // Does the file exist? If not, initialize a new XML document.
            if (File.Exists(_fileName))
            {
                _doc.Load(_fileName);
            }
            else
            {
                this.InitializeNew(_doc);
            }

            _rootNode = _doc.SelectSingleNode("/" + _rootElementName);
        }

        public string FileName
        {
            get { return _fileName; }
        }

        public void Flush()
        {
            lock (_doc)
            {
                _doc.Save(_fileName);
            }
        }

        public string GetValue(string name)
        {
            lock (_doc)
            {
                var nodes = _rootNode.SelectNodes(_settingElementName + "[@" + _nameAttributeName + "='" + name + "']");

                if (nodes.Count == 0)
                    return null;

                return nodes[0].InnerText;
            }
        }

        private void InitializeNew(XmlDocument doc)
        {
            doc.RemoveAll();
            doc.AppendChild(doc.CreateElement(_rootElementName));
        }

        public void Reset()
        {
            lock (_doc)
            {
                this.InitializeNew(_doc);
                _doc.Save(_fileName);
            }
        }

        public void SetValue(string name, string value)
        {
            lock (_doc)
            {
                var nodes = _rootNode.SelectNodes(_settingElementName + "[@" + _nameAttributeName + "='" + name + "']");

                if (nodes.Count != 0)
                {
                    nodes[0].InnerText = value;
                }
                else
                {
                    var settingElement = _doc.CreateElement(_settingElementName);
                    var nameAttribute = _doc.CreateAttribute(_nameAttributeName);

                    nameAttribute.Value = name;
                    settingElement.Attributes.Append(nameAttribute);
                    settingElement.InnerText = value;

                    _rootNode.AppendChild(settingElement);
                }
            }
        }
    }
}