posemesh_domain_http/
lib.rs

1use futures::channel::mpsc::Receiver;
2
3#[cfg(target_family = "wasm")]
4use crate::domain_data::UploadDomainData;
5use crate::domain_data::{download_v1_stream, DomainData, DownloadQuery};
6
7mod auth;
8pub mod config;
9pub mod domain_data;
10pub mod discovery;
11#[cfg(target_family = "wasm")]
12pub mod wasm;
13
14use crate::discovery::DiscoveryService;
15
16#[derive(Debug, Clone)]
17pub struct DomainClient {
18    discovery_client: DiscoveryService,
19    pub client_id: String,
20}
21
22impl DomainClient {
23    fn new(api_url: &str, dds_url: &str, client_id: &str) -> Self {
24        if client_id.is_empty() {
25            panic!("client_id is empty");
26        }
27        Self {
28            discovery_client: DiscoveryService::new(api_url, dds_url, client_id),
29            client_id: client_id.to_string(),
30        }
31    }
32
33    pub async fn new_with_app_credential(api_url: &str, dds_url: &str, client_id: &str, app_key: &str, app_secret: &str) -> Result<Self, Box<dyn std::error::Error>> {
34        let mut dc = DomainClient::new(api_url, dds_url, client_id);
35        let _ = dc.discovery_client.sign_in_as_auki_app(app_key, app_secret).await?;
36        Ok(dc)
37    }
38
39    pub async fn new_with_user_credential(api_url: &str, dds_url: &str, client_id: &str, email: &str, password: &str) -> Result<Self, Box<dyn std::error::Error>> {
40        let mut dc = DomainClient::new(api_url, dds_url, client_id);
41        let _ = dc.discovery_client.sign_in_with_auki_account(email, password).await?;
42        Ok(dc)
43    }
44
45    pub async fn download_domain_data(&self, domain_id: &str, query: &DownloadQuery) -> Result<Receiver<Result<DomainData, Box<dyn std::error::Error + Send + Sync>>>, Box<dyn std::error::Error>> {
46        let domain = self.discovery_client.auth_domain(domain_id).await?;
47        let rx = download_v1_stream(&domain.domain.domain_server.url, &self.client_id, &domain.access_token, domain_id, query).await?;
48        Ok(rx)
49    }
50
51    #[cfg(not(target_family = "wasm"))]
52    pub async fn upload_domain_data(&self, domain_id: &str, data: Receiver<domain_data::UploadDomainData>) -> Result<Vec<DomainData>, Box<dyn std::error::Error>> {
53        use crate::domain_data::upload_v1_stream;
54        let domain = self.discovery_client.auth_domain(domain_id).await?;
55        upload_v1_stream(&domain.domain.domain_server.url, &domain.access_token, domain_id, data).await
56    }
57
58    #[cfg(target_family = "wasm")]
59    pub async fn upload_domain_data(&self, domain_id: &str, data: Vec<UploadDomainData>) -> Result<(), Box<dyn std::error::Error>> {
60        let domain = self.discovery_client.auth_domain(domain_id).await?;
61        upload_v1(&domain.domain.domain_server.url, &domain.access_token, domain_id, data).await
62    }
63}
64
65#[cfg(not(target_family = "wasm"))]
66#[cfg(test)]
67mod tests {
68    use crate::domain_data::{UpdateDomainData, UploadDomainData};
69
70    use super::*;
71    use futures::{channel::mpsc, StreamExt};
72    use tokio::spawn;
73
74    fn get_config() -> config::Config {
75        if std::path::Path::new("../.env.local").exists() {
76            dotenvy::from_filename("../.env.local").ok();
77            dotenvy::dotenv().ok();
78        }
79        let config = config::Config::from_env().unwrap();
80        config
81    }
82
83    #[tokio::test]
84    async fn test_download_domain_data_with_app_credential() {
85        // Create a test client
86        let config = get_config();
87        let client = DomainClient::new_with_app_credential(&config.api_url, &config.dds_url, &config.client_id, &config.app_key, &config.app_secret).await.expect("Failed to create client");
88        
89        // Create a test query
90        let query = DownloadQuery {
91            ids: vec![],
92            name: None,
93            data_type: Some("dmt_accel_csv".to_string()),
94        };
95
96        // Test the download function
97        let result = client.download_domain_data(
98            &config.domain_id,
99            &query
100        ).await;
101
102        assert!(result.is_ok(), "error message : {:?}", result.err());
103
104        let mut count = 0;
105        let mut rx = result.unwrap();
106        while let Some(Ok(_)) = rx.next().await {
107            count += 1;
108        }
109        assert!(count > 0);
110    }
111
112    #[tokio::test]
113    async fn test_upload_domain_data_with_user_credential() {
114        use futures::SinkExt;
115        let config = get_config();
116        let user_email = std::env::var("USER_EMAIL").unwrap();
117        let user_password = std::env::var("USER_PASSWORD").unwrap();
118        let client = DomainClient::new_with_user_credential(&config.api_url, &config.dds_url, &config.client_id, &user_email, &user_password).await.expect("Failed to create client");
119
120        let data = vec![UploadDomainData {
121            create: None,
122            update: Some(UpdateDomainData {
123                id: "a84a36e5-312b-4f80-974a-06f5d19c1e16".to_string(),
124            }),
125            data: "{\"test\": \"test updated\"}".as_bytes().to_vec(),
126        }, UploadDomainData {
127            create: None,
128            update: Some(UpdateDomainData {
129                id: "a08dc12f-c79e-4f5e-b388-c09ad6d8cfd8".to_string(),
130            }),
131            data: "{\"test\": \"test updated\"}".as_bytes().to_vec(),
132        }];
133        let (mut tx, rx) = mpsc::channel(10);
134        spawn(async move {
135            for d in data {
136                tx.send(d).await.unwrap();
137            }
138            tx.close().await.unwrap();
139        });
140        let result = client.upload_domain_data(&config.domain_id, rx).await;
141
142        assert!(result.is_ok(), "error message : {:?}", result.err());
143        assert_eq!(result.unwrap().len(), 2);
144    }
145}
146
posemesh_domain_http/lib.rs

posemesh_domain_http/
lib.rs
