src/main/java/de/uapcore/lightpit/modules/ProjectsModule.java

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright 2018 Mike Becker. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *   1. Redistributions of source code must retain the above copyright
 *      notice, this list of conditions and the following disclaimer.
 *
 *   2. Redistributions in binary form must reproduce the above copyright
 *      notice, this list of conditions and the following disclaimer in the
 *      documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 *
 */
package de.uapcore.lightpit.modules;


import de.uapcore.lightpit.*;
import de.uapcore.lightpit.dao.DataAccessObjects;
import de.uapcore.lightpit.entities.*;
import de.uapcore.lightpit.viewmodel.*;
import de.uapcore.lightpit.viewmodel.util.IssueSorter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.sql.Date;
import java.sql.SQLException;
import java.util.NoSuchElementException;
import java.util.stream.Collectors;
import java.util.stream.Stream;

@WebServlet(
        name = "ProjectsModule",
        urlPatterns = "/projects/*"
)
public final class ProjectsModule extends AbstractLightPITServlet {

    private static final Logger LOG = LoggerFactory.getLogger(ProjectsModule.class);

    @Override
    protected String getResourceBundleName() {
        return "localization.projects";
    }

    private void populate(ProjectView viewModel, PathParameters pathParameters, DataAccessObjects dao) throws SQLException {
        final var projectDao = dao.getProjectDao();
        final var versionDao = dao.getVersionDao();
        final var componentDao = dao.getComponentDao();

        projectDao.list().stream().map(ProjectInfo::new).forEach(viewModel.getProjectList()::add);

        if (pathParameters == null)
            return;

        // Select Project
        final int pid = Functions.parseIntOrZero(pathParameters.get("project"));
        if (pid > 0) {
            final var project = projectDao.find(pid);
            if (project != null) {
                final var info = new ProjectInfo(project);
                info.setVersions(versionDao.list(project));
                info.setComponents(componentDao.list(project));
                info.setIssueSummary(projectDao.getIssueSummary(project));
                viewModel.setProjectInfo(info);
            }
        }

        // Select Version
        final int vid = Functions.parseIntOrZero(pathParameters.get("version"));
        if (vid > 0) {
            viewModel.setVersionFilter(versionDao.find(vid));
        }
        // TODO: don't treat unknown == unassigned - send 404 for unknown and introduce special word for unassigned

        // Select Component
        final int cid = Functions.parseIntOrZero(pathParameters.get("component"));
        if (cid > 0) {
            viewModel.setComponentFilter(componentDao.find(cid));
        }

        // TODO: distinguish all/unassigned for components
    }

    private ResponseType forwardView(HttpServletRequest req, ProjectView viewModel, String name) {
        setViewModel(req, viewModel);
        setContentPage(req, name);
        setStylesheet(req, "projects");
        setNavigationMenu(req, "project-navmenu");
        return ResponseType.HTML;
    }

    @RequestMapping(method = HttpMethod.GET)
    public ResponseType index(HttpServletRequest req, DataAccessObjects dao) throws SQLException {
        final var viewModel = new ProjectView();
        populate(viewModel, null, dao);

        final var projectDao = dao.getProjectDao();
        final var versionDao = dao.getVersionDao();

        for (var info : viewModel.getProjectList()) {
            info.setVersions(versionDao.list(info.getProject()));
            info.setIssueSummary(projectDao.getIssueSummary(info.getProject()));
        }

        return forwardView(req, viewModel, "projects");
    }

    private void configureProjectEditor(ProjectEditView viewModel, Project project, DataAccessObjects dao) throws SQLException {
        viewModel.setProject(project);
        viewModel.setUsers(dao.getUserDao().list());
    }

    @RequestMapping(requestPath = "$project/edit", method = HttpMethod.GET)
    public ResponseType edit(HttpServletRequest req, HttpServletResponse resp, PathParameters pathParams, DataAccessObjects dao) throws IOException, SQLException {
        final var viewModel = new ProjectEditView();
        populate(viewModel, pathParams, dao);

        if (viewModel.getProjectInfo() == null) {
            resp.sendError(HttpServletResponse.SC_NOT_FOUND);
            return ResponseType.NONE;
        }

        configureProjectEditor(viewModel, viewModel.getProjectInfo().getProject(), dao);
        return forwardView(req, viewModel, "project-form");
    }

    @RequestMapping(requestPath = "create", method = HttpMethod.GET)
    public ResponseType create(HttpServletRequest req, DataAccessObjects dao) throws SQLException {
        final var viewModel = new ProjectEditView();
        populate(viewModel, null, dao);
        configureProjectEditor(viewModel, new Project(-1), dao);
        return forwardView(req, viewModel, "project-form");
    }

    @RequestMapping(requestPath = "commit", method = HttpMethod.POST)
    public ResponseType commit(HttpServletRequest req, HttpServletResponse resp, DataAccessObjects dao) throws IOException {

        try {
            final var project = new Project(getParameter(req, Integer.class, "pid").orElseThrow());
            project.setName(getParameter(req, String.class, "name").orElseThrow());
            getParameter(req, String.class, "description").ifPresent(project::setDescription);
            getParameter(req, String.class, "repoUrl").ifPresent(project::setRepoUrl);
            getParameter(req, Integer.class, "owner").map(
                    ownerId -> ownerId >= 0 ? new User(ownerId) : null
            ).ifPresent(project::setOwner);

            dao.getProjectDao().saveOrUpdate(project);

            setRedirectLocation(req, "./projects/");
            setContentPage(req, Constants.JSP_COMMIT_SUCCESSFUL);
            LOG.debug("Successfully updated project {}", project.getName());

            return ResponseType.HTML;
        } catch (NoSuchElementException | IllegalArgumentException | SQLException ex) {
            resp.sendError(HttpServletResponse.SC_NOT_IMPLEMENTED);
            // TODO: implement - fix issue #21
            return ResponseType.NONE;
        }
    }

    @RequestMapping(requestPath = "$project/versions/$version", method = HttpMethod.GET)
    public ResponseType view(HttpServletRequest req, HttpServletResponse resp, PathParameters pathParams, DataAccessObjects dao) throws SQLException, IOException {
        final var viewModel = new ProjectDetailsView();
        populate(viewModel, pathParams, dao);
        final var version = viewModel.getVersionFilter();

        if (viewModel.getProjectInfo() == null || version == null) {
            resp.sendError(HttpServletResponse.SC_NOT_FOUND);
            return ResponseType.NONE;
        }

        final var issueDao = dao.getIssueDao();

        final var detailView = viewModel.getProjectDetails();
        final var issues = issueDao.list(version);
        for (var issue : issues) issueDao.joinVersionInformation(issue);
        issues.sort(new IssueSorter(
                new IssueSorter.Criteria(IssueSorter.Field.PHASE, true),
                new IssueSorter.Criteria(IssueSorter.Field.ETA, true),
                new IssueSorter.Criteria(IssueSorter.Field.UPDATED, false)
        ));
        detailView.updateDetails(issues, version);

        return forwardView(req, viewModel, "project-details");
    }

    @RequestMapping(requestPath = "$project/versions/", method = HttpMethod.GET)
    public ResponseType versions(HttpServletRequest req, HttpServletResponse resp, PathParameters pathParameters, DataAccessObjects dao) throws IOException, SQLException {
        final var viewModel = new VersionsView();
        populate(viewModel, pathParameters, dao);

        final var projectInfo = viewModel.getProjectInfo();
        if (projectInfo == null) {
            resp.sendError(HttpServletResponse.SC_NOT_FOUND);
            return ResponseType.NONE;
        }

        final var issueDao = dao.getIssueDao();
        final var issues = issueDao.list(projectInfo.getProject());
        for (var issue : issues) issueDao.joinVersionInformation(issue);
        viewModel.update(projectInfo.getVersions(), issues);

        return forwardView(req, viewModel, "versions");
    }

    @RequestMapping(requestPath = "$project/versions/$version/edit", method = HttpMethod.GET)
    public ResponseType editVersion(HttpServletRequest req, HttpServletResponse resp, PathParameters pathParameters, DataAccessObjects dao) throws IOException, SQLException {
        final var viewModel = new VersionEditView();
        populate(viewModel, pathParameters, dao);

        if (viewModel.getProjectInfo() == null || viewModel.getVersionFilter() == null) {
            resp.sendError(HttpServletResponse.SC_NOT_FOUND);
            return ResponseType.NONE;
        }

        viewModel.setVersion(viewModel.getVersionFilter());

        return forwardView(req, viewModel, "version-form");
    }

    @RequestMapping(requestPath = "$project/create-version", method = HttpMethod.GET)
    public ResponseType createVersion(HttpServletRequest req, HttpServletResponse resp, PathParameters pathParameters, DataAccessObjects dao) throws IOException, SQLException {
        final var viewModel = new VersionEditView();
        populate(viewModel, pathParameters, dao);

        if (viewModel.getProjectInfo() == null) {
            resp.sendError(HttpServletResponse.SC_NOT_FOUND);
            return ResponseType.NONE;
        }

        viewModel.setVersion(viewModel.getVersionFilter());

        return forwardView(req, viewModel, "version-form");
    }

    @RequestMapping(requestPath = "commit-version", method = HttpMethod.POST)
    public ResponseType commitVersion(HttpServletRequest req, HttpServletResponse resp, DataAccessObjects dao) throws IOException {

        try {
            final var project = new Project(getParameter(req, Integer.class, "pid").orElseThrow());
            final var version = new Version(getParameter(req, Integer.class, "id").orElseThrow());
            version.setName(getParameter(req, String.class, "name").orElseThrow());
            getParameter(req, Integer.class, "ordinal").ifPresent(version::setOrdinal);
            version.setStatus(VersionStatus.valueOf(getParameter(req, String.class, "status").orElseThrow()));
            dao.getVersionDao().saveOrUpdate(version, project);

            // TODO: improve building the redirect location
            setRedirectLocation(req, "./projects/" + project.getId() + "/versions/");
            setContentPage(req, Constants.JSP_COMMIT_SUCCESSFUL);
        } catch (NoSuchElementException | IllegalArgumentException | SQLException ex) {
            resp.sendError(HttpServletResponse.SC_NOT_IMPLEMENTED);
            // TODO: implement - fix issue #21
            return ResponseType.NONE;
        }

        return ResponseType.HTML;
    }

    private void configureProjectEditor(IssueEditView viewModel, Issue issue, DataAccessObjects dao) throws SQLException {
        issue.setProject(viewModel.getProjectInfo().getProject());
        viewModel.setIssue(issue);
        viewModel.configureVersionSelectors(viewModel.getProjectInfo().getVersions());
        viewModel.setUsers(dao.getUserDao().list());
        if (issue.getId() >= 0) {
            viewModel.setComments(dao.getIssueDao().listComments(issue));
        }
    }

    @RequestMapping(requestPath = "$project/issues/$issue/edit", method = HttpMethod.GET)
    public ResponseType editIssue(HttpServletRequest req, HttpServletResponse resp, PathParameters pathParameters, DataAccessObjects dao) throws IOException, SQLException {
        final var viewModel = new IssueEditView();
        populate(viewModel, pathParameters, dao);

        final var projectInfo = viewModel.getProjectInfo();
        if (projectInfo == null) {
            resp.sendError(HttpServletResponse.SC_NOT_FOUND);
            return ResponseType.NONE;
        }

        final var issueDao = dao.getIssueDao();
        final var issue = issueDao.find(Functions.parseIntOrZero(pathParameters.get("issue")));
        if (issue == null) {
            resp.sendError(HttpServletResponse.SC_NOT_FOUND);
            return ResponseType.NONE;
        }

        issueDao.joinVersionInformation(issue);
        configureProjectEditor(viewModel, issue, dao);

        return forwardView(req, viewModel, "issue-form");
    }

    @RequestMapping(requestPath = "$project/create-issue", method = HttpMethod.GET)
    public ResponseType createIssue(HttpServletRequest req, HttpServletResponse resp, PathParameters pathParameters, DataAccessObjects dao) throws IOException, SQLException {
        final var viewModel = new IssueEditView();
        populate(viewModel, pathParameters, dao);

        final var projectInfo = viewModel.getProjectInfo();
        if (projectInfo == null) {
            resp.sendError(HttpServletResponse.SC_NOT_FOUND);
            return ResponseType.NONE;
        }

        final var issue = new Issue(-1);
        issue.setProject(projectInfo.getProject());
        configureProjectEditor(viewModel, issue, dao);

        return forwardView(req, viewModel, "issue-form");
    }

    @RequestMapping(requestPath = "commit-issue", method = HttpMethod.POST)
    public ResponseType commitIssue(HttpServletRequest req, HttpServletResponse resp, DataAccessObjects dao) throws IOException {
        try {
            final var issue = new Issue(getParameter(req, Integer.class, "id").orElseThrow());
            issue.setProject(new Project(getParameter(req, Integer.class, "pid").orElseThrow()));
            getParameter(req, String.class, "category").map(IssueCategory::valueOf).ifPresent(issue::setCategory);
            getParameter(req, String.class, "status").map(IssueStatus::valueOf).ifPresent(issue::setStatus);
            issue.setSubject(getParameter(req, String.class, "subject").orElseThrow());
            getParameter(req, Integer.class, "assignee").map(
                    userid -> userid >= 0 ? new User(userid) : null
            ).ifPresent(issue::setAssignee);
            getParameter(req, String.class, "description").ifPresent(issue::setDescription);
            getParameter(req, Date.class, "eta").ifPresent(issue::setEta);

            getParameter(req, Integer[].class, "affected")
                    .map(Stream::of)
                    .map(stream ->
                            stream.map(Version::new).collect(Collectors.toList())
                    ).ifPresent(issue::setAffectedVersions);
            getParameter(req, Integer[].class, "resolved")
                    .map(Stream::of)
                    .map(stream ->
                            stream.map(Version::new).collect(Collectors.toList())
                    ).ifPresent(issue::setResolvedVersions);

            dao.getIssueDao().saveOrUpdate(issue, issue.getProject());

            // TODO: fix issue #14
            setRedirectLocation(req, "./projects/" + issue.getProject().getId() + "/versions/");
            setContentPage(req, Constants.JSP_COMMIT_SUCCESSFUL);

            return ResponseType.HTML;
        } catch (NoSuchElementException | IllegalArgumentException | SQLException ex) {
            resp.sendError(HttpServletResponse.SC_NOT_IMPLEMENTED);
            // TODO: implement - fix issue #21
            return ResponseType.NONE;
        }
    }

    @RequestMapping(requestPath = "commit-issue-comment", method = HttpMethod.POST)
    public ResponseType commentIssue(HttpServletRequest req, HttpServletResponse resp, DataAccessObjects dao) throws SQLException, IOException {
        final var issueIdParam = getParameter(req, Integer.class, "issueid");
        if (issueIdParam.isEmpty()) {
            resp.sendError(HttpServletResponse.SC_FORBIDDEN, "Detected manipulated form.");
            return ResponseType.NONE;
        }
        final var issue = new Issue(issueIdParam.get());
        try {
            final var issueComment = new IssueComment(getParameter(req, Integer.class, "commentid").orElse(-1), issue);
            issueComment.setComment(getParameter(req, String.class, "comment").orElse(""));

            if (issueComment.getComment().isBlank()) {
                throw new IllegalArgumentException("comment.null");
            }

            LOG.debug("User {} is commenting on issue #{}", req.getRemoteUser(), issue.getId());
            if (req.getRemoteUser() != null) {
                dao.getUserDao().findByUsername(req.getRemoteUser()).ifPresent(issueComment::setAuthor);
            }

            dao.getIssueDao().saveComment(issueComment);

            // TODO: fix redirect location (e.g. after fixing #24)
            setRedirectLocation(req, "./projects/" + issue.getProject().getId()+"/issues/"+issue.getId()+"/edit");
            setContentPage(req, Constants.JSP_COMMIT_SUCCESSFUL);

            return ResponseType.HTML;
        } catch (NoSuchElementException | IllegalArgumentException | SQLException ex) {
            resp.sendError(HttpServletResponse.SC_NOT_IMPLEMENTED);
            // TODO: implement - fix issue #21
            return ResponseType.NONE;
        }
    }
}