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

use async_trait::async_trait;
use filament_chain::{Account, Address};
use filament_encoding::StateReadDecode;
use serde::{Deserialize, Serialize};

use crate::{
    component::accounts::AccountsRead as _,
    query::{self, Prefix},
};

#[derive(Clone, Debug, PartialEq, Deserialize, Serialize)]
pub enum Query {
    AccountByAddress(Address),
}

impl query::Query for Query {
    const PREFIX: Prefix = Prefix::Accounts;
}

#[async_trait]
impl<S> query::Respond<S> for Query
where
    S: StateReadDecode,
{
    type Key = Vec<u8>;
    type Response = Response;

    async fn respond(&self, state: &S) -> eyre::Result<(Vec<u8>, Response)> {
        match self {
            Self::AccountByAddress(address) => {
                let account = state
                    .account(address)
                    .await?
                    .ok_or(eyre::eyre!("account not found"))?;

                Ok((
                    filament_encoding::to_bytes(&address)?.to_vec(),
                    Response::Account(account),
                ))
            },
        }
    }
}

#[derive(Debug, Deserialize, Serialize)]
pub enum Response {
    Account(Account),
}