GitHubRemoteFeatures.cs

Go to the documentation of this file.

using System;
using System.Threading;
using System.Threading.Tasks;
 
using Microsoft.Extensions.Logging;
 
using Octokit;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Host.Utils.GitHub;
 
namespace Tgstation.Server.Host.Components.Repository
{
    sealed class GitHubRemoteFeatures : GitRemoteFeaturesBase
    {
        public override string TestMergeRefSpecFormatter => "pull/{0}/head:{1}";
 
        public override string TestMergeLocalBranchNameFormatter => "pull/{0}/headrefs/heads/{1}";
 
        public override RemoteGitProvider? RemoteGitProvider => Api.Models.RemoteGitProvider.GitHub;
 
        readonly IGitHubServiceFactory gitHubServiceFactory;
 
        public GitHubRemoteFeatures(IGitHubServiceFactory gitHubServiceFactory, ILogger<GitHubRemoteFeatures> logger, Uri remoteUrl)
            : base(logger, remoteUrl)
        {
            this.gitHubServiceFactory = gitHubServiceFactory ?? throw new ArgumentNullException(nameof(gitHubServiceFactory));
        }
 
        public override async ValueTask<string?> TransformRepositoryPassword(string? rawPassword, CancellationToken cancellationToken)
        {
            if (rawPassword == null)
                return null;
 
            var gitHubService = await gitHubServiceFactory.CreateService(
                rawPassword,
                new RepositoryIdentifier(this),
                cancellationToken);
            return gitHubService?.GetGitPassword() ?? rawPassword;
        }
 
        protected override async ValueTask<Models.TestMerge> GetTestMergeImpl(
            TestMergeParameters parameters,
            RepositorySettings repositorySettings,
            CancellationToken cancellationToken)
        {
            var gitHubService = repositorySettings.AccessToken != null
                ? await gitHubServiceFactory.CreateService(
                    repositorySettings.AccessToken,
                    new RepositoryIdentifier(this),
                    cancellationToken)
                : await gitHubServiceFactory.CreateService(cancellationToken);
 
            PullRequest? pr = null;
            ApiException? exception = null;
            string? errorMessage = null;
            if (gitHubService == null)
                errorMessage = "GITHUB API ERROR: AUTH FAILURE";
            else
                try
                {
                    pr = await gitHubService.GetPullRequest(RemoteRepositoryOwner, RemoteRepositoryName, parameters.Number, cancellationToken);
                }
                catch (RateLimitExceededException ex)
                {
                    // you look at your anonymous access and sigh
                    errorMessage = "GITHUB API ERROR: RATE LIMITED";
                    exception = ex;
                }
                catch (AuthorizationException ex)
                {
                    errorMessage = "GITHUB API ERROR: BAD CREDENTIALS";
                    exception = ex;
                }
                catch (NotFoundException ex)
                {
                    // you look at your shithub and sigh
                    errorMessage = "GITHUB API ERROR: PULL REQUEST NOT FOUND";
                    exception = ex;
                }
 
            if (exception != null)
                Logger.LogWarning(exception, "Error retrieving pull request metadata!");
 
            var revisionToUse = parameters.TargetCommitSha == null
                || pr?.Head.Sha.StartsWith(parameters.TargetCommitSha, StringComparison.OrdinalIgnoreCase) == true
                ? pr?.Head.Sha
                : parameters.TargetCommitSha;
 
            var testMerge = new Models.TestMerge
            {
                Author = pr?.User.Login ?? errorMessage,
                BodyAtMerge = pr?.Body ?? errorMessage ?? String.Empty,
                TitleAtMerge = pr?.Title ?? errorMessage ?? String.Empty,
                Comment = parameters.Comment,
                Number = parameters.Number,
                TargetCommitSha = revisionToUse,
                Url = pr?.HtmlUrl ?? $"https://github.com/{RemoteRepositoryOwner}/{RemoteRepositoryName}/pull/{parameters.Number}",
            };
 
            return testMerge;
        }
    }
}