`baeb3670f725`

source

cmds/jrb/src/main.rs5.1 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}106107fn main() {108	tracing_subscriber::fmt().init();109110	let opts = Opts::parse();111112	match opts.command {113		Command::Init => {114			let path = Path::new(MANIFEST);115			if path.exists() {116				warn!("{MANIFEST} already exists");117				exit(1);118			}119			let jf = JsonnetFile {120				version: 1,121				dependencies: Vec::new(),122				legacy_imports: true,123			};124			save_json(path, &jf);125		}126		Command::Install { uris, dry_run } => {127			let mut manifest = load_manifest();128129			for uri in &uris {130				let dep = GitSource::parse(uri).unwrap_or_else(|| {131					eprintln!("failed to parse URI: {uri}");132					exit(1);133				});134				let is_new = !manifest.dependencies.iter().any(|d| {135					std::mem::discriminant(&d.source) == std::mem::discriminant(&dep.source)136						&& d.canonical_name() == dep.canonical_name()137				});138				if is_new {139					manifest.dependencies.push(dep);140				}141			}142143			if !uris.is_empty() {144				save_json(Path::new(MANIFEST), &manifest);145			}146147			let lock = load_lockfile();148			do_install(&manifest, lock.as_ref(), &opts.jsonnetpkg_home, dry_run);149		}150		Command::Update { uris, dry_run } => {151			let mut manifest = load_manifest();152153			if !uris.is_empty() {154				for uri in &uris {155					let dep = GitSource::parse(uri).unwrap_or_else(|| {156						eprintln!("failed to parse URI: {uri}");157						exit(1);158					});159					if let Some(existing) = manifest160						.dependencies161						.iter_mut()162						.find(|d| d.canonical_name() == dep.canonical_name())163					{164						*existing = dep;165					} else {166						manifest.dependencies.push(dep);167					}168				}169				save_json(Path::new(MANIFEST), &manifest);170			}171172			do_install(&manifest, None, &opts.jsonnetpkg_home, dry_run);173		}174		Command::Remove { names, dry_run } => {175			let mut manifest = load_manifest();176177			let matched: Vec<_> = manifest178				.dependencies179				.iter()180				.filter(|dep| {181					names.iter().any(|name| {182						dep.canonical_name() == *name || dep.legacy_link_name() == *name183					})184				})185				.cloned()186				.collect::<Vec<_>>();187188			if matched.is_empty() {189				eprintln!("no matching dependencies found");190				exit(1);191			}192193			for dep in &matched {194				let canonical = dep.canonical_name();195				let dir = opts.jsonnetpkg_home.join(&canonical);196				let legacy = dep.legacy_link_name();197				let link = opts.jsonnetpkg_home.join(&legacy);198				if dry_run {199					info!("would remove: {canonical} ({})", dir.display());200				} else {201					info!("removing: {canonical}");202					if dir.exists() {203						let _ = std::fs::remove_dir_all(&dir);204					}205					if link.symlink_metadata().is_ok() {206						let _ = std::fs::remove_file(&link);207					}208				}209			}210211			if !dry_run {212				manifest.dependencies.retain(|dep| {213					!names.iter().any(|name| {214						dep.canonical_name() == *name || dep.legacy_link_name() == *name215					})216				});217				save_json(Path::new(MANIFEST), &manifest);218				save_json(Path::new(LOCKFILE), &manifest);219			}220		}221	}222}