using modular_avatar_tests;
using nadena.dev.modular_avatar.animation;
using nadena.dev.modular_avatar.core;
using nadena.dev.modular_avatar.core.editor;
using NUnit.Framework;
using UnityEngine;
using UnityEngine.Animations;

public class WorldFixedObjectTest : TestBase
{
    [Test]
    public void SimpleTest()
    {
        var avatar = CreatePrefab("Simple.prefab");

        var fixedObject = avatar.transform.Find("FixedObject");

        // initialize context
        var buildContext = new BuildContext(avatar);
        buildContext.PluginBuildContext.ActivateExtensionContext<AnimationServicesContext>();

        new WorldFixedObjectProcessor().Process(buildContext);

        var fixedRoot = avatar.transform.Find("(MA WorldFixedRoot)");
        var movedFixedObject = avatar.transform.Find("(MA WorldFixedRoot)/FixedObject");

        // fixed root is created
        Assert.That(fixedRoot, Is.Not.Null);
        bool isVrcAvatar = false;
        System.Type vrcParentConstraintType = null;
        #if MA_VRCSDK3_AVATARS
        isVrcAvatar = avatar.TryGetComponent(out VRC.SDKBase.VRC_AvatarDescriptor _);
        vrcParentConstraintType = System.Type.GetType("VRC.SDK3.Dynamics.Constraint.Components.VRCParentConstraint, VRC.SDK3.Dynamics.Constraint");
        #endif
        Component constraint = isVrcAvatar && vrcParentConstraintType != null ?
            fixedRoot.GetComponent(vrcParentConstraintType) :
            fixedRoot.GetComponent<ParentConstraint>();
        Assert.That(constraint, Is.Not.Null);

        // objects are moved to fixed root
        Assert.That(movedFixedObject, Is.Not.Null);
        Assert.That(movedFixedObject, Is.EqualTo(fixedObject));
    }

    [Test]
    public void NestedTest()
    {
        var avatar = CreatePrefab("Nested.prefab");

        var fixedObject = avatar.transform.Find("FixedObject");
        var nestedFixed = avatar.transform.Find("FixedObject/NestedFixed");

        // initialize context
        var buildContext = new BuildContext(avatar);
        buildContext.PluginBuildContext.ActivateExtensionContext<AnimationServicesContext>();

        new WorldFixedObjectProcessor().Process(buildContext);

        var fixedRoot = avatar.transform.Find("(MA WorldFixedRoot)");
        var movedFixedObject = avatar.transform.Find("(MA WorldFixedRoot)/FixedObject");
        var nestedFixedObject = avatar.transform.Find("(MA WorldFixedRoot)/NestedFixed");

        // fixed root is created
        Assert.That(fixedRoot, Is.Not.Null);
        bool isVrcAvatar = false;
        System.Type vrcParentConstraintType = null;
        #if MA_VRCSDK3_AVATARS
        isVrcAvatar = avatar.TryGetComponent(out VRC.SDKBase.VRC_AvatarDescriptor _);
        vrcParentConstraintType = System.Type.GetType("VRC.SDK3.Dynamics.Constraint.Components.VRCParentConstraint, VRC.SDK3.Dynamics.Constraint");
        #endif
        Component constraint = isVrcAvatar && vrcParentConstraintType != null ?
            fixedRoot.GetComponent(vrcParentConstraintType) :
            fixedRoot.GetComponent<ParentConstraint>();
        Assert.That(constraint, Is.Not.Null);

        // objects are moved to fixed root
        Assert.That(movedFixedObject, Is.Not.Null);
        Assert.That(movedFixedObject, Is.EqualTo(fixedObject));

        // objects are moved to fixed root
        Assert.That(nestedFixedObject, Is.Not.Null);
        Assert.That(nestedFixedObject, Is.EqualTo(nestedFixed));
    }

    [Test]
    public void NameCollisions()
    {
        var avatar = CreateRoot("Avatar");
        var target1 = CreateChild(avatar, "Target");
        var target2 = CreateChild(avatar, "Target");

        target1.AddComponent<ModularAvatarWorldFixedObject>();
        target2.AddComponent<ModularAvatarWorldFixedObject>();
        
        // initialize context
        var buildContext = new BuildContext(avatar);
        var animationServices = buildContext.PluginBuildContext.ActivateExtensionContext<AnimationServicesContext>();

        new WorldFixedObjectProcessor().Process(buildContext);
        
        Assert.AreSame(target1.transform.parent, target2.transform.parent);
        Assert.AreNotSame(target1.transform.parent, avatar.transform);
        Assert.AreNotSame(target1.gameObject.name, target2.gameObject.name);
    }
}