use std::{env, result}; use std::array::TryFromSliceError; use std::ops::Deref; use json::{object, JsonValue}; use once_cell::sync::Lazy; use thiserror::Error; use tokio::task::{spawn_blocking, JoinError}; use crate::grpc::account::Error::SomethingHappened; static API_KEY: Lazy = Lazy::new(||{ let key = env::var("ACCOUNT_GQL_API_KEY") .expect("no graphql ip specified"); key }); static CLIENT_URI: Lazy = Lazy::new(||{ env::var("ACCOUNT_GQL_URL") .ok() .and_then(|s| s.parse().ok()) .expect("no graphql ip specified") }); #[derive(Error, Debug)] pub enum Error{ #[error(transparent)] RequestError(#[from] ureq::Error), #[error(transparent)] Json(#[from] json::Error), //#[error(transparent)] //Status(#[from] tonic::Status), #[error("invalid password size: {0}")] PasswordConversion(#[from] TryFromSliceError), #[error("something happened")] SomethingHappened, #[error("error joining blocking task: {0}")] Join(#[from] JoinError) } pub type Result = result::Result; pub struct Client;//(reqwest::Client); impl Client{ pub async fn new() -> Result { //Ok(Self(reqwest::ClientBuilder::new().build()?)) Ok(Self) } async fn do_request(&self, request_data: JsonValue) -> Result{ let request = ureq::post(CLIENT_URI.as_str()) .header("X-API-Key", API_KEY.deref()) .content_type("application/json"); let mut response = spawn_blocking(move || request.send(request_data.to_string())).await??; let str_body = response.body_mut().read_to_string()?; Ok(json::parse(&str_body)?) /* let mut request = reqwest::Request::new(Method::POST, Url::from_str(CLIENT_URI.as_str()).unwrap()); *(request.body_mut()) = Some(Body::from(request_data.to_string())); request.headers_mut().insert("X-API-Key", HeaderValue::from_str(&API_KEY).unwrap()); request.headers_mut().insert("Content-Type", HeaderValue::from_str("application/json").unwrap()); let response = self.0.execute(request).await?; Ok(json::parse(&response.text().await?)?) */ } pub async fn get_nex_password(&mut self , pid: u32) -> Result<[u8; 16]>{ let req = self.do_request(object!{ "query": r"query($pid: Int!){ userByPid(pid: $pid){ nexPassword } }", "variables": { "pid": pid } }).await?; let Some(val) = req.entries() .find(|v| v.0 == "data") .ok_or(SomethingHappened)?.1 .entries() .find(|v| v.0 == "userByPid") .ok_or(SomethingHappened)?.1 .entries() .find(|v| v.0 == "nexPassword") .ok_or(SomethingHappened)?.1 .as_str() else { return Err(SomethingHappened); }; Ok(val.as_bytes().try_into().map_err(|_| SomethingHappened)?) } /*pub async fn get_user_data(&mut self , pid: u32) -> Result{ let req = Request::new(GetUserDataRequest{ pid }); let response = self.0.get_user_data(req).await?.into_inner(); Ok(response) }*/ } /* pub struct Client(AccountClient>); impl Client{ pub async fn new() -> Result{ let channel = Channel::from_static(&*CLIENT_URI).connect().await?; let func = Box::new(&|mut req: Request<()>|{ req.metadata_mut().insert("x-api-key", API_KEY.clone()); Ok(req) }) as InterceptorFunc; let client = AccountClient::with_interceptor(channel, func); Ok(Self(client)) } pub async fn get_nex_password(&mut self , pid: u32) -> Result<[u8; 16]>{ let req = Request::new(GetNexPasswordRequest{ pid }); let response = self.0.get_nex_password(req).await?.into_inner(); Ok(response.password.as_bytes().try_into()?) } pub async fn get_user_data(&mut self , pid: u32) -> Result{ let req = Request::new(GetUserDataRequest{ pid }); let response = self.0.get_user_data(req).await?.into_inner(); Ok(response) } } */