/* 
 * Copyright (c) 2009 Craig Sutherland
 * 
 * Permission is hereby granted, free of charge, to any person
 * obtaining a copy of this software and associated documentation
 * files (the "Software"), to deal in the Software without
 * restriction, including without limitation the rights to use,
 * copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following
 * conditions:
 * 
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 */

using System.Collections.Generic;
using System.ComponentModel;
using System.Windows.Forms;
using FastForward.WinCore.Configuration;
using ThoughtWorks.CruiseControl.Remote.Monitor;

namespace FastForward.WinCore
{
    /// <summary>
    /// A <see cref="TreeNode"/> that automatically displays server details.
    /// </summary>
    public class ServerTreeNode
        : TreeNode
    {
        #region Private fields
        private Server server;
        private ServerConfig configuration;
        private Dictionary<Project, TreeNode> projects = new Dictionary<Project, TreeNode>();
        private Dictionary<BuildQueue, TreeNode> queues = new Dictionary<BuildQueue, TreeNode>();
        private TreeNode projectsNode = new TreeNode("Projects");
        private TreeNode queuesNode = new TreeNode("Queues");
        #endregion

        #region Constructors
        /// <summary>
        /// Initialse a new <see cref="ServerTreeNode"/>.
        /// </summary>
        /// <param name="server"></param>
        /// <param name="configuration"></param>
        public ServerTreeNode(Server server, ServerConfig configuration)
        {
            this.server = server;
            this.configuration = configuration;
            configuration.PropertyChanged += configuration_PropertyChanged;
            configuration.Server.PropertyChanged += configuration_PropertyChanged;

            // Set the main properties
            SetNodeText();
            if (server.Version != null)
            {
                Text += string.Format(" [{0}]", server.Version);
            }

            // Link up the projects
            server.ProjectAdded += server_ProjectAdded;
            server.ProjectRemoved += server_ProjectRemoved;
            server.LoginChanged += (o, e) => SetNodeImage();
            foreach (var project in server.Projects)
            {
                AddProject(project);
            }

            // Link up the build queues
            server.BuildQueueAdded += server_BuildQueueAdded;
            server.BuildQueueRemoved += server_BuildQueueRemoved;
            foreach (var buildQueue in server.BuildQueues)
            {
                AddQueue(buildQueue);
            }

            // Add the sub-nodes
            Nodes.Add(projectsNode);
            Nodes.Add(queuesNode);
            Expand();
            SetNodeImage();
            queuesNode.ImageKey = "Queues";
            queuesNode.SelectedImageKey = "Queues";
            projectsNode.ImageKey = "Project";
            projectsNode.SelectedImageKey = "Project";
        }
        #endregion

        #region Public properties
        #region Server
        /// <summary>
        /// The underlying server.
        /// </summary>
        public Server Server
        {
            get { return server; }
        }
        #endregion
        #endregion

        #region Private methods
        #region SetNodeImage()
        /// <summary>
        /// Set the image for the server.
        /// </summary>
        private void SetNodeImage()
        {
            var image = server.IsLoggedIn ? "SecureServer" : "Server";
            ImageKey = image;
            SelectedImageKey = image;
        }
        #endregion

        #region SetNodeText()
        /// <summary>
        /// Set the main text.
        /// </summary>
        private void SetNodeText()
        {
            Text = configuration.Server.GetDisplayName();
        }
        #endregion

        #region AddProject()
        /// <summary>
        /// Add a new project.
        /// </summary>
        /// <param name="project"></param>
        private void AddProject(Project project)
        {
            var projectNode = new ProjectTreeNode(project);
            projects.Add(project, projectNode);
            projectsNode.Nodes.Add(projectNode);
        }
        #endregion

        #region AddQueue()
        /// <summary>
        /// Add a new queue.
        /// </summary>
        /// <param name="queue"></param>
        private void AddQueue(BuildQueue queue)
        {
            var queueNode = new BuildQueueTreeNode(queue);
            queues.Add(queue, queueNode);
            queuesNode.Nodes.Add(queueNode);
        }
        #endregion
        #endregion

        #region Event handlers
        #region configuration_PropertyChanged()
        /// <summary>
        /// Handle a change in one of the properties for the underlying server.
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void configuration_PropertyChanged(object sender, PropertyChangedEventArgs e)
        {
            switch (e.PropertyName)
            {
                case "Name":
                case "DisplayName":
                    SetNodeText();
                    break;
            }
        }
        #endregion

        #region server_ProjectAdded()
        /// <summary>
        /// Handle when a new project is added.
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void server_ProjectAdded(object sender, ProjectChangedArgs e)
        {
            TreeView.SafeInvoke(() =>
            {
                AddProject(e.Project);
            });
        }
        #endregion

        #region server_ProjectRemoved()
        /// <summary>
        /// Handle when a project is removed
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void server_ProjectRemoved(object sender, ProjectChangedArgs e)
        {
            TreeView.SafeInvoke(() =>
            {
                if (projects.ContainsKey(e.Project))
                {
                    projects[e.Project].Remove();
                }
            });
        }
        #endregion

        #region server_BuildQueueAdded()
        /// <summary>
        /// Handle when a new build queue is added.
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void server_BuildQueueAdded(object sender, BuildQueueChangedArgs e)
        {
            TreeView.SafeInvoke(() =>
            {
                AddQueue(e.BuildQueue);
            });
        }
        #endregion

        #region server_BuildQueueRemoved()
        /// <summary>
        /// Handle when a build queue is removed.
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void server_BuildQueueRemoved(object sender, BuildQueueChangedArgs e)
        {
            TreeView.SafeInvoke(() =>
            {
                if (queues.ContainsKey(e.BuildQueue))
                {
                    queues[e.BuildQueue].Remove();
                }
            });
        }
        #endregion
        #endregion
    }
}