Skip to content

Commit

Permalink
新規作成
Browse files Browse the repository at this point in the history
  • Loading branch information
@tanapl
tanapl committed Nov 25, 2024
1 parent 0456a2f commit 6b7ea05
Show file tree
Hide file tree
Showing 2 changed files with 308 additions and 158 deletions.
374 changes: 216 additions & 158 deletions pkgs/contract/contracts/splitscreator/SplitsCreator.sol
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,163 +11,221 @@ import { IHatsTimeFrameModule } from "../timeframe/IHatsTimeFrameModule.sol";
import { ERC2771Context } from "@openzeppelin/contracts/metatx/ERC2771Context.sol";
import { Clone } from "solady/src/utils/Clone.sol";

import "hardhat/console.sol";

contract SplitsCreator is ISplitsCreator, Clone {
function TRUSTED_FORWARDER() public pure returns (address) {
return _getArgAddress(12);
}

function HATS() public pure returns (IHats) {
return IHats(_getArgAddress(44));
}

function SPLIT_FACTORY_V2() public pure returns (ISplitFactoryV2) {
return ISplitFactoryV2(_getArgAddress(76));
}

function HATS_TIME_FRAME_MODULE()
public
pure
returns (IHatsTimeFrameModule)
{
return IHatsTimeFrameModule(_getArgAddress(108));
}

function FRACTION_TOKEN() public pure returns (IFractionToken) {
return IFractionToken(_getArgAddress(140));
}

function create(
SplitsInfo[] memory _splitsInfo
) external returns (address) {
uint256 numOfShareHolders = 0;
for (uint i = 0; i < _splitsInfo.length; i++) {
SplitsInfo memory _splitInfo = _splitsInfo[i];
for (uint si = 0; si < _splitInfo.wearers.length; si++) {
uint256 tokenId = FRACTION_TOKEN().getTokenId(
_splitInfo.hatId,
_splitInfo.wearers[si]
);
address[] memory recepients = FRACTION_TOKEN()
.getTokenRecipients(tokenId);
numOfShareHolders += recepients.length;
}
}

address[] memory shareHolders = new address[](numOfShareHolders);
address[] memory wearers = new address[](numOfShareHolders);
uint256[] memory hatIdsOfShareHolders = new uint256[](
numOfShareHolders
);
uint256[] memory roleMultipliersOfShareHolders = new uint256[](
numOfShareHolders
);
uint256[] memory hatsTimeFrameMultipliersOfShareHolders = new uint256[](
numOfShareHolders
);

uint256 shareHolderIndex = 0;

for (uint i = 0; i < _splitsInfo.length; i++) {
SplitsInfo memory _splitInfo = _splitsInfo[i];
uint256 roleMultiplier = _splitInfo.multiplierTop /
_splitInfo.multiplierBottom;
for (uint si = 0; si < _splitInfo.wearers.length; si++) {
address wearer = _splitInfo.wearers[si];

require(
HATS().balanceOf(wearer, _splitInfo.hatId) > 0,
"Invalid wearer"
);

uint256 tokenId = FRACTION_TOKEN().getTokenId(
_splitInfo.hatId,
wearer
);
uint256 hatsTimeFrameMultiplier = _getHatsTimeFrameMultiplier(
wearer,
_splitInfo.hatId
);

// FractionTokenのホルダーに対する分配の計算
address[] memory recipients = FRACTION_TOKEN()
.getTokenRecipients(tokenId);
for (uint j = 0; j < recipients.length; j++) {
shareHolders[shareHolderIndex] = recipients[j];
wearers[shareHolderIndex] = wearer;
hatIdsOfShareHolders[shareHolderIndex] = _splitInfo.hatId;
roleMultipliersOfShareHolders[
shareHolderIndex
] = roleMultiplier;
hatsTimeFrameMultipliersOfShareHolders[
shareHolderIndex
] = hatsTimeFrameMultiplier;
shareHolderIndex++;
}
}
}

uint256[] memory balanceOfShareHolders = FRACTION_TOKEN()
.balanceOfBatch(shareHolders, wearers, hatIdsOfShareHolders);

uint256 totalAllocation = 0;
uint256[] memory allocations = new uint256[](shareHolderIndex);
for (uint256 i = 0; i < shareHolderIndex; i++) {
uint256 share = balanceOfShareHolders[i] *
roleMultipliersOfShareHolders[i] *
hatsTimeFrameMultipliersOfShareHolders[i];
totalAllocation += share;
allocations[i] = share;
}

SplitV2Lib.Split memory _splitParams = SplitV2Lib.Split({
recipients: shareHolders,
allocations: allocations,
totalAllocation: totalAllocation,
distributionIncentive: 0
});

address split = SPLIT_FACTORY_V2().createSplitDeterministic(
_splitParams,
address(this),
msg.sender,
_generateSalt(_splitsInfo)
);

emit SplitsCreated(split, shareHolders, allocations, totalAllocation);

return split;
}

function _getHatsTimeFrameMultiplier(
address _wearer,
uint256 _hatId
) internal view returns (uint256) {
if (address(HATS_TIME_FRAME_MODULE()) == address(0)) return 1;
return
_sqrt(
HATS_TIME_FRAME_MODULE().getWearingElapsedTime(_wearer, _hatId)
);
}

function _sqrt(uint256 y) internal pure returns (uint256 z) {
if (y > 3) {
z = y;
uint256 x = y / 2 + 1;
while (x < z) {
z = x;
x = (y / x + x) / 2;
}
} else if (y != 0) {
z = 1;
}
// else z = 0 (default value)
}

function _generateSalt(
SplitsInfo[] memory splitsInfo
) internal pure returns (bytes32) {
return keccak256(abi.encode(splitsInfo));
}
function TRUSTED_FORWARDER() public pure returns (address) {
return _getArgAddress(12);
}

function HATS() public pure returns (IHats) {
return IHats(_getArgAddress(44));
}

function SPLIT_FACTORY_V2() public pure returns (ISplitFactoryV2) {
return ISplitFactoryV2(_getArgAddress(76));
}

function HATS_TIME_FRAME_MODULE()
public
pure
returns (IHatsTimeFrameModule)
{
return IHatsTimeFrameModule(_getArgAddress(108));
}

function FRACTION_TOKEN() public pure returns (IFractionToken) {
return IFractionToken(_getArgAddress(140));
}

/**
* @notice Creates a new split contract based on the provided splits information.
* @param _splitsInfo An array of SplitsInfo structs containing details about roles, wearers, and multipliers.
* @return The address of the newly created split contract.
*/
function create(
SplitsInfo[] memory _splitsInfo
) external returns (address) {
(
address[] memory shareHolders,
uint256[] memory allocations,
uint256 totalAllocation
) = _calculateAllocations(_splitsInfo);

SplitV2Lib.Split memory _splitParams = SplitV2Lib.Split({
recipients: shareHolders,
allocations: allocations,
totalAllocation: totalAllocation,
distributionIncentive: 0
});

address split = SPLIT_FACTORY_V2().createSplitDeterministic(
_splitParams,
address(this),
msg.sender,
_generateSalt(_splitsInfo)
);

emit SplitsCreated(split, shareHolders, allocations, totalAllocation);

return split;
}

/**
* @notice Previews the allocations without creating a split contract.
* @param _splitsInfo An array of SplitsInfo structs containing details about roles, wearers, and multipliers.
* @return shareHolders An array of shareholder addresses.
* @return percentages Corresponding allocation percentages (with 18 decimals) for each shareholder.
*/
function preview(
SplitsInfo[] memory _splitsInfo
)
external
view
returns (
address[] memory shareHolders,
uint256[] memory percentages
)
{
(
address[] memory _shareHolders,
uint256[] memory allocations,
uint256 totalAllocation
) = _calculateAllocations(_splitsInfo);

percentages = new uint256[](allocations.length);
for (uint256 i = 0; i < allocations.length; i++) {
percentages[i] = (allocations[i] * 1e18) / totalAllocation;
}

return (_shareHolders, percentages);
}

/**
* @dev Internal function to calculate allocations for shareholders.
* @param _splitsInfo An array of SplitsInfo structs containing details about roles, wearers, and multipliers.
* @return shareHolders An array of shareholder addresses.
* @return allocations Corresponding allocations for each shareholder.
* @return totalAllocation Sum of all allocations.
*/
function _calculateAllocations(
SplitsInfo[] memory _splitsInfo
)
internal
view
returns (
address[] memory shareHolders,
uint256[] memory allocations,
uint256 totalAllocation
)
{
uint256 numOfShareHolders = 0;
for (uint256 i = 0; i < _splitsInfo.length; i++) {
SplitsInfo memory _splitInfo = _splitsInfo[i];
for (uint256 si = 0; si < _splitInfo.wearers.length; si++) {
uint256 tokenId = FRACTION_TOKEN().getTokenId(
_splitInfo.hatId,
_splitInfo.wearers[si]
);
address[] memory recipients = FRACTION_TOKEN()
.getTokenRecipients(tokenId);
numOfShareHolders += recipients.length;
}
}

shareHolders = new address[](numOfShareHolders);
address[] memory wearers = new address[](numOfShareHolders);
uint256[] memory hatIdsOfShareHolders = new uint256[](numOfShareHolders);
uint256[] memory roleMultipliersOfShareHolders = new uint256[](
numOfShareHolders
);
uint256[] memory hatsTimeFrameMultipliersOfShareHolders = new uint256[](
numOfShareHolders
);

uint256 shareHolderIndex = 0;

for (uint256 i = 0; i < _splitsInfo.length; i++) {
SplitsInfo memory _splitInfo = _splitsInfo[i];
uint256 roleMultiplier = _splitInfo.multiplierTop /
_splitInfo.multiplierBottom;
for (uint256 si = 0; si < _splitInfo.wearers.length; si++) {
address wearer = _splitInfo.wearers[si];

require(
HATS().balanceOf(wearer, _splitInfo.hatId) > 0,
"Invalid wearer"
);

uint256 tokenId = FRACTION_TOKEN().getTokenId(
_splitInfo.hatId,
wearer
);
uint256 hatsTimeFrameMultiplier = _getHatsTimeFrameMultiplier(
wearer,
_splitInfo.hatId
);

// Get the recipients from FractionToken
address[] memory recipients = FRACTION_TOKEN()
.getTokenRecipients(tokenId);
for (uint256 j = 0; j < recipients.length; j++) {
shareHolders[shareHolderIndex] = recipients[j];
wearers[shareHolderIndex] = wearer;
hatIdsOfShareHolders[shareHolderIndex] = _splitInfo.hatId;
roleMultipliersOfShareHolders[
shareHolderIndex
] = roleMultiplier;
hatsTimeFrameMultipliersOfShareHolders[
shareHolderIndex
] = hatsTimeFrameMultiplier;
shareHolderIndex++;
}
}
}

uint256[] memory balanceOfShareHolders = FRACTION_TOKEN()
.balanceOfBatch(shareHolders, wearers, hatIdsOfShareHolders);

totalAllocation = 0;
allocations = new uint256[](shareHolderIndex);
for (uint256 i = 0; i < shareHolderIndex; i++) {
uint256 share = balanceOfShareHolders[i] *
roleMultipliersOfShareHolders[i] *
hatsTimeFrameMultipliersOfShareHolders[i];
totalAllocation += share;
allocations[i] = share;
}

return (shareHolders, allocations, totalAllocation);
}

function _getHatsTimeFrameMultiplier(
address _wearer,
uint256 _hatId
) internal view returns (uint256) {
if (address(HATS_TIME_FRAME_MODULE()) == address(0)) return 1;
return
_sqrt(
HATS_TIME_FRAME_MODULE().getWearingElapsedTime(_wearer, _hatId)
);
}

function _sqrt(uint256 y) internal pure returns (uint256 z) {
if (y > 3) {
z = y;
uint256 x = y / 2 + 1;
while (x < z) {
z = x;
x = (y / x + x) / 2;
}
} else if (y != 0) {
z = 1;
}
// else z = 0 (default value)
}

function _generateSalt(
SplitsInfo[] memory splitsInfo
) internal pure returns (bytes32) {
return keccak256(abi.encode(splitsInfo));
}
}
Loading

0 comments on commit 6b7ea05

Please sign in to comment.