Rasmus Rosengren
3 years ago
parent
cc573c56d6
commit
9155a46de6
GPG Key ID:
A13BC7BC4F81CF5F
@ -0,0 +1,71 @@ |
||||
use std::path::PathBuf; |
||||
|
||||
use async_recursion::async_recursion; |
||||
use futures::future::join_all; |
||||
use tokio::fs::{copy, create_dir, read_dir}; |
||||
|
||||
use crate::Context; |
||||
|
||||
pub async fn build_tree(context: &Context) -> std::io::Result<()> { |
||||
dir(context, PathBuf::new()).await |
||||
} |
||||
|
||||
#[async_recursion] |
||||
async fn dir(context: &Context, relative_path: PathBuf) -> std::io::Result<()> { |
||||
let source_path = context.tree_source_dir.join(&relative_path); |
||||
let build_path = context.tree_build_dir.join(&relative_path); |
||||
|
||||
match create_dir(build_path).await { |
||||
Err(e) if e.kind() != std::io::ErrorKind::AlreadyExists => { |
||||
return Err(e); |
||||
} |
||||
_ => (), |
||||
} |
||||
|
||||
let mut dir_walker = read_dir(source_path).await?; |
||||
|
||||
let mut dir_tasks = vec![]; |
||||
let mut file_tasks = vec![]; |
||||
|
||||
while let Some(entry) = dir_walker.next_entry().await? { |
||||
let metadata = entry.metadata().await?; |
||||
let os_name = entry.file_name(); |
||||
let name = os_name.to_str(); |
||||
if name == Some(".git") { |
||||
continue; |
||||
} |
||||
|
||||
let relative_path = relative_path.join(entry.file_name()); |
||||
|
||||
if metadata.is_dir() { |
||||
dir_tasks.push(dir(context, relative_path)); |
||||
} else if metadata.is_file() { |
||||
file_tasks.push(file(context, relative_path)); |
||||
} |
||||
} |
||||
|
||||
let dir_tasks = async { |
||||
join_all(dir_tasks) |
||||
.await |
||||
.into_iter() |
||||
.collect::<std::io::Result<Vec<_>>>() |
||||
}; |
||||
let file_tasks = async { |
||||
join_all(file_tasks) |
||||
.await |
||||
.into_iter() |
||||
.collect::<std::io::Result<Vec<_>>>() |
||||
}; |
||||
|
||||
tokio::try_join!(dir_tasks, file_tasks)?; |
||||
|
||||
Ok(()) |
||||
} |
||||
|
||||
async fn file(context: &Context, relative_path: PathBuf) -> std::io::Result<()> { |
||||
let source_path = context.tree_source_dir.join(&relative_path); |
||||
let build_path = context.tree_build_dir.join(&relative_path); |
||||
|
||||
copy(source_path, build_path).await?; |
||||
Ok(()) |
||||
} |
||||
@ -0,0 +1,36 @@ |
||||
use std::path::PathBuf; |
||||
|
||||
use tokio::fs::read; |
||||
|
||||
use crate::{tree::get_tree_files, Context}; |
||||
|
||||
pub async fn check_tree( |
||||
context: &Context, |
||||
) -> std::io::Result<(Vec<PathBuf>, Vec<PathBuf>, Vec<PathBuf>)> { |
||||
let source_files = get_tree_files(&context.tree_source_dir).await?; |
||||
let installed_files = get_tree_files(&context.tree_install_dir).await?; |
||||
|
||||
let mut deleted_files = vec![]; |
||||
let mut changed_files = vec![]; |
||||
for installed_file in installed_files.iter() { |
||||
if source_files.contains(installed_file) { |
||||
let source_file_content = read(context.tree_build_dir.join(installed_file)).await?; |
||||
let installed_file_content = |
||||
read(context.tree_install_dir.join(installed_file)).await?; |
||||
if source_file_content != installed_file_content { |
||||
changed_files.push(installed_file.clone()); |
||||
} |
||||
} else { |
||||
deleted_files.push(installed_file.clone()); |
||||
} |
||||
} |
||||
|
||||
let mut added_files = vec![]; |
||||
for source_file in source_files.iter() { |
||||
if !installed_files.contains(source_file) { |
||||
added_files.push(source_file.clone()); |
||||
} |
||||
} |
||||
|
||||
Ok((added_files, changed_files, deleted_files)) |
||||
} |
||||
@ -0,0 +1,52 @@ |
||||
use std::path::PathBuf; |
||||
|
||||
use tokio::fs::{copy, create_dir_all, remove_file, symlink}; |
||||
|
||||
use crate::{check::check_tree, Context}; |
||||
|
||||
pub async fn link_tree(context: &Context) -> std::io::Result<()> { |
||||
let (add, update, remove) = check_tree(context).await?; |
||||
for file in add { |
||||
copy_and_link(context, file).await?; |
||||
} |
||||
|
||||
for file in update { |
||||
copy_and_link(context, file).await?; |
||||
} |
||||
|
||||
for file in remove { |
||||
remove_file(context.link_root_dir.join(file)).await?; |
||||
} |
||||
|
||||
Ok(()) |
||||
} |
||||
|
||||
async fn copy_and_link(context: &Context, relative_path: PathBuf) -> std::io::Result<()> { |
||||
// Create all necessary directories that are accessed
|
||||
let mut current_install_dir = context.tree_install_dir.join(&relative_path); |
||||
current_install_dir.pop(); |
||||
let mut current_link_dir = context.link_root_dir.join(&relative_path); |
||||
current_link_dir.pop(); |
||||
create_dir_all(current_install_dir).await?; |
||||
create_dir_all(current_link_dir).await?; |
||||
|
||||
// Copy from build to install
|
||||
copy( |
||||
context.tree_build_dir.join(&relative_path), |
||||
context.tree_install_dir.join(&relative_path), |
||||
) |
||||
.await?; |
||||
|
||||
// Make sure symlink doesn't exist before attempting to symlink
|
||||
match remove_file(context.link_root_dir.join(&relative_path)).await { |
||||
Ok(_) => {} |
||||
Err(e) if e.kind() == std::io::ErrorKind::NotFound => {} |
||||
Err(e) => return Err(e), |
||||
}; |
||||
symlink( |
||||
context.tree_install_dir.join(&relative_path), |
||||
context.link_root_dir.join(&relative_path), |
||||
) |
||||
.await?; |
||||
Ok(()) |
||||
} |
||||
@ -1,47 +1,88 @@ |
||||
use async_recursion::async_recursion; |
||||
use tokio::fs; |
||||
mod build; |
||||
mod check; |
||||
mod link; |
||||
mod tree; |
||||
|
||||
use std::path::PathBuf; |
||||
|
||||
use build::build_tree; |
||||
use check::check_tree; |
||||
use link::link_tree; |
||||
|
||||
const APP_VERSION: &'static str = env!("CARGO_PKG_VERSION"); |
||||
|
||||
#[tokio::main] |
||||
async fn main() -> std::io::Result<()> { |
||||
let home_dir = std::env::var("HOME").expect("Could not get home dir"); |
||||
symlink_folder(&format!("{}/.df", home_dir), &home_dir).await; |
||||
Ok(()) |
||||
} |
||||
let app = clap::App::new("DotFiles Manager") |
||||
.version(APP_VERSION) |
||||
.author("Rasmus Rosengren <rasmus.rosengren@protonmail.com>") |
||||
.about("Utility to manage dotfiles") |
||||
.arg( |
||||
clap::Arg::with_name("dry-run") |
||||
.long("dry-run") |
||||
.short("d") |
||||
.help("Report changes without installing"), |
||||
) |
||||
.arg( |
||||
clap::Arg::with_name("repo-path") |
||||
.long("repo-path") |
||||
.short("r") |
||||
.default_value(".df") |
||||
.help("Location of repo, relative to $HOME"), |
||||
); |
||||
|
||||
let matches = app.get_matches(); |
||||
|
||||
let home_dir = std::env::var("HOME").expect("$HOME variable not found"); |
||||
let xdg_dirs = xdg::BaseDirectories::with_prefix("dfm").expect("xdg dirs"); |
||||
|
||||
#[async_recursion] |
||||
async fn symlink_folder(src_dir: &str, target_dir: &str) { |
||||
fs::create_dir_all(target_dir) |
||||
.await |
||||
.unwrap_or_else(|_| panic!("Could not create {}", target_dir)); |
||||
let mut dir_contents = fs::read_dir(src_dir) |
||||
.await |
||||
.unwrap_or_else(|_| panic!("Could read dir {}", src_dir)); |
||||
while let Ok(Some(entry)) = dir_contents.next_entry().await { |
||||
let os_name = entry.file_name(); |
||||
let name = os_name.to_str().unwrap(); |
||||
if name == ".git" { |
||||
continue; |
||||
let context = Context { |
||||
tree_source_dir: format!("{}/{}", home_dir, matches.value_of("repo-path").unwrap()).into(), |
||||
tree_build_dir: xdg_dirs |
||||
.create_cache_directory("tree") |
||||
.expect("xdg cache dir"), |
||||
tree_install_dir: xdg_dirs |
||||
.create_config_directory("tree") |
||||
.expect("xdg config dir"), |
||||
link_root_dir: home_dir.into(), |
||||
}; |
||||
|
||||
build_tree(&context).await?; |
||||
if matches.is_present("dry-run") { |
||||
let (add, change, delete) = check_tree(&context).await?; |
||||
if !add.is_empty() { |
||||
println!("The following files have been added:"); |
||||
for file in add { |
||||
println!("\t{}", file.to_string_lossy()); |
||||
} |
||||
println!(); |
||||
} |
||||
|
||||
if let Ok(metadata) = entry.metadata().await { |
||||
if metadata.is_dir() { |
||||
symlink_folder( |
||||
&format!("{}/{}", src_dir, name), |
||||
&format!("{}/{}", target_dir, name), |
||||
) |
||||
.await; |
||||
if !change.is_empty() { |
||||
println!("The following files have been changed:"); |
||||
for file in change { |
||||
println!("\t{}", file.to_string_lossy()); |
||||
} |
||||
println!(); |
||||
} |
||||
|
||||
if metadata.is_file() { |
||||
#[allow(unused_must_use)] |
||||
let _ = fs::remove_file(&format!("{}/{}", target_dir, name)).await; |
||||
fs::symlink( |
||||
&format!("{}/{}", src_dir, name), |
||||
&format!("{}/{}", target_dir, name), |
||||
) |
||||
.await |
||||
.unwrap(); |
||||
if !delete.is_empty() { |
||||
println!("The following files have been deleted:"); |
||||
for file in delete { |
||||
println!("\t{}", file.to_string_lossy()); |
||||
} |
||||
println!(); |
||||
} |
||||
} else { |
||||
link_tree(&context).await?; |
||||
} |
||||
|
||||
Ok(()) |
||||
} |
||||
|
||||
pub struct Context { |
||||
tree_source_dir: PathBuf, |
||||
tree_build_dir: PathBuf, |
||||
tree_install_dir: PathBuf, |
||||
link_root_dir: PathBuf, |
||||
} |
||||
|
||||
@ -0,0 +1,44 @@ |
||||
use std::{ |
||||
collections::HashSet, |
||||
path::{Path, PathBuf}, |
||||
}; |
||||
|
||||
use async_recursion::async_recursion; |
||||
use futures::future::try_join_all; |
||||
use tokio::fs::read_dir; |
||||
|
||||
pub async fn get_tree_files(tree_root_path: &Path) -> std::io::Result<HashSet<PathBuf>> { |
||||
dir(tree_root_path, PathBuf::new()).await |
||||
} |
||||
|
||||
#[async_recursion] |
||||
async fn dir(tree_root_path: &Path, relative_path: PathBuf) -> std::io::Result<HashSet<PathBuf>> { |
||||
let current_path = tree_root_path.join(&relative_path); |
||||
|
||||
let mut dir_walker = read_dir(¤t_path).await?; |
||||
let mut dir_tasks = vec![]; |
||||
|
||||
let mut files = HashSet::new(); |
||||
|
||||
while let Some(entry) = dir_walker.next_entry().await? { |
||||
let metadata = entry.metadata().await?; |
||||
let os_name = entry.file_name(); |
||||
let name = os_name.to_string_lossy().to_string(); |
||||
if name == ".git" { |
||||
continue; |
||||
} |
||||
|
||||
if metadata.is_dir() { |
||||
dir_tasks.push(dir(tree_root_path, relative_path.join(name))); |
||||
} else if metadata.is_file() { |
||||
files.insert(relative_path.join(name)); |
||||
} |
||||
} |
||||
|
||||
let file_sets = try_join_all(dir_tasks).await?; |
||||
for file_set in file_sets { |
||||
files.extend(file_set); |
||||
} |
||||
|
||||
Ok(files) |
||||
} |
||||
Loading…
Reference in new issue