#![cfg_attr(not(feature = "std"), no_std)]
use ink::prelude::vec::Vec;
use ink::storage;
#[ink::contract]
mod private_voting {
#[ink(storage)]
pub struct PrivateVoting {
votes: storage::Mapping<AccountId, String>,
voter_whitelist: Vec<AccountId>,
is_open: bool,
}
impl PrivateVoting {
#[ink(constructor)]
pub fn new(voters: Vec<AccountId>) -> Self {
Self {
votes: storage::Mapping::default(),
voter_whitelist: voters,
is_open: true,
}
}
#[ink(message)]
pub fn cast_vote(&mut self, choice: String) {
let caller = self.env().caller();
assert!(self.voter_whitelist.contains(&caller), "Unauthorized voter");
assert!(self.is_open, "Voting is closed");
self.votes.insert(caller, &choice);
}
#[ink(message)]
pub fn close_voting(&mut self) {
self.is_open = false;
}
#[ink(message)]
pub fn get_vote(&self, voter: AccountId) -> Option<String> {
self.votes.get(voter)
}
}
}
|