topgrade/src/main.rs

extern crate console;
extern crate directories;
extern crate env_logger;
extern crate failure;
extern crate failure_derive;
#[cfg(unix)]
extern crate lazy_static;
extern crate log;
#[cfg(unix)]
extern crate nix;
#[cfg(feature = "self-update")]
extern crate self_update as self_update_crate;
extern crate serde;
extern crate serde_derive;
extern crate shellexpand;
extern crate structopt;
extern crate toml;
extern crate walkdir;
extern crate which;

#[cfg(target_os = "freebsd")]
mod freebsd;
#[cfg(target_os = "linux")]
mod linux;
#[cfg(target_os = "macos")]
mod macos;
#[cfg(unix)]
mod tmux;
#[cfg(unix)]
mod unix;
#[cfg(target_os = "windows")]
mod windows;

mod config;
mod ctrlc;
mod executor;
mod generic;
mod git;
mod node;
mod report;
#[cfg(feature = "self-update")]
mod self_update;
mod terminal;
mod utils;
mod vim;

use self::config::Config;
use self::git::{Git, Repositories};
use self::report::Report;
use self::terminal::Terminal;
use failure::Error;
use failure_derive::Fail;
use std::borrow::Cow;
use std::env;
use std::io::ErrorKind;
use std::process::exit;
use structopt::StructOpt;

#[derive(Fail, Debug)]
#[fail(display = "A step failed")]
struct StepFailed;

#[derive(Fail, Debug)]
#[fail(display = "Cannot find the user base directories")]
struct NoBaseDirectories;

#[derive(Fail, Debug)]
#[fail(display = "Process Interrupted")]
pub struct Interrupted;

struct ExecutionContext {
    terminal: Terminal,
}

fn execute<'a, F, M>(func: F, execution_context: &mut ExecutionContext) -> Result<Option<(M, bool)>, Error>
where
    M: Into<Cow<'a, str>>,
    F: Fn(&mut Terminal) -> Option<(M, bool)>,
{
    while let Some((key, success)) = func(&mut execution_context.terminal) {
        if success {
            return Ok(Some((key, success)));
        }

        let running = ctrlc::running();
        if !running {
            ctrlc::set_running(true);
        }

        let should_retry = execution_context.terminal.should_retry(running).map_err(|e| {
            if e.kind() == ErrorKind::Interrupted {
                Error::from(Interrupted)
            } else {
                Error::from(e)
            }
        })?;

        if !should_retry {
            return Ok(Some((key, success)));
        }
    }

    Ok(None)
}

fn run() -> Result<(), Error> {
    ctrlc::set_handler();

    let opt = config::Opt::from_args();

    if opt.run_in_tmux && env::var("TMUX").is_err() {
        #[cfg(unix)]
        {
            tmux::run_in_tmux();
        }
    }

    env_logger::init();

    let mut execution_context = ExecutionContext {
        terminal: Terminal::new(),
    };

    let base_dirs = directories::BaseDirs::new().ok_or(NoBaseDirectories)?;
    let git = Git::new();
    let mut git_repos = Repositories::new(&git);

    let config = Config::read(&base_dirs)?;
    let mut report = Report::new();

    #[cfg(any(target_os = "freebsd", target_os = "linux"))]
    let sudo = utils::which("sudo");

    #[cfg(feature = "self-update")]
    {
        if !opt.dry_run {
            if let Err(e) = self_update::self_update(&mut execution_context.terminal) {
                execution_context
                    .terminal
                    .print_warning(format!("Self update error: {}", e));
            }
        }
    }

    if let Some(commands) = config.pre_commands() {
        for (name, command) in commands {
            generic::run_custom_command(&name, &command, &mut execution_context.terminal, opt.dry_run)?;
        }
    }

    #[cfg(windows)]
    let powershell = windows::Powershell::new();

    #[cfg(windows)]
    report.push_result(execute(
        |terminal| powershell.update_modules(terminal, opt.dry_run),
        &mut execution_context,
    )?);

    #[cfg(target_os = "linux")]
    let distribution = linux::Distribution::detect();

    #[cfg(target_os = "linux")]
    {
        if !opt.no_system {
            match &distribution {
                Ok(distribution) => {
                    report.push_result(execute(
                        |terminal| distribution.upgrade(&sudo, terminal, opt.dry_run),
                        &mut execution_context,
                    )?);
                }
                Err(e) => {
                    println!("Error detecting current distribution: {}", e);
                }
            }
            report.push_result(execute(
                |terminal| linux::run_etc_update(&sudo, terminal, opt.dry_run),
                &mut execution_context,
            )?);
        }
    }

    #[cfg(windows)]
    report.push_result(execute(
        |terminal| windows::run_chocolatey(terminal, opt.dry_run),
        &mut execution_context,
    )?);

    #[cfg(windows)]
    report.push_result(execute(
        |terminal| windows::run_scoop(terminal, opt.dry_run),
        &mut execution_context,
    )?);

    #[cfg(unix)]
    report.push_result(execute(
        |terminal| unix::run_homebrew(terminal, opt.cleanup, opt.dry_run),
        &mut execution_context,
    )?);
    #[cfg(target_os = "freebsd")]
    report.push_result(execute(
        |terminal| freebsd::upgrade_packages(&sudo, terminal, opt.dry_run),
        &mut execution_context,
    )?);
    #[cfg(unix)]
    report.push_result(execute(
        |terminal| unix::run_nix(terminal, opt.dry_run),
        &mut execution_context,
    )?);

    if !opt.no_emacs {
        git_repos.insert(base_dirs.home_dir().join(".emacs.d"));
    }

    git_repos.insert(base_dirs.home_dir().join(".vim"));
    git_repos.insert(base_dirs.home_dir().join(".config/nvim"));

    #[cfg(unix)]
    {
        git_repos.insert(base_dirs.home_dir().join(".zshrc"));
        git_repos.insert(base_dirs.home_dir().join(".oh-my-zsh"));
        git_repos.insert(base_dirs.home_dir().join(".tmux"));
        git_repos.insert(base_dirs.home_dir().join(".config/fish"));
        git_repos.insert(base_dirs.config_dir().join("openbox"));
    }

    #[cfg(windows)]
    {
        if let Some(profile) = powershell.profile() {
            git_repos.insert(profile);
        }
    }

    if !opt.no_git_repos {
        if let Some(custom_git_repos) = config.git_repos() {
            for git_repo in custom_git_repos {
                git_repos.insert(git_repo);
            }
        }
    }
    for repo in git_repos.repositories() {
        report.push_result(execute(
            |terminal| git.pull(&repo, terminal, opt.dry_run),
            &mut execution_context,
        )?);
    }

    #[cfg(unix)]
    {
        report.push_result(execute(
            |terminal| unix::run_zplug(&base_dirs, terminal, opt.dry_run),
            &mut execution_context,
        )?);
        report.push_result(execute(
            |terminal| unix::run_fisher(&base_dirs, terminal, opt.dry_run),
            &mut execution_context,
        )?);
        report.push_result(execute(
            |terminal| tmux::run_tpm(&base_dirs, terminal, opt.dry_run),
            &mut execution_context,
        )?);
    }

    report.push_result(execute(
        |terminal| generic::run_rustup(&base_dirs, terminal, opt.dry_run),
        &mut execution_context,
    )?);
    report.push_result(execute(
        |terminal| generic::run_cargo_update(terminal, opt.dry_run),
        &mut execution_context,
    )?);

    if !opt.no_emacs {
        report.push_result(execute(
            |terminal| generic::run_emacs(&base_dirs, terminal, opt.dry_run),
            &mut execution_context,
        )?);
    }

    report.push_result(execute(
        |terminal| generic::run_opam_update(terminal, opt.dry_run),
        &mut execution_context,
    )?);
    report.push_result(execute(
        |terminal| generic::run_vcpkg_update(terminal, opt.dry_run),
        &mut execution_context,
    )?);
    report.push_result(execute(
        |terminal| generic::run_pipx_update(terminal, opt.dry_run),
        &mut execution_context,
    )?);
    report.push_result(execute(
        |terminal| generic::run_jetpack(terminal, opt.dry_run),
        &mut execution_context,
    )?);
    report.push_result(execute(
        |terminal| vim::upgrade_vim(&base_dirs, terminal, opt.dry_run),
        &mut execution_context,
    )?);
    report.push_result(execute(
        |terminal| vim::upgrade_neovim(&base_dirs, terminal, opt.dry_run),
        &mut execution_context,
    )?);
    report.push_result(execute(
        |terminal| node::run_npm_upgrade(&base_dirs, terminal, opt.dry_run),
        &mut execution_context,
    )?);
    report.push_result(execute(
        |terminal| generic::run_composer_update(&base_dirs, terminal, opt.dry_run),
        &mut execution_context,
    )?);
    report.push_result(execute(
        |terminal| node::yarn_global_update(terminal, opt.dry_run),
        &mut execution_context,
    )?);

    #[cfg(not(any(
        target_os = "freebsd",
        target_os = "openbsd",
        target_os = "netbsd",
        target_os = "dragonfly"
    )))]
    report.push_result(execute(
        |terminal| generic::run_apm(terminal, opt.dry_run),
        &mut execution_context,
    )?);
    report.push_result(execute(
        |terminal| generic::run_gem(&base_dirs, terminal, opt.dry_run),
        &mut execution_context,
    )?);

    #[cfg(target_os = "linux")]
    {
        report.push_result(execute(
            |terminal| linux::flatpak_user_update(terminal, opt.dry_run),
            &mut execution_context,
        )?);
        report.push_result(execute(
            |terminal| linux::flatpak_global_update(&sudo, terminal, opt.dry_run),
            &mut execution_context,
        )?);
        report.push_result(execute(
            |terminal| linux::run_snap(&sudo, terminal, opt.dry_run),
            &mut execution_context,
        )?);
    }

    if let Some(commands) = config.commands() {
        for (name, command) in commands {
            report.push_result(execute(
                |terminal| {
                    Some((
                        name,
                        generic::run_custom_command(&name, &command, terminal, opt.dry_run).is_ok(),
                    ))
                },
                &mut execution_context,
            )?);
        }
    }

    #[cfg(target_os = "linux")]
    {
        report.push_result(execute(
            |terminal| linux::run_fwupdmgr(terminal, opt.dry_run),
            &mut execution_context,
        )?);
        report.push_result(execute(
            |terminal| linux::run_needrestart(&sudo, terminal, opt.dry_run),
            &mut execution_context,
        )?);
    }

    #[cfg(target_os = "macos")]
    {
        if !opt.no_system {
            report.push_result(execute(
                |terminal| macos::upgrade_macos(terminal, opt.dry_run),
                &mut execution_context,
            )?);
        }
    }

    #[cfg(target_os = "freebsd")]
    {
        if !opt.no_system {
            report.push_result(execute(
                |terminal| freebsd::upgrade_freebsd(&sudo, terminal, opt.dry_run),
                &mut execution_context,
            )?);
        }
    }

    #[cfg(windows)]
    {
        if !opt.no_system {
            report.push_result(execute(
                |terminal| powershell.windows_update(terminal, opt.dry_run),
                &mut execution_context,
            )?);
        }
    }

    if !report.data().is_empty() {
        execution_context.terminal.print_separator("Summary");

        for (key, succeeded) in report.data() {
            execution_context.terminal.print_result(key, *succeeded);
        }

        #[cfg(target_os = "linux")]
        {
            if let Ok(distribution) = &distribution {
                distribution.show_summary();
            }
        }

        #[cfg(target_os = "freebsd")]
        freebsd::audit_packages(&sudo).ok();
    }

    if report.data().iter().all(|(_, succeeded)| *succeeded) {
        Ok(())
    } else {
        Err(StepFailed.into())
    }
}

fn main() {
    match run() {
        Ok(()) => {
            exit(0);
        }
        Err(error) => {
            if (error.downcast_ref::<StepFailed>().is_some()) || (error.downcast_ref::<Interrupted>().is_some()) {
            } else {
                println!("ERROR: {}", error)
            }
            exit(1);
        }
    }
}