Test-Multiplayer/Assets/Scripts/Chat/ChatNetworkManager.cs

using UnityEngine;
using Mirror;
using System.Collections.Generic;
using UnityEngine.SceneManagement;

[AddComponentMenu("")]
public class ChatNetworkManager : NetworkManager
{
    [SerializeField] private GameObject chatUIPrefab;
    [SerializeField] private GameObject riveLoginCanvas;

    // Server-side authoritative store of chat messages
    private readonly List<string> chatMessages = new();

    // Server-side map of connections to player names (moved from the old ChatUI)
    internal static readonly Dictionary<NetworkConnectionToClient, string> connNames = new();

    public void SetHostname(string hostname)
    {
        networkAddress = hostname;
    }

    public override void OnServerDisconnect(NetworkConnectionToClient conn)
    {
        Debug.Log($"[ChatNetworkManager] Server disconnecting connection {conn.connectionId}. Active scene: '{SceneManager.GetActiveScene().name}'");

        if (conn.authenticationData != null)
            ChatAuthenticator.playerNames.Remove((string)conn.authenticationData);

        connNames.Remove(conn);

        base.OnServerDisconnect(conn);
    }

    public override void OnClientDisconnect()
    {
        Debug.Log($"[ChatNetworkManager] Client disconnected. Active scene: '{SceneManager.GetActiveScene().name}'");
        base.OnClientDisconnect();

        // Destroy ChatUI instance when disconnecting
        ChatUI existingChatUI = FindFirstObjectByType<ChatUI>();
        if (existingChatUI != null)
        {
            Destroy(existingChatUI.gameObject);
        }

        // Show LoginUI again
        if (LoginUI.instance != null)
        {
            LoginUI.instance.gameObject.SetActive(true);
            LoginUI.instance.usernameInput.text = "";
            LoginUI.instance.usernameInput.ActivateInputField();
        }
    }

    public override void OnClientConnect()
    {
        base.OnClientConnect();

        Debug.Log($"[ChatNetworkManager] OnClientConnect called. Active scene: '{SceneManager.GetActiveScene().name}'");

        // Spawn ChatUI prefab locally when client successfully connects
        if (chatUIPrefab == null)
        {
            Debug.LogError("ChatNetworkManager: ChatUI prefab is not assigned in the inspector!");
            return;
        }

        if (riveLoginCanvas == null)
        {
            Debug.LogError("ChatNetworkManager: Rive Login Canvas is not assigned in the inspector!");
        }

        // Check if ChatUI already exists to avoid duplicates
        ChatUI existingChatUI = FindFirstObjectByType<ChatUI>();
        if (existingChatUI == null)
        {
            GameObject chatUIInstance = Instantiate(chatUIPrefab);
            if (chatUIInstance == null)
            {
                Debug.LogError("ChatNetworkManager: Failed to instantiate ChatUI prefab!");
                return;
            }

            chatUIInstance.SetActive(true);
        }

        // Hide LoginUI when successfully connected
        if (LoginUI.instance != null)
        {
            LoginUI.instance.gameObject.SetActive(false);
        }

        if (riveLoginCanvas != null)
            riveLoginCanvas.SetActive(false);

        Debug.Log($"[ChatNetworkManager] Client connection setup completed. Active scene: '{SceneManager.GetActiveScene().name}'");
    }

    [Server]
    public void AddServerMessage(string message)
    {
        if (string.IsNullOrWhiteSpace(message)) return;
        chatMessages.Add(message);
    }

    [Server]
    public string[] GetHistory()
    {
        return chatMessages.ToArray();
    }
}