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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

use std::sync::Arc;
use codec::Codec;
use sp_blockchain::HeaderBackend;
use sp_runtime::{generic::BlockId, traits::Block as BlockT};
use jsonrpc_core::Result;
use jsonrpc_derive::rpc;
use sp_api::ProvideRuntimeApi;
use pallet_utils::{SpaceId, rpc::map_rpc_error};
use pallet_permissions::SpacePermission;

pub use roles_runtime_api::RolesApi as RolesRuntimeApi;

#[rpc]
pub trait RolesApi<BlockHash, AccountId> {
    #[rpc(name = "roles_getSpacePermissionsByAccount")]
    fn get_space_permissions_by_account(
        &self,
        at: Option<BlockHash>,
        account: AccountId,
        space_id: SpaceId
    ) -> Result<Vec<SpacePermission>>;

    #[rpc(name = "roles_getAccountsWithAnyRoleInSpace")]
    fn get_accounts_with_any_role_in_space(
        &self,
        at: Option<BlockHash>,
        space_id: SpaceId
    ) -> Result<Vec<AccountId>>;

    #[rpc(name = "roles_getSpaceIdsForAccountWithAnyRole")]
    fn get_space_ids_for_account_with_any_role(
        &self,
        at: Option<BlockHash>,
        account_id: AccountId
    ) -> Result<Vec<SpaceId>>;
}

pub struct Roles<C, M> {
    client: Arc<C>,
    _marker: std::marker::PhantomData<M>,
}

impl<C, M> Roles<C, M> {
    pub fn new(client: Arc<C>) -> Self {
        Self {
            client,
            _marker: Default::default(),
        }
    }
}

impl<C, Block, AccountId> RolesApi<<Block as BlockT>::Hash, AccountId>
    for Roles<C, Block>
where
    Block: BlockT,
    AccountId: Codec,
    C: Send + Sync + 'static + ProvideRuntimeApi<Block> + HeaderBackend<Block>,
    C::Api: RolesRuntimeApi<Block, AccountId>,
{
    fn get_space_permissions_by_account(
        &self, at:
        Option<<Block as BlockT>::Hash>,
        account: AccountId,
        space_id: SpaceId
    ) -> Result<Vec<SpacePermission>> {
        let api = self.client.runtime_api();
        let at = BlockId::hash(at.unwrap_or_else(|| self.client.info().best_hash));

        let runtime_api_result = api.get_space_permissions_by_account(&at, account, space_id);
        runtime_api_result.map_err(map_rpc_error)
    }

    fn get_accounts_with_any_role_in_space(
        &self, at:
        Option<<Block as BlockT>::Hash>,
        space_id: SpaceId
    ) -> Result<Vec<AccountId>> {
        let api = self.client.runtime_api();
        let at = BlockId::hash(at.unwrap_or_else(|| self.client.info().best_hash));

        let runtime_api_result = api.get_accounts_with_any_role_in_space(&at, space_id);
        runtime_api_result.map_err(map_rpc_error)
    }

    fn get_space_ids_for_account_with_any_role(
        &self, at:
        Option<<Block as BlockT>::Hash>,
        account_id: AccountId
    ) -> Result<Vec<SpaceId>> {
        let api = self.client.runtime_api();
        let at = BlockId::hash(at.unwrap_or_else(|| self.client.info().best_hash));

        let runtime_api_result = api.get_space_ids_for_account_with_any_role(&at, account_id);
        runtime_api_result.map_err(map_rpc_error)
    }
}