gtav-src/tools_ng/bin/CruiseControl/WebDashboard/Dashboard/VelocityProjectGridAction.cs

using System;
using System.Collections;
using System.Globalization;
using ThoughtWorks.CruiseControl.Core;
using ThoughtWorks.CruiseControl.Core.Reporting.Dashboard.Navigation;
using ThoughtWorks.CruiseControl.WebDashboard.MVC;
using ThoughtWorks.CruiseControl.WebDashboard.MVC.View;
using ThoughtWorks.CruiseControl.WebDashboard.ServerConnection;

namespace ThoughtWorks.CruiseControl.WebDashboard.Dashboard
{
	// ToDo - Test!
	public class VelocityProjectGridAction : IProjectGridAction
	{
		private readonly IFarmService farmService;
		private readonly IUrlBuilder urlBuilder;
		private readonly ICruiseUrlBuilder cruiseUrlBuilder;
		private readonly IVelocityViewGenerator viewGenerator;
		private readonly IProjectGrid projectGrid;

		public VelocityProjectGridAction(IFarmService farmService, IUrlBuilder urlBuilder, ICruiseUrlBuilder cruiseUrlBuilder, IVelocityViewGenerator viewGenerator, IProjectGrid projectGrid)
		{
			this.farmService = farmService;
			this.urlBuilder = urlBuilder;
			this.cruiseUrlBuilder = cruiseUrlBuilder;
			this.viewGenerator = viewGenerator;
			this.projectGrid = projectGrid;
		}

		public IResponse Execute(string actionName, IRequest request)
		{
			return GenerateView(farmService.GetProjectStatusListAndCaptureExceptions(), actionName, request, null);
		}

		public IResponse Execute(string actionName, IServerSpecifier serverSpecifier, IRequest request)
		{
			//Added code so since defaultServerSpecifier only sets the name of the server - not the actual config
			serverSpecifier = farmService.GetServerConfiguration(serverSpecifier.ServerName);
			return GenerateView(farmService.GetProjectStatusListAndCaptureExceptions(serverSpecifier), actionName, request, serverSpecifier);
		}

		private HtmlFragmentResponse GenerateView(ProjectStatusListAndExceptions projectStatusListAndExceptions, string actionName, IRequest request, IServerSpecifier serverSpecifier)
		{
			Hashtable velocityContext = new Hashtable();
			velocityContext["forceBuildMessage"] = ForceBuildIfNecessary(request);

			velocityContext["wholeFarm"] = serverSpecifier == null ?  true : false;

			string category = request.GetText("Category");
			velocityContext["showCategoryColumn"] = string.IsNullOrEmpty(category) ? true : false;

			ProjectGridSortColumn sortColumn = GetSortColumn(request);
			bool sortReverse = SortAscending(request);

			velocityContext["projectNameSortLink"] = GenerateSortLink(serverSpecifier, actionName, ProjectGridSortColumn.Name, sortColumn, sortReverse);
			velocityContext["buildStatusSortLink"] = GenerateSortLink(serverSpecifier, actionName, ProjectGridSortColumn.BuildStatus, sortColumn, sortReverse);
			velocityContext["lastBuildDateSortLink"] = GenerateSortLink(serverSpecifier, actionName, ProjectGridSortColumn.LastBuildDate, sortColumn, sortReverse);
			velocityContext["serverNameSortLink"] = GenerateSortLink(serverSpecifier, actionName, ProjectGridSortColumn.ServerName, sortColumn, sortReverse);
			velocityContext["projectCategorySortLink"] = GenerateSortLink(serverSpecifier, actionName, ProjectGridSortColumn.Category, sortColumn, sortReverse);
			velocityContext["projectGrid"] = projectGrid.GenerateProjectGridRows(
				projectStatusListAndExceptions.StatusAndServerList, actionName, sortColumn, sortReverse, category);
			velocityContext["exceptions"] = projectStatusListAndExceptions.Exceptions;

            // DW - a hack! - I dunno what I'm doing yet - so trying to work with this in a hacky way to start with.
            if (actionName == "ViewCodeFarmReport")
            {                
                DateTime now = System.DateTime.Now;
                velocityContext["tomorrowDate"] = now.AddDays(1).ToString("yyyy-MM-dd");
                velocityContext["yesterdayDate"] = now.AddDays(-1).ToString("yyyy-MM-dd");
                velocityContext["currDate"] = now.ToString("yyyy-MM-dd");
                velocityContext["currTime"] = now.ToString("HH:mm");
                velocityContext["currDateTime"] = now;
                return viewGenerator.GenerateView(@"CodeProjectGrid.vm", velocityContext);
            }

			return viewGenerator.GenerateView(@"ProjectGrid.vm", velocityContext);
		}

		private bool SortAscending(IRequest request)
		{
			return request.FindParameterStartingWith("ReverseSort") == string.Empty;
		}

		private ProjectGridSortColumn GetSortColumn(IRequest request)
		{
			string columnName = request.GetText("SortColumn");
			if (columnName == string.Empty)
			{
				columnName = "Name";
			}
			try
			{
				return (ProjectGridSortColumn) Enum.Parse(typeof(ProjectGridSortColumn), columnName);	
			}
			catch (Exception)
			{
				throw new CruiseControlException(string.Format("Error attempting to calculate column to sort. Specified column name was [{0}]", columnName));
			}
		}

		private object GenerateSortLink(IServerSpecifier serverSpecifier, string action, ProjectGridSortColumn column, ProjectGridSortColumn currentColumn, bool currentReverse)
		{
			string queryString = "SortColumn=" + column.ToString();
			if (column == currentColumn && !currentReverse)
			{
				queryString += "&ReverseSort=ReverseSort";
			}
			if (serverSpecifier == null)
			{
				return urlBuilder.BuildUrl(action, queryString);
			}
			else
			{
				return cruiseUrlBuilder.BuildServerUrl(action, serverSpecifier, queryString);
			}
		}

		private string ForceBuildIfNecessary(IRequest request)
		{
			if (request.FindParameterStartingWith("StopBuild") != string.Empty)
			{
				farmService.Stop(ProjectSpecifier(request));
				return string.Format("Stopping project {0}", SelectedProject(request));
			}
			else if (request.FindParameterStartingWith("StartBuild") != string.Empty)
			{
				farmService.Start(ProjectSpecifier(request));
				return string.Format("Starting project {0}", SelectedProject(request));				
			}
			else if (request.FindParameterStartingWith("ForceBuild") != string.Empty)
			{
				farmService.ForceBuild(ProjectSpecifier(request),"Dashboard");
				return string.Format("Build successfully forced for {0}", SelectedProject(request));
			}
			else if (request.FindParameterStartingWith("AbortBuild") != string.Empty)
			{
				farmService.AbortBuild(ProjectSpecifier(request), "Dashboard");
				return string.Format("Abort successfully forced for {0}", SelectedProject(request));
			}
			else
			{
				return "";
			}
		}

		private DefaultProjectSpecifier ProjectSpecifier(IRequest request)
		{
			return new DefaultProjectSpecifier(
				farmService.GetServerConfiguration(request.GetText("serverName")), SelectedProject(request));
		}

		private static string SelectedProject(IRequest request)
		{
			return request.GetText("projectName");
		}
	}
}