contract CollectorPool is Ownable, ReentrancyGuard {
  // Ore token contract interface
  IOre public ore;

  // Determines if collectors can deposit ore to be burnt
  bool private _openForBurning;
  // Keeps track fo the currently active launch counter
  uint256 private _currentLaunch;

  // Keeps track of whitelisted fund sources
  mapping (address => bool) public fundAddresses;
  // Keeps track of the total claimed rewards by an address for each launch
  mapping (address => mapping (uint256 => uint256)) private _claimedByAddress;
  // Keeps track of the total burnt ore by an address for each launch
  mapping (address => mapping (uint256 => uint256)) private _burntOreByAddress;
  // Keeps track of the total burnt ore in each launch
  mapping (uint256 => uint256) private _totalBurntOreByLaunch;
  // Keeps track of the total funds received in each launch
  mapping (uint256 => uint256) private _totalFundsByLaunch;

  constructor(address _ore) {
    ore = IOre(_ore);
  }

  function currentLaunch() external view returns (uint256) {
    return _currentLaunch;
  }

  function openForBurning() external view returns (bool) {
    return _openForBurning;
  }

  function setBurningState(bool _state) external onlyOwner {
    require(_openForBurning != _state, "Invalid State");
    _openForBurning = _state;
  }

  function setCurrentLaunch(uint256 _launch) external onlyOwner {
    require(_currentLaunch < _launch, "Invalid Launch Number");
    _currentLaunch = _launch;
  }

  function setFundAddress(address _address, bool _state) external onlyOwner {
    require(_address != address(0), "Invalid Address");

    if (fundAddresses[_address] != _state) {
      fundAddresses[_address] = _state;
    }
  }

  function claimedByAddress(address _address, uint256 _launch) external view returns (uint256) {
    return _claimedByAddress[_address][_launch];
  }

  function burntOreByAddress(address _address, uint256 _launch) external view returns (uint256) {
    return _burntOreByAddress[_address][_launch];
  }

  function totalBurntOreByLaunch(uint256 _launch) external view returns (uint256) {
    return _totalBurntOreByLaunch[_launch];
  }

  function totalFundsByLaunch(uint256 _launch) external view returns (uint256) {
    return _totalFundsByLaunch[_launch];
  }

  // Deposits ore to be burned
  function burnOre(uint256 _amount) external nonReentrant {
    require(_openForBurning, "Not Open For Burning");
    require(_amount > 0, "Invalid Ore Amount");
    require(ore.balanceOf(msg.sender) >= _amount, "Insufficient Ore");

    ore.burn(msg.sender, _amount);

    _burntOreByAddress[msg.sender][_currentLaunch] += _amount;
    _totalBurntOreByLaunch[_currentLaunch] += _amount;
  }

  // Calculates and returns the total amount of claimable funds for the specified account and period
  function totalClaimableByAccount(
    address _account,
    uint256 _fromLaunch,
    uint256 _toLaunch
  ) public view returns (
    uint256,
    uint256[] memory
  ) {
    require(_fromLaunch > 0 && _toLaunch <= _currentLaunch && _fromLaunch <= _toLaunch, "Invalid Launch Period");

    // Calculate the total claimable amount along with the detail for each launch in the specified period range
    uint256 totalClaimable = 0;
    uint256 periodSize = _toLaunch - _fromLaunch + 1;
    uint256[] memory maxPerLaunch = new uint256[](periodSize);
    for (uint256 i = _fromLaunch; i <= _toLaunch ; i++) {
      uint256 maxAmount = 0;

      if (_totalBurntOreByLaunch[i] > 0 && _totalFundsByLaunch[i] > 0) {
        maxAmount = (_burntOreByAddress[_account][i] * _totalFundsByLaunch[i]) / _totalBurntOreByLaunch[i];
        uint256 claimable = maxAmount - _claimedByAddress[_account][i];

        if (claimable > 0) {
          totalClaimable += claimable;
        }
      }

      uint256 index = i - _fromLaunch;
      maxPerLaunch[index] = maxAmount;
    }

    return (totalClaimable, maxPerLaunch);
  }

  // Can be called by collectors for claiming funds from the ore burning mechanism
  function claim(uint256 _fromLaunch, uint256 _toLaunch) external nonReentrant {
    uint256 totalClaimable;
    uint256[] memory maxPerLaunch;
    (totalClaimable, maxPerLaunch) = totalClaimableByAccount(msg.sender, _fromLaunch, _toLaunch);

    // Check the claimable amount and update the total claimed so far for the account
    require(totalClaimable > 0, "Insufficient Claimable Funds");
    for (uint256 i = _fromLaunch; i <= _toLaunch; i++) {
      uint256 index = i - _fromLaunch;
      if (_claimedByAddress[msg.sender][i] < maxPerLaunch[index]) {
        _claimedByAddress[msg.sender][i] = maxPerLaunch[index];
      }
    }

    payable(msg.sender).transfer(totalClaimable);
  }

  // Handles funds received for the pool
  receive() external payable {
    // Make sure enough funds are sent, and the source is whitelisted
    require(msg.value > 0, "Insufficient Funds");
    require(fundAddresses[msg.sender], "Invalid Fund Address");

    // Update the total received funds for the currently open launch
    _totalFundsByLaunch[_currentLaunch] += msg.value;
  }
}
function totalClaimableByAccount(
  address _account,
  uint256 _fromLaunch,
  uint256 _toLaunch
) public view returns (
  uint256,
  uint256[] memory
) {
  require(_fromLaunch > 0 && _toLaunch <= _currentLaunch && _fromLaunch <= _toLaunch, "Invalid Launch Period");

  // Calculate the total claimable amount along with the detail for each launch in the specified period range
  uint256 totalClaimable = 0;
  uint256 periodSize = _toLaunch - _fromLaunch + 1;
  uint256[] memory maxPerLaunch = new uint256[](periodSize);
  for (uint256 i = _fromLaunch; i <= _toLaunch ; i++) {
    uint256 maxAmount = 0;

    if (_totalBurntOreByLaunch[i] > 0 && _totalFundsByLaunch[i] > 0) {
      maxAmount = (_burntOreByAddress[_account][i] * _totalFundsByLaunch[i]) / _totalBurntOreByLaunch[i];
      uint256 claimable = maxAmount - _claimedByAddress[_account][i];

      if (claimable > 0) {
        totalClaimable += claimable;
      }
    }

    uint256 index = i - _fromLaunch;
    maxPerLaunch[index] = maxAmount;
  }

  return (totalClaimable, maxPerLaunch);
}

// Calculates and returns the total amount of claimable funds for the specified account and period