Merge branch 'main' into chunk-data

app/MindWork AI Studio/Tools/Services/EnterpriseEnvironmentService.cs

@@ -14,7 +14,7 @@ public sealed class EnterpriseEnvironmentService(ILogger<EnterpriseEnvironmentSe
     
     private static EnterpriseSecretSnapshot CURRENT_SECRET_SNAPSHOT;
 
-    private readonly record struct EnterpriseEnvironmentSnapshot(Guid ConfigurationId, string ConfigurationServerUrl, string? ETag);
+    private readonly record struct EnterpriseEnvironmentSnapshot(Guid ConfigurationId, string ConfigurationServerUrl, string Source, string SourceDetail, string Slot, string? ETag);
     
     private readonly record struct EnterpriseSecretSnapshot(bool HasSecret, string Fingerprint);
     
@@ -224,6 +224,9 @@ public sealed class EnterpriseEnvironmentService(ILogger<EnterpriseEnvironmentSe
             .Select(environment => new EnterpriseEnvironmentSnapshot(
                 environment.ConfigurationId,
                 NormalizeServerUrl(environment.ConfigurationServerUrl),
+                environment.Source,
+                environment.SourceDetail,
+                environment.Slot,
                 environment.ETag?.ToString()))
             .OrderBy(environment => environment.ConfigurationId)
             .ToList();

app/MindWork AI Studio/Tools/Services/RustService.App.cs

@@ -120,6 +120,12 @@ public sealed partial class RustService
         return await response.Content.ReadAsStringAsync();
     }
 
+    public async Task<RuntimeInfoResponse> GetRuntimeInfo()
+    {
+        var response = await this.http.GetFromJsonAsync<RuntimeInfoResponse>("/system/runtime/info", this.jsonRustSerializerOptions);
+        return response;
+    }
+
     /// <summary>
     /// Requests the Rust runtime to exit the entire desktop application.
     /// </summary>

app/MindWork AI Studio/Tools/Services/RustService.Databases.cs

@@ -1,44 +1,53 @@
-using AIStudio.Tools.Databases;
-using AIStudio.Tools.Rust;
-
 namespace AIStudio.Tools.Services;
 
 public sealed partial class RustService
 {
-    public async Task<QdrantInfo> GetQdrantInfo(CancellationToken cancellationToken = default)
+    public async Task<TDatabaseInfo> GetDatabaseInfo<TDatabaseInfo>(
+        string databaseName,
+        string infoPath,
+        Func<string, TDatabaseInfo> unavailableFactory,
+        CancellationToken cancellationToken = default)
     {
         try
         {
             using var cts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken);
             cts.CancelAfter(TimeSpan.FromSeconds(45));
-            
-            return await this.http.GetFromJsonAsync<QdrantInfo>("/system/qdrant/info", this.jsonRustSerializerOptions, cts.Token);
+
+            var databaseInfo = await this.http.GetFromJsonAsync<TDatabaseInfo>(infoPath, this.jsonRustSerializerOptions, cts.Token);
+            return databaseInfo ?? unavailableFactory("The database information response was empty.");
         }
         catch (OperationCanceledException) when (cancellationToken.IsCancellationRequested)
         {
             if(this.logger is not null)
-                this.logger.LogWarning("Fetching Qdrant info from Rust service was cancelled by caller.");
+                this.logger.LogWarning("Fetching {DatabaseName} info from Rust service was cancelled by caller.", databaseName);
             else
-                Console.WriteLine("Fetching Qdrant info from Rust service was cancelled by caller.");
-            
-            return new QdrantInfo
-            {
-                Status = QdrantStatus.UNAVAILABLE,
-                UnavailableReason = "Operation cancelled by caller."
-            };
+                Console.WriteLine($"Fetching {databaseName} info from Rust service was cancelled by caller.");
+
+            return unavailableFactory("Operation cancelled by caller.");
         }
         catch (Exception e)
         {
             if(this.logger is not null)
-                this.logger.LogError(e, "Error while fetching Qdrant info from Rust service.");
+                this.logger.LogError(e, "Error while fetching {DatabaseName} info from Rust service.", databaseName);
             else
-                Console.WriteLine($"Error while fetching Qdrant info from Rust service: '{e}'.");
-            
-            return new QdrantInfo
-            {
-                Status = QdrantStatus.UNAVAILABLE,
-                UnavailableReason = e.Message
-            };
+                Console.WriteLine($"Error while fetching {databaseName} info from Rust service: '{e}'.");
+
+            return unavailableFactory(e.Message);
         }
     }
-}
+
+    public async Task ExecuteDatabaseOperation<TRequest>(string databaseName, string path, TRequest request, CancellationToken cancellationToken = default)
+    {
+        using var cts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken);
+        cts.CancelAfter(TimeSpan.FromMinutes(5));
+
+        using var response = await this.http.PostAsJsonAsync(path, request, this.jsonRustSerializerOptions, cts.Token);
+        response.EnsureSuccessStatusCode();
+
+        var operation = await response.Content.ReadFromJsonAsync<DatabaseOperationResponse>(this.jsonRustSerializerOptions, cts.Token);
+        if (operation is not { Success: true })
+            throw new InvalidOperationException(operation?.Issue ?? $"The {databaseName} operation failed.");
+    }
+
+    private sealed record DatabaseOperationResponse(bool Success, string Issue);
+}

app/MindWork AI Studio/Tools/Services/RustService.Enterprise.cs

@@ -47,7 +47,7 @@ public sealed partial class RustService
         foreach (var config in configs)
         {
             if (Guid.TryParse(config.Id, out var id))
-                environments.Add(new EnterpriseEnvironment(config.ServerUrl, id, null));
+                environments.Add(new EnterpriseEnvironment(config.ServerUrl, id, config.Source, config.SourceDetail, config.Slot, null));
             else
                 this.logger!.LogWarning($"Skipping enterprise config with invalid ID: '{config.Id}'.");
         }