check_run.go

// Copyright 2018 Palantir Technologies, 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 handler

import (
	"context"
	"encoding/json"

	"github.com/google/go-github/v43/github"
	"github.com/palantir/go-githubapp/githubapp"
	"github.com/pkg/errors"

	"github.com/ridge/bulldozer/pull"
)

type CheckRun struct {
	Config *ServerConfig
}

func (h *CheckRun) Handles() []string {
	return []string{"check_run"}
}

func handleCheckRun(config *ServerConfig, event github.CheckRunEvent) {
	ctx := context.Background()

	repo := event.GetRepo()
	installationID := githubapp.GetInstallationIDFromEvent(&event)

	ctx, logger := githubapp.PrepareRepoContext(ctx, installationID, repo)

	if event.GetAction() != "completed" {
		logger.Debug().Msgf("Doing nothing since check_run action was %q instead of 'completed'", event.GetAction())
	}

	client, err := config.ClientCreator.NewInstallationClient(installationID)
	if err != nil {
		logger.Error().Err(err).Msg("failed to instantiate github client")
		return
	}

	prs := event.GetCheckRun().PullRequests
	if len(prs) == 0 {
		logger.Debug().Msg("Doing nothing since status change event affects no open pull requests")
		return
	}

	for _, pr := range prs {
		// The PR included in the CheckRun response is very slim on information.
		// It does not contain the owner information or label information we
		// need to process the pull request.

		fullPR, _, err := client.PullRequests.Get(ctx, repo.GetOwner().GetLogin(), repo.GetName(), pr.GetNumber())
		if err != nil {
			logger.Error().Err(err).Msgf("failed to fetch PR number %q for CheckRun", pr.GetNumber())
			continue
		}
		pullCtx := pull.NewGithubContext(client, fullPR)

		logger := logger.With().Int(githubapp.LogKeyPRNum, pr.GetNumber()).Logger()
		if err := ProcessPullRequest(logger.WithContext(ctx), config, pullCtx, client, fullPR.GetBase().GetRef()); err != nil {
			logger.Error().Err(err).Msg("Error processing pull request")
		}
	}
}

func (h *CheckRun) Handle(ctx context.Context, eventType, deliveryID string, payload []byte) error {
	var event github.CheckRunEvent

	if err := json.Unmarshal(payload, &event); err != nil {
		return errors.Wrap(err, "failed to parse check_run event payload")
	}

	go handleCheckRun(h.Config, event)

	return nil
}

// type assertion
var _ githubapp.EventHandler = &CheckRun{}