SlackNotificationIndexPageController.java

package slacknotifications.teamcity.extension;

import jetbrains.buildServer.controllers.BaseController;
import jetbrains.buildServer.serverSide.SBuildServer;
import jetbrains.buildServer.serverSide.SBuildType;
import jetbrains.buildServer.serverSide.SProject;
import jetbrains.buildServer.serverSide.auth.Permission;
import jetbrains.buildServer.serverSide.settings.ProjectSettingsManager;
import jetbrains.buildServer.users.SUser;
import jetbrains.buildServer.web.openapi.PluginDescriptor;
import jetbrains.buildServer.web.openapi.WebControllerManager;
import jetbrains.buildServer.web.util.SessionUser;
import org.jetbrains.annotations.Nullable;
import org.springframework.web.servlet.ModelAndView;
import slacknotifications.teamcity.TeamCityIdResolver;
import slacknotifications.teamcity.extension.bean.ProjectSlackNotificationsBean;
import slacknotifications.teamcity.extension.bean.ProjectSlackNotificationsBeanJsonSerialiser;
import slacknotifications.teamcity.payload.SlackNotificationPayloadManager;
import slacknotifications.teamcity.settings.SlackNotificationConfig;
import slacknotifications.teamcity.settings.SlackNotificationMainSettings;
import slacknotifications.teamcity.settings.SlackNotificationProjectSettings;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.HashMap;
import java.util.List;


public class SlackNotificationIndexPageController extends BaseController {

		private static final String SHOW_FURTHER_READING = "ShowFurtherReading";
		private static final String SLACK_NOTIFICATIONS = "slackNotifications";
		private static final String NO_SLACK_NOTIFICATIONS = "noSlackNotifications";
		private static final String PROJECT_ID = "projectId";
		private static final String FALSE = "false";
		private static final String ERROR_REASON = "errorReason";
		private static final String PROJECT_SLACK_NOTIFICATION_AS_JSON = "projectSlackNotificationsAsJson";
		
		private final WebControllerManager myWebManager;
	    private final SlackNotificationMainSettings myMainSettings;
	    private SBuildServer myServer;
	    private ProjectSettingsManager mySettings;
	    private PluginDescriptor myPluginDescriptor;
	    private final SlackNotificationPayloadManager myManager;

	    public SlackNotificationIndexPageController(SBuildServer server, WebControllerManager webManager,
                                                    ProjectSettingsManager settings, PluginDescriptor pluginDescriptor, SlackNotificationPayloadManager manager,
                                                    SlackNotificationMainSettings configSettings) {
	        super(server);
	        myWebManager = webManager;
	        myServer = server;
	        mySettings = settings;
	        myPluginDescriptor = pluginDescriptor;
	        myMainSettings = configSettings;
	        myManager = manager;
	    }

	    public void register(){
	      myWebManager.registerController("/slacknotifications/index.html", this);
	    }

		@Override
	    @Nullable
	    protected ModelAndView doHandle(HttpServletRequest request, HttpServletResponse response) throws Exception {
	    	
	        HashMap<String,Object> params = new HashMap<String,Object>();
	        params.put("jspHome",this.myPluginDescriptor.getPluginResourcesPath());
        	params.put("includeJquery", Boolean.toString(this.myServer.getServerMajorVersion() < 7));
        	params.put("rootContext", myServer.getServerRootPath());
	        
	    	if (myMainSettings.getInfoUrl() != null && myMainSettings.getInfoUrl().length() > 0){
	    		params.put("moreInfoText", "<li><a href=\"" + myMainSettings.getInfoUrl() + "\">" + myMainSettings.getInfoText() + "</a></li>");
	    		if (myMainSettings.getSlackNotificationShowFurtherReading()){
	    			params.put(SHOW_FURTHER_READING, "ALL");
	    		} else {
	    			params.put(SHOW_FURTHER_READING, "SINGLE");
	    		}
	    	} else if (myMainSettings.getSlackNotificationShowFurtherReading()){
	    		params.put(SHOW_FURTHER_READING, "DEFAULT");
	    	} else {
	    		params.put(SHOW_FURTHER_READING, "NONE");
	    	}
	        
	        if(request.getParameter(PROJECT_ID) != null){
	        	
	        	SProject project = TeamCityIdResolver.findProjectById(this.myServer.getProjectManager(), request.getParameter("projectId"));
	        	if (project != null){
	        		
			    	SlackNotificationProjectSettings projSettings = (SlackNotificationProjectSettings)
			    			mySettings.getSettings(project.getProjectId(), SLACK_NOTIFICATIONS);
			    	
			        SUser myUser = SessionUser.getUser(request);
			        params.put("hasPermission", myUser.isPermissionGrantedForProject(project.getProjectId(), Permission.EDIT_PROJECT));
			    	
			    	String message = projSettings.getSlackNotificationsAsString();
			    	
			    	params.put("haveProject", "true");
			    	params.put("messages", message);
			    	params.put(PROJECT_ID, project.getProjectId());
			    	params.put("buildTypeList", project.getBuildTypes());
			    	params.put("projectExternalId", TeamCityIdResolver.getExternalProjectId(project));
			    	params.put("projectName", project.getName());
			    	
			    	logger.debug(myMainSettings.getInfoText() + myMainSettings.getInfoUrl() + myMainSettings.getProxySettingsAsString());
			    	
			    	params.put("slackNotificationCount", projSettings.getSlackNotificationsCount());
			    	
			    	if (projSettings.getSlackNotificationsCount() == 0){
			    		params.put(NO_SLACK_NOTIFICATIONS, "true");
			    		params.put(SLACK_NOTIFICATIONS, FALSE);
			    		params.put(PROJECT_SLACK_NOTIFICATION_AS_JSON, ProjectSlackNotificationsBeanJsonSerialiser.serialise(ProjectSlackNotificationsBean.build(projSettings, project, myMainSettings)));
			    	} else {
			    		params.put(NO_SLACK_NOTIFICATIONS, FALSE);
			    		params.put(SLACK_NOTIFICATIONS, "true");
			    		params.put("slackNotificationList", projSettings.getSlackNotificationsAsList());
			    		params.put("slackNotificationsDisabled", !projSettings.isEnabled());
			    		params.put("slackNotificationsEnabledAsChecked", projSettings.isEnabledAsChecked());
			    		params.put(PROJECT_SLACK_NOTIFICATION_AS_JSON, ProjectSlackNotificationsBeanJsonSerialiser.serialise(ProjectSlackNotificationsBean.build(projSettings, project, myMainSettings)));

			    	}
		    	} else {
		    		params.put("haveProject", FALSE);
		    		params.put(ERROR_REASON, "The project requested does not appear to be valid.");
		    	}
        	} else if (request.getParameter("buildTypeId") != null){
        		SBuildType sBuildType = TeamCityIdResolver.findBuildTypeById(this.myServer.getProjectManager(), request.getParameter("buildTypeId"));
        		if (sBuildType != null){
		        	SProject project = sBuildType.getProject();
		        	if (project != null){
		        		
				    	SlackNotificationProjectSettings projSettings = (SlackNotificationProjectSettings)
				    			mySettings.getSettings(project.getProjectId(), SLACK_NOTIFICATIONS);
				    	
				    	SUser myUser = SessionUser.getUser(request);
				        params.put("hasPermission", myUser.isPermissionGrantedForProject(project.getProjectId(), Permission.EDIT_PROJECT));
				    	
				    	List<SlackNotificationConfig> configs = projSettings.getBuildSlackNotificationsAsList(sBuildType);
				    	params.put("slackNotificationList", configs);
				    	params.put("slackNotificationsDisabled", !projSettings.isEnabled());
				    	params.put(PROJECT_ID, project.getProjectId());
				    	params.put("haveProject", "true");
				    	params.put("projectName", project.getName());
				    	params.put("projectExternalId", TeamCityIdResolver.getExternalProjectId(project));
				    	params.put("haveBuild", "true");
				    	params.put("buildName", sBuildType.getName());
				    	params.put("buildExternalId", TeamCityIdResolver.getExternalBuildId(sBuildType));
				    	params.put("buildTypeList", project.getBuildTypes());
			    		params.put(NO_SLACK_NOTIFICATIONS, configs.isEmpty());
			    		params.put(SLACK_NOTIFICATIONS, !configs.isEmpty());
				    	
			    		params.put(PROJECT_SLACK_NOTIFICATION_AS_JSON, ProjectSlackNotificationsBeanJsonSerialiser.serialise(ProjectSlackNotificationsBean.build(projSettings, sBuildType, project, myMainSettings)));
		        	}
        		} else {
		    		params.put("haveProject", FALSE);
		    		params.put(ERROR_REASON, "The build requested does not appear to be valid.");
        		}
	        } else {
	        	params.put("haveProject", FALSE);
	        	params.put(ERROR_REASON, "No project specified.");
	        }

	        return new ModelAndView(myPluginDescriptor.getPluginResourcesPath() + "SlackNotification/index.jsp", params);
	    }
}