108 lines
2.6 KiB
Solidity
108 lines
2.6 KiB
Solidity
|
pragma solidity ^0.5.0;
|
||
|
|
|
||
|
|
// @notice Contract to create posts
|
||
|
|
contract DReddit {
|
||
|
|
|
||
|
|
enum Ballot { NONE, UPVOTE, DOWNVOTE }
|
||
|
|
|
||
|
|
struct Post {
|
||
|
|
uint creationDate;
|
||
|
|
string description;
|
||
|
|
address owner;
|
||
|
|
uint upvotes;
|
||
|
|
uint downvotes;
|
||
|
|
mapping(address => Ballot) voters;
|
||
|
|
}
|
||
|
|
|
||
|
|
Post[] public posts;
|
||
|
|
|
||
|
|
event NewPost (
|
||
|
|
uint indexed postId,
|
||
|
|
address owner,
|
||
|
|
string description
|
||
|
|
);
|
||
|
|
|
||
|
|
event Vote(
|
||
|
|
uint indexed postId,
|
||
|
|
address voter,
|
||
|
|
uint8 vote
|
||
|
|
);
|
||
|
|
|
||
|
|
// @notice Number of posts created
|
||
|
|
// @return Num of posts
|
||
|
|
function numPosts()
|
||
|
|
public
|
||
|
|
view
|
||
|
|
returns(uint)
|
||
|
|
{
|
||
|
|
return posts.length;
|
||
|
|
}
|
||
|
|
|
||
|
|
// @notice Create Post
|
||
|
|
// @param _ipfsHash IPFS hash of the content of the post
|
||
|
|
function create(string memory _ipfsHash) public
|
||
|
|
{
|
||
|
|
require(bytes(_ipfsHash).length > 0, "IPFS hash is required");
|
||
|
|
|
||
|
|
posts.push(Post({
|
||
|
|
creationDate: now,
|
||
|
|
description: _ipfsHash,
|
||
|
|
owner: msg.sender,
|
||
|
|
upvotes: 0,
|
||
|
|
downvotes: 0
|
||
|
|
}));
|
||
|
|
|
||
|
|
emit NewPost(posts.length - 1, msg.sender, _ipfsHash);
|
||
|
|
}
|
||
|
|
|
||
|
|
// @notice Vote on a post
|
||
|
|
// @param _postId Id of the post to up/downvote
|
||
|
|
// @param _vote Vote selection: 0 -> none, 1 -> upvote, 2 -> downvote
|
||
|
|
function vote(uint _postId, uint8 _vote)
|
||
|
|
public
|
||
|
|
{
|
||
|
|
Post storage p = posts[_postId];
|
||
|
|
require(p.creationDate != 0, 'Post does not exist.');
|
||
|
|
require(p.voters[msg.sender] == Ballot.NONE, 'You already voted on this post.');
|
||
|
|
|
||
|
|
Ballot b = Ballot(_vote);
|
||
|
|
require(b != Ballot.NONE, 'Invalid vote');
|
||
|
|
|
||
|
|
if (b == Ballot.UPVOTE) {
|
||
|
|
p.upvotes++;
|
||
|
|
} else {
|
||
|
|
p.downvotes++;
|
||
|
|
}
|
||
|
|
p.voters[msg.sender] = b;
|
||
|
|
|
||
|
|
emit Vote(_postId, msg.sender, _vote);
|
||
|
|
}
|
||
|
|
|
||
|
|
// @notice Determine if the sender can vote on a post
|
||
|
|
// @param _postId Id of the post
|
||
|
|
// @return bool that indicates if the sender can vote or not
|
||
|
|
function canVote(uint _postId)
|
||
|
|
public
|
||
|
|
view
|
||
|
|
returns (bool)
|
||
|
|
{
|
||
|
|
if (_postId > posts.length - 1) return false;
|
||
|
|
|
||
|
|
Post storage p = posts[_postId];
|
||
|
|
return (p.voters[msg.sender] == Ballot.NONE);
|
||
|
|
}
|
||
|
|
|
||
|
|
// @notice Obtain vote for specific post
|
||
|
|
// @param _postId Id of the post
|
||
|
|
// @return uint that represents the vote: 0 -> none, 1 -> upvote, 2 -> downvote
|
||
|
|
function getVote(uint _postId)
|
||
|
|
public
|
||
|
|
view
|
||
|
|
returns (uint8)
|
||
|
|
{
|
||
|
|
Post storage p = posts[_postId];
|
||
|
|
return uint8(p.voters[msg.sender]);
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|