add pgt simulate

Author: Jicheng Lu

src/Plugins/BotSharp.Plugin.Membase/Controllers/MembaseController.cs

@@ -448,4 +448,51 @@ public async Task<IActionResult> DeleteEdge(string graphId, string edgeId)
                 new { message = "An error occurred while deleting the edge.", error = ex.Message });
         }
     }
+
+    /// <summary>
+    /// Simulate a PGT definition
+    /// </summary>
+    /// <param name="graphId">The graph identifier</param>
+    /// <param name="definitionId">The PGT definition identifier</param>
+    /// <param name="request">The simulation request containing start node and options</param>
+    /// <returns>Simulation result with trace log and visited nodes</returns>
+#if DEBUG
+    [AllowAnonymous]
+#endif
+    [HttpPost("/membase/{graphId}/pgt-definitions/{definitionId}/simulate")]
+    [ProducesResponseType(StatusCodes.Status200OK)]
+    [ProducesResponseType(StatusCodes.Status400BadRequest)]
+    [ProducesResponseType(StatusCodes.Status500InternalServerError)]
+    public async Task<IActionResult> SimulatePgtDefinition(string graphId, string definitionId, [FromBody] PgtSimulationRequest request)
+    {
+        if (string.IsNullOrWhiteSpace(graphId))
+        {
+            return BadRequest("Graph ID cannot be empty.");
+        }
+
+        if (string.IsNullOrWhiteSpace(definitionId))
+        {
+            return BadRequest("Definition ID cannot be empty.");
+        }
+
+        if (string.IsNullOrWhiteSpace(request?.StartId))
+        {
+            return BadRequest("Start ID cannot be empty.");
+        }
+
+        try
+        {
+            request.Options ??= new();
+            request.Options.RunId = $"sim-{Guid.NewGuid()}";
+            var result = await _membaseApi.SimulatePgtDefinitionAsync(graphId, definitionId, request);
+            result.RunId = request.Options.RunId;
+            return Ok(result);
+        }
+        catch (Exception ex)
+        {
+            return StatusCode(
+                StatusCodes.Status500InternalServerError,
+                new { message = "An error occurred while simulating the PGT definition.", error = ex.Message });
+        }
+    }
 }

src/Plugins/BotSharp.Plugin.Membase/Interfaces/IMembaseApi.cs

@@ -45,4 +45,9 @@ public interface IMembaseApi
     [Delete("/graph/{graphId}/edge/{edgeId}")]
     Task DeleteEdgeAsync(string graphId, string edgeId);
     #endregion
+
+    #region PGT
+    [Post("/graph/{graphId}/pgt-definitions/{definitionId}/simulate")]
+    Task<PgtSimulationResponse> SimulatePgtDefinitionAsync(string graphId, string definitionId, [Body] PgtSimulationRequest request);
+    #endregion
 }

src/Plugins/BotSharp.Plugin.Membase/Models/Requests/PgtSimulationRequest.cs

@@ -0,0 +1,47 @@
+using System.Text.Json.Serialization;
+
+namespace BotSharp.Plugin.Membase.Models;
+
+public class PgtSimulationRequest
+{
+    public string StartId { get; set; } = string.Empty;
+
+    [JsonPropertyName("options")]
+    public PgtSimulationOptions? Options { get; set; }
+}
+
+public class PgtSimulationOptions
+{
+    [JsonPropertyName("max_depth")]
+    public int? MaxDepth { get; set; }
+
+    [JsonPropertyName("timeout_ms")]
+    public int? TimeoutMs { get; set; }
+
+    [JsonPropertyName("strategy")]
+    public string? Strategy { get; set; }
+
+    [JsonPropertyName("initial_context")]
+    public Dictionary<string, object>? InitialContext { get; set; }
+
+    [JsonPropertyName("stream")]
+    public bool Stream { get; set; }
+
+    [JsonPropertyName("run_id")]
+    public string? RunId { get; set; }
+
+    [JsonPropertyName("persist_run")]
+    public bool PersistRun { get; set; }
+
+    [JsonPropertyName("debug")]
+    public bool Debug { get; set; }
+
+    [JsonPropertyName("pause_on")]
+    public string[]? PauseOn { get; set; }
+
+    [JsonPropertyName("debug_idle_timeout_ms")]
+    public int? DebugIdleTimeoutMs { get; set; }
+
+    [JsonPropertyName("node_execute_hooks")]
+    public Dictionary<string, object>? NodeExecuteHooks { get; set; }
+}

src/Plugins/BotSharp.Plugin.Membase/Models/Responses/PgtSimulationResponse.cs

@@ -0,0 +1,79 @@
+using System.Text.Json.Serialization;
+
+namespace BotSharp.Plugin.Membase.Models;
+
+public class PgtSimulationResponse
+{
+    public string[] Columns { get; set; } = [];
+    public PgtSimulationDataItem[] Data { get; set; } = [];
+
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public PgtSimulationStatistics? Statistics { get; set; }
+
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public object? ExecutionPlan { get; set; }
+    public CypherNotification[] Notifications { get; set; } = [];
+    public int RowCount { get; set; }
+
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public DateTime? ExecutedAt { get; set; }
+
+    public string RunId { get; set; }
+}
+
+public class PgtSimulationDataItem
+{
+    [JsonPropertyName("final_context")]
+    public Dictionary<string, object>? FinalContext { get; set; }
+
+    [JsonPropertyName("visited")]
+    public string[] Visited { get; set; } = [];
+
+    [JsonPropertyName("trace_log")]
+    public PgtTraceLogEntry[] TraceLog { get; set; } = [];
+
+    [JsonPropertyName("halted")]
+    public string Halted { get; set; } = string.Empty;
+
+    [JsonPropertyName("halt_reason")]
+    public string HaltReason { get; set; } = string.Empty;
+}
+
+public class PgtTraceLogEntry
+{
+    public string Event { get; set; } = string.Empty;
+
+    [JsonPropertyName("node_id")]
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public string? NodeId { get; set; }
+
+    [JsonPropertyName("edge_id")]
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public string? EdgeId { get; set; }
+
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public string? Source { get; set; }
+
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public string? Target { get; set; }
+
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public string? Type { get; set; }
+
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public bool? Allowed { get; set; }
+
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public int? Depth { get; set; }
+
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public Dictionary<string, object>? Output { get; set; }
+
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public Dictionary<string, object>? Input { get; set; }
+}
+
+public class PgtSimulationStatistics
+{
+    public long ExecutionTimeMs { get; set; }
+}