UiTalkHandler.cs

// <copyright file="UiTalkHandler.cs" company="lokinmodar">
// Copyright (c) lokinmodar. All rights reserved.
// Licensed under the Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International Public License license.
// </copyright>

using System;
using System.Collections.Concurrent;
using System.IO;
using System.Threading;
using System.Threading.Tasks;

using Dalamud.Game.Text.SeStringHandling;
using Dalamud.Utility;
using Echoglossian.EFCoreSqlite.Models;
using Echoglossian.Properties;
using FFXIVClientStructs.FFXIV.Component.GUI;
using XivCommon.Functions;

namespace Echoglossian
{
  public partial class Echoglossian
  {
    public bool talkTracker { get; set; } = false;

    private ConcurrentQueue<TalkMessage> _talkMessageQueue = new();
    private ConcurrentDictionary<int, TalkMessage> _talkConcurrentDictionary = new();

    private void HandleTalkAsync()
    {
      Thread t = new(new ThreadStart(this.TalkProc));
      t.Start();
    }

    public void TalkProc()
    {
      PluginLog.Information("Translation Engine Started");
      while (this.configuration.TranslateTalk)
      {
        if (this._talkMessageQueue.Count > 0)
        {
          this._talkMessageQueue.TryDequeue(out TalkMessage dialogue);
          this.talkTracker = true;
          string nameTranslation;
          string messageTranslation;
          if (this.configuration.TranslateNpcNames)
          {
            nameTranslation = this.Translate(dialogue?.SenderName);
            PluginLog.Warning($"Name translation async: {nameTranslation}");
          }

          messageTranslation = this.Translate(dialogue?.OriginalTalkMessage);
          PluginLog.Warning($"Message translation async: {messageTranslation}");
        }
      }
    }

    /*    IntPtr talk = GameGui.GetAddonByName("Talk", 1);
          if (talk != IntPtr.Zero)
        {
          unsafe
          {
            AtkUnitBase* talkMaster = (AtkUnitBase*)talk;
            if (talkMaster->IsVisible)
            {
              AtkTextNode* textNode = null;
              for (int i = 0; i<talkMaster->UldManager.NodeListCount; i++)
              {
                if (talkMaster->UldManager.NodeList[i]->Type != NodeType.Text)
                {
                  continue;
                }

                textNode = (AtkTextNode*) talkMaster->UldManager.NodeList[i];
                break;
              }

              textNode->SetText(messageTranslation);
            }
          }
        }*/

    private unsafe void TalkHandler(string addonName, int index)
    {
      IntPtr talk = GameGui.GetAddonByName(addonName, index);
      if (talk != IntPtr.Zero)
      {
        AtkUnitBase* talkMaster = (AtkUnitBase*)talk;
        if (talkMaster->IsVisible)
        {
          this.talkDisplayTranslation = true;
          this.talkTextDimensions.X = talkMaster->RootNode->Width * talkMaster->Scale;
          this.talkTextDimensions.Y = talkMaster->RootNode->Height * talkMaster->Scale;
          this.talkTextPosition.X = talkMaster->RootNode->X;
          this.talkTextPosition.Y = talkMaster->RootNode->Y;
#if DEBUG
          // PluginLog.Verbose("Inside Talk Handler.");
#endif
        }
        else
        {
          this.talkDisplayTranslation = false;
        }
      }
      else
      {
        this.talkDisplayTranslation = false;
      }
    }

    private void GetTalk(ref SeString name, ref SeString text, ref TalkStyle style)
    {
      if (!this.configuration.TranslateTalk)
      {
        return;
      }
#if DEBUG
      using StreamWriter logStream = new(this.configDir + "GetTalkLog.txt", append: true);
#endif

      try
      {
#if DEBUG
        PluginLog.Information(name.TextValue + ": " + text.TextValue);
#endif

        string nameToTranslate = !name.TextValue.IsNullOrEmpty() ? name.TextValue : string.Empty;
        string textToTranslate = text.TextValue;

        TalkMessage talkMessage = this.FormatTalkMessage(nameToTranslate, textToTranslate);
        this._talkMessageQueue.Enqueue(talkMessage);

#if DEBUG
        PluginLog.Fatal($"Before DB Query attempt: {talkMessage}");
#endif
        bool findings = Echoglossian.FindTalkMessage(talkMessage);
#if DEBUG
        PluginLog.Fatal(
          $"After DB Query attempt: {(findings ? "Message found in Db." : "Message not found in Db")}");
#endif

        // If the dialogue is not saved
        if (!findings)
        {
          if (!this.configuration.UseImGuiForTalk)
          {
            string translatedText = this.Translate(textToTranslate);
            string nameTranslation = nameToTranslate.IsNullOrEmpty() ? string.Empty : this.Translate(nameToTranslate);
#if DEBUG
            PluginLog.Warning(translatedText);
#endif
            if (this.configuration.TranslateNpcNames)
            {
              name = nameTranslation == string.Empty ? name : nameTranslation;
              text = translatedText;

              TalkMessage translatedTalkData = new TalkMessage(
                nameToTranslate,
                textToTranslate,
                LangIdentify(textToTranslate),
                LangIdentify(nameToTranslate),
                nameTranslation,
                translatedText,
                langDict[languageInt].Code,
                this.configuration.ChosenTransEngine,
                DateTime.Now,
                DateTime.Now);
#if DEBUG
              logStream.WriteLineAsync($"Before Talk Messages table data insertion:  {translatedTalkData}");
#endif
              string result = Echoglossian.InsertTalkData(translatedTalkData);
#if DEBUG
              PluginLog.Verbose($"Talk Message DB Insert operation result: {result}");
#endif
            }
            else
            {
              text = translatedText;

              TalkMessage translatedTalkData = new TalkMessage(
                nameToTranslate,
                textToTranslate,
                LangIdentify(textToTranslate),
                LangIdentify(nameToTranslate),
                string.Empty,
                translatedText,
                langDict[languageInt].Code,
                this.configuration.ChosenTransEngine,
                DateTime.Now,
                DateTime.Now);

              string result = Echoglossian.InsertTalkData(translatedTalkData);
#if DEBUG
              PluginLog.Verbose(result);
#endif
            }
#if DEBUG
            PluginLog.Information($"Using Talk Replace - {name.TextValue}: {text.TextValue}");
#endif
          }
          else
          {
            if (!this.configuration.SwapTextsUsingImGui)
            {
              if (this.configuration.TranslateNpcNames)
              {
                this.currentNameTranslationId = Environment.TickCount;
                this.currentNameTranslation = Resources.WaitingForTranslation;
                Task.Run(() =>
                {
                  int nameId = this.currentNameTranslationId;
                  string nameTranslation = this.Translate(nameToTranslate);
                  this.nameTranslationSemaphore.Wait();
                  if (nameId == this.currentNameTranslationId)
                  {
                    this.currentNameTranslation = nameTranslation;
                    if (this.configuration.Lang == 2)
                    {
                      this.currentNameTranslationTexture =
                        (ImGuiScene.TextureWrap)PluginInterface.UiBuilder.LoadImage(this.TranslationImageConverter(this.DrawText(this.currentNameTranslation)));
                    }
                  }

                  this.nameTranslationSemaphore.Release();
                });
              }

              this.currentTalkTranslationId = Environment.TickCount;
              this.currentTalkTranslation = Resources.WaitingForTranslation;
              Task.Run(() =>
              {
                int id = this.currentTalkTranslationId;
                string translation = this.Translate(textToTranslate);
                this.talkTranslationSemaphore.Wait();
                if (id == this.currentTalkTranslationId)
                {
                  this.currentTalkTranslation = translation;
                  // TODO: check image creation logic for RTL
                  /*                  if (this.configuration.Lang == 2)
                                    {
                  #if DEBUG
                                      PluginLog.Warning("Lang is 2!");
                  #endif
                                      var textAsImage = this.DrawText(this.currentTalkTranslation);
                                      var textImageAsBytes = this.TranslationImageConverter(textAsImage);
                  #if DEBUG
                                      PluginLog.Warning($"image bytes: {textImageAsBytes}");
                  #endif
                                      this.currentTalkTranslationTexture =
                                        pluginInterface.UiBuilder.LoadImage(textImageAsBytes);
                                    }*/
                }

                this.talkTranslationSemaphore.Release();

#if DEBUG
                PluginLog.Verbose($"Before if talk translation: {this.currentTalkTranslation}");
#endif
                if (this.currentNameTranslation != Resources.WaitingForTranslation &&
                    this.currentTalkTranslation != Resources.WaitingForTranslation)
                {
                  TalkMessage translatedTalkData = new TalkMessage(
                    nameToTranslate,
                    textToTranslate,
                    LangIdentify(textToTranslate),
                    LangIdentify(nameToTranslate),
                    this.configuration.TranslateNpcNames ? this.currentNameTranslation : string.Empty,
                    this.currentTalkTranslation,
                    langDict[languageInt].Code,
                    this.configuration.ChosenTransEngine,
                    DateTime.Now,
                    DateTime.Now);
                  string result = Echoglossian.InsertTalkData(translatedTalkData);
#if DEBUG
                  PluginLog.Verbose($"Talk Message DB Insert operation result: {result}");
#endif
                }
              });
            }
            else
            {
#if DEBUG
              PluginLog.Warning("Using Swap text for translation");
#endif
              string translatedText = this.Translate(textToTranslate);
              string nameTranslation = nameToTranslate.IsNullOrEmpty() ? string.Empty : this.Translate(nameToTranslate);
              if (this.configuration.TranslateNpcNames)
              {
                name = nameTranslation == string.Empty ? name : nameTranslation;
                text = translatedText;

                this.currentNameTranslationId = Environment.TickCount;
                this.currentNameTranslation = Resources.WaitingForTranslation;
                Task.Run(() =>
                {
                  int nameId = this.currentNameTranslationId;
                  this.nameTranslationSemaphore.Wait();
                  if (nameId == this.currentNameTranslationId)
                  {
                    this.currentNameTranslation = nameToTranslate;
                  }

                  this.nameTranslationSemaphore.Release();
                });

                this.currentTalkTranslationId = Environment.TickCount;
                this.currentTalkTranslation = Resources.WaitingForTranslation;
                Task.Run(() =>
                {
                  int id = this.currentTalkTranslationId;
                  this.talkTranslationSemaphore.Wait();
                  if (id == this.currentTalkTranslationId)
                  {
                    this.currentTalkTranslation = textToTranslate;
                  }

                  this.talkTranslationSemaphore.Release();
                });

                TalkMessage translatedTalkData = new TalkMessage(
                  nameToTranslate,
                  textToTranslate,
                  LangIdentify(textToTranslate),
                  LangIdentify(nameToTranslate),
                  nameTranslation,
                  translatedText,
                  langDict[languageInt].Code,
                  this.configuration.ChosenTransEngine,
                  DateTime.Now,
                  DateTime.Now);
#if DEBUG
                logStream.WriteLineAsync($"Before Talk Messages table data insertion:  {translatedTalkData}");
#endif
                string result = Echoglossian.InsertTalkData(translatedTalkData);
#if DEBUG
                PluginLog.Verbose($"Talk Message DB Insert operation result: {result}");
#endif
              }
              else
              {
                text = translatedText;

                this.currentTalkTranslationId = Environment.TickCount;
                this.currentTalkTranslation = Resources.WaitingForTranslation;
                Task.Run(() =>
                {
                  int id = this.currentTalkTranslationId;
                  this.talkTranslationSemaphore.Wait();
                  if (id == this.currentTalkTranslationId)
                  {
                    this.currentTalkTranslation = textToTranslate;
                  }

                  this.talkTranslationSemaphore.Release();
                  /*#if DEBUG
                                    PluginLog.Verbose($"Before if talk translation: {this.currentTalkTranslation}");
                  #endif*/
                  /*if (this.currentNameTranslation != Resources.WaitingForTranslation &&
                      this.currentTalkTranslation != Resources.WaitingForTranslation)
                  {
                    var translatedTalkData = new TalkMessage(this.currentNameTranslation, this.currentTalkTranslation,
                      LangIdentify(this.currentTalkTranslation),
                      LangIdentify(this.currentNameTranslation),
                      this.configuration.TranslateNPCNames ? Translate(this.currentNameTranslation) : string.Empty,
                      Translate(this.currentTalkTranslation), langDict[languageInt].Code,
                      this.configuration.ChosenTransEngine, DateTime.Now, DateTime.Now);
                    var result = this.InsertTalkData(translatedTalkData);
#if DEBUG
                    PluginLog.Verbose($"Talk Message DB Insert operation result: {result}");
#endif
                  }*/
                });

                TalkMessage translatedTalkData = new TalkMessage(
                  nameToTranslate,
                  textToTranslate,
                  LangIdentify(textToTranslate),
                  LangIdentify(nameToTranslate),
                  string.Empty,
                  translatedText,
                  langDict[languageInt].Code,
                  this.configuration.ChosenTransEngine,
                  DateTime.Now,
                  DateTime.Now);

                string result = Echoglossian.InsertTalkData(translatedTalkData);
#if DEBUG
                PluginLog.Verbose(result);
#endif
              }
            }
          }
        }
        else
        {
          if (!this.configuration.UseImGuiForTalk)
          {
            string translatedText = FoundTalkMessage.TranslatedTalkMessage;
            string nameTranslation = FoundTalkMessage.TranslatedSenderName;
#if DEBUG
            PluginLog.Warning($"From database - Name: {nameTranslation}, Message: {translatedText}");
#endif
            if (this.configuration.TranslateNpcNames)
            {
              name = nameTranslation == string.Empty || nameTranslation == null || nameTranslation == string.Empty ? name : nameTranslation;
              text = translatedText;
            }
            else
            {
              text = translatedText;
            }
#if DEBUG
            PluginLog.Information(name.TextValue + ": " + text.TextValue);
#endif
          }
          else
          {
            if (!this.configuration.SwapTextsUsingImGui)
            {
              if (this.configuration.TranslateNpcNames)
              {
                this.currentNameTranslationId = Environment.TickCount;
                this.currentNameTranslation = Resources.WaitingForTranslation;
                Task.Run(() =>
                {
                  int nameId = this.currentNameTranslationId;
                  string nameTranslation = FoundTalkMessage.TranslatedSenderName;
                  this.nameTranslationSemaphore.Wait();
                  if (nameId == this.currentNameTranslationId)
                  {
                    this.currentNameTranslation = nameTranslation;
#if DEBUG
                    PluginLog.Error($"Using overlay - name found in DB: {nameTranslation} ");
#endif
                  }

                  this.nameTranslationSemaphore.Release();
                });
              }

              this.currentTalkTranslationId = Environment.TickCount;
              this.currentTalkTranslation = Resources.WaitingForTranslation;
              Task.Run(() =>
              {
                int id = this.currentTalkTranslationId;
                string translatedTalkMessage = FoundTalkMessage.TranslatedTalkMessage;
                this.talkTranslationSemaphore.Wait();
                if (id == this.currentTalkTranslationId)
                {
                  this.currentTalkTranslation = translatedTalkMessage;
#if DEBUG
                  PluginLog.Error($"Using overlay - message found in DB: {translatedTalkMessage} ");
#endif
                }

                this.talkTranslationSemaphore.Release();
              });
            }
            else
            {
              if (this.configuration.TranslateNpcNames)
              {
                this.currentNameTranslationId = Environment.TickCount;
                this.currentNameTranslation = Resources.WaitingForTranslation;
                Task.Run(() =>
                {
                  int nameId = this.currentNameTranslationId;
                  string nameTranslation = FoundTalkMessage.SenderName;
                  this.nameTranslationSemaphore.Wait();
                  if (nameId == this.currentNameTranslationId)
                  {
                    this.currentNameTranslation = nameTranslation;
#if DEBUG
                    PluginLog.Error($"Using overlay - name found in DB: {nameTranslation} ");
#endif
                  }

                  this.nameTranslationSemaphore.Release();
                });
              }

              this.currentTalkTranslationId = Environment.TickCount;
              this.currentTalkTranslation = Resources.WaitingForTranslation;
              Task.Run(() =>
              {
                int id = this.currentTalkTranslationId;
                string translatedTalkMessage = FoundTalkMessage.OriginalTalkMessage;
                this.talkTranslationSemaphore.Wait();
                if (id == this.currentTalkTranslationId)
                {
                  this.currentTalkTranslation = translatedTalkMessage;
#if DEBUG
                  PluginLog.Error($"Using overlay - message found in DB: {translatedTalkMessage} ");
#endif
                }

                this.talkTranslationSemaphore.Release();
              });
              name = FoundTalkMessage.TranslatedSenderName;
              text = FoundTalkMessage.TranslatedTalkMessage;
            }
          }
        }
      }
      catch (Exception e)
      {
        PluginLog.Information("Exception: " + e.StackTrace);
        throw;
      }
    }
  }
}