add trait ApiCall

This commit is contained in:
grimhilt 2023-10-21 21:47:48 +02:00
parent 07f6405b26
commit 9ea1d01c27
13 changed files with 292 additions and 210 deletions

View File

@ -1,4 +1,7 @@
use std::env; use std::env;
use lazy_static::lazy_static;
use std::sync::Mutex;
use std::path::PathBuf;
use dotenv::dotenv; use dotenv::dotenv;
use reqwest::Client; use reqwest::Client;
use reqwest::RequestBuilder; use reqwest::RequestBuilder;
@ -8,6 +11,14 @@ use reqwest::header::{HeaderValue, CONTENT_TYPE, HeaderMap, IntoHeaderName};
use crate::utils::api::ApiProps; use crate::utils::api::ApiProps;
use crate::commands::config; use crate::commands::config;
use crate::commands::clone::get_url_props; use crate::commands::clone::get_url_props;
use crate::services::request_manager::get_request_manager;
use crate::services::api_call::ApiCall;
use super::login::Login;
lazy_static! {
static ref HTTP_TOKEN: Mutex<String> = Mutex::new(String::new());
}
#[derive(Debug)] #[derive(Debug)]
pub enum ApiError { pub enum ApiError {
@ -20,7 +31,9 @@ pub enum ApiError {
pub struct ApiBuilder { pub struct ApiBuilder {
client: Client, client: Client,
request: Option<RequestBuilder>, request: Option<RequestBuilder>,
headers: Option<HeaderMap> headers: Option<HeaderMap>,
auth_set: bool,
host: Option<String>,
} }
impl ApiBuilder { impl ApiBuilder {
@ -29,24 +42,14 @@ impl ApiBuilder {
client: Client::new(), client: Client::new(),
request: None, request: None,
headers: None, headers: None,
auth_set: false,
host: None,
} }
} }
pub fn set_url(&mut self, method: Method, url: &str) -> &mut ApiBuilder { pub fn set_url(&mut self, method: Method, url: &str) -> &mut ApiBuilder {
let remote = match config::get("remote") { self.request = Some(self.client.request(method, url));
Some(r) => r,
None => {
eprintln!("fatal: unable to find a remote");
std::process::exit(1);
}
};
let (host, _, _) = get_url_props(&remote);
let mut u = String::from(host);
u.push_str(url);
self.request = Some(self.client.request(method, u));
self self
} }
pub fn build_request(&mut self, method: Method, path: &str) -> &mut ApiBuilder { pub fn build_request(&mut self, method: Method, path: &str) -> &mut ApiBuilder {
@ -58,6 +61,7 @@ impl ApiBuilder {
} }
}; };
let (host, username, root) = get_url_props(&remote); let (host, username, root) = get_url_props(&remote);
self.host = Some(host.clone());
let mut url = String::from(host); let mut url = String::from(host);
url.push_str("/remote.php/dav/files/"); url.push_str("/remote.php/dav/files/");
url.push_str(username.unwrap()); url.push_str(username.unwrap());
@ -71,6 +75,7 @@ impl ApiBuilder {
} }
pub fn set_req(&mut self, meth: Method, p: &str, api_props: &ApiProps) -> &mut ApiBuilder { pub fn set_req(&mut self, meth: Method, p: &str, api_props: &ApiProps) -> &mut ApiBuilder {
self.host = Some(api_props.clone().host.clone());
let mut url = String::from(&api_props.host); let mut url = String::from(&api_props.host);
url.push_str("/remote.php/dav/files/"); url.push_str("/remote.php/dav/files/");
url.push_str("/"); url.push_str("/");
@ -84,20 +89,41 @@ impl ApiBuilder {
self self
} }
fn set_auth(&mut self) -> &mut ApiBuilder { pub fn set_basic_auth(&mut self, login: String, pwd: String) -> &mut ApiBuilder {
// todo if not exist
dotenv().ok();
let password = env::var("PASSWORD").unwrap();
let username = env::var("USERNAME").unwrap();
match self.request.take() { match self.request.take() {
None => { None => {
eprintln!("fatal: incorrect request"); eprintln!("fatal: incorrect request");
std::process::exit(1); std::process::exit(1);
}, },
Some(req) => { Some(req) => {
self.request = Some(req.basic_auth(username, Some(password))); self.request = Some(req.basic_auth(login, Some(pwd)));
} }
} }
self.auth_set = true;
self
}
fn set_auth(&mut self) -> &mut ApiBuilder {
// check .config
//let config_file = PathBuf::from("~/.nextsync/config");
//if config_file.exists() {
//
//} else {
// let res = Login::new()
// .set_host(self.host.clone())
// .ask_auth()
// .send_with_err();
// if let Err(err) = res {
// eprintln!("fatal: authentification failed");
// std::process::exit(1);
// }
//}
//// todo if not exist
//dotenv().ok();
//let password = env::var("PASSWORD").unwrap();
//let username = env::var("USERNAME").unwrap();
self self
} }
@ -148,8 +174,63 @@ impl ApiBuilder {
self self
} }
pub async fn send(&mut self) -> Result<Response, Error> { pub fn send(&mut self, need_text: bool) -> Result<Option<String>, ApiError> {
self.set_auth(); let mut request_manager = get_request_manager().lock().unwrap();
let request_manager = request_manager.as_mut().unwrap();
if !self.host.is_none()
{
request_manager.set_host(self.host.clone().unwrap());
}
if !self.auth_set {
//self.set_auth();
self.set_header("TOKEN", HeaderValue::from_str(&request_manager.get_token()).unwrap());
}
tokio::runtime::Runtime::new().unwrap().block_on(async {
let res = match self.request.take() {
None => {
eprintln!("fatal: incorrect request");
std::process::exit(1);
},
Some(req) => {
if let Some(headers) = &self.headers {
req.headers(headers.clone())
.send().await.map_err(ApiError::RequestError)?
} else {
req.send().await.map_err(ApiError::RequestError)?
}
},
};
if res.status().is_success() {
if need_text {
let body = res.text().await.map_err(|err| ApiError::EmptyError(err))?;
Ok(Some(body))
} else {
Ok(None)
}
} else {
Err(ApiError::IncorrectRequest(res))
}
})
}
pub async fn old_send(&mut self) -> Result<Response, Error> {
let mut request_manager = get_request_manager().lock().unwrap();
let request_manager = request_manager.as_mut().unwrap();
if !self.host.is_none()
{
request_manager.set_host(self.host.clone().unwrap());
}
if !self.auth_set {
//self.set_auth();
self.set_header("TOKEN", HeaderValue::from_str(&request_manager.get_token()).unwrap());
}
match self.request.take() { match self.request.take() {
None => { None => {
eprintln!("fatal: incorrect request"); eprintln!("fatal: incorrect request");

13
src/services/api_call.rs Normal file
View File

@ -0,0 +1,13 @@
use crate::services::api::ApiError;
pub trait ApiCall {
fn new() -> Self where Self: Sized {
unimplemented!()
}
fn set_url(&mut self, url: &str) -> &mut Self {
self
}
fn send(&mut self) -> Result<Option<String>, ApiError> {
unimplemented!()
}
}

View File

@ -1,2 +0,0 @@

View File

@ -1,20 +1,27 @@
use reqwest::{Method, Response, Error, header::HeaderValue}; use reqwest::{Method, header::HeaderValue};
use crate::services::api::{ApiBuilder, ApiError}; use crate::services::api::{ApiBuilder, ApiError};
use crate::clone::get_url_props; use crate::clone::get_url_props;
use crate::commands::config; use crate::commands::config;
use crate::services::api_call::ApiCall;
pub struct Copy { pub struct Copy {
api_builder: ApiBuilder, api_builder: ApiBuilder,
} }
impl Copy { impl ApiCall for Copy {
pub fn new() -> Self { fn new() -> Self {
Copy { Copy {
api_builder: ApiBuilder::new(), api_builder: ApiBuilder::new(),
} }
} }
pub fn set_url(&mut self, url: &str, destination: &str) -> &mut Copy { fn send(&mut self) -> Result<Option<String>, ApiError> {
self.api_builder.send(true)
}
}
impl Copy {
pub fn set_url_copy(&mut self, url: &str, destination: &str) -> &mut Copy {
self.api_builder.build_request(Method::from_bytes(b"COPY").unwrap(), url); self.api_builder.build_request(Method::from_bytes(b"COPY").unwrap(), url);
let remote = match config::get("remote") { let remote = match config::get("remote") {
@ -37,25 +44,10 @@ impl Copy {
self self
} }
pub async fn send(&mut self) -> Result<Response, Error> {
self.api_builder.send().await
}
pub fn _overwrite(&mut self, overwrite: bool) -> &mut Copy { pub fn _overwrite(&mut self, overwrite: bool) -> &mut Copy {
self.api_builder.set_header("Overwrite", HeaderValue::from_str({ self.api_builder.set_header("Overwrite", HeaderValue::from_str({
if overwrite { "T" } else { "F" } if overwrite { "T" } else { "F" }
}).unwrap()); }).unwrap());
self self
} }
pub fn send_with_err(&mut self) -> Result<(), ApiError> {
let res = tokio::runtime::Runtime::new().unwrap().block_on(async {
self.send().await
}).map_err(ApiError::RequestError)?;
if res.status().is_success() {
Ok(())
} else {
Err(ApiError::IncorrectRequest(res))
}
}
} }

View File

@ -1,34 +1,24 @@
use reqwest::{Method, Response, Error}; use reqwest::Method;
use crate::services::api::{ApiBuilder, ApiError}; use crate::services::api::{ApiBuilder, ApiError};
use crate::services::api_call::ApiCall;
pub struct CreateFolder { pub struct CreateFolder {
api_builder: ApiBuilder, api_builder: ApiBuilder,
} }
impl CreateFolder { impl ApiCall for CreateFolder {
pub fn new() -> Self { fn new() -> Self {
CreateFolder { CreateFolder {
api_builder: ApiBuilder::new(), api_builder: ApiBuilder::new(),
} }
} }
pub fn set_url(&mut self, url: &str) -> &mut CreateFolder { fn set_url(&mut self, url: &str) -> &mut CreateFolder {
self.api_builder.build_request(Method::from_bytes(b"MKCOL").unwrap(), url); self.api_builder.build_request(Method::from_bytes(b"MKCOL").unwrap(), url);
self self
} }
pub async fn send(&mut self) -> Result<Response, Error> { fn send(&mut self) -> Result<Option<String>, ApiError> {
self.api_builder.send().await self.api_builder.send(false)
}
pub fn send_with_err(&mut self) -> Result<(), ApiError> {
let res = tokio::runtime::Runtime::new().unwrap().block_on(async {
self.send().await
}).map_err(ApiError::RequestError)?;
if res.status().is_success() {
Ok(())
} else {
Err(ApiError::IncorrectRequest(res))
}
} }
} }

View File

@ -1,38 +1,24 @@
use reqwest::{Method, Response, Error}; use reqwest::Method;
use crate::services::api::{ApiBuilder, ApiError}; use crate::services::api::{ApiBuilder, ApiError};
use crate::services::api_call::ApiCall;
pub struct DeletePath { pub struct DeletePath {
api_builder: ApiBuilder, api_builder: ApiBuilder,
} }
impl DeletePath { impl ApiCall for DeletePath {
pub fn new() -> Self { fn new() -> Self {
DeletePath { DeletePath {
api_builder: ApiBuilder::new(), api_builder: ApiBuilder::new(),
} }
} }
pub fn set_url(&mut self, url: &str) -> &mut DeletePath { fn set_url(&mut self, url: &str) -> &mut DeletePath {
self.api_builder.build_request(Method::DELETE, url); self.api_builder.build_request(Method::DELETE, url);
self self
} }
pub async fn send(&mut self) -> Result<Response, Error> { fn send(&mut self) -> Result<Option<String>, ApiError> {
self.api_builder.send().await self.api_builder.send(true)
}
pub fn send_with_err(&mut self) -> Result<String, ApiError> {
let res = tokio::runtime::Runtime::new().unwrap().block_on(async {
self.send().await
}).map_err(ApiError::RequestError)?;
if res.status().is_success() {
let body = tokio::runtime::Runtime::new().unwrap().block_on(async {
res.text().await
}).map_err(ApiError::EmptyError)?;
Ok(body)
} else {
Err(ApiError::IncorrectRequest(res))
}
} }
} }

View File

@ -6,38 +6,32 @@ use std::io::{self, Write};
use reqwest::{Method, Response, Error}; use reqwest::{Method, Response, Error};
use crate::utils::api::ApiProps; use crate::utils::api::ApiProps;
use crate::services::api::{ApiBuilder, ApiError}; use crate::services::api::{ApiBuilder, ApiError};
use crate::services::api_call::ApiCall;
pub struct DownloadFiles { pub struct DownloadFiles {
api_builder: ApiBuilder, api_builder: ApiBuilder,
relative_ps: String, relative_ps: String,
} }
impl DownloadFiles { impl ApiCall for DownloadFiles {
pub fn new() -> Self { fn new() -> Self {
DownloadFiles { DownloadFiles {
api_builder: ApiBuilder::new(), api_builder: ApiBuilder::new(),
relative_ps: String::new(), relative_ps: String::new(),
} }
} }
}
pub fn set_url(&mut self, relative_ps: &str, api_props: &ApiProps) -> &mut DownloadFiles { impl DownloadFiles {
// todo make it beautiful
pub fn set_url_download(&mut self, relative_ps: &str, api_props: &ApiProps) -> &mut DownloadFiles {
self.relative_ps = relative_ps.to_string(); self.relative_ps = relative_ps.to_string();
self.api_builder.set_req(Method::GET, relative_ps, api_props); self.api_builder.set_req(Method::GET, relative_ps, api_props);
self self
} }
pub async fn send(&mut self) -> Result<Response, Error> { pub async fn send_download(&mut self) -> Result<Response, Error> {
self.api_builder.send().await self.api_builder.old_send().await
}
pub async fn _send_with_err(mut self) -> Result<Vec<u8>, ApiError> {
let res = self.send().await.map_err(ApiError::RequestError)?;
if res.status().is_success() {
let body = res.bytes().await.map_err(ApiError::EmptyError)?;
Ok(body.to_vec())
} else {
Err(ApiError::IncorrectRequest(res))
}
} }
pub fn save_stream(&mut self, ref_p: PathBuf, callback: Option<impl Fn(u64)>) -> Result<(), ApiError> { pub fn save_stream(&mut self, ref_p: PathBuf, callback: Option<impl Fn(u64)>) -> Result<(), ApiError> {
@ -45,7 +39,7 @@ impl DownloadFiles {
let mut file = File::create(abs_p).unwrap(); let mut file = File::create(abs_p).unwrap();
tokio::runtime::Runtime::new().unwrap().block_on(async { tokio::runtime::Runtime::new().unwrap().block_on(async {
let res = self.send().await.map_err(ApiError::RequestError)?; let res = self.send_download().await.map_err(ApiError::RequestError)?;
if res.status().is_success() { if res.status().is_success() {
let mut stream = res.bytes_stream(); let mut stream = res.bytes_stream();
@ -70,7 +64,7 @@ impl DownloadFiles {
pub fn save(&mut self, ref_p: PathBuf) -> Result<(), ApiError> { pub fn save(&mut self, ref_p: PathBuf) -> Result<(), ApiError> {
tokio::runtime::Runtime::new().unwrap().block_on(async { tokio::runtime::Runtime::new().unwrap().block_on(async {
let p = ref_p.join(PathBuf::from(self.relative_ps.clone())); let p = ref_p.join(PathBuf::from(self.relative_ps.clone()));
let res = self.send().await.map_err(ApiError::RequestError)?; let res = self.send_download().await.map_err(ApiError::RequestError)?;
if res.status().is_success() { if res.status().is_success() {
let body = res.bytes().await.map_err(ApiError::EmptyError)?; let body = res.bytes().await.map_err(ApiError::EmptyError)?;
match Self::write_file(p, &body.to_vec()) { match Self::write_file(p, &body.to_vec()) {

View File

@ -2,6 +2,7 @@ use std::path::PathBuf;
use indicatif::{ProgressBar, MultiProgress, ProgressStyle, HumanBytes}; use indicatif::{ProgressBar, MultiProgress, ProgressStyle, HumanBytes};
use crate::utils::api::ApiProps; use crate::utils::api::ApiProps;
use crate::services::api_call::ApiCall;
use crate::services::api::ApiError; use crate::services::api::ApiError;
use crate::services::download_files::DownloadFiles; use crate::services::download_files::DownloadFiles;
use crate::services::req_props::ObjProps; use crate::services::req_props::ObjProps;
@ -106,7 +107,7 @@ impl Downloader {
for file in self.files.clone() { for file in self.files.clone() {
let relative_s = &file.clone().relative_s.unwrap(); let relative_s = &file.clone().relative_s.unwrap();
let mut download = DownloadFiles::new(); let mut download = DownloadFiles::new();
download.set_url(&relative_s, &self.api_props.clone().unwrap()); download.set_url_download(&relative_s, &self.api_props.clone().unwrap());
let should_use_stream = { let should_use_stream = {
if let Some(size) = file.contentlength { if let Some(size) = file.contentlength {

View File

@ -2,10 +2,9 @@ use std::io;
use std::io::Cursor; use std::io::Cursor;
use std::io::prelude::*; use std::io::prelude::*;
use xml::reader::{EventReader, XmlEvent}; use xml::reader::{EventReader, XmlEvent};
use reqwest::{Response, Error, header::HeaderValue, Method}; use reqwest::{header::HeaderValue, Method};
use rpassword; use rpassword;
use crate::services::api_call::ApiCall; use crate::services::api_call::ApiCall;
use crate::services::api::{ApiBuilder, ApiError}; use crate::services::api::{ApiBuilder, ApiError};
pub struct Login { pub struct Login {
@ -16,7 +15,7 @@ pub struct Login {
} }
impl ApiCall for Login { impl ApiCall for Login {
pub fn new() -> Self { fn new() -> Self {
Login { Login {
api_builder: ApiBuilder::new(), api_builder: ApiBuilder::new(),
login: String::new(), login: String::new(),
@ -25,21 +24,7 @@ impl ApiCall for Login {
} }
} }
pub fn ask_auth(&mut self) -> &mut Login { fn send(&mut self) -> Result<Option<String>, ApiError> {
println!("Please enter your username/email: ");
let stdin = io::stdin();
self.login = stdin.lock().lines().next().unwrap().unwrap();
println!("Please enter your password: ");
self.password = rpassword::read_password().unwrap();
self
}
pub fn set_host(&mut self, host: Option<String>) -> &mut Login {
self.host = host;
self
}
pub async fn send(&mut self) -> Result<Response, Error> {
let url = match self.host.clone() { let url = match self.host.clone() {
Some(h) => { Some(h) => {
@ -54,28 +39,29 @@ impl ApiCall for Login {
self.api_builder.set_url(Method::GET, &url); self.api_builder.set_url(Method::GET, &url);
self.api_builder.set_header("OCS-APIRequest", HeaderValue::from_str("true").unwrap()); self.api_builder.set_header("OCS-APIRequest", HeaderValue::from_str("true").unwrap());
self.api_builder.set_basic_auth(self.login.clone(), self.password.clone()); self.api_builder.set_basic_auth(self.login.clone(), self.password.clone());
self.api_builder.send().await self.api_builder.send(true)
} }
}
pub async fn send_with_err(&mut self) -> Result<String, ApiError> {
match self.send().await { impl Login {
Err(res) => Err(ApiError::RequestError(res)), pub fn ask_auth(&mut self) -> &mut Login {
Ok(res) if res.status().is_success() => { println!("Please enter your username/email: ");
let body = res let stdin = io::stdin();
.text() self.login = stdin.lock().lines().next().unwrap().unwrap();
.await println!("Please enter your password: ");
.map_err(|err| ApiError::EmptyError(err))?; self.password = rpassword::read_password().unwrap();
Ok(body) self
},
Ok(res) => {
Err(ApiError::IncorrectRequest(res))
}
}
} }
pub async fn send_login(&mut self) -> Result<String, ApiError> { pub fn set_host(&mut self, host: Option<String>) -> &mut Login {
match self.send_with_err().await { self.host = host;
Ok(body) => Ok(self.parse(body)), self
}
pub fn send_login(&mut self) -> Result<String, ApiError> {
match self.send() {
Ok(Some(body)) => Ok(self.parse(body)),
Ok(None) => Err(ApiError::Unexpected(String::from("Empty after tested"))),
Err(err) => Err(err), Err(err) => Err(err),
} }
} }

View File

@ -1,20 +1,28 @@
use reqwest::{Method, Response, Error, header::HeaderValue}; use reqwest::{Method, header::HeaderValue};
use crate::services::api::{ApiBuilder, ApiError}; use crate::services::api::{ApiBuilder, ApiError};
use crate::clone::get_url_props; use crate::clone::get_url_props;
use crate::commands::config; use crate::commands::config;
use crate::services::api_call::ApiCall;
pub struct Move { pub struct Move {
api_builder: ApiBuilder, api_builder: ApiBuilder,
} }
impl Move { impl ApiCall for Move {
pub fn new() -> Self { fn new() -> Self {
Move { Move {
api_builder: ApiBuilder::new(), api_builder: ApiBuilder::new(),
} }
} }
pub fn set_url(&mut self, url: &str, destination: &str) -> &mut Move {
fn send(&mut self) -> Result<Option<String>, ApiError> {
self.api_builder.send(false)
}
}
impl Move {
pub fn set_url_move(&mut self, url: &str, destination: &str) -> &mut Move {
self.api_builder.build_request(Method::from_bytes(b"MOVE").unwrap(), url); self.api_builder.build_request(Method::from_bytes(b"MOVE").unwrap(), url);
let remote = match config::get("remote") { let remote = match config::get("remote") {
@ -37,25 +45,10 @@ impl Move {
self self
} }
pub async fn send(&mut self) -> Result<Response, Error> {
self.api_builder.send().await
}
pub fn _overwrite(&mut self, overwrite: bool) -> &mut Move { pub fn _overwrite(&mut self, overwrite: bool) -> &mut Move {
self.api_builder.set_header("Overwrite", HeaderValue::from_str({ self.api_builder.set_header("Overwrite", HeaderValue::from_str({
if overwrite { "T" } else { "F" } if overwrite { "T" } else { "F" }
}).unwrap()); }).unwrap());
self self
} }
pub fn send_with_err(&mut self) -> Result<(), ApiError> {
let res = tokio::runtime::Runtime::new().unwrap().block_on(async {
self.send().await
}).map_err(ApiError::RequestError)?;
if res.status().is_success() {
Ok(())
} else {
Err(ApiError::IncorrectRequest(res))
}
}
} }

View File

@ -1,6 +1,6 @@
use std::io::Cursor; use std::io::Cursor;
use chrono::{Utc, DateTime}; use chrono::{Utc, DateTime};
use reqwest::{Method, Response, Error}; use reqwest::Method;
use xml::reader::{EventReader, XmlEvent}; use xml::reader::{EventReader, XmlEvent};
use reqwest::header::HeaderValue; use reqwest::header::HeaderValue;
use crate::commands::clone::get_url_props; use crate::commands::clone::get_url_props;
@ -8,6 +8,7 @@ use crate::commands::config;
use crate::utils::time::parse_timestamp; use crate::utils::time::parse_timestamp;
use crate::utils::api::{get_relative_s, ApiProps}; use crate::utils::api::{get_relative_s, ApiProps};
use crate::services::api::{ApiBuilder, ApiError}; use crate::services::api::{ApiBuilder, ApiError};
use crate::services::api_call::ApiCall;
#[derive(Debug)] #[derive(Debug)]
pub struct ObjProps { pub struct ObjProps {
@ -55,8 +56,8 @@ pub struct ReqProps {
api_props: Option<ApiProps> api_props: Option<ApiProps>
} }
impl ReqProps { impl ApiCall for ReqProps {
pub fn new() -> Self { fn new() -> Self {
ReqProps { ReqProps {
api_builder: ApiBuilder::new(), api_builder: ApiBuilder::new(),
xml_balises: vec![], xml_balises: vec![],
@ -65,7 +66,7 @@ impl ReqProps {
} }
} }
pub fn set_url(&mut self, url: &str) -> &mut ReqProps { fn set_url(&mut self, url: &str) -> &mut ReqProps {
let remote = match config::get("remote") { let remote = match config::get("remote") {
Some(r) => r, Some(r) => r,
None => { None => {
@ -83,6 +84,13 @@ impl ReqProps {
self self
} }
fn send(&mut self) -> Result<Option<String>, ApiError> {
self.validate_xml();
self.api_builder.send(true)
}
}
impl ReqProps {
pub fn set_request(&mut self, p: &str, api_props: &ApiProps) -> &mut ReqProps { pub fn set_request(&mut self, p: &str, api_props: &ApiProps) -> &mut ReqProps {
self.api_props = Some(api_props.clone()); self.api_props = Some(api_props.clone());
self.api_builder.set_req(Method::from_bytes(b"PROPFIND").unwrap(), p, api_props); self.api_builder.set_req(Method::from_bytes(b"PROPFIND").unwrap(), p, api_props);
@ -145,32 +153,10 @@ impl ReqProps {
self self
} }
pub async fn send(&mut self) -> Result<Response, Error> {
self.validate_xml();
self.api_builder.send().await
}
pub fn send_with_err(&mut self) -> Result<String, ApiError> {
tokio::runtime::Runtime::new().unwrap().block_on(async {
match self.send().await {
Err(res) => Err(ApiError::RequestError(res)),
Ok(res) if res.status().is_success() => {
let body = res
.text()
.await
.map_err(|err| ApiError::EmptyError(err))?;
Ok(body)
},
Ok(res) => {
Err(ApiError::IncorrectRequest(res))
}
}
})
}
pub fn send_req_multiple(&mut self) -> Result<Vec<ObjProps>, ApiError> { pub fn send_req_multiple(&mut self) -> Result<Vec<ObjProps>, ApiError> {
match self.send_with_err() { match self.send() {
Ok(body) => Ok(self.parse(body, true)), Ok(Some(body)) => Ok(self.parse(body, true)),
Ok(None) => Err(ApiError::Unexpected(String::from("Empty after tested"))),
Err(err) => Err(err), Err(err) => Err(err),
} }
} }
@ -178,12 +164,13 @@ impl ReqProps {
pub fn send_req_single(&mut self) -> Result<ObjProps, ApiError> { pub fn send_req_single(&mut self) -> Result<ObjProps, ApiError> {
// set depth to 0 as we only need one element // set depth to 0 as we only need one element
self.set_depth("0"); self.set_depth("0");
match self.send_with_err() { match self.send() {
Ok(body) => { Ok(Some(body)) => {
let objs = self.parse(body, false); let objs = self.parse(body, false);
let obj = objs[0].clone(); let obj = objs[0].clone();
Ok(obj) Ok(obj)
}, },
Ok(None) => Err(ApiError::Unexpected(String::from("Empty after tested"))),
Err(err) => Err(err), Err(err) => Err(err),
} }
} }

View File

@ -0,0 +1,73 @@
use lazy_static::lazy_static;
use std::sync::Mutex;
use crate::services::login::Login;
use crate::commands::config;
use crate::commands::clone::get_url_props;
use crate::services::api_call::ApiCall;
lazy_static! {
static ref REQUEST_MANAGER: Mutex<Option<RequestManager>> = Mutex::new(None);
}
pub fn get_request_manager() -> &'static Mutex<Option<RequestManager>> {
if REQUEST_MANAGER.lock().unwrap().is_none() {
*REQUEST_MANAGER.lock().unwrap() = Some(RequestManager::new());
}
&REQUEST_MANAGER
}
pub struct RequestManager {
token: Option<String>,
host: Option<String>,
}
impl RequestManager {
pub fn new() -> Self {
RequestManager {
token: None,
host: None,
}
}
pub fn set_host(&mut self, host: String) {
self.host = Some(host);
}
pub fn get_host(&mut self) -> String
{
if self.host.is_none()
{
let remote = match config::get("remote") {
Some(r) => r,
None => {
// todo ask user instead
eprintln!("fatal: unable to find a remote");
std::process::exit(1);
}
};
let (host, _, _) = get_url_props(&remote);
self.host = Some(host.clone());
// todo ask user
}
self.host.clone().unwrap()
}
pub fn get_token(&mut self) -> String {
if self.token.is_none() {
// todo check in config
let get_token = Login::new()
.ask_auth()
.set_host(Some(self.get_host()))
.send_login();
// todo deal with error cases
self.token = Some(get_token.unwrap());
}
self.token.clone().unwrap()
}
pub fn create_request()
{
}
}

View File

@ -1,25 +1,32 @@
use std::fs::File; use std::fs::File;
use std::io::Read; use std::io::Read;
use std::path::PathBuf; use std::path::PathBuf;
use reqwest::{Method, Response, Error}; use reqwest::Method;
use crate::services::api::{ApiBuilder, ApiError}; use crate::services::api::{ApiBuilder, ApiError};
use crate::services::api_call::ApiCall;
pub struct UploadFile { pub struct UploadFile {
api_builder: ApiBuilder, api_builder: ApiBuilder,
} }
impl UploadFile { impl ApiCall for UploadFile {
pub fn new() -> Self { fn new() -> Self {
UploadFile { UploadFile {
api_builder: ApiBuilder::new(), api_builder: ApiBuilder::new(),
} }
} }
pub fn set_url(&mut self, url: &str) -> &mut UploadFile { fn set_url(&mut self, url: &str) -> &mut UploadFile {
self.api_builder.build_request(Method::PUT, url); self.api_builder.build_request(Method::PUT, url);
self self
} }
fn send(&mut self) -> Result<Option<String>, ApiError> {
self.api_builder.send(true)
}
}
impl UploadFile {
pub fn set_file(&mut self, path: PathBuf) -> &mut UploadFile { pub fn set_file(&mut self, path: PathBuf) -> &mut UploadFile {
// todo large file // todo large file
// todo small files // todo small files
@ -29,23 +36,4 @@ impl UploadFile {
self.api_builder.set_body(buffer); self.api_builder.set_body(buffer);
self self
} }
pub async fn send(&mut self) -> Result<Response, Error> {
self.api_builder.send().await
}
pub fn send_with_err(&mut self) -> Result<String, ApiError> {
let res = tokio::runtime::Runtime::new().unwrap().block_on(async {
self.send().await
}).map_err(ApiError::RequestError)?;
if res.status().is_success() {
let body = tokio::runtime::Runtime::new().unwrap().block_on(async {
res.text().await
}).map_err(ApiError::EmptyError)?;
Ok(body)
} else {
Err(ApiError::IncorrectRequest(res))
}
}
} }