feat: добавить поддержку GitLab (api, clients, tools) и обновить документацию

This commit is contained in:
2026-04-14 12:57:47 +03:00
parent e96bab114e
commit b5fe2623b3
17 changed files with 3479 additions and 39 deletions

View File

@@ -0,0 +1,358 @@
using System.ComponentModel;
using System.Text.Json;
using LazyBear.MCP.Services.ToolRegistry;
using ModelContextProtocol.Server;
using RestSharp;
namespace LazyBear.MCP.Services.GitLab;
public sealed class GitLabBranchTools(
GitLabClientProvider provider,
IConfiguration configuration,
ToolRegistryService registry)
{
private readonly string _token = configuration["GitLab:Token"] ?? string.Empty;
private const string ModuleName = "GitLab";
private bool TryCheckEnabled(string toolName, out string error)
{
if (!registry.IsToolEnabled(ModuleName, toolName))
{
error = $"Инструмент '{toolName}' модуля GitLab отключён в TUI.";
return false;
}
error = string.Empty;
return true;
}
private bool TryGetClient(out RestSharp.IRestClient client, out string error)
{
if (provider.InitializationError is not null)
{
client = null!;
error = $"GitLab клиент не инициализирован. Детали: {provider.InitializationError}";
return false;
}
var clientInstance = provider.GetClient();
if (clientInstance is null)
{
client = null!;
error = "GitLab клиент не создан.";
return false;
}
client = clientInstance.RestClient;
error = string.Empty;
return true;
}
private string FormatResponseError(string toolName, RestResponse response, string? resource = null)
{
var resourcePart = string.IsNullOrWhiteSpace(resource) ? string.Empty : $", resource='{resource}'";
var body = string.IsNullOrWhiteSpace(response.Content) ? "-" : response.Content;
return $"Ошибка GitLab в tool '{toolName}'{resourcePart}: status={(int)response.StatusCode}, details={body}";
}
private string FormatException(string toolName, Exception exception, string? resource = null)
{
var resourcePart = string.IsNullOrWhiteSpace(resource) ? string.Empty : $", resource='{resource}'";
return $"Ошибка GitLab в tool '{toolName}'{resourcePart}: {exception.GetType().Name}: {exception.Message}";
}
private static string? GetNestedString(JsonElement element, params string[] path)
{
var current = element;
foreach (var segment in path)
{
if (!current.TryGetProperty(segment, out current))
{
return null;
}
}
return current.ValueKind == JsonValueKind.String ? current.GetString() : current.ToString();
}
[McpServerTool, Description("Получить список веток GitLab проекта")]
public async Task<string> ListBranches(
[Description("ID проекта")] int projectId,
CancellationToken cancellationToken = default)
{
if (!TryCheckEnabled("ListBranches", out var enabledError)) return enabledError;
if (projectId <= 0)
{
return "ID проекта GitLab некорректно задан.";
}
if (!TryGetClient(out var client, out var error)) return error;
try
{
var request = new RestRequest($"/projects/{projectId}/repository/branches", Method.Get);
request.AddHeader("Accept", "application/json");
request.AddHeader("PRIVATE-TOKEN", _token);
request.AddQueryParameter("per_page", "100");
var response = await client.ExecuteAsync(request, cancellationToken);
if (!response.IsSuccessful || string.IsNullOrWhiteSpace(response.Content))
{
return FormatResponseError("list_branches", response, $"/projects/{projectId}/repository/branches");
}
using var document = JsonDocument.Parse(response.Content);
var root = document.RootElement;
if (root.ValueKind != JsonValueKind.Array || root.GetArrayLength() == 0)
{
return $"Ветки в проекте #{projectId} не найдены.";
}
var lines = new List<string>();
foreach (var branch in root.EnumerateArray())
{
var name = GetNestedString(branch, "name") ?? "-";
var isDefault = GetNestedString(branch, "default") ?? "false";
var isProtected = GetNestedString(branch, "protected") ?? "false";
var commit = GetNestedString(branch, "commit", "short_id") ?? GetNestedString(branch, "commit", "id") ?? "-";
lines.Add($"{name} (default={isDefault}, protected={isProtected}, commit={commit})");
}
return $"Ветки проекта #{projectId} ({root.GetArrayLength()} шт.):{Environment.NewLine}{string.Join(Environment.NewLine, lines)}";
}
catch (Exception ex)
{
return FormatException("list_branches", ex);
}
}
[McpServerTool, Description("Получить ветку GitLab проекта")]
public async Task<string> GetBranch(
[Description("ID проекта")] int projectId,
[Description("Имя ветки")] string branchName,
CancellationToken cancellationToken = default)
{
if (!TryCheckEnabled("GetBranch", out var enabledError)) return enabledError;
if (projectId <= 0)
{
return "ID проекта GitLab некорректно задан.";
}
if (string.IsNullOrWhiteSpace(branchName))
{
return "Имя ветки GitLab не может быть пустым.";
}
if (!TryGetClient(out var client, out var error)) return error;
try
{
var encoded = Uri.EscapeDataString(branchName);
var request = new RestRequest($"/projects/{projectId}/repository/branches/{encoded}", Method.Get);
request.AddHeader("Accept", "application/json");
request.AddHeader("PRIVATE-TOKEN", _token);
var response = await client.ExecuteAsync(request, cancellationToken);
if (!response.IsSuccessful || string.IsNullOrWhiteSpace(response.Content))
{
return FormatResponseError("get_branch", response, $"/projects/{projectId}/repository/branches/{encoded}");
}
using var document = JsonDocument.Parse(response.Content);
var root = document.RootElement;
var name = GetNestedString(root, "name") ?? branchName;
var isDefault = GetNestedString(root, "default") ?? "false";
var isProtected = GetNestedString(root, "protected") ?? "false";
var canPush = GetNestedString(root, "can_push") ?? "false";
var commitId = GetNestedString(root, "commit", "id") ?? "-";
return $"Ветка '{name}' проекта #{projectId}:{Environment.NewLine}default={isDefault}, protected={isProtected}, can_push={canPush}{Environment.NewLine}commit={commitId}";
}
catch (Exception ex)
{
return FormatException("get_branch", ex);
}
}
[McpServerTool, Description("Создать ветку GitLab проекта")]
public async Task<string> CreateBranch(
[Description("ID проекта")] int projectId,
[Description("Имя новой ветки")] string branchName,
[Description("Ветка или SHA-реф источника")] string @ref,
CancellationToken cancellationToken = default)
{
if (!TryCheckEnabled("CreateBranch", out var enabledError)) return enabledError;
if (projectId <= 0)
{
return "ID проекта GitLab некорректно задан.";
}
if (string.IsNullOrWhiteSpace(branchName))
{
return "Имя новой ветки GitLab не может быть пустым.";
}
if (string.IsNullOrWhiteSpace(@ref))
{
return "Источник ветки (ref) GitLab не может быть пустым.";
}
if (!TryGetClient(out var client, out var error)) return error;
try
{
var request = new RestRequest($"/projects/{projectId}/repository/branches", Method.Post);
request.AddHeader("Accept", "application/json");
request.AddHeader("PRIVATE-TOKEN", _token);
request.AddHeader("Content-Type", "application/json");
request.AddJsonBody(new { branch = branchName, @ref });
var response = await client.ExecuteAsync(request, cancellationToken);
if (!response.IsSuccessful || string.IsNullOrWhiteSpace(response.Content))
{
return FormatResponseError("create_branch", response, $"/projects/{projectId}/repository/branches");
}
using var document = JsonDocument.Parse(response.Content);
var root = document.RootElement;
var name = GetNestedString(root, "name") ?? branchName;
var commit = GetNestedString(root, "commit", "short_id") ?? GetNestedString(root, "commit", "id") ?? "-";
return $"Ветка '{name}' успешно создана в проекте #{projectId}. commit={commit}";
}
catch (Exception ex)
{
return FormatException("create_branch", ex);
}
}
[McpServerTool, Description("Удалить ветку GitLab проекта")]
public async Task<string> DeleteBranch(
[Description("ID проекта")] int projectId,
[Description("Имя ветки")] string branchName,
CancellationToken cancellationToken = default)
{
if (!TryCheckEnabled("DeleteBranch", out var enabledError)) return enabledError;
if (projectId <= 0)
{
return "ID проекта GitLab некорректно задан.";
}
if (string.IsNullOrWhiteSpace(branchName))
{
return "Имя ветки GitLab не может быть пустым.";
}
if (!TryGetClient(out var client, out var error)) return error;
try
{
var encoded = Uri.EscapeDataString(branchName);
var request = new RestRequest($"/projects/{projectId}/repository/branches/{encoded}", Method.Delete);
request.AddHeader("Accept", "application/json");
request.AddHeader("PRIVATE-TOKEN", _token);
var response = await client.ExecuteAsync(request, cancellationToken);
if (!response.IsSuccessful)
{
return FormatResponseError("delete_branch", response, $"/projects/{projectId}/repository/branches/{encoded}");
}
return $"Ветка '{branchName}' успешно удалена из проекта #{projectId}.";
}
catch (Exception ex)
{
return FormatException("delete_branch", ex);
}
}
[McpServerTool, Description("Защитить ветку GitLab проекта")]
public async Task<string> ProtectBranch(
[Description("ID проекта")] int projectId,
[Description("Имя ветки")] string branchName,
CancellationToken cancellationToken = default)
{
if (!TryCheckEnabled("ProtectBranch", out var enabledError)) return enabledError;
if (projectId <= 0)
{
return "ID проекта GitLab некорректно задан.";
}
if (string.IsNullOrWhiteSpace(branchName))
{
return "Имя ветки GitLab не может быть пустым.";
}
if (!TryGetClient(out var client, out var error)) return error;
try
{
var request = new RestRequest($"/projects/{projectId}/protected_branches", Method.Post);
request.AddHeader("Accept", "application/json");
request.AddHeader("PRIVATE-TOKEN", _token);
request.AddHeader("Content-Type", "application/json");
request.AddJsonBody(new { name = branchName });
var response = await client.ExecuteAsync(request, cancellationToken);
if (!response.IsSuccessful || string.IsNullOrWhiteSpace(response.Content))
{
return FormatResponseError("protect_branch", response, $"/projects/{projectId}/protected_branches");
}
return $"Ветка '{branchName}' успешно защищена в проекте #{projectId}.";
}
catch (Exception ex)
{
return FormatException("protect_branch", ex);
}
}
[McpServerTool, Description("Снять защиту с ветки GitLab проекта")]
public async Task<string> UnprotectBranch(
[Description("ID проекта")] int projectId,
[Description("Имя ветки")] string branchName,
CancellationToken cancellationToken = default)
{
if (!TryCheckEnabled("UnprotectBranch", out var enabledError)) return enabledError;
if (projectId <= 0)
{
return "ID проекта GitLab некорректно задан.";
}
if (string.IsNullOrWhiteSpace(branchName))
{
return "Имя ветки GitLab не может быть пустым.";
}
if (!TryGetClient(out var client, out var error)) return error;
try
{
var encoded = Uri.EscapeDataString(branchName);
var request = new RestRequest($"/projects/{projectId}/protected_branches/{encoded}", Method.Delete);
request.AddHeader("Accept", "application/json");
request.AddHeader("PRIVATE-TOKEN", _token);
var response = await client.ExecuteAsync(request, cancellationToken);
if (!response.IsSuccessful)
{
return FormatResponseError("unprotect_branch", response, $"/projects/{projectId}/protected_branches/{encoded}");
}
return $"Защита с ветки '{branchName}' успешно снята в проекте #{projectId}.";
}
catch (Exception ex)
{
return FormatException("unprotect_branch", ex);
}
}
}