PullRequestsBuildListener.java

/**
 * Copyright 2014 ArcBees Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */

package com.arcbees.pullrequest;

import java.io.IOException;
import java.util.concurrent.ExecutorService;
import java.util.logging.Level;
import java.util.logging.Logger;

import org.jetbrains.annotations.NotNull;

import jetbrains.buildServer.buildTriggers.BuildTriggerDescriptor;
import jetbrains.buildServer.serverSide.Branch;
import jetbrains.buildServer.serverSide.BuildServerAdapter;
import jetbrains.buildServer.serverSide.BuildServerListener;
import jetbrains.buildServer.serverSide.SBuildType;
import jetbrains.buildServer.serverSide.SRunningBuild;
import jetbrains.buildServer.serverSide.executors.ExecutorServices;
import jetbrains.buildServer.util.EventDispatcher;
import jetbrains.buildServer.util.ExceptionUtil;

public class PullRequestsBuildListener {
    private static final Logger LOGGER = Logger.getLogger(PullRequestsBuildListener.class.getName());

    private final PullRequestStatusHandler statusHandler;
    private final ExecutorService executorService;

    public PullRequestsBuildListener(EventDispatcher<BuildServerListener> listener,
                                     ExecutorServices executorServices,
                                     PullRequestStatusHandler statusHandler) {
        this.statusHandler = statusHandler;
        executorService = executorServices.getLowPriorityExecutorService();

        listener.addListener(new BuildServerAdapter() {
            @Override
            public void buildStarted(@NotNull SRunningBuild build) {
                onBuildStatusChanged(build, BuildStatus.STARTING);
            }

            @Override
            public void buildInterrupted(@NotNull SRunningBuild build) {
                onBuildStatusChanged(build, BuildStatus.INTERRUPTED);
            }

            @Override
            public void buildFinished(@NotNull SRunningBuild build) {
                onBuildStatusChanged(build, BuildStatus.FINISHED);
            }
        });
    }

    private void onBuildStatusChanged(final SRunningBuild build,
                                      final BuildStatus buildStatus) {
        final BuildTriggerDescriptor trigger = getTrigger(build);

        if (trigger != null) {
            Branch branch = build.getBranch();
            if (branch != null) {
                handleBuildStatus(build, buildStatus, trigger);
            } else {
                LOGGER.severe("Unknown branch name");
            }
        }
    }

    private void handleBuildStatus(final SRunningBuild build,
                                   final BuildStatus buildStatus,
                                   final BuildTriggerDescriptor trigger) {
        executorService.submit(ExceptionUtil.catchAll("PullRequest Handler", new Runnable() {
            @Override
            public void run() {
                try {
                    statusHandler.handle(build, trigger, buildStatus);
                } catch (IOException e) {
                    LOGGER.log(Level.SEVERE, "Error updating pull request status.", e);
                }
            }
        }));
    }

    private BuildTriggerDescriptor getTrigger(SRunningBuild build) {
        SBuildType buildType = build.getBuildType();
        if (buildType == null) {
            return null;
        }

        for (BuildTriggerDescriptor trigger : buildType.getResolvedSettings().getBuildTriggersCollection()) {
            if (trigger.getType().equals(PullRequestsFeature.NAME)) {
                return trigger;
            }
        }

        return null;
    }
}