auto_submit/lib/service/approver_service.dart

// Copyright 2022 The Flutter Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

import 'package:cocoon_server/logging.dart';
import 'package:github/github.dart' as github;

import '../service/config.dart';

/// Function signature for a [ApproverService] provider.
typedef ApproverServiceProvider = ApproverService Function(Config config);

/// Provides github PR approval services.
class ApproverService {
  const ApproverService(this.config);

  final Config config;

  /// Creates and returns a [ApproverService] using [config].
  static ApproverService defaultProvider(Config config) {
    return ApproverService(config);
  }

  /// Get the auto approval accounts from the configuration is any are supplied.
  Future<Set<String>> getAutoApprovalAccounts(
    github.RepositorySlug slug,
  ) async {
    final repositoryConfiguration = await config.getRepositoryConfiguration(
      slug,
    );
    final approvalAccounts = repositoryConfiguration.autoApprovalAccounts;
    return approvalAccounts;
  }

  Future<void> autoApproval(github.PullRequest pullRequest) async {
    final author = pullRequest.user!.login;
    final prNumber = pullRequest.number!;
    final slug = pullRequest.base!.repo!.slug();
    final approvalAccounts = await getAutoApprovalAccounts(
      github.RepositorySlug.full(pullRequest.base!.repo!.fullName),
    );

    log.info(
      'Determining auto approval of $author on ${slug.fullName}/$prNumber.',
    );

    // If there are auto_approvers let them approve the pull request.
    if (!approvalAccounts.contains(author)) {
      log.info(
        'Auto-review ignored for $author on ${slug.fullName}/$prNumber.',
      );
    } else {
      log.info('Auto approval detected on ${slug.fullName}/$prNumber.');
      await _approve(pullRequest, author);
    }
  }

  Future<void> _approve(github.PullRequest pullRequest, String? author) async {
    final slug = pullRequest.base!.repo!.slug();
    final botClient = await config.createFlutterGitHubBotClient(slug);

    final reviews = botClient.pullRequests.listReviews(
      slug,
      pullRequest.number!,
    );
    // TODO(ricardoamador) this will need to be refactored to make this code more general and
    // not applicable to only flutter.
    await for (github.PullRequestReview review in reviews) {
      if (review.user?.login == 'fluttergithubbot' &&
          review.state == 'APPROVED') {
        // Already approved.
        return;
      }
    }

    final review = github.CreatePullRequestReview(
      slug.owner,
      slug.name,
      pullRequest.number!,
      'APPROVE',
    );
    await botClient.pullRequests.createReview(slug, review);
    log.info('Review for ${slug.fullName}/${pullRequest.number} complete');
  }
}