GitLabRemoteFeatures.cs

Go to the documentation of this file.

using System;
using System.Globalization;
using System.Threading;
using System.Threading.Tasks;
 
using Microsoft.Extensions.Logging;
 
using StrawberryShake;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Host.Utils.GitLab.GraphQL;
 
namespace Tgstation.Server.Host.Components.Repository
{
    sealed class GitLabRemoteFeatures : GitRemoteFeaturesBase
    {
        public const string GitLabUrl = "https://gitlab.com";
 
        public override string TestMergeRefSpecFormatter => "merge-requests/{0}/head:{1}";
 
        public override string TestMergeLocalBranchNameFormatter => "merge-requests/{0}/headrefs/heads/{1}";
 
        public override RemoteGitProvider? RemoteGitProvider => Api.Models.RemoteGitProvider.GitLab;
 
        public GitLabRemoteFeatures(ILogger<GitLabRemoteFeatures> logger, Uri remoteUrl)
            : base(logger, remoteUrl)
        {
        }
 
        protected override async ValueTask<Models.TestMerge> GetTestMergeImpl(
            TestMergeParameters parameters,
            RepositorySettings repositorySettings,
            CancellationToken cancellationToken)
        {
            await using var client = await GraphQLGitLabClientFactory.CreateClient(repositorySettings.AccessToken);
            try
            {
                var operationResult = await client.GraphQL.GetMergeRequest.ExecuteAsync(
                    $"{RemoteRepositoryOwner}/{RemoteRepositoryName}",
                    parameters.Number.ToString(CultureInfo.InvariantCulture),
                    cancellationToken);
 
                operationResult.EnsureNoErrors();
                var mr = operationResult.Data?.Project?.MergeRequest ?? throw new InvalidOperationException("GitLab MergeRequest check returned null!");
 
                return new Models.TestMerge
                {
                    Author = mr.Author?.Username,
                    BodyAtMerge = mr.Description,
                    TitleAtMerge = mr.Title,
                    Comment = parameters.Comment,
                    Number = parameters.Number,
                    TargetCommitSha = mr.DiffHeadSha,
                    Url = mr.WebUrl,
                };
            }
            catch (Exception ex)
            {
                Logger.LogWarning(ex, "Error retrieving merge request metadata!");
 
                return new Models.TestMerge
                {
                    Author = ex.Message,
                    BodyAtMerge = ex.Message,
                    TitleAtMerge = ex.Message,
                    Comment = parameters.Comment,
                    Number = parameters.Number,
                    TargetCommitSha = parameters.TargetCommitSha,
                    Url = $"https://gitlab.com/{RemoteRepositoryOwner}/{RemoteRepositoryName}/-/merge_requests/{parameters.Number}",
                };
            }
        }
    }
}