#nullable enable

using System.Collections;
using System.Reflection;
using Demo2.Domain;
using Demo2.ECS;
using Demo2.Runtime;
using NUnit.Framework;
using ShrinkContext.AppAdapter;
using Unity.Entities;
using UnityEngine;
using UnityEngine.SceneManagement;
using UnityEngine.TestTools;
using UnityEngine.UIElements;

namespace Demo2.Tests
{
    public sealed class Demo2PlayModeTests
    {
        [UnityTest]
        public IEnumerator SceneEntry_OpensTitleAndStartsTheRealFactoryScreen()
        {
            yield return SceneManager.LoadSceneAsync("Demo2", LoadSceneMode.Single);
            yield return null;
            var controller = Object.FindObjectOfType<Demo2DemoController>();
            Assert.That(controller, Is.Not.Null);
            for (var frame = 0; frame < 30 &&
                 (ShrinkAppLoaderBootstrapper.Instance == null || !ShrinkAppLoaderBootstrapper.Instance.Host.IsRunning);
                 frame++)
                yield return null;
            var appHost = ShrinkAppLoaderBootstrapper.Instance;
            Assert.That(appHost, Is.Not.Null);
            Assert.That(appHost!.Host.IsRunning, Is.True);
            Assert.That(appHost.Host.ActiveModuleIds, Does.Contain("demo2.rule-factory"));
            Assert.That(appHost.Host.Services.TryGet<Demo2GameService>(out var appService), Is.True);
            Assert.That(appService, Is.Not.Null);
            var document = controller!.GetComponentInChildren<UIDocument>();
            Assert.That(document, Is.Not.Null);
            var solo = document!.rootVisualElement.Q<Button>("solo-button");
            Assert.That(solo, Is.Not.Null);

            var simulateClick = typeof(Clickable).GetMethod("SimulateSingleClick", BindingFlags.Instance | BindingFlags.NonPublic);
            Assert.That(simulateClick, Is.Not.Null);
            simulateClick!.Invoke(solo.clickable, new object?[] { null, 0 });
            yield return null;

            var factory = document.rootVisualElement.Q<VisualElement>("factory-screen");
            Assert.That(factory.resolvedStyle.display, Is.EqualTo(DisplayStyle.Flex));
            Assert.That(Demo2GameService.Current?.Phase, Is.EqualTo(Demo2Phase.Build));
        }

        [UnityTest]
        public IEnumerator CompleteFiveCycleFlow_ReachesVictory()
        {
            using var service = new Demo2GameService();
            service.StartNewGame(20260817);
            for (var cycle = 1; cycle <= Demo2Protocol.CycleCount; cycle++)
            {
                service.SetReady("test", true);
                var world = service.EcsWorld!;
                var economy = world.EntityManager.GetComponentData<Demo2EconomyState>(world.Singleton);
                economy.Payload = 1_000_000;
                world.EntityManager.SetComponentData(world.Singleton, economy);
                service.StartWaveImmediately();
                for (var frame = 0; frame < 202; frame++) service.AdvanceFrame(0.1f);
                Assert.That(service.Phase, Is.EqualTo(Demo2Phase.Reward), "cycle " + cycle);
                Assert.That(service.RewardChoices.Count, Is.EqualTo(3));
                Assert.That(service.ChooseReward(service.RewardChoices[0]), Is.True);
                yield return null;
            }

            Assert.That(service.Phase, Is.EqualTo(Demo2Phase.Victory));
        }
    }
}