com.cradiator.TeamCityStatusPlugin.BuildMonitorController.java Source code

Introduction

Here is the source code for com.cradiator.TeamCityStatusPlugin.BuildMonitorController.java

Source

/*
 *    Cradiator TeamCity Plugin Copyright (c) 2009 Peter van der Woude, James Summerton
 *
 *    This file is part of Cradiator - http://cradiator.codeplex.com/
 *
 *    Cradiator 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.
 *
 *    Cradiator 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 this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 *    Original idea and TeamCity plugin from Team Piazza Copyright (c) 2007-2009 Nat Pryce.
 */
package com.cradiator.TeamCityStatusPlugin;

import jetbrains.buildServer.controllers.BaseController;
import jetbrains.buildServer.serverSide.ProjectManager;
import jetbrains.buildServer.serverSide.SBuildServer;
import jetbrains.buildServer.serverSide.SProject;
import jetbrains.buildServer.serverSide.SBuildType;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;

public class BuildMonitorController extends BaseController {
    public static final String PROJECT_ID = "projectId";
    public static final String BUILD_TYPE_ID = "buildTypeId";

    private final ProjectManager projectManager;
    private final Cradiator cradiator;
    private SBuildServer server;

    public BuildMonitorController(SBuildServer server, ProjectManager projectManager, Cradiator cradiator) {
        super(server);
        this.server = server;
        this.projectManager = projectManager;
        this.cradiator = cradiator;
    }

    protected ModelAndView doHandle(HttpServletRequest request, HttpServletResponse response) throws Exception {

        if (requestHasParameter(request, PROJECT_ID))
            return showProject(request.getParameter(PROJECT_ID), response);
        else if (requestHasParameter(request, BUILD_TYPE_ID))
            return showBuild(request.getParameter(BUILD_TYPE_ID), response);
        else {
            response.sendError(HttpServletResponse.SC_BAD_REQUEST, "no project id or buildTypeId specified");
            return null;
        }
    }

    private ModelAndView showProject(String projectId, HttpServletResponse response) throws IOException {
        SProject project = projectManager.findProjectById(projectId);
        if (project == null) {
            response.sendError(HttpServletResponse.SC_NOT_FOUND, "no project with id " + projectId);
            return null;
        }
        return modelWithView("project-status.jsp").addObject("project", new ProjectMonitorViewState(project));
    }

    private ModelAndView showBuild(String buildTypeId, HttpServletResponse response) throws IOException {
        SBuildType buildType = projectManager.findBuildTypeById(buildTypeId);
        if (buildType == null) {
            response.sendError(HttpServletResponse.SC_NOT_FOUND, "no buildType with id " + buildTypeId);
            return null;
        }
        return modelWithView("build-status.jsp").addObject("build",
                new BuildTypeMonitorViewState(server, buildType));
    }

    private ModelAndView modelWithView(String viewJSP) {
        return new ModelAndView(cradiator.resourcePath(viewJSP)).addObject("version", cradiator.version())
                .addObject("resourceRoot", cradiator.resourcePath(""));
    }

    private boolean requestHasParameter(HttpServletRequest request, String parameterName) {
        return request.getParameterMap().containsKey(parameterName);
    }
}