quantum-space-buddies/QSB/ConversationSync/ConversationManager.cs

using Cysharp.Threading.Tasks;
using OWML.Common;
using QSB.ConversationSync.Messages;
using QSB.ConversationSync.WorldObjects;
using QSB.Messaging;
using QSB.Player;
using QSB.Utility;
using QSB.WorldSync;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using UnityEngine;
using UnityEngine.UI;

namespace QSB.ConversationSync;

public class ConversationManager : WorldObjectManager
{
	public override WorldObjectScene WorldObjectScene => WorldObjectScene.Both;

	public static ConversationManager Instance { get; private set; }
	public Dictionary<CharacterDialogueTree, GameObject> BoxMappings { get; } = new();

	private GameObject _boxPrefab;

	public void Start()
	{
		Instance = this;

		_boxPrefab = QSBCore.ConversationAssetBundle.LoadAsset<GameObject>("assets/Prefabs/dialoguebubble.prefab");

		var font = (Font)Resources.Load(@"fonts\english - latin\HVD Fonts - BrandonGrotesque-Bold_Dynamic");
		if (font == null)
		{
			DebugLog.ToConsole("Error - Font is null!", MessageType.Error);
		}

		_boxPrefab.GetComponent<Text>().font = font;
		_boxPrefab.GetComponent<Text>().color = Color.white;
	}

	public override async UniTask BuildWorldObjects(OWScene scene, CancellationToken ct)
	{
		QSBWorldSync.Init<QSBRemoteDialogueTrigger, RemoteDialogueTrigger>();
		QSBWorldSync.Init<QSBCharacterDialogueTree, CharacterDialogueTree>();
	}

	public uint GetPlayerTalkingToTree(CharacterDialogueTree tree) =>
		QSBPlayerManager.PlayerList.FirstOrDefault(x => x.CurrentCharacterDialogueTree?.AttachedObject == tree)
			?.PlayerId ?? uint.MaxValue;

	public void SendPlayerOption(string text)
		=> new ConversationMessage(ConversationType.Player, (int)QSBPlayerManager.LocalPlayerId, text).Send();

	public void SendCharacterDialogue(int id, string text)
		=> new ConversationMessage(ConversationType.Character, id, text).Send();

	public void CloseBoxPlayer() =>
		new ConversationMessage(ConversationType.ClosePlayer, (int)QSBPlayerManager.LocalPlayerId).Send();

	public void CloseBoxCharacter(int id) =>
		new ConversationMessage(ConversationType.CloseCharacter, id).Send();

	public void SendConvState(QSBCharacterDialogueTree tree, bool state)
		=> tree.SendMessage(new ConversationStartEndMessage(QSBPlayerManager.LocalPlayerId, state));

	public void DisplayPlayerConversationBox(uint playerId, string text)
	{
		if (playerId == QSBPlayerManager.LocalPlayerId)
		{
			DebugLog.ToConsole("Error - Cannot display conversation box for local player!", MessageType.Error);
			return;
		}

		var player = QSBPlayerManager.GetPlayer(playerId);

		// Destroy old box if it exists
		var playerBox = player.CurrentDialogueBox;
		if (playerBox != null)
		{
			Destroy(playerBox);
		}

		QSBPlayerManager.GetPlayer(playerId).CurrentDialogueBox = CreateBox(player.Body.transform, 2, text);
	}

	public void DisplayCharacterConversationBox(int index, string text)
	{
		// Remove old box if it exists
		var oldDialogueTree = index.GetWorldObject<QSBCharacterDialogueTree>().AttachedObject;

		if (BoxMappings.ContainsKey(oldDialogueTree))
		{
			Destroy(BoxMappings[oldDialogueTree]);
			BoxMappings.Remove(oldDialogueTree);
		}

		BoxMappings.Add(oldDialogueTree, CreateBox(oldDialogueTree.gameObject.transform, 2, text));
	}

	private GameObject CreateBox(Transform parent, float vertOffset, string text)
	{
		var newBox = Instantiate(_boxPrefab);
		newBox.SetActive(false);
		newBox.transform.SetParent(parent);
		newBox.transform.localPosition = new Vector3(0, vertOffset, 0);
		newBox.transform.rotation = parent.rotation;
		newBox.GetComponent<Text>().text = text;
		newBox.SetActive(true);
		return newBox;
	}
}