Merge pull request #25 from sum1t-here/feat/voting

feat: add functionality to end election

Foundry_Contracts/src/Voting.sol

@@ -17,6 +17,7 @@ struct Candidate{
 
 address public owner;
 string public electionName;
+bool public electionEnded = false;
 
 mapping(address => Voter) public voters;
 Candidate[] public candidates;
@@ -39,6 +40,12 @@ modifier ownerOnly(){
  _;
 }
 
+
+modifier electionOngoing() {
+    require(!electionEnded, "Election has ended");
+    _;
+}
+
 function addCandidate(string memory _name) public ownerOnly {
     // Check if a candidate with the same name already exists
     for (uint i = 0; i < candidates.length; i++) {
@@ -60,17 +67,22 @@ function authorize(address _person) public ownerOnly {
   voters[_person].authorized = true;
 }
 
-function vote(uint _voteIndex) public {
-  if (voters[msg.sender].voted) revert Voting__AlreadyVoted();
-  if (!voters[msg.sender].authorized) revert Voting__NotAuthorized();
-  if (_voteIndex >= candidates.length) revert Voting__IncorrectVoteIndex();
 
+function vote(uint _voteIndex) public electionOngoing() {
+  require(!voters[msg.sender].voted, Voting__AlreadyVoted());
+  require(voters[msg.sender].authorized, Voting__NotAuthorized());
+  require(_voteIndex<candidates.length, Voting__IncorrectVoteIndex());
+
   voters[msg.sender].vote = _voteIndex;
   voters[msg.sender].voted = true;
   candidates[_voteIndex].voteCount += 1;
   totalVotes +=1;
 }
 
+function endElection() public ownerOnly { 
+        electionEnded = true;
+    }
+
 function getCandidates() public view returns (Candidate[] memory) {
   return candidates;
 }