using FluentTerminal.Models; using FluentTerminal.Models.Enums; using FluentTerminal.Models.Requests; using FluentTerminal.Models.Responses; using Newtonsoft.Json; using System; using System.Collections.Generic; using System.Threading.Tasks; namespace FluentTerminal.App.Services.Implementation { public class TrayProcessCommunicationService : ITrayProcessCommunicationService { private readonly ISettingsService _settingsService; private IAppServiceConnection _appServiceConnection; private readonly Dictionary> _terminalOutputHandlers; private int _nextTerminalId = 0; public event EventHandler TerminalExited; public TrayProcessCommunicationService(ISettingsService settingsService) { _settingsService = settingsService; _terminalOutputHandlers = new Dictionary>(); } public int GetNextTerminalId() { return _nextTerminalId++; } public async Task GetAvailablePort() { var request = new GetAvailablePortRequest(); var responseMessage = await _appServiceConnection.SendMessageAsync(CreateMessage(request)); var response = JsonConvert.DeserializeObject(responseMessage[MessageKeys.Content]); Logger.Instance.Debug("Received GetAvailablePortResponse: {@response}", response); return response; } public async Task CreateTerminal(int id, TerminalSize size, ShellProfile shellProfile, SessionType sessionType) { var request = new CreateTerminalRequest { Id = id, Size = size, Profile = shellProfile, SessionType = sessionType }; Logger.Instance.Debug("Sending CreateTerminalRequest: {@request}", request); var responseMessage = await _appServiceConnection.SendMessageAsync(CreateMessage(request)); var response = JsonConvert.DeserializeObject(responseMessage[MessageKeys.Content]); Logger.Instance.Debug("Received CreateTerminalResponse: {@response}", response); return response; } public void Initialize(IAppServiceConnection appServiceConnection) { _appServiceConnection = appServiceConnection; _appServiceConnection.MessageReceived += OnMessageReceived; } private void OnMessageReceived(object sender, IDictionary e) { var messageType = e[MessageKeys.Type]; var messageContent = e[MessageKeys.Content]; if (messageType == nameof(DisplayTerminalOutputRequest)) { var request = JsonConvert.DeserializeObject(messageContent); if (_terminalOutputHandlers.ContainsKey(request.TerminalId)) { _terminalOutputHandlers[request.TerminalId].Invoke(request.Output); } else { Logger.Instance.Error("Received output for unknown terminal Id {id}", request.TerminalId); } } else if (messageType == nameof(TerminalExitedRequest)) { var request = JsonConvert.DeserializeObject(messageContent); Logger.Instance.Debug("Received TerminalExitedRequest: {@request}", request); TerminalExited?.Invoke(this, request.TerminalId); } } public Task ResizeTerminal(int id, TerminalSize size) { var request = new ResizeTerminalRequest { TerminalId = id, NewSize = size }; return _appServiceConnection.SendMessageAsync(CreateMessage(request)); } public void SubscribeForTerminalOutput(int terminalId, Action callback) { _terminalOutputHandlers[terminalId] = callback; } public Task UpdateToggleWindowKeyBindings() { var keyBindings = _settingsService.GetCommandKeyBindings()[nameof(Command.ToggleWindow)]; var request = new SetToggleWindowKeyBindingsRequest { KeyBindings = keyBindings }; return _appServiceConnection.SendMessageAsync(CreateMessage(request)); } public Task Write(int id, byte[] data) { var request = new WriteDataRequest { TerminalId = id, Data = data }; return _appServiceConnection.SendMessageAsync(CreateMessage(request)); } public Task CloseTerminal(int terminalId) { var request = new TerminalExitedRequest { TerminalId = terminalId }; Logger.Instance.Debug("Sending TerminalExitedRequest: {@request}", request); return _appServiceConnection.SendMessageAsync(CreateMessage(request)); } private IDictionary CreateMessage(object content) { return new Dictionary { [MessageKeys.Type] = content.GetType().Name, [MessageKeys.Content] = JsonConvert.SerializeObject(content) }; } } }