GitLabRemoteFeatures.cs

Go to the documentation of this file.

using System;
using System.Threading;
using System.Threading.Tasks;
 
using GitLabApiClient;
using Microsoft.Extensions.Logging;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Api.Models.Internal;
using Tgstation.Server.Host.Extensions;
 
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 override string RemoteRepositoryOwner { get; }
 
        public override string RemoteRepositoryName { get; }
 
        public GitLabRemoteFeatures(ILogger<GitLabRemoteFeatures> logger, Uri remoteUrl)
            : base(logger, remoteUrl)
        {
            RemoteRepositoryOwner = remoteUrl.Segments[1].TrimEnd('/');
            RemoteRepositoryName = remoteUrl.Segments[2].TrimEnd('/');
            if (RemoteRepositoryName.EndsWith(".git", StringComparison.OrdinalIgnoreCase))
                RemoteRepositoryName = RemoteRepositoryName[0..^4];
        }
 
        protected override async Task<Models.TestMerge> GetTestMergeImpl(
            TestMergeParameters parameters,
            RepositorySettings repositorySettings,
            CancellationToken cancellationToken)
        {
            var client = repositorySettings.AccessToken != null
                ? new GitLabClient(GitLabUrl, repositorySettings.AccessToken)
                : new GitLabClient(GitLabUrl);
 
            try
            {
                var mr = await client
                    .MergeRequests
                    .GetAsync($"{RemoteRepositoryOwner}/{RemoteRepositoryName}", parameters.Number)
                    .WithToken(cancellationToken);
 
                var revisionToUse = parameters.TargetCommitSha == null
                    || mr.Sha.StartsWith(parameters.TargetCommitSha, StringComparison.OrdinalIgnoreCase)
                    ? mr.Sha
                    : parameters.TargetCommitSha;
 
                return new Models.TestMerge
                {
                    Author = mr.Author.Username,
                    BodyAtMerge = mr.Description,
                    TitleAtMerge = mr.Title,
                    Comment = parameters.Comment,
                    Number = parameters.Number,
                    TargetCommitSha = mr.Sha,
                    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 = ex.Message,
                };
            }
        }
    }
}