// SPDX-License-Identifier: MIT
pragma solidity ^0.8.20;

import "forge-std/Test.sol";
import "../../src/AccountWalletRegistry.sol";
import "../../src/interfaces/IAccountWalletRegistry.sol";

contract AccountWalletRegistryTest is Test {
    AccountWalletRegistry public registry;
    address public admin;
    address public accountManager;

    bytes32 public accountRefId1 = keccak256("account1");
    bytes32 public walletRefId1 = keccak256("wallet1");
    bytes32 public walletRefId2 = keccak256("wallet2");
    bytes32 public provider1 = keccak256("METAMASK");
    bytes32 public provider2 = keccak256("FIREBLOCKS");

    function setUp() public {
        admin = address(0x1);
        accountManager = address(0x2);

        registry = new AccountWalletRegistry(admin);

        vm.startPrank(admin);
        registry.grantRole(registry.ACCOUNT_MANAGER_ROLE(), accountManager);
        vm.stopPrank();
    }

    function test_linkAccountToWallet() public {
        vm.expectEmit(true, true, false, true);
        emit IAccountWalletRegistry.AccountWalletLinked(accountRefId1, walletRefId1, provider1, uint64(block.timestamp));

        vm.prank(accountManager);
        registry.linkAccountToWallet(accountRefId1, walletRefId1, provider1);

        assertTrue(registry.isLinked(accountRefId1, walletRefId1));
        assertTrue(registry.isActive(accountRefId1, walletRefId1));

        IAccountWalletRegistry.WalletLink[] memory wallets = registry.getWallets(accountRefId1);
        assertEq(wallets.length, 1);
        assertEq(wallets[0].walletRefId, walletRefId1);
        assertEq(wallets[0].provider, provider1);
        assertTrue(wallets[0].active);
    }

    function test_linkMultipleWallets() public {
        vm.prank(accountManager);
        registry.linkAccountToWallet(accountRefId1, walletRefId1, provider1);

        vm.prank(accountManager);
        registry.linkAccountToWallet(accountRefId1, walletRefId2, provider2);

        IAccountWalletRegistry.WalletLink[] memory wallets = registry.getWallets(accountRefId1);
        assertEq(wallets.length, 2);
        assertEq(wallets[0].walletRefId, walletRefId1);
        assertEq(wallets[1].walletRefId, walletRefId2);
    }

    function test_unlinkAccountFromWallet() public {
        vm.prank(accountManager);
        registry.linkAccountToWallet(accountRefId1, walletRefId1, provider1);

        assertTrue(registry.isActive(accountRefId1, walletRefId1));

        vm.expectEmit(true, true, false, false);
        emit IAccountWalletRegistry.AccountWalletUnlinked(accountRefId1, walletRefId1);

        vm.prank(accountManager);
        registry.unlinkAccountFromWallet(accountRefId1, walletRefId1);

        assertTrue(registry.isLinked(accountRefId1, walletRefId1)); // Still linked
        assertFalse(registry.isActive(accountRefId1, walletRefId1)); // But inactive
    }

    function test_getAccounts() public {
        bytes32 accountRefId2 = keccak256("account2");

        vm.prank(accountManager);
        registry.linkAccountToWallet(accountRefId1, walletRefId1, provider1);

        vm.prank(accountManager);
        registry.linkAccountToWallet(accountRefId2, walletRefId1, provider1);

        bytes32[] memory accounts = registry.getAccounts(walletRefId1);
        assertEq(accounts.length, 2);
    }

    function test_linkAccountToWallet_reactivate() public {
        vm.prank(accountManager);
        registry.linkAccountToWallet(accountRefId1, walletRefId1, provider1);

        vm.prank(accountManager);
        registry.unlinkAccountFromWallet(accountRefId1, walletRefId1);

        assertFalse(registry.isActive(accountRefId1, walletRefId1));

        // Reactivate
        vm.prank(accountManager);
        registry.linkAccountToWallet(accountRefId1, walletRefId1, provider1);

        assertTrue(registry.isActive(accountRefId1, walletRefId1));
    }
}