using BotSharp.Abstraction.Conversations.Models; using BotSharp.Abstraction.Repositories; using System.IO; namespace BotSharp.Core.Conversations.Services; /// /// Maintain the conversation state /// public class ConversationStateService : IConversationStateService, IDisposable { private readonly ILogger _logger; private readonly IServiceProvider _services; private ConversationState _state; private MyDatabaseSettings _dbSettings; private string _conversationId; private string _file; public ConversationStateService(ILogger logger, IServiceProvider services, MyDatabaseSettings dbSettings) { _logger = logger; _services = services; _dbSettings = dbSettings; } public void SetState(string name, string value) { var hooks = _services.GetServices(); string preValue = _state.ContainsKey(name) ? _state[name] : ""; if (!_state.ContainsKey(name) || _state[name] != value) { var currentValue = value; _state[name] = currentValue; _logger.LogInformation($"Set state: {name} = {value}"); foreach (var hook in hooks) { hook.OnStateChanged(name, preValue, currentValue).Wait(); } } } public void SetConversation(string conversationId) { _conversationId = conversationId; } public ConversationState Load() { if (_state != null) { return _state; } _state = new ConversationState(); _file = GetStorageFile(_conversationId); if (File.Exists(_file)) { var dict = File.ReadAllLines(_file); foreach (var line in dict) { _state[line.Split('=')[0]] = line.Split('=')[1]; } } _logger.LogInformation($"Loaded state {_conversationId}"); var hooks = _services.GetServices(); foreach (var hook in hooks) { hook.OnStateLoaded(_state).Wait(); } return _state; } public void Save() { var states = new List(); foreach (var dic in _state) { states.Add($"{dic.Key}={dic.Value}"); } File.WriteAllLines(_file, states); _logger.LogInformation($"Saved state {_conversationId}"); } public void CleanState() { File.Delete(_file); } private string GetStorageFile(string conversationId) { var dir = Path.Combine(_dbSettings.FileRepository, "conversations", conversationId); if (!Directory.Exists(dir)) { Directory.CreateDirectory(dir); } return Path.Combine(dir, "state.dict"); } public string GetState(string name) { if (!_state.ContainsKey(name)) { _state[name] = ""; } return _state[name]; } public void Dispose() { Save(); } }