status test for differences works

This commit is contained in:
grimhilt
2023-06-01 20:41:54 +02:00
parent 709f0f3b84
commit 4d390b131e
7 changed files with 256 additions and 3 deletions

View File

@@ -4,12 +4,21 @@ use std::env;
pub fn init() {
let builder = DirBuilder::new();
let mut path = env::current_dir().unwrap();
// .nextsync folder
path.push(".nextsync");
match builder.create(path.clone()) {
Ok(()) => println!("Directory successfuly created"),
Err(_) => println!("Error: cannot create directory"),
}
path.push("HEAD");
match File::create(path.clone()) {
Ok(_) => println!("File successfuly created"),
Err(_) => println!("Error: cannot create .nextsyncignore"),
}
path.pop();
path.pop();
path.push(".nextsyncignore");
@@ -17,4 +26,5 @@ pub fn init() {
Ok(_) => println!("File successfuly created"),
Err(_) => println!("Error: cannot create .nextsyncignore"),
}
}

141
src/commands/status.rs Normal file
View File

@@ -0,0 +1,141 @@
use std::env;
use std::path::Path;
use std::fs::File;
use std::io::{self, BufRead};
use std::path::PathBuf;
use crypto::digest::Digest;
use crypto::sha1::Sha1;
use std::collections::HashSet;
pub fn status() {
let mut a = vec![];
{
let b = 2;
a.push(b.clone());
}
println!("Vector a: {:?}", a);
let mut new_files: Vec<PathBuf> = Vec::new();
let mut hashes = HashSet::new();
let mut objects: Vec<&str> = vec![];
let path = env::current_dir().unwrap();
let mut next_sync_path = path.clone();
next_sync_path.push(".nextsync");
if let Ok(lines) = read_head(next_sync_path.clone()) {
for line in lines {
if let Ok(ip) = line {
hashes.insert(String::from(ip).as_str());
}
}
}
if let Ok(entries) = read_folder(path.clone()) {
for entry in entries {
if !is_nextsync_config(entry.clone()) {
let object_path = entry.strip_prefix(path.clone()).unwrap();
objects.push(object_path.to_str().unwrap().clone());
}
}
}
find_missing_elements(&mut hashes, &mut objects);
dbg!(hashes);
dbg!(objects);
}
fn find_missing_elements(hashes: &mut HashSet<&str>, objects: &mut Vec<&str>) {
let mut hasher = Sha1::new();
let mut to_remove: Vec<usize> = vec![];
let mut i = 0;
for object in &mut *objects {
// hash the object
hasher.input_str(object);
let hash = hasher.result_str();
hasher.reset();
// find it on the list of hashes
if hashes.contains(hash.as_str()) {
hashes.remove(hash.as_str());
to_remove.push(i);
}
i += 1;
}
// remove all objects existing in the list of hashes
i = 0;
for index in to_remove {
objects.remove(index-i);
i += 1;
}
}
fn is_nextsync_config(path: PathBuf) -> bool {
path.ends_with(".nextsync") || path.ends_with(".nextsyncignore")
}
fn read_head(mut path: PathBuf) -> io::Result<io::Lines<io::BufReader<File>>> {
path.push("HEAD");
read_lines(path)
}
use std::fs;
fn read_folder(path: PathBuf) -> io::Result<Vec<PathBuf>> {
let mut entries = fs::read_dir(path)?
.map(|res| res.map(|e| e.path()))
.collect::<Result<Vec<_>, io::Error>>()?;
entries.sort();
Ok(entries)
}
// The output is wrapped in a Result to allow matching on errors
// Returns an Iterator to the Reader of the lines of the file.
fn read_lines<P>(filename: P) -> io::Result<io::Lines<io::BufReader<File>>>
where P: AsRef<Path>, {
let file = File::open(filename)?;
Ok(io::BufReader::new(file).lines())
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_find_missing_elements() {
let mut hasher = Sha1::new();
hasher.input_str("file1");
let hash1 = hasher.result_str();
hasher.reset();
let mut hasher = Sha1::new();
hasher.input_str("file2");
let hash2 = hasher.result_str();
hasher.reset();
let mut hasher = Sha1::new();
hasher.input_str("file4");
let hash4 = hasher.result_str();
hasher.reset();
let mut hashes = HashSet::new();
hashes.insert(hash1.as_str());
hashes.insert(hash2.as_str());
hashes.insert(hash4.as_str());
let mut objects: Vec<&str> = vec![];
objects.push("file1");
objects.push("file2");
objects.push("file3");
find_missing_elements(&mut hashes, &mut objects);
dbg!(hashes.clone());
dbg!(objects.clone());
assert_eq!(hashes.contains(hash4.as_str()), true);
assert_eq!(hashes.len(), 1);
assert_eq!(objects, vec!["file3"]);
}
}