1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

#![no_std]
#![allow(clippy::string_lit_as_bytes)]

#[macro_use]
extern crate elrond_wasm;

imports!();

/// The module deals with temporarily pausing contract operations.
/// It provides a flag that contracts can use to check if owner decided to users the entire contract.
/// Use the features module for more granular on/off switches.
#[elrond_wasm_derive::module(UsersModuleImpl)]
pub trait UsersModule {
	/// Each user gets a user id. This is in order to be able to iterate over their data.
	/// This is a mapping from user address to user id.
	/// The key is the bytes "user_id" concatenated with their public key.
	/// The value is the user id.
	#[view(getUserId)]
	#[storage_get("user_id")]
	fn get_user_id(&self, address: &Address) -> usize;

	#[storage_set("user_id")]
	fn set_user_id(&self, address: &Address, user_id: usize);

	#[view(getUserAddress)]
	#[storage_get("user_address")]
	fn get_user_address(&self, user_id: usize) -> Address;

	#[storage_set("user_address")]
	fn set_user_address(&self, user_id: usize, address: &Address);

	/// Retrieves the number of delegtors, including the owner,
	/// even if they no longer have anything in the contract.
	#[view(getNumUsers)]
	#[storage_get("num_users")]
	fn get_num_users(&self) -> usize;

	/// Yields how accounts are registered in the contract.
	/// Note that not all of them must have stakes greater than zero.
	#[storage_set("num_users")]
	fn set_num_users(&self, num_users: usize);

	fn get_or_create_user(&self, address: &Address) -> usize {
		let mut user_id = self.get_user_id(&address);
		if user_id == 0 {
			let mut num_users = self.get_num_users();
			num_users += 1;
			self.set_num_users(num_users);
			user_id = num_users;
			self.set_user_id(&address, user_id);
			self.set_user_address(user_id, &address);
		}
		user_id
	}

	#[endpoint(updateUserAddress)]
	fn update_user_address(&self, #[var_args] addresses: VarArgs<Address>) -> SCResult<()> {
		for address in addresses.into_vec() {
			let user_id = self.get_user_id(&address);
			require!(user_id > 0, "unknown address");
			self.set_user_address(user_id, &address);
		}
		Ok(())
	}
}