Update oh-my-zsh plugins (fix #360) (#363)

src/steps/git.rs

@@ -1,4 +1,5 @@
 use crate::error::{SkipStep, TopgradeError};
+use crate::execution_context::ExecutionContext;
 use crate::executor::{CommandExt, RunType};
 use crate::terminal::print_separator;
 use crate::utils::{which, PathExt};
@@ -95,22 +96,19 @@ impl Git {
 
         None
     }
-
-    pub fn multi_pull(
-        &self,
-        repositories: &Repositories,
-        run_type: RunType,
-        extra_arguments: &Option<String>,
-    ) -> Result<()> {
+    pub fn multi_pull_step(&self, repositories: &Repositories, ctx: &ExecutionContext) -> Result<()> {
         if repositories.repositories.is_empty() {
             return Err(SkipStep.into());
         }
 
+        print_separator("Git repositories");
+        self.multi_pull(repositories, ctx)
+    }
+
+    pub fn multi_pull(&self, repositories: &Repositories, ctx: &ExecutionContext) -> Result<()> {
         let git = self.git.as_ref().unwrap();
 
-        print_separator("Git repositories");
-
-        if let RunType::Dry = run_type {
+        if let RunType::Dry = ctx.run_type() {
             repositories
                 .repositories
                 .iter()
@@ -133,7 +131,7 @@ impl Git {
 
                 command.args(&["pull", "--ff-only"]).current_dir(&repo);
 
-                if let Some(extra_arguments) = extra_arguments {
+                if let Some(extra_arguments) = ctx.config().git_arguments() {
                     command.args(extra_arguments.split_whitespace());
                 }
 
@@ -259,4 +257,15 @@ impl<'a> Repositories<'a> {
             error!("Bad glob pattern: {}", pattern);
         }
     }
+
+    #[cfg(unix)]
+    pub fn is_empty(&self) -> bool {
+        self.repositories.is_empty()
+    }
+
+    #[cfg(unix)]
+    pub fn remove(&mut self, path: &str) {
+        let _removed = self.repositories.remove(path);
+        debug_assert!(_removed);
+    }
 }

src/steps/zsh.rs

@@ -1,10 +1,15 @@
-use crate::executor::RunType;
+use crate::execution_context::ExecutionContext;
+use crate::executor::{CommandExt, RunType};
+use crate::git::Repositories;
 use crate::terminal::print_separator;
 use crate::utils::{require, PathExt};
 use anyhow::Result;
 use directories::BaseDirs;
+use log::debug;
 use std::env;
+use std::fs;
 use std::path::{Path, PathBuf};
+use std::process::Command;
 
 pub fn run_zr(base_dirs: &BaseDirs, run_type: RunType) -> Result<()> {
     let zsh = require("zsh")?;
@@ -78,13 +83,35 @@ pub fn run_zinit(base_dirs: &BaseDirs, run_type: RunType) -> Result<()> {
     run_type.execute(zsh).args(&["-i", "-c", cmd.as_str()]).check_run()
 }
 
-pub fn run_oh_my_zsh(base_dirs: &BaseDirs, run_type: RunType) -> Result<()> {
+pub fn run_oh_my_zsh(ctx: &ExecutionContext) -> Result<()> {
     require("zsh")?;
-    let oh_my_zsh = base_dirs.home_dir().join(".oh-my-zsh").require()?;
+    let oh_my_zsh = ctx.base_dirs().home_dir().join(".oh-my-zsh").require()?;
 
     print_separator("oh-my-zsh");
 
-    run_type
+    let mut custom_dir = PathBuf::from(
+        Command::new("zsh")
+            .args(&["-i", "-c", "echo -n $ZSH_CUSTOM"])
+            .check_output()?,
+    );
+    custom_dir.push("plugins");
+
+    debug!("oh-my-zsh custom dir: {}", custom_dir.display());
+
+    let mut custom_plugins = Repositories::new(ctx.git());
+    for entry in fs::read_dir(custom_dir)? {
+        let entry = entry?;
+        custom_plugins.insert_if_repo(entry.path());
+    }
+
+    custom_plugins.remove(&oh_my_zsh.to_string_lossy());
+
+    if !custom_plugins.is_empty() {
+        println!("Pulling custom plugins");
+        ctx.git().multi_pull(&custom_plugins, ctx)?;
+    }
+
+    ctx.run_type()
         .execute("sh")
         .env("ZSH", &oh_my_zsh)
         .arg(&oh_my_zsh.join("tools/upgrade.sh"))