org.sonar.plugins.testtrack.reviews.LinkFunction.java Source code

Introduction

Here is the source code for org.sonar.plugins.testtrack.reviews.LinkFunction.java

Source

/*
 * Sonar, open source software quality management tool.
 * Copyright (C) 2009 SonarSource
 * mailto:contact AT sonarsource DOT com
 *
 * Sonar is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * Sonar 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with Sonar; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02
 */
package org.sonar.plugins.testtrack.reviews;

import com.atlassian.jira.rpc.soap.client.RemoteIssue;
import org.apache.commons.lang.StringUtils;
import org.sonar.api.ServerExtension;
import org.sonar.api.workflow.function.Function;
import org.sonar.api.workflow.Comment;
import org.sonar.api.workflow.MutableReview;
import org.sonar.api.workflow.Review;
import org.sonar.api.workflow.WorkflowContext;
import org.sonar.plugins.testtrack.TestTrackConstants;

import java.rmi.RemoteException;
import java.util.Map;

public class LinkFunction extends Function implements ServerExtension {

    private final TestTrackIssueCreator jiraIssueCreator;

    public LinkFunction(TestTrackIssueCreator jiraIssueCreator) {
        this.jiraIssueCreator = jiraIssueCreator;
    }

    @Override
    public void doExecute(MutableReview review, Review initialReview, WorkflowContext context,
            Map<String, String> parameters) {
        RemoteIssue issue;
        try {
            issue = jiraIssueCreator.createIssue(initialReview, context.getProjectSettings(),
                    parameters.get("text"));
        } catch (RemoteException e) {
            throw new IllegalStateException(
                    "Impossible to create an issue on JIRA. A problem occured with the remote server: "
                            + e.getMessage(),
                    e);
        }

        createComment(issue, review, context, parameters);
        // and add the property
        review.setProperty(TestTrackConstants.REVIEW_DATA_PROPERTY_KEY, issue.getKey());
    }

    protected void createComment(RemoteIssue issue, MutableReview review, WorkflowContext context,
            Map<String, String> parameters) {
        Comment newComment = review.createComment();
        newComment.setUserId(context.getUserId());
        newComment.setMarkdownText(generateCommentText(issue, context, parameters));
    }

    protected String generateCommentText(RemoteIssue issue, WorkflowContext context,
            Map<String, String> parameters) {
        StringBuilder message = new StringBuilder();
        String text = parameters.get("text");
        if (!StringUtils.isBlank(text)) {
            message.append(text);
            message.append("\n\n");
        }
        message.append("Review linked to JIRA issue: ");
        message.append(context.getProjectSettings().getString(TestTrackConstants.SERVER_URL_PROPERTY));
        message.append("/browse/");
        message.append(issue.getKey());
        return message.toString();
    }

}