using Solana.Unity.Bolt.Test;
using Solana.Unity.Rpc.Models;
using Solana.Unity.Wallet;
using Solana.Unity.Bolt;
using Solana.Unity.Rpc;
using System;
using System.Threading.Tasks;
using Solana.Unity.Wallet.Bip39;
using World.Program;
using System.Diagnostics;
using Solana.Unity.Rpc.Types;
namespace ECSTest {
    public class Test {
        public static async Task Run(Framework framework) {
            await Profiler.Run("AddEntity1", async () => {
                await AddEntity1(framework);
            });
            await Profiler.Run("AddEntity2", async () => {
                await AddEntity2(framework);
            });
            await Profiler.Run("AddEntity3", async () => {
                await AddEntity3(framework);
            });
            await Profiler.Run("AddEntity4WithSeed", async () => {
                await AddEntity4WithSeed(framework);
            });
            await Profiler.Run("InitializeVelocityComponentOnEntity1WithSeed", async () => {
                await InitializeVelocityComponentOnEntity1WithSeed(framework);
            });
            await Profiler.Run("InitializePositionComponentOnEntity1", async () => {
                await InitializePositionComponentOnEntity1(framework);
            });
            await Profiler.Run("InitializePositionComponentOnEntity2", async () => {
                await InitializePositionComponentOnEntity2(framework);
            });
            await Profiler.Run("InitializePositionComponentOnEntity4", async () => {
                await InitializePositionComponentOnEntity4(framework);
            });
            await Profiler.Run("CheckPositionOnEntity1IsDefault", async () => {
                await CheckPositionOnEntity1IsDefault(framework);
            });
            await Profiler.Run("ApplySimpleMovementSystemUpOnEntity1", async () => {
                await ApplySimpleMovementSystemUpOnEntity1(framework);
            });
            await Profiler.Run("ApplySimpleMovementSystemRightOnEntity1", async () => {
                await ApplySimpleMovementSystemRightOnEntity1(framework);
            });
            await Profiler.Run("DestroyVelocityComponentOnEntity1", async () => {
                await DestroyVelocityComponentOnEntity1(framework);
            });
        }

        public static async Task AddEntity1(Framework framework) {
            var addEntity = await Bolt.World.AddEntity(framework.Client, framework.WorldPda, framework.Wallet.Account.PublicKey, Commitment.Processed);
            framework.Entity1Pda = addEntity.Pda;
            await framework.SendAndConfirmInstruction(addEntity.Instruction);
        }
        
        public static async Task AddEntity2(Framework framework) {
            var addEntity = await Bolt.World.AddEntity(framework.Client, framework.WorldPda, framework.Wallet.Account.PublicKey, Commitment.Processed);
            framework.Entity2Pda = addEntity.Pda;
            await framework.SendAndConfirmInstruction(addEntity.Instruction);
        }

        public static async Task AddEntity3(Framework framework) {
            var addEntity = await Bolt.World.AddEntity(framework.Client, framework.WorldPda, framework.Wallet.Account.PublicKey, Commitment.Processed);
            await framework.SendAndConfirmInstruction(addEntity.Instruction);
        }

        public static async Task AddEntity4WithSeed(Framework framework) {
            var addEntity = await Bolt.World.AddEntity(framework.Client, framework.WorldPda, framework.Wallet.Account.PublicKey, "custom-seed", Commitment.Processed);
            framework.Entity4Pda = addEntity.Pda;
            await framework.SendAndConfirmInstruction(addEntity.Instruction);
        }

        public static async Task InitializeVelocityComponentOnEntity1WithSeed(Framework framework) {
            var initializeComponent = await Bolt.World.InitializeComponent(framework.Wallet.Account.PublicKey, framework.Entity1Pda, framework.ExampleComponentVelocity, "component-velocity", framework.Wallet.Account.PublicKey);
            framework.ComponentVelocityEntity1Pda = initializeComponent.Pda;
            await framework.SendAndConfirmInstruction(initializeComponent.Instruction);
        }

        public static async Task InitializePositionComponentOnEntity1(Framework framework) {
            var initializeComponent = await Bolt.World.InitializeComponent(framework.Wallet.Account.PublicKey, framework.Entity1Pda, framework.ExampleComponentPosition);
            framework.ComponentPositionEntity1Pda = initializeComponent.Pda;
            await framework.SendAndConfirmInstruction(initializeComponent.Instruction);
        }

        public static async Task InitializePositionComponentOnEntity2(Framework framework) {
            var initializeComponent = await Bolt.World.InitializeComponent(framework.Wallet.Account.PublicKey, framework.Entity2Pda, framework.ExampleComponentPosition);
            framework.ComponentPositionEntity2Pda = initializeComponent.Pda;
            await framework.SendAndConfirmInstruction(initializeComponent.Instruction);
        }

        public static async Task InitializePositionComponentOnEntity4(Framework framework) {
            var initializeComponent = await Bolt.World.InitializeComponent(framework.Wallet.Account.PublicKey, framework.Entity4Pda, framework.ExampleComponentPosition);
            framework.ComponentPositionEntity4Pda = initializeComponent.Pda;
            await framework.SendAndConfirmInstruction(initializeComponent.Instruction);
        }

        public static async Task CheckPositionOnEntity1IsDefault(Framework framework) {
            var accountInfo = await framework.GetAccountInfo(framework.ComponentPositionEntity1Pda);
            var data = Convert.FromBase64String(accountInfo.Data[0]);
            var position = Position.Accounts.Position.Deserialize(data);
            Debug.Assert(0 == position.X, "X is not equal to 0");
            Debug.Assert(0 == position.Y, "Y is not equal to 0");
            Debug.Assert(0 == position.Z, "Z is not equal to 0");
        }

        public static async Task ApplySimpleMovementSystemUpOnEntity1(Framework framework) {
            var apply = new ApplyAccounts() {
                Authority = framework.Wallet.Account.PublicKey,
                BoltSystem = framework.SystemSimpleMovement,
                World = framework.WorldPda,
            };
            var instruction = WorldProgram.Apply(apply, Bolt.World.SerializeArgs(new { direction = "Up" }));
            instruction.Keys.Add(AccountMeta.ReadOnly(framework.ExampleComponentPosition, false));
            instruction.Keys.Add(AccountMeta.Writable(framework.ComponentPositionEntity1Pda, false));
            await framework.SendAndConfirmInstruction(instruction);

            var accountInfo = await framework.GetAccountInfo(framework.ComponentPositionEntity1Pda);
            var data = Convert.FromBase64String(accountInfo.Data[0]);
            var position = Position.Accounts.Position.Deserialize(data);
            Debug.Assert(0 == position.X, "X is not equal to 0");
            Debug.Assert(1 == position.Y, "Y is not equal to 1");
            Debug.Assert(0 == position.Z, "Z is not equal to 0");
        }

        public static async Task ApplySimpleMovementSystemRightOnEntity1(Framework framework) {
            var instruction = Bolt.World.ApplySystem(
                framework.WorldPda,
                framework.SystemSimpleMovement,
                new Bolt.World.EntityType[] {
                    new Bolt.World.EntityType(framework.Entity1Pda, new PublicKey[] { framework.ExampleComponentPosition })
                },
                new { direction = "Right" },
                framework.Wallet.Account.PublicKey
            );
            await framework.SendAndConfirmInstruction(instruction);

            var accountInfo = await framework.GetAccountInfo(framework.ComponentPositionEntity1Pda);
            var data = Convert.FromBase64String(accountInfo.Data[0]);
            var position = Position.Accounts.Position.Deserialize(data);
            Debug.Assert(1 == position.X, "X is not equal to 1");
            Debug.Assert(1 == position.Y, "Y is not equal to 1");
            Debug.Assert(0 == position.Z, "Z is not equal to 0");
        }

        public static async Task DestroyVelocityComponentOnEntity1(Framework framework) {
            var receiver = new Wallet(new Mnemonic(WordList.English, WordCount.Twelve));

            var componentBalance = await framework.Client.GetBalanceAsync(framework.ComponentVelocityEntity1Pda);

            var destroyComponent = await Bolt.World.DestroyComponent(framework.Wallet.Account.PublicKey, receiver.Account.PublicKey, framework.Entity1Pda, framework.ExampleComponentVelocity, "component-velocity");
            await framework.SendAndConfirmInstruction(destroyComponent.Instruction);

            var receiverBalance = await framework.Client.GetBalanceAsync(receiver.Account.PublicKey);
            Debug.Assert(componentBalance.Result.Value == receiverBalance.Result.Value, "Component balance is not equal to receiver balance");
        }
   }
}