push file

This commit is contained in:
grimhilt 2023-06-15 16:22:07 +02:00
parent 851f395236
commit 418ca68745
4 changed files with 182 additions and 16 deletions

View File

@ -2,12 +2,12 @@
## Blob object ## Blob object
``` ```
file_name hash timestamp file_name timestamp size hash
``` ```
## Tree object ## Tree object
``` ```
folder_name folder_name timestamp
tree hash_path folder_name tree hash_path folder_name
blob hash_path file_name blob hash_path file_name
``` ```

View File

@ -1,7 +1,11 @@
use crate::commands::{status, config}; use crate::commands::{status, config};
use crate::services::req_props::ReqProps;
use crate::services::api::ApiError;
use crate::services::upload_file::UploadFile;
use crate::commands::status::{State, Obj};
pub fn push() { pub fn push() {
dbg!(status::get_diff()); dbg!(status::get_all_staged());
let remote = match config::get("remote") { let remote = match config::get("remote") {
Some(r) => r, Some(r) => r,
@ -10,9 +14,149 @@ pub fn push() {
std::process::exit(1); std::process::exit(1);
} }
}; };
let (staged_obj, new_obj, del_obj) = status::get_diff();
let staged_objs = status::get_all_staged();
// todo sort folder first
for obj in staged_objs {
if obj.otype == String::from("tree") {
dbg!("should push folder");
} else {
let push_factory = PushFactory.new(obj.clone());
match push_factory.can_push() {
PushState::Valid => push_factory.push(),
_ => todo!(),
}
}
}
// read index // read index
// if dir upload dir // if dir upload dir
} }
#[derive(Debug)]
enum PushState {
Valid,
Conflict,
Error,
}
trait PushChange {
fn can_push(&self) -> PushState;
fn push(&self);
}
struct New {
obj: Obj,
}
impl PushChange for New {
fn can_push(&self) -> PushState {
// check if exist on server
let file_infos = tokio::runtime::Runtime::new().unwrap().block_on(async {
let res = ReqProps::new()
.set_url(&self.obj.path.to_str().unwrap())
.getlastmodified()
.send_with_err()
.await;
match res {
Ok(data) => Ok(data),
Err(ApiError::IncorrectRequest(err)) => {
if err.status() == 404 {
Ok(vec![])
} else {
Err(())
}
},
Err(_) => Err(()),
}
});
if let Ok(infos) = file_infos {
if infos.len() == 0 {
// file doesn't exist on remote
PushState::Valid
} else {
// check date
PushState::Conflict
}
} else {
PushState::Error
}
}
fn push(&self) {
let obj = &self.obj;
tokio::runtime::Runtime::new().unwrap().block_on(async {
let res = UploadFile::new()
.set_url(obj.path.to_str().unwrap())
.set_file(obj.path.clone())
.send_with_err()
.await;
match res {
Err(ApiError::IncorrectRequest(err)) => {
eprintln!("fatal: error pushing file {}: {}", obj.name, err.status());
std::process::exit(1);
},
Err(ApiError::RequestError(_)) => {
eprintln!("fatal: request error pushing file {}", obj.name);
std::process::exit(1);
}
_ => (),
}
// todo manage err
// todo remove index
});
}
}
struct PushFactory;
impl PushFactory {
fn new(&self, obj: Obj) -> Box<dyn PushChange> {
match obj.state {
State::New => Box::new(New { obj: obj.clone() }),
State::Renamed => todo!(),
State::Modified => todo!(),
State::Deleted => todo!(),
State::Default => todo!(),
}
}
}
fn can_push_file(obj: Obj) -> PushState {
dbg!(obj.clone());
// check if exist on server
let file_infos = tokio::runtime::Runtime::new().unwrap().block_on(async {
let res = ReqProps::new()
.set_url(obj.path.to_str().unwrap())
.getlastmodified()
.send_with_err()
.await;
match res {
Ok(data) => Ok(data),
Err(ApiError::IncorrectRequest(err)) => {
if err.status() == 404 {
Ok(vec![])
} else {
Err(())
}
},
Err(_) => Err(()),
}
});
if let Ok(infos) = file_infos {
if infos.len() == 0 {
// file doesn't exist on remote
PushState::Valid
} else {
// check date
PushState::Conflict
}
} else {
PushState::Error
}
}

View File

@ -18,7 +18,7 @@ enum RemoveSide {
#[derive(PartialEq)] #[derive(PartialEq)]
#[derive(Debug)] #[derive(Debug)]
#[derive(Clone)] #[derive(Clone)]
enum State { pub enum State {
Default, Default,
New, New,
Renamed, Renamed,
@ -42,10 +42,23 @@ pub fn status() {
#[derive(Debug)] #[derive(Debug)]
#[derive(Clone)] #[derive(Clone)]
pub struct Obj { pub struct Obj {
otype: String, pub otype: String,
name: String, pub name: String,
path: PathBuf, pub path: PathBuf,
state: State, pub state: State,
}
pub fn get_all_staged() -> Vec<Obj> {
// todo opti getting staged and then finding differences ?
// todo opti return folder
let (mut new_objs, mut del_objs) = get_diff();
let mut renamed_objs = get_renamed(&mut new_objs, &mut del_objs);
// get copy, modified
let mut objs = new_objs;
objs.append(&mut del_objs);
objs.append(&mut renamed_objs);
let staged_objs = get_staged(&mut objs);
staged_objs
} }
fn get_renamed(new_obj: &mut Vec<Obj>, del_obj: &mut Vec<Obj>) -> Vec<Obj> { fn get_renamed(new_obj: &mut Vec<Obj>, del_obj: &mut Vec<Obj>) -> Vec<Obj> {
@ -62,7 +75,6 @@ fn get_staged(objs: &mut Vec<Obj>) -> Vec<Obj> {
let nextsync_path = utils::path::nextsync().unwrap(); let nextsync_path = utils::path::nextsync().unwrap();
if let Ok(entries) = store::index::read_line(nextsync_path.clone()) { if let Ok(entries) = store::index::read_line(nextsync_path.clone()) {
for entry in entries { for entry in entries {
// todo hash this
indexes.insert(entry.unwrap()); indexes.insert(entry.unwrap());
} }
} }
@ -88,7 +100,7 @@ fn get_staged(objs: &mut Vec<Obj>) -> Vec<Obj> {
staged_objs staged_objs
} }
pub fn get_diff() -> (Vec<Obj>, Vec<Obj>) { fn get_diff() -> (Vec<Obj>, Vec<Obj>) {
let mut hashes = HashMap::new(); let mut hashes = HashMap::new();
let mut objs: Vec<String> = vec![]; let mut objs: Vec<String> = vec![];
@ -119,7 +131,7 @@ pub fn get_diff() -> (Vec<Obj>, Vec<Obj>) {
while obj_to_analyse.len() > 0 { while obj_to_analyse.len() > 0 {
let cur_obj = obj_to_analyse.pop().unwrap(); let cur_obj = obj_to_analyse.pop().unwrap();
let cur_path = PathBuf::from(&cur_obj); let cur_path = PathBuf::from(&cur_obj);
dbg!(cur_path.clone());
let obj_path = root.clone().join(cur_path.clone()); let obj_path = root.clone().join(cur_path.clone());
if obj_path.is_dir() { if obj_path.is_dir() {
@ -149,17 +161,26 @@ pub fn get_diff() -> (Vec<Obj>, Vec<Obj>) {
}).collect(); }).collect();
let new_objs: Vec<Obj> = objs.iter().map(|x| { let new_objs: Vec<Obj> = objs.iter().map(|x| {
// todo otype and name let p = PathBuf::from(x.to_string());
// todo name
Obj { Obj {
otype: String::from(""), otype: get_type(p.clone()),
name: x.to_string(), name: x.to_string(),
path: PathBuf::from(x.to_string()), path: p,
state: State::New state: State::New
} }
}).collect(); }).collect();
(new_objs, del_objs) (new_objs, del_objs)
} }
fn get_type(p: PathBuf) -> String {
if p.is_dir() {
String::from("tree")
} else {
String::from("blob")
}
}
fn add_to_hashmap(lines: Lines<BufReader<File>>, hashes: &mut HashMap<String, Obj>, path: PathBuf) { fn add_to_hashmap(lines: Lines<BufReader<File>>, hashes: &mut HashMap<String, Obj>, path: PathBuf) {
for line in lines { for line in lines {
if let Ok(ip) = line { if let Ok(ip) = line {

View File

@ -2,3 +2,4 @@ pub mod api;
pub mod list_folders; pub mod list_folders;
pub mod download_files; pub mod download_files;
pub mod req_props; pub mod req_props;
pub mod upload_file;