From e7b3ea37f6d63c6ed990a00e82b0bad7f8688ba7 Mon Sep 17 00:00:00 2001
From: Ezerous <ezerous@gmail.com>
Date: Mon, 16 Nov 2020 13:13:39 +0200
Subject: [PATCH] Voting contract init

---
 .../concordia-contracts/contracts/Forum.sol   | 23 +++--
 .../contracts/Migrations.sol                  |  5 +-
 .../concordia-contracts/contracts/Voting.sol  | 90 +++++++++++++++++++
 .../migrations/2_deploy_contracts.js          |  3 +-
 .../concordia-contracts/truffle-config.js     |  2 +-
 5 files changed, 110 insertions(+), 13 deletions(-)
 create mode 100644 packages/concordia-contracts/contracts/Voting.sol

diff --git a/packages/concordia-contracts/contracts/Forum.sol b/packages/concordia-contracts/contracts/Forum.sol
index 13ad2ea..ae6ec07 100644
--- a/packages/concordia-contracts/contracts/Forum.sol
+++ b/packages/concordia-contracts/contracts/Forum.sol
@@ -1,5 +1,5 @@
 //SPDX-License-Identifier: MIT
-pragma solidity 0.7.1;
+pragma solidity 0.7.4;
 
 contract Forum {
 
@@ -133,10 +133,10 @@ contract Forum {
     function getTopic(uint topicID) public view returns (address, string memory, uint, uint[] memory) {
         require(topicID<numTopics);
         return (
-            topics[topicID].author,
-            users[topics[topicID].author].username,
-            topics[topicID].timestamp,
-            topics[topicID].postIDs
+        topics[topicID].author,
+        users[topics[topicID].author].username,
+        topics[topicID].timestamp,
+        topics[topicID].postIDs
         );
     }
 
@@ -145,13 +145,18 @@ contract Forum {
         return topics[topicID].postIDs;
     }
 
+    function getTopicAuthor(uint topicID) public view returns (address) {
+        require(topicID<numTopics); // Topic should exist
+        return topics[topicID].author;
+    }
+
     function getPost(uint postID) public view returns (address, string memory, uint, uint) {
         require(postID<numPosts);
         return (
-            posts[postID].author,
-            users[posts[postID].author].username,
-            posts[postID].timestamp,
-            posts[postID].topicID
+        posts[postID].author,
+        users[posts[postID].author].username,
+        posts[postID].timestamp,
+        posts[postID].topicID
         );
     }
 }
diff --git a/packages/concordia-contracts/contracts/Migrations.sol b/packages/concordia-contracts/contracts/Migrations.sol
index 8c29f66..503020e 100644
--- a/packages/concordia-contracts/contracts/Migrations.sol
+++ b/packages/concordia-contracts/contracts/Migrations.sol
@@ -1,10 +1,11 @@
-pragma solidity 0.7.1;
+//SPDX-License-Identifier: MIT
+pragma solidity 0.7.4;
 
 contract Migrations {
     address public owner;
     uint public last_completed_migration;
 
-    constructor() public {
+    constructor() {
         owner = msg.sender;
     }
 
diff --git a/packages/concordia-contracts/contracts/Voting.sol b/packages/concordia-contracts/contracts/Voting.sol
new file mode 100644
index 0000000..debf752
--- /dev/null
+++ b/packages/concordia-contracts/contracts/Voting.sol
@@ -0,0 +1,90 @@
+//SPDX-License-Identifier: MIT
+pragma solidity 0.7.4;
+
+import "./Forum.sol";
+
+contract Voting {
+    Forum public forum;
+
+    constructor(Forum addr) {
+        forum = Forum(addr);
+    }
+
+    struct Poll {
+        uint topicID;
+        uint numOptions;
+        string dataHash;
+        mapping (address => uint) voters;
+        uint[] voteCounts;  // First element will hold total count
+        uint timestamp;
+    }
+
+    mapping (uint => Poll) polls;
+
+    event PollCreated(uint topicID);
+    event UserVoted(address userAddress);
+
+    function createPoll(uint topicID, uint numOptions, string memory dataHash) public returns (uint) {
+        require(forum.hasUserSignedUp(msg.sender));  // Only registered users can create polls
+        require(topicID<forum.getNumberOfTopics()); // Only allow poll creation if topic exists
+        require (forum.getTopicAuthor(topicID) == msg.sender); // Only allow poll creation from the author of the topic
+        require(polls[topicID].timestamp == 0); // Only allow poll creation if it doesn't exist yet
+
+        Poll storage poll = polls[topicID];
+        poll.topicID = topicID;
+        poll.numOptions = numOptions;
+        poll.dataHash = dataHash;
+        poll.voteCounts = new uint[](numOptions+1);
+        poll.timestamp = block.timestamp;
+
+        emit PollCreated(topicID);
+        return topicID;
+    }
+
+    // Verify that poll exists
+    function isPollExistent(uint topicID) public view returns (bool) {
+        if (polls[topicID].timestamp != 0)
+            return true;
+        return false;
+    }
+
+    function hasVoted(uint topicID, address voter) public view returns (bool) {
+        require(isPollExistent(topicID));
+        if (polls[topicID].voters[voter] != 0)
+            return true;
+        return false;
+    }
+
+    function getPollInfo(uint topicID) public view returns (uint, string memory, uint, uint) {
+        require(isPollExistent(topicID));
+        return (
+        polls[topicID].numOptions,
+        polls[topicID].dataHash,
+        polls[topicID].timestamp,
+        polls[topicID].voteCounts[0]
+        );
+    }
+
+    // Gets vote count for a specific option (option 0 will return total count)
+    function getVoteCount(uint topicID, uint option) public view returns (uint) {
+        require(isPollExistent(topicID)); // Verify that poll exists
+        return (polls[topicID].voteCounts[option]);
+    }
+
+    function vote(uint topicID, uint option) public {
+        require(isPollExistent(topicID));
+        Poll storage poll = polls[topicID];
+        require(option > 0 && option <= poll.numOptions); // Verify that this option exists
+        address voter = msg.sender;
+        uint currentVote = poll.voters[voter];
+        if(currentVote == option)
+            return;
+        if(currentVote == 0)   // Voter hadn't voted before
+            poll.voteCounts[0]++;
+        else
+            poll.voteCounts[currentVote]--;
+        poll.voteCounts[option]++;
+        poll.voters[voter] = option;
+        emit UserVoted(voter);
+    }
+}
diff --git a/packages/concordia-contracts/migrations/2_deploy_contracts.js b/packages/concordia-contracts/migrations/2_deploy_contracts.js
index ec5722e..3e81c0e 100644
--- a/packages/concordia-contracts/migrations/2_deploy_contracts.js
+++ b/packages/concordia-contracts/migrations/2_deploy_contracts.js
@@ -1,6 +1,7 @@
 const Forum = artifacts.require('Forum');
+const Voting = artifacts.require('Voting');
 
 // eslint-disable-next-line func-names
 module.exports = function (deployer) {
-  deployer.deploy(Forum);
+  deployer.deploy(Forum).then((forum) => deployer.deploy(Voting, forum.address));
 };
diff --git a/packages/concordia-contracts/truffle-config.js b/packages/concordia-contracts/truffle-config.js
index 3ef72a1..d66ff67 100644
--- a/packages/concordia-contracts/truffle-config.js
+++ b/packages/concordia-contracts/truffle-config.js
@@ -8,7 +8,7 @@ module.exports = {
   // to customize your Truffle configuration!
   compilers: {
     solc: {
-      version: '0.7.1',
+      version: '0.7.4',
     },
   },
   contracts_build_directory: path.join(__dirname, 'build/'),