feat: rocks pin

rocks-bin/src/main.rs

@@ -9,6 +9,7 @@ use info::Info;
 use install::Install;
 use list::ListCmd;
 use outdated::Outdated;
+use pin::Pin;
 use remove::Remove;
 use rocks_lib::config::{ConfigBuilder, LuaVersion};
 use search::Search;
@@ -24,6 +25,7 @@ mod install;
 mod install_lua;
 mod list;
 mod outdated;
+mod pin;
 mod project;
 mod purge;
 mod remove;
@@ -129,6 +131,8 @@ enum Commands {
     Pack,
     /// Return the currently configured package path.
     Path,
+    /// Pin an existing rock, preventing any updates to the package.
+    Pin(Pin),
     /// Remove all installed rocks from a tree.
     Purge,
     /// Uninstall a rock.
@@ -197,6 +201,7 @@ async fn main() {
         Commands::Remove(remove_args) => remove::remove(remove_args, config).await.unwrap(),
         Commands::Update(_update_args) => update::update(config).await.unwrap(),
         Commands::Info(info_data) => info::info(info_data, config).await.unwrap(),
+        Commands::Pin(pin_data) => pin::pin(pin_data, config).unwrap(),
         _ => unimplemented!(),
     }
 }

rocks-bin/src/pin.rs

@@ -0,0 +1,24 @@
+use clap::Args;
+use eyre::eyre;
+use eyre::Result;
+use rocks_lib::operations;
+use rocks_lib::{
+    config::{Config, LuaVersion},
+    package::PackageReq,
+    tree::Tree,
+};
+
+#[derive(Args)]
+pub struct Pin {
+    package: PackageReq,
+}
+
+pub fn pin(data: Pin, config: Config) -> Result<()> {
+    let tree = Tree::new(config.tree().clone(), LuaVersion::from(&config)?)?;
+
+    if let Some(mut rock) = tree.has_rock(&data.package) {
+        operations::pin(&mut rock, &tree)
+    } else {
+        Err(eyre!("Rock {} not found!", data.package))
+    }
+}

rocks-bin/src/update.rs

@@ -1,7 +1,7 @@
 use clap::Args;
-use eyre::OptionExt;
 use eyre::Result;
 use indicatif::MultiProgress;
+use rocks_lib::config::LuaVersion;
 use rocks_lib::{
     config::Config,
     manifest::{manifest_from_server, ManifestMetadata},
@@ -14,13 +14,7 @@ use rocks_lib::{
 pub struct Update {}
 
 pub async fn update(config: Config) -> Result<()> {
-    let tree = Tree::new(
-        config.tree().clone(),
-        config
-            .lua_version()
-            .cloned()
-            .ok_or_eyre("lua version not supplied!")?,
-    )?;
+    let tree = Tree::new(config.tree().clone(), LuaVersion::from(&config)?)?;
 
     let lockfile = tree.lockfile()?;
     let rocks = lockfile.rocks();

rocks-lib/Cargo.toml

@@ -46,6 +46,7 @@ infer = "0.16.0"
 indicatif = "0.17.8"
 sha2 = "0.10.8"
 hex = { version = "0.4.3" }
+fs_extra = "1.3.0"
 
 [dev-dependencies]
 httptest = { version = "0.16.1" }

rocks-lib/src/operations/mod.rs

@@ -1,13 +1,15 @@
 mod download;
 mod fetch;
 mod install;
+mod pin;
 mod remove;
 mod unpack;
 mod update;
 
 pub use download::*;
 pub use fetch::*;
 pub use install::*;
+pub use pin::*;
 pub use remove::*;
 pub use unpack::*;
 pub use update::*;

rocks-lib/src/operations/pin.rs

@@ -0,0 +1,32 @@
+use eyre::{bail, Result};
+use fs_extra::dir::CopyOptions;
+use itertools::Itertools;
+
+use crate::{lockfile::LocalPackage, tree::Tree};
+
+pub fn pin(package: &mut LocalPackage, tree: &Tree) -> Result<()> {
+    if package.pinned() {
+        bail!("Rock {} is already pinned!", package.to_package());
+    }
+
+    let mut lockfile = tree.lockfile()?;
+    let old_package = package.clone();
+    let items = std::fs::read_dir(tree.root_for(package))?
+        .filter_map(Result::ok)
+        .map(|dir| dir.path())
+        .collect_vec();
+
+    package.pinned = true;
+
+    let new_root = tree.root_for(package);
+
+    std::fs::create_dir_all(&new_root)?;
+
+    fs_extra::move_items(&items, new_root, &CopyOptions::new())?;
+
+    lockfile.remove(&old_package);
+    lockfile.add(package);
+    lockfile.flush()?;
+
+    Ok(())
+}