cli/repo: Plug in repo update

Signed-off-by: Ikey Doherty <[email protected]>

moss/src/cli/repo.rs

@@ -21,6 +21,8 @@ enum Action<'a> {
     Add(&'a Path, String, Url, String, Priority),
     // Root, Id
     Remove(&'a Path, String),
+    // Root, Id
+    Update(&'a Path, Option<String>),
 }
 
 /// Return a command for handling `repo` subcommands
@@ -60,6 +62,12 @@ pub fn command() -> Command {
                 .about("Remove a repository for the system")
                 .arg(arg!(<NAME> "repo name").value_parser(clap::value_parser!(String))),
         )
+        .subcommand(
+            Command::new("update")
+                .about("Update the system repositories")
+                .long_about("If no repository is named, update them all")
+                .arg(arg!([NAME] "repo name").value_parser(clap::value_parser!(String))),
+        )
 }
 
 /// Handle subcommands to `repo`
@@ -76,6 +84,9 @@ pub async fn handle(args: &ArgMatches, root: &Path) -> Result<(), Error> {
         Some(("remove", cmd_args)) => {
             Action::Remove(root, cmd_args.get_one::<String>("NAME").cloned().unwrap())
         }
+        Some(("update", cmd_args)) => {
+            Action::Update(root, cmd_args.get_one::<String>("NAME").cloned())
+        }
         _ => unreachable!(),
     };
 
@@ -86,6 +97,7 @@ pub async fn handle(args: &ArgMatches, root: &Path) -> Result<(), Error> {
             add(root, name, uri, comment, priority).await
         }
         Action::Remove(_, _) => unimplemented!(),
+        Action::Update(root, name) => update(root, name).await,
     }
 }
 
@@ -137,6 +149,19 @@ async fn list(root: &Path) -> Result<(), Error> {
     Ok(())
 }
 
+/// Update specific repos or all
+async fn update(root: &Path, which: Option<String>) -> Result<(), Error> {
+    let installation = Installation::open(root);
+    let mut manager = repository::Manager::new(installation).await?;
+
+    match which {
+        Some(repo) => manager.refresh(&repository::Id::new(repo)).await?,
+        None => manager.refresh_all().await?,
+    }
+
+    Ok(())
+}
+
 #[derive(Debug, Error)]
 pub enum Error {
     #[error("repo manager")]

moss/src/repository/manager.rs

@@ -95,6 +95,15 @@ impl Manager {
         Ok(())
     }
 
+    /// Refresh a [`Repository`] by Id
+    pub async fn refresh(&mut self, id: &repository::Id) -> Result<(), Error> {
+        if let Some(repo) = self.repositories.get(id) {
+            refresh_index(id, repo, &self.installation).await
+        } else {
+            Err(Error::UnknownRepo(id.clone()))
+        }
+    }
+
     /// Returns the active repositories held by this manager
     pub(crate) fn active(&self) -> impl Iterator<Item = repository::Active> + '_ {
         self.repositories.values().cloned()
@@ -208,6 +217,8 @@ pub enum Error {
     Database(#[from] meta::Error),
     #[error("save config")]
     SaveConfig(#[source] config::SaveError),
+    #[error("unknown repo")]
+    UnknownRepo(repository::Id),
 }
 
 impl From<package::MissingMetaError> for Error {