// SPDX-License-Identifier: MIT

pragma solidity 0.8.19;

import "../interfaces/ITroveManager.sol";
import "../interfaces/IFactory.sol";

/*  Helper contract for grabbing Trove data for the front end. Not part of the PropelCore Propel system. */
contract TroveManagerGetters {
	struct Collateral {
		IERC20 collateral;
		address[] troveManagers;
	}

	IFactory public immutable factory;

	constructor(IFactory _factory) {
		factory = _factory;
	}

	/**
        @notice Returns all active system trove managers and collaterals, as an
        `       array of tuples of [(collateral, [troveManager, ...]), ...]
     */
	function getAllCollateralsAndTroveManagers() external view returns (Collateral[] memory) {
		uint256 length = factory.troveManagerCount();
		address[2][] memory troveManagersAndCollaterals = new address[2][](length);
		address[] memory uniqueCollaterals = new address[](length);
		uint256 collateralCount;
		for (uint i = 0; i < length; i++) {
			address troveManager = factory.troveManagers(i);
			IERC20 collateral = ITroveManager(troveManager).collateralToken();
			troveManagersAndCollaterals[i] = [troveManager, address(collateral)];
			for (uint x = 0; x < length; x++) {
				if (uniqueCollaterals[x] == address(collateral)) break;
				if (uniqueCollaterals[x] == address(0)) {
					uniqueCollaterals[x] = address(collateral);
					collateralCount++;
					break;
				}
			}
		}
		Collateral[] memory collateralMap = new Collateral[](collateralCount);
		for (uint i = 0; i < collateralCount; i++) {
			collateralMap[i].collateral = IERC20(uniqueCollaterals[i]);
			uint tmCollCount = 0;
			address[] memory troveManagers = new address[](length);
			for (uint x = 0; x < length; x++) {
				if (troveManagersAndCollaterals[x][1] == uniqueCollaterals[i]) {
					troveManagers[tmCollCount] = troveManagersAndCollaterals[x][0];
					tmCollCount++;
				}
			}
			collateralMap[i].troveManagers = new address[](tmCollCount);
			for (uint x = 0; x < tmCollCount; x++) {
				collateralMap[i].troveManagers[x] = troveManagers[x];
			}
		}

		return collateralMap;
	}

	/**
        @notice Returns a list of trove managers where `account` has an existing trove
     */
	function getActiveTroveManagersForAccount(address account) external view returns (address[] memory) {
		uint256 length = factory.troveManagerCount();
		address[] memory troveManagers = new address[](length);
		uint256 tmCount;
		for (uint i = 0; i < length; i++) {
			address troveManager = factory.troveManagers(i);
			if (ITroveManager(troveManager).getTroveStatus(account) > 0) {
				troveManagers[tmCount] = troveManager;
				tmCount++;
			}
		}
		assembly {
			mstore(troveManagers, tmCount)
		}
		return troveManagers;
	}
}