Добавлен модуль Kubernetes MCP с DI и диагностикой ошибок
This commit is contained in:
205
LazyBear.MCP/Services/Kubernetes/K8sConfigTools.cs
Normal file
205
LazyBear.MCP/Services/Kubernetes/K8sConfigTools.cs
Normal file
@@ -0,0 +1,205 @@
|
||||
using System.ComponentModel;
|
||||
using k8s;
|
||||
using k8s.Autorest;
|
||||
using ModelContextProtocol.Server;
|
||||
|
||||
namespace LazyBear.MCP.Services.Kubernetes;
|
||||
|
||||
[McpServerToolType]
|
||||
public sealed class K8sConfigTools(K8sClientProvider clientProvider, IConfiguration configuration)
|
||||
{
|
||||
private readonly IKubernetes? _client = clientProvider.Client;
|
||||
private readonly string? _clientInitializationError = clientProvider.InitializationError;
|
||||
private readonly string _defaultNamespace = configuration["Kubernetes:DefaultNamespace"] ?? "default";
|
||||
|
||||
[McpServerTool, Description("Список ConfigMap в namespace")]
|
||||
public async Task<string> ListConfigMaps(
|
||||
[Description("Namespace Kubernetes")] string? @namespace = null,
|
||||
CancellationToken cancellationToken = default)
|
||||
{
|
||||
var ns = ResolveNamespace(@namespace);
|
||||
if (!TryGetClient(out var client, out var clientError))
|
||||
{
|
||||
return clientError;
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
var configMaps = await client.CoreV1.ListNamespacedConfigMapAsync(ns, cancellationToken: cancellationToken);
|
||||
|
||||
if (configMaps.Items.Count == 0)
|
||||
{
|
||||
return $"В namespace '{ns}' configMap не найдены.";
|
||||
}
|
||||
|
||||
var lines = configMaps.Items.Select(cm =>
|
||||
{
|
||||
var name = cm.Metadata?.Name ?? "unknown";
|
||||
var keyCount = cm.Data?.Count ?? 0;
|
||||
return $"{name}: keys={keyCount}";
|
||||
});
|
||||
|
||||
return $"ConfigMap namespace '{ns}':\n{string.Join('\n', lines)}";
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
return FormatError("list_config_maps", ns, ex);
|
||||
}
|
||||
}
|
||||
|
||||
[McpServerTool, Description("Получить данные ConfigMap")]
|
||||
public async Task<string> GetConfigMapData(
|
||||
[Description("Имя ConfigMap")] string name,
|
||||
[Description("Namespace Kubernetes")] string? @namespace = null,
|
||||
CancellationToken cancellationToken = default)
|
||||
{
|
||||
var ns = ResolveNamespace(@namespace);
|
||||
if (!TryGetClient(out var client, out var clientError))
|
||||
{
|
||||
return clientError;
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
var configMap = await client.CoreV1.ReadNamespacedConfigMapAsync(name, ns, cancellationToken: cancellationToken);
|
||||
|
||||
if (configMap.Data is null || configMap.Data.Count == 0)
|
||||
{
|
||||
return $"ConfigMap '{name}' в namespace '{ns}' не содержит данных.";
|
||||
}
|
||||
|
||||
var lines = configMap.Data.Select(item => $"{item.Key}={item.Value}");
|
||||
return $"Данные ConfigMap '{name}' namespace '{ns}':\n{string.Join('\n', lines)}";
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
return FormatError("get_config_map_data", ns, ex, name);
|
||||
}
|
||||
}
|
||||
|
||||
[McpServerTool, Description("Список Secret в namespace (без значений)")]
|
||||
public async Task<string> ListSecrets(
|
||||
[Description("Namespace Kubernetes")] string? @namespace = null,
|
||||
CancellationToken cancellationToken = default)
|
||||
{
|
||||
var ns = ResolveNamespace(@namespace);
|
||||
if (!TryGetClient(out var client, out var clientError))
|
||||
{
|
||||
return clientError;
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
var secrets = await client.CoreV1.ListNamespacedSecretAsync(ns, cancellationToken: cancellationToken);
|
||||
|
||||
if (secrets.Items.Count == 0)
|
||||
{
|
||||
return $"В namespace '{ns}' secret не найдены.";
|
||||
}
|
||||
|
||||
var lines = secrets.Items.Select(secret =>
|
||||
{
|
||||
var name = secret.Metadata?.Name ?? "unknown";
|
||||
var type = secret.Type ?? "Opaque";
|
||||
return $"{name}: type={type}";
|
||||
});
|
||||
|
||||
return $"Secret namespace '{ns}':\n{string.Join('\n', lines)}";
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
return FormatError("list_secrets", ns, ex);
|
||||
}
|
||||
}
|
||||
|
||||
[McpServerTool, Description("Ключи Secret без значений")]
|
||||
public async Task<string> GetSecretKeys(
|
||||
[Description("Имя Secret")] string name,
|
||||
[Description("Namespace Kubernetes")] string? @namespace = null,
|
||||
CancellationToken cancellationToken = default)
|
||||
{
|
||||
var ns = ResolveNamespace(@namespace);
|
||||
if (!TryGetClient(out var client, out var clientError))
|
||||
{
|
||||
return clientError;
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
var secret = await client.CoreV1.ReadNamespacedSecretAsync(name, ns, cancellationToken: cancellationToken);
|
||||
|
||||
var keys = new HashSet<string>(StringComparer.Ordinal);
|
||||
if (secret.Data is not null)
|
||||
{
|
||||
foreach (var key in secret.Data.Keys)
|
||||
{
|
||||
keys.Add(key);
|
||||
}
|
||||
}
|
||||
|
||||
if (secret.StringData is not null)
|
||||
{
|
||||
foreach (var key in secret.StringData.Keys)
|
||||
{
|
||||
keys.Add(key);
|
||||
}
|
||||
}
|
||||
|
||||
if (keys.Count == 0)
|
||||
{
|
||||
return $"Secret '{name}' в namespace '{ns}' не содержит ключей.";
|
||||
}
|
||||
|
||||
return $"Ключи Secret '{name}' namespace '{ns}': {string.Join(", ", keys.OrderBy(k => k))}";
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
return FormatError("get_secret_keys", ns, ex, name);
|
||||
}
|
||||
}
|
||||
|
||||
private string ResolveNamespace(string? @namespace)
|
||||
{
|
||||
return string.IsNullOrWhiteSpace(@namespace) ? _defaultNamespace : @namespace;
|
||||
}
|
||||
|
||||
private static string BuildClientInitializationError()
|
||||
{
|
||||
return "Kubernetes клиент не инициализирован. Проверьте Kubernetes:KubeconfigPath или in-cluster окружение (KUBERNETES_SERVICE_HOST/KUBERNETES_SERVICE_PORT).";
|
||||
}
|
||||
|
||||
private bool TryGetClient(out IKubernetes client, out string error)
|
||||
{
|
||||
if (_client is null)
|
||||
{
|
||||
client = null!;
|
||||
var details = string.IsNullOrWhiteSpace(_clientInitializationError)
|
||||
? string.Empty
|
||||
: $" Детали: {_clientInitializationError}";
|
||||
error = BuildClientInitializationError() + details;
|
||||
return false;
|
||||
}
|
||||
|
||||
client = _client;
|
||||
error = string.Empty;
|
||||
return true;
|
||||
}
|
||||
|
||||
private static string FormatError(string toolName, string @namespace, Exception exception, string? resourceName = null)
|
||||
{
|
||||
var resourcePart = string.IsNullOrWhiteSpace(resourceName) ? string.Empty : $", resource='{resourceName}'";
|
||||
|
||||
if (exception is 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}";
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user