WizardFirstTry.sol 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147
  1. // SPDX-License-Identifier: MIT
  2. pragma solidity ^0.8.2;
  3. import "../../contracts/governance/Governor.sol";
  4. import "../../contracts/governance/extensions/GovernorCountingSimple.sol";
  5. import "../../contracts/governance/extensions/GovernorVotes.sol";
  6. import "../../contracts/governance/extensions/GovernorVotesQuorumFraction.sol";
  7. import "../../contracts/governance/extensions/GovernorTimelockCompound.sol";
  8. /*
  9. Wizard options:
  10. ERC20Votes
  11. TimelockCompound
  12. */
  13. contract WizardFirstTry is Governor, GovernorCountingSimple, GovernorVotes, GovernorVotesQuorumFraction, GovernorTimelockCompound {
  14. constructor(ERC20Votes _token, ICompoundTimelock _timelock, string memory name, uint256 quorumFraction)
  15. Governor(name)
  16. GovernorVotes(_token)
  17. GovernorVotesQuorumFraction(quorumFraction)
  18. GovernorTimelockCompound(_timelock)
  19. {}
  20. function isExecuted(uint256 proposalId) public view returns (bool) {
  21. return _proposals[proposalId].executed;
  22. }
  23. function isCanceled(uint256 proposalId) public view returns (bool) {
  24. return _proposals[proposalId].canceled;
  25. }
  26. uint256 _votingDelay;
  27. function votingDelay() public view override virtual returns (uint256) { // HARNESS: pure -> view
  28. return _votingDelay;
  29. }
  30. uint256 _votingPeriod;
  31. function votingPeriod() public view override virtual returns (uint256) { // HARNESS: pure -> view
  32. return _votingPeriod;
  33. }
  34. mapping(uint256 => uint256) public ghost_sum_vote_power_by_id;
  35. function _castVote(
  36. uint256 proposalId,
  37. address account,
  38. uint8 support,
  39. string memory reason
  40. ) internal override virtual returns (uint256) {
  41. uint256 deltaWeight = super._castVote(proposalId, account, support, reason); //HARNESS
  42. ghost_sum_vote_power_by_id[proposalId] += deltaWeight;
  43. return deltaWeight;
  44. }
  45. function callPropose(address[] memory targets,
  46. uint256[] memory values,
  47. bytes[] memory calldatas) public virtual returns (uint256) {
  48. return super.propose(targets, values, calldatas, "");
  49. }
  50. // Harness of castVoteWithReason to be able to impose requirement on the proposal ID.
  51. uint256 public _pId_Harness;
  52. function castVoteWithReason(uint256 proposalId, uint8 support, string calldata reason)
  53. public
  54. override(IGovernor, Governor)
  55. returns (uint256)
  56. {
  57. require(proposalId == _pId_Harness);
  58. return super.castVoteWithReason(proposalId, support, reason);
  59. }
  60. // The following functions are overrides required by Solidity.
  61. function quorum(uint256 blockNumber)
  62. public
  63. view
  64. override(IGovernor, GovernorVotesQuorumFraction)
  65. returns (uint256)
  66. {
  67. return super.quorum(blockNumber);
  68. }
  69. function getVotes(address account, uint256 blockNumber)
  70. public
  71. view
  72. override(IGovernor, GovernorVotes)
  73. returns (uint256)
  74. {
  75. return super.getVotes(account, blockNumber);
  76. }
  77. function state(uint256 proposalId)
  78. public
  79. view
  80. override(Governor, GovernorTimelockCompound)
  81. returns (ProposalState)
  82. {
  83. return super.state(proposalId);
  84. }
  85. function propose(address[] memory targets, uint256[] memory values, bytes[] memory calldatas, string memory description)
  86. public
  87. override(Governor, IGovernor)
  88. returns (uint256)
  89. {
  90. return super.propose(targets, values, calldatas, description);
  91. }
  92. function _execute(uint256 proposalId, address[] memory targets, uint256[] memory values, bytes[] memory calldatas, bytes32 descriptionHash)
  93. internal
  94. override(Governor, GovernorTimelockCompound)
  95. {
  96. super._execute(proposalId, targets, values, calldatas, descriptionHash);
  97. }
  98. function _cancel(address[] memory targets, uint256[] memory values, bytes[] memory calldatas, bytes32 descriptionHash)
  99. internal
  100. override(Governor, GovernorTimelockCompound)
  101. returns (uint256)
  102. {
  103. return super._cancel(targets, values, calldatas, descriptionHash);
  104. }
  105. function _executor()
  106. internal
  107. view
  108. override(Governor, GovernorTimelockCompound)
  109. returns (address)
  110. {
  111. return super._executor();
  112. }
  113. function supportsInterface(bytes4 interfaceId)
  114. public
  115. view
  116. override(Governor, GovernorTimelockCompound)
  117. returns (bool)
  118. {
  119. return super.supportsInterface(interfaceId);
  120. }
  121. }