108 lines
5.0 KiB
C#
108 lines
5.0 KiB
C#
using System.ComponentModel;
|
|
using System.Text.RegularExpressions;
|
|
using k8s;
|
|
using Microsoft.Extensions.Logging;
|
|
using ModelContextProtocol.Server;
|
|
|
|
namespace LazyBear.MCP.Services.Kubernetes;
|
|
|
|
[McpServerToolType]
|
|
public abstract class KubernetesToolsBase(K8sClientProvider clientProvider, IConfiguration configuration, ILogger? logger = null)
|
|
{
|
|
protected readonly IKubernetes? _client = clientProvider.Client;
|
|
protected readonly string? _clientInitializationError = clientProvider.InitializationError;
|
|
protected readonly string _defaultNamespace = configuration["Kubernetes:DefaultNamespace"] ?? "default";
|
|
protected readonly ILogger? _logger = logger;
|
|
|
|
protected static readonly Regex NamespaceRegex = new(@"^[a-z0-9]([-a-z0-9]*[a-z0-9])?$", RegexOptions.Compiled, TimeSpan.FromSeconds(1));
|
|
protected static readonly Regex ResourceNameRegex = new(@"^[a-z0-9]([-a-z0-9]*[a-z0-9])?$", RegexOptions.Compiled, TimeSpan.FromSeconds(1));
|
|
|
|
protected string ResolveNamespace(string? @namespace)
|
|
{
|
|
return string.IsNullOrWhiteSpace(@namespace) ? _defaultNamespace : @namespace;
|
|
}
|
|
|
|
protected void ValidateNamespace(string value, string parameterName)
|
|
{
|
|
if (string.IsNullOrWhiteSpace(value))
|
|
{
|
|
throw new ArgumentException($"Namespace не может быть пустым", parameterName);
|
|
}
|
|
if (value.Length > 63)
|
|
{
|
|
throw new ArgumentException($"Namespace не может быть длиннее 63 символов", parameterName);
|
|
}
|
|
if (!NamespaceRegex.IsMatch(value))
|
|
{
|
|
throw new ArgumentException($"Невалидный формат namespace: '{value}'. Только lowercase alphanumetic, -, не может начинаться/заканчиваться на -", parameterName);
|
|
}
|
|
}
|
|
|
|
protected void ValidateResourceName(string value, string parameterName)
|
|
{
|
|
if (string.IsNullOrWhiteSpace(value))
|
|
{
|
|
throw new ArgumentException($"Resource name не может быть пустым", parameterName);
|
|
}
|
|
if (value.Length > 253)
|
|
{
|
|
throw new ArgumentException($"Resource name не может быть длиннее 253 символов", parameterName);
|
|
}
|
|
if (!ResourceNameRegex.IsMatch(value))
|
|
{
|
|
throw new ArgumentException($"Невалидный формат: '{value}'. Только lowercase alphanumetic, -, не может начинаться/заканчиваться на -", parameterName);
|
|
}
|
|
}
|
|
|
|
protected bool TryGetClient(out IKubernetes client, out string error)
|
|
{
|
|
if (_client is null)
|
|
{
|
|
client = null!;
|
|
var details = string.IsNullOrWhiteSpace(_clientInitializationError)
|
|
? string.Empty
|
|
: $" Детали: {_clientInitializationError}";
|
|
error = "Kubernetes клиент не инициализирован. Проверьте Kubernetes:KubeconfigPath или in-cluster окружение (KUBERNETES_SERVICE_HOST/KUBERNETES_SERVICE_PORT)." + details;
|
|
return false;
|
|
}
|
|
|
|
client = _client;
|
|
error = string.Empty;
|
|
return true;
|
|
}
|
|
|
|
protected string FormatError(string toolName, string @namespace, Exception exception, string? resourceName = null)
|
|
{
|
|
_logger?.LogError(exception, "Ошибка Kubernetes в tool '{ToolName}' (namespace='{Namespace}'{ResourcePart}): {ExceptionMessage}",
|
|
toolName, @namespace,
|
|
string.IsNullOrWhiteSpace(resourceName) ? string.Empty : $", resource='{resourceName}'",
|
|
exception.Message);
|
|
|
|
var resourcePart = string.IsNullOrWhiteSpace(resourceName) ? string.Empty : $", resource='{resourceName}'";
|
|
|
|
if (exception is global::k8s.Autorest.HttpOperationException httpException)
|
|
{
|
|
var statusCode = httpException.Response?.StatusCode;
|
|
var reason = httpException.Response?.ReasonPhrase;
|
|
var body = string.IsNullOrWhiteSpace(httpException.Response?.Content)
|
|
? "-"
|
|
: httpException.Response!.Content;
|
|
|
|
return $"Ошибка Kubernetes в tool '{toolName}' (namespace='{@namespace}'{resourcePart}): status={(int?)statusCode ?? 0} {reason}, details={body}";
|
|
}
|
|
|
|
return $"Ошибка Kubernetes в tool '{toolName}' (namespace='{@namespace}'{resourcePart}): {exception.GetType().Name}: {exception.Message}";
|
|
}
|
|
|
|
protected string FormatException(string toolName, Exception exception, string? resource = null)
|
|
{
|
|
_logger?.LogError(exception, "Ошибка exception в tool '{ToolName}'{ResourcePart}: {ExceptionMessage}",
|
|
toolName,
|
|
string.IsNullOrWhiteSpace(resource) ? string.Empty : $", resource='{resource}'",
|
|
exception.Message);
|
|
|
|
var resourcePart = string.IsNullOrWhiteSpace(resource) ? string.Empty : $", resource='{resource}'";
|
|
return $"Ошибка в tool '{toolName}'{resourcePart}: {exception.GetType().Name}: {exception.Message}";
|
|
}
|
|
}
|