SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

add hook

Browse source
This commit is contained in:
Jicheng Lu 2024-04-07 13:37:16 -05:00
parent 7c061af7f5
commit 8fd6f6a99e
2 changed files with 29 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/Infrastructure/BotSharp.Abstraction/Conversations/Models/StateChangeModel.cs
View file

@ -12,13 +12,13 @@ public class StateChangeModel
public string Name { get; set; }
[JsonPropertyName("before_value")]
public string BeforeValue { get; set; }
public string? BeforeValue { get; set; }
[JsonPropertyName("before_active_rounds")]
public int? BeforeActiveRounds { get; set; }
[JsonPropertyName("after_value")]
public string AfterValue { get; set; }
public string? AfterValue { get; set; }
[JsonPropertyName("after_active_rounds")]
public int? AfterActiveRounds { get; set; }

32
src/Infrastructure/BotSharp.Core/Conversations/Services/ConversationStateService.cs
View file

@ -1,5 +1,8 @@
using Amazon.Auth.AccessControlPolicy;
using AspectInjector.Broker;
using BotSharp.Abstraction.Conversations.Enums;
using BotSharp.Abstraction.Users.Enums;
using System;
namespace BotSharp.Core.Conversations.Services;
@ -201,9 +204,9 @@ public class ConversationStateService : IConversationStateService, IDisposable
var key = pair.Key;
var curValue = pair.Value;
if (!_historyStates.TryGetValue(key, out var historyValue)
|| historyValue == null
|| historyValue.Values.IsNullOrEmpty()
if (!_historyStates.TryGetValue(key, out var historyValue)
|| historyValue == null
|| historyValue.Values.IsNullOrEmpty()
|| !curValue.Versioning)
{
states.Add(curValue);
@ -232,8 +235,9 @@ public class ConversationStateService : IConversationStateService, IDisposable
if (!ContainsState(name)) return false;
var routingCtx = _services.GetRequiredService<IRoutingContext>();
var leafNode = _curStates[name].Values?.LastOrDefault();
if (leafNode == null) return false;
var value = _curStates[name];
var leafNode = value?.Values?.LastOrDefault();
if (value == null || !value.Versioning || leafNode == null) return false;
_curStates[name].Values.Add(new StateValue
{
@ -246,6 +250,24 @@ public class ConversationStateService : IConversationStateService, IDisposable
UpdateTime = DateTime.UtcNow
});
var hooks = _services.GetServices<IConversationHook>();
foreach (var hook in hooks)
{
hook.OnStateChanged(new StateChangeModel
{
ConversationId = _conversationId,
MessageId = routingCtx.MessageId,
Name = name,
BeforeValue = leafNode.Data,
BeforeActiveRounds = leafNode.ActiveRounds,
AfterValue = null,
AfterActiveRounds = leafNode.ActiveRounds,
DataType = leafNode.DataType,
Source = leafNode.Source,
Readonly = _curStates[name].Readonly
}).Wait();
}
return true;
}