97 lines
2.3 KiB
Rust
97 lines
2.3 KiB
Rust
|
#[derive(Debug)]
|
||
|
|
pub enum GetError{
|
||
|
|
ParseError(url::ParseError),
|
||
|
|
Reqwest(reqwest::Error),
|
||
|
|
JSON(serde_json::Error),
|
||
|
|
}
|
||
|
|
impl std::fmt::Display for GetError{
|
||
|
|
fn fmt(&self,f:&mut std::fmt::Formatter<'_>)->std::fmt::Result{
|
||
|
|
write!(f,"{self:?}")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
impl std::error::Error for GetError{}
|
||
|
|
|
||
|
|
#[derive(serde::Deserialize)]
|
||
|
|
pub struct ScriptID(i64);
|
||
|
|
|
||
|
|
#[allow(nonstandard_style)]
|
||
|
|
pub struct GetScriptRequest{
|
||
|
|
pub ScriptID:ScriptID,
|
||
|
|
}
|
||
|
|
#[allow(nonstandard_style)]
|
||
|
|
#[derive(serde::Deserialize)]
|
||
|
|
pub struct ScriptResponse{
|
||
|
|
pub ID:i64,
|
||
|
|
pub Hash:String,
|
||
|
|
pub Source:String,
|
||
|
|
pub SubmissionID:i64,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(serde::Deserialize)]
|
||
|
|
#[repr(i32)]
|
||
|
|
pub enum Policy{
|
||
|
|
Allowed=0,
|
||
|
|
Blocked=1,
|
||
|
|
Delete=2,
|
||
|
|
Replace=3,
|
||
|
|
}
|
||
|
|
|
||
|
|
pub struct ScriptPolicyHashRequest{
|
||
|
|
pub hash:String,
|
||
|
|
}
|
||
|
|
#[allow(nonstandard_style)]
|
||
|
|
#[derive(serde::Deserialize)]
|
||
|
|
pub struct ScriptPolicyResponse{
|
||
|
|
pub ID:i64,
|
||
|
|
pub Hash:String,
|
||
|
|
pub ScriptID:ScriptID,
|
||
|
|
pub Policy:Policy
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Clone)]
|
||
|
|
pub struct Context{
|
||
|
|
base_url:String,
|
||
|
|
client:reqwest::Client,
|
||
|
|
}
|
||
|
|
|
||
|
|
pub type ReqwestError=reqwest::Error;
|
||
|
|
|
||
|
|
impl Context{
|
||
|
|
pub fn new(
|
||
|
|
base_url:String,
|
||
|
|
// cert:reqwest::Certificate,
|
||
|
|
// identity:reqwest::Identity,
|
||
|
|
)->reqwest::Result<Self>{
|
||
|
|
Ok(Self{
|
||
|
|
base_url,
|
||
|
|
client:reqwest::Client::builder()
|
||
|
|
.use_rustls_tls()
|
||
|
|
//.tls_built_in_root_certs(false)
|
||
|
|
//.add_root_certificate(cert)
|
||
|
|
//.identity(identity)
|
||
|
|
.https_only(true)
|
||
|
|
.build()?,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
async fn get(&self,url:impl reqwest::IntoUrl)->Result<reqwest::Response,reqwest::Error>{
|
||
|
|
self.client.get(url)
|
||
|
|
.send().await
|
||
|
|
}
|
||
|
|
pub async fn get_script(&self,config:GetScriptRequest)->Result<ScriptResponse,GetError>{
|
||
|
|
let url_raw=format!("{}/scripts/{}",self.base_url,config.ScriptID.0);
|
||
|
|
let url=reqwest::Url::parse(url_raw.as_str()).map_err(GetError::ParseError)?;
|
||
|
|
|
||
|
|
self.get(url).await.map_err(GetError::Reqwest)?
|
||
|
|
.error_for_status().map_err(GetError::Reqwest)?
|
||
|
|
.json().await.map_err(GetError::Reqwest)
|
||
|
|
}
|
||
|
|
pub async fn get_script_policy_from_hash(&self,config:ScriptPolicyHashRequest)->Result<ScriptPolicyResponse,GetError>{
|
||
|
|
let url_raw=format!("{}/script-policy/hash/{}",self.base_url,config.hash);
|
||
|
|
let url=reqwest::Url::parse(url_raw.as_str()).map_err(GetError::ParseError)?;
|
||
|
|
|
||
|
|
self.get(url).await.map_err(GetError::Reqwest)?
|
||
|
|
.error_for_status().map_err(GetError::Reqwest)?
|
||
|
|
.json().await.map_err(GetError::Reqwest)
|
||
|
|
}
|
||
|
|
}
|