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

use candid::Principal;
use ic_agent::Agent;
use ic_oss_types::{bucket::Token, cluster::ClusterInfo};
use serde_bytes::ByteBuf;
use std::{collections::BTreeSet, sync::Arc};

use crate::agent::{query_call, update_call};

#[derive(Clone)]
pub struct Client {
    agent: Arc<Agent>,
    cluster: Principal,
}

impl Client {
    pub fn new(agent: Arc<Agent>, cluster: Principal) -> Client {
        Client { agent, cluster }
    }

    /// the caller of agent should be canister controller
    pub async fn admin_set_managers(&self, args: BTreeSet<Principal>) -> Result<(), String> {
        update_call(&self.agent, &self.cluster, "admin_set_managers", (args,)).await?
    }

    /// the caller of agent should be canister manager
    pub async fn admin_sign_access_token(&self, args: Token) -> Result<ByteBuf, String> {
        update_call(
            &self.agent,
            &self.cluster,
            "admin_sign_access_token",
            (args,),
        )
        .await?
    }

    /// the caller of agent should be canister manager
    pub async fn admin_attach_policies(&self, args: Token) -> Result<(), String> {
        update_call(&self.agent, &self.cluster, "admin_attach_policies", (args,)).await?
    }

    /// the caller of agent should be canister manager
    pub async fn admin_detach_policies(&self, args: Token) -> Result<(), String> {
        update_call(&self.agent, &self.cluster, "admin_detach_policies", (args,)).await?
    }

    pub async fn access_token(&self, audience: Principal) -> Result<ByteBuf, String> {
        update_call(&self.agent, &self.cluster, "access_token", (audience,)).await?
    }

    pub async fn get_cluster_info(&self) -> Result<ClusterInfo, String> {
        query_call(&self.agent, &self.cluster, "get_cluster_info", ()).await?
    }
}