`de9f0cb0f75a`

source

cmds/jrb/src/main.rs5.2 KiBsource history
1use std::{2	path::{Path, PathBuf},3	process::exit,4};56use clap::{Parser, Subcommand};7use jrsonnet_pkg::{8	install,9	jsonnet_bundler::{GitSource, JsonnetFile},10};11use tracing::{error, info, warn};1213#[derive(Parser)]14#[clap(about = "A jsonnet package manager")]15struct Opts {16	/// The directory used to cache packages in.17	#[clap(long, default_value = "vendor")]18	jsonnetpkg_home: PathBuf,19	#[clap(subcommand)]20	command: Command,21}2223#[derive(Subcommand)]24enum Command {25	/// Initialize a new empty jsonnetfile26	Init,27	/// Install new dependencies. Existing ones are silently skipped28	Install {29		/// Package URIs to install30		uris: Vec<String>,31		/// Show what would be done without making changes32		#[clap(long)]33		dry_run: bool,34	},35	/// Update all or specific dependencies36	Update {37		/// Package URIs to update (all if empty)38		uris: Vec<String>,39		/// Show what would be done without making changes40		#[clap(long)]41		dry_run: bool,42	},43	/// Remove dependencies by name44	Remove {45		/// Dependency names (matched against both canonical and legacy names)46		names: Vec<String>,47		/// Show what would be removed without making changes48		#[clap(long)]49		dry_run: bool,50	},51}5253const MANIFEST: &str = "jsonnetfile.json";54const LOCKFILE: &str = "jsonnetfile.lock.json";5556fn load_manifest() -> JsonnetFile {57	let path = Path::new(MANIFEST);58	if path.exists() {59		JsonnetFile::load(path).unwrap_or_else(|e| {60			error!("failed to load {MANIFEST}: {e}");61			exit(1);62		})63	} else {64		JsonnetFile {65			version: 1,66			dependencies: Vec::new(),67			legacy_imports: true,68		}69	}70}7172fn save_json(path: &Path, value: &impl serde::Serialize) {73	let json = serde_json::to_string_pretty(value).expect("serialization failed");74	std::fs::write(path, format!("{json}\n")).unwrap_or_else(|e| {75		error!("failed to write {}: {e}", path.display());76		exit(1);77	});78}7980fn load_lockfile() -> Option<JsonnetFile> {81	let path = Path::new(LOCKFILE);82	if path.exists() {83		Some(JsonnetFile::load(path).unwrap_or_else(|e| {84			error!("failed to load {LOCKFILE}: {e}");85			exit(1);86		}))87	} else {88		None89	}90}9192fn do_install(93	manifest: &JsonnetFile,94	lock: Option<&JsonnetFile>,95	vendor_dir: &Path,96	dry_run: bool,97) {98	let new_lock = install::install(manifest, lock, vendor_dir, dry_run).unwrap_or_else(|e| {99		error!("install failed: {e}");100		exit(1);101	});102	if !dry_run {103		save_json(Path::new(LOCKFILE), &new_lock);104	}105}106107#[allow(clippy::too_many_lines)]108fn main() {109	tracing_subscriber::fmt().init();110111	let opts = Opts::parse();112113	match opts.command {114		Command::Init => {115			let path = Path::new(MANIFEST);116			if path.exists() {117				warn!("{MANIFEST} already exists");118				exit(1);119			}120			let jf = JsonnetFile {121				version: 1,122				dependencies: Vec::new(),123				legacy_imports: true,124			};125			save_json(path, &jf);126		}127		Command::Install { uris, dry_run } => {128			let mut manifest = load_manifest();129130			for uri in &uris {131				let dep = GitSource::parse(uri).unwrap_or_else(|| {132					eprintln!("failed to parse URI: {uri}");133					exit(1);134				});135				let is_new = !manifest.dependencies.iter().any(|d| {136					std::mem::discriminant(&d.source) == std::mem::discriminant(&dep.source)137						&& d.canonical_name() == dep.canonical_name()138				});139				if is_new {140					manifest.dependencies.push(dep);141				}142			}143144			if !uris.is_empty() {145				save_json(Path::new(MANIFEST), &manifest);146			}147148			let lock = load_lockfile();149			do_install(&manifest, lock.as_ref(), &opts.jsonnetpkg_home, dry_run);150		}151		Command::Update { uris, dry_run } => {152			let mut manifest = load_manifest();153154			if !uris.is_empty() {155				for uri in &uris {156					let dep = GitSource::parse(uri).unwrap_or_else(|| {157						eprintln!("failed to parse URI: {uri}");158						exit(1);159					});160					if let Some(existing) = manifest161						.dependencies162						.iter_mut()163						.find(|d| d.canonical_name() == dep.canonical_name())164					{165						*existing = dep;166					} else {167						manifest.dependencies.push(dep);168					}169				}170				save_json(Path::new(MANIFEST), &manifest);171			}172173			do_install(&manifest, None, &opts.jsonnetpkg_home, dry_run);174		}175		Command::Remove { names, dry_run } => {176			let mut manifest = load_manifest();177178			let matched: Vec<_> = manifest179				.dependencies180				.iter()181				.filter(|dep| {182					names.iter().any(|name| {183						dep.canonical_name() == *name || dep.legacy_link_name() == *name184					})185				})186				.cloned()187				.collect::<Vec<_>>();188189			if matched.is_empty() {190				eprintln!("no matching dependencies found");191				exit(1);192			}193194			for dep in &matched {195				let canonical = dep.canonical_name();196				let dir = opts.jsonnetpkg_home.join(&canonical);197				let legacy = dep.legacy_link_name();198				let link = opts.jsonnetpkg_home.join(&legacy);199				if dry_run {200					info!("would remove: {canonical} ({})", dir.display());201				} else {202					info!("removing: {canonical}");203					if dir.exists() {204						let _ = std::fs::remove_dir_all(&dir);205					}206					if link.symlink_metadata().is_ok() {207						let _ = std::fs::remove_file(&link);208					}209				}210			}211212			if !dry_run {213				manifest.dependencies.retain(|dep| {214					!names.iter().any(|name| {215						dep.canonical_name() == *name || dep.legacy_link_name() == *name216					})217				});218				save_json(Path::new(MANIFEST), &manifest);219				save_json(Path::new(LOCKFILE), &manifest);220			}221		}222	}223}