// SPDX-License-Identifier: MIT
pragma solidity 0.8.24;

import {KeystoneForwarder} from "../KeystoneForwarder.sol";
import {Receiver} from "./mocks/Receiver.sol";
import {Test} from "forge-std/Test.sol";

contract BaseTest is Test {
  address internal ADMIN = address(1);
  address internal constant TRANSMITTER = address(50);
  uint256 internal constant MAX_ORACLES = 31;
  uint32 internal DON_ID = 0x01020304;
  uint8 internal F = 1;
  uint32 internal CONFIG_VERSION = 1;

  struct Signer {
    uint256 mockPrivateKey;
    address signerAddress;
  }

  Signer[MAX_ORACLES] internal s_signers;
  KeystoneForwarder internal s_forwarder;
  KeystoneForwarder internal s_router;
  Receiver internal s_receiver;

  function setUp() public virtual {
    vm.startPrank(ADMIN);
    s_router = new KeystoneForwarder();
    s_forwarder = new KeystoneForwarder();
    s_router.addForwarder(address(s_forwarder));
    s_receiver = new Receiver();

    uint256 seed = 0;

    for (uint256 i; i < MAX_ORACLES; ++i) {
      uint256 mockPK = seed + i + 1;
      s_signers[i].mockPrivateKey = mockPK;
      s_signers[i].signerAddress = vm.addr(mockPK);
    }
  }

  function _getSignerAddresses() internal view returns (address[] memory) {
    address[] memory signerAddrs = new address[](s_signers.length);
    for (uint256 i = 0; i < signerAddrs.length; ++i) {
      signerAddrs[i] = s_signers[i].signerAddress;
    }
    return signerAddrs;
  }

  function _getSignerAddresses(
    uint256 limit
  ) internal view returns (address[] memory) {
    address[] memory signerAddrs = new address[](limit);
    for (uint256 i = 0; i < limit; ++i) {
      signerAddrs[i] = s_signers[i].signerAddress;
    }
    return signerAddrs;
  }

  function _signReport(
    bytes memory report,
    bytes memory reportContext,
    uint256 requiredSignatures
  ) internal view returns (bytes[] memory signatures) {
    signatures = new bytes[](requiredSignatures);
    for (uint256 i = 0; i < requiredSignatures; ++i) {
      (uint8 v, bytes32 r, bytes32 s) =
        vm.sign(s_signers[i].mockPrivateKey, keccak256(abi.encodePacked(keccak256(report), reportContext)));
      signatures[i] = bytes.concat(r, s, bytes1(v - 27));
    }
    return signatures;
  }
}