BuildCommitBuildListener.java

/*
 * Copyright 2016 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.feature;

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 com.arcbees.pullrequest.BuildStatus;

import jetbrains.buildServer.serverSide.Branch;
import jetbrains.buildServer.serverSide.BuildServerAdapter;
import jetbrains.buildServer.serverSide.BuildServerListener;
import jetbrains.buildServer.serverSide.SBuildFeatureDescriptor;
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 BuildCommitBuildListener {
    private static final Logger LOGGER = Logger.getLogger(BuildCommitBuildListener.class.getName());

    private final BuildCommitStatusHandler statusHandler;
    private final ExecutorService executorService;

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

        listener.addListener(new BuildServerAdapter() {
            @Override
            public void changesLoaded(@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(
            SRunningBuild build,
            BuildStatus buildStatus) {
        SBuildFeatureDescriptor feature = getFeature(build);

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

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

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

        for (SBuildFeatureDescriptor feature : buildType.getResolvedSettings().getBuildFeatures()) {
            if (BuildCommitFeature.NAME.equals(feature.getType())) {
                return feature;
            }
        }

        return null;
    }
}