mirror of
https://github.com/thegeneralist01/archivr
synced 2026-07-22 03:05:32 +02:00
TheGeneralist
351e74df09
- capture.rs: add archive_id: Option<&str> to perform_capture; when Some, call font_extractor::extract_and_rewrite before hashing HTML, register each font as a deduplicated blob + 'font' artifact - main.rs: pass None as archive_id (CLI keeps fonts embedded) - routes.rs: add GET /api/archives/:id/blobs/:sha256 (serve_blob handler), pass Some(&archive_id) to perform_capture in capture_handler, add ApiError::internal constructor
102 lines
2.8 KiB
Rust
102 lines
2.8 KiB
Rust
use anyhow::{Context, Result};
|
|
use archivr_core::archive;
|
|
use clap::{Parser, Subcommand};
|
|
use std::{
|
|
env,
|
|
path::Path,
|
|
process,
|
|
};
|
|
|
|
#[derive(Parser, Debug)]
|
|
#[command(version, about, long_about = None)]
|
|
struct Args {
|
|
#[command(subcommand)]
|
|
command: Command,
|
|
}
|
|
|
|
#[derive(Subcommand, Debug)]
|
|
enum Command {
|
|
/// Archive the specified file or directory
|
|
Archive {
|
|
/// URL or Path to archive
|
|
path: String,
|
|
},
|
|
Init {
|
|
/// Path to initialize the archive in
|
|
#[arg(default_value = ".")]
|
|
path: String,
|
|
|
|
/// Store path - path to store the archived files in.
|
|
/// Structure will be:
|
|
/// store_path/
|
|
/// temp/
|
|
/// ...
|
|
/// raw/
|
|
/// ...
|
|
/// raw_tweets/
|
|
/// ...
|
|
/// structured/
|
|
/// ...
|
|
#[arg(default_value = "./.archivr/store")]
|
|
store_path: String,
|
|
|
|
/// Name of the archive
|
|
#[arg(short, long)]
|
|
name: String,
|
|
|
|
/// Wipe existing .archivr repository data
|
|
#[arg(long = "force-with-info-removal")]
|
|
force_with_info_removal: bool,
|
|
},
|
|
}
|
|
|
|
fn main() -> Result<()> {
|
|
let args = Args::parse();
|
|
|
|
match args.command {
|
|
Command::Archive { ref path } => {
|
|
let archive_path = match archive::find_archive_path()? {
|
|
Some(path) => path,
|
|
None => {
|
|
eprintln!("Not in an archive. Use 'archivr init' to create one.");
|
|
process::exit(1);
|
|
}
|
|
};
|
|
let archive_paths = archive::read_archive_paths(&archive_path)?;
|
|
let result = archivr_core::capture::perform_capture(&archive_paths, path, None)?;
|
|
println!("Archived: run {}", result.run_uid);
|
|
Ok(())
|
|
}
|
|
|
|
Command::Init {
|
|
path: ref archive_path_string,
|
|
store_path: ref store_path_string,
|
|
name: ref archive_name,
|
|
force_with_info_removal,
|
|
} => {
|
|
let archive_parent = Path::new(&archive_path_string);
|
|
let store_path = if Path::new(&store_path_string).is_relative() {
|
|
env::current_dir()
|
|
.context("failed to read current working directory")?
|
|
.join(store_path_string)
|
|
} else {
|
|
Path::new(store_path_string).to_path_buf()
|
|
};
|
|
|
|
let paths = archive::initialize_archive(
|
|
archive_parent,
|
|
&store_path,
|
|
archive_name,
|
|
force_with_info_removal,
|
|
)?;
|
|
|
|
println!(
|
|
"Initialized empty archive in {}",
|
|
paths.archive_path.display()
|
|
);
|
|
|
|
Ok(())
|
|
} // _ => eprintln!("Unknown command: {:?}", args.command),
|
|
}
|
|
}
|
|
|