path: root/src/main.rs

extern crate serde;
extern crate serde_json;
extern crate toml;
extern crate url;
extern crate xdg;
extern crate flexi_logger;
extern crate filters;
extern crate boolinator;
extern crate itertools;
extern crate semver;

#[cfg(feature = "compare_csv")]
extern crate csv;

#[macro_use] extern crate serde_derive;
#[macro_use] extern crate log;
#[macro_use] extern crate anyhow;
#[macro_use] extern crate prettytable;

mod config;
mod backend;
mod frontend;
mod cli;
mod compare;

use std::path::PathBuf;
use std::cmp::Ordering;

#[cfg(feature = "compare_csv")]
use std::io::Cursor;

use anyhow::Error;
use anyhow::Context;
use anyhow::Result;
use clap::ArgMatches;
use filters::filter::Filter;
use boolinator::Boolinator;
use itertools::Itertools;
use semver::Version as SemverVersion;

use config::Configuration;
use compare::ComparePackage;
use librepology::v1::api::Api;
use librepology::v1::types::Repo;
use librepology::v1::types::Package;

fn initialize_logging(app: &ArgMatches) -> Result<()> {
    let verbosity = app.occurrences_of("verbose");
    let quietness = app.occurrences_of("quiet");
    let sum = verbosity as i64 - quietness as i64;
    let mut level_filter = flexi_logger::LevelFilter::Info;

    if sum == 1 {
        level_filter = flexi_logger::LevelFilter::Debug;
    } else if sum >= 2 {
        level_filter = flexi_logger::LevelFilter::Trace;
    } else if sum == -1 {
        level_filter = flexi_logger::LevelFilter::Warn;
    } else if sum <= -2 {
        level_filter = flexi_logger::LevelFilter::Error;
    }

    let mut builder = flexi_logger::LogSpecBuilder::new();
    builder.default(level_filter);

    flexi_logger::Logger::with(builder.build())
        .start()
        .map(|_| {
            debug!("Logger initialized!");
        })
        .map_err(Error::from)
}

fn deserialize_package_list(s: String, filepath: &str) -> Result<Vec<ComparePackage>> {
    let pb = PathBuf::from(filepath);
    let ext = pb
        .extension()
        .ok_or_else(|| format_err!("Couldn't get file extension: {}", filepath))?
        .to_str()
        .ok_or_else(|| format_err!("Not valid Unicode: {}", filepath))?;

    match ext {
        "json" => {
            serde_json::from_str(&s).map_err(Error::from)
        },

        #[cfg(feature = "compare_csv")]
        "csv" => {
            let cursor = Cursor::new(s);
            let mut v : Vec<ComparePackage> = vec![];
            let mut reader = csv::ReaderBuilder::new()
                .has_headers(true)
                .delimiter(b';')
                .from_reader(cursor);

            for element in reader.deserialize() {
                v.push(element?);
            }
            Ok(v)
        },

        other => Err(format_err!("Unknown file extension: {}", other)),
    }
}

fn app() -> Result<()> {
    let app = cli::build_cli().get_matches();
    initialize_logging(&app)?;
    let config : Configuration = {
        let path = if let Some(path) = app
            .value_of("config")
            .map(PathBuf::from)
            {
                debug!("Found passed configuration file at {}", path.display());
                Ok(path)
            } else {
                debug!("Searching for configuration in XDG");
                xdg::BaseDirectories::new()?
                    .find_config_file("repolocli.toml")
                    .ok_or_else(|| anyhow!("Cannot find repolocli.toml"))
            }?;

        debug!("Parsing configuration from file: {}", path.display());

        let buffer = std::fs::read_to_string(path).map_err(Error::from)?;
        trace!("Config read into memory");
        toml::de::from_str(&buffer)
            .map_err(Error::from)
            .context("Configuration file parsing")
    }?;
    trace!("Config deserialized");

    debug!("Initializing Backend");
    let backend = crate::backend::new_backend(&app, &config)?;
    debug!("Backend initialized");

    debug!("Initializing Frontend");
    let frontend = crate::frontend::new_frontend(&app, &config)?;
    debug!("Frontend initialized");

    let repository_filter = {
        let blacklist_filter = |repo: &Repo| -> bool {
            if config.blacklist().contains(repo) {
                trace!("In Blacklist: {:?} -> false", repo);
                false
            } else {
                trace!("Not in Blacklist: {:?} -> true", repo);
                true
            }
        };

        let whitelist_filter = |repo: &Repo| -> bool {
            if config.whitelist().contains(repo) {
                trace!("In Whitelist: {:?} -> true", repo);
                true
            } else {
                trace!("Not in Whitelist: {:?} -> false", repo);
                false
            }
        };

        blacklist_filter.or(whitelist_filter)
    };
    debug!("Repository filter constructed successfully");

    match app.subcommand() {
        ("project", Some(mtch)) => {
            debug!("Subcommand: 'project'");
            trace!("sort-versions:   {}", mtch.is_present("sort-version"));
            trace!("sort-repository: {}", mtch.is_present("sort-repo"));

            let name = if app.is_present("input_stdin") {
                // Ugly, but works:
                // If we have "--stdin" on CLI, we have a CLI/Stdin backend, which means that we can query
                // _any_ "project", and get the stdin anyways. This is really not like it should be, but
                // works for now
                ""
            } else {
                mtch.value_of("project_name").unwrap()  // safe by clap
            };

            let mut packages: Vec<Package> = {
                debug!("Fetching packages");
                let iter = backend
                    .project(&name)?
                    .into_iter()
                    .filter(|package| repository_filter.filter(package.repo()));

                if mtch.is_present("sort-version"){
                    trace!("Sorting by version");
                    iter.sorted_by(|a, b| Ord::cmp(a.version(), b.version()))
                        .collect()
                } else if mtch.is_present("sort-repo") {
                    trace!("Sorting by repository");
                    iter.sorted_by(|a, b| Ord::cmp(a.repo(), b.repo()))
                        .collect()
                } else {
                    trace!("Not sorting");
                    iter.collect()
                }
            };

            let packages = if mtch.is_present("latest") {
                if mtch.is_present("semver") {
                    let comp = |a: &Package, b: &Package| {
                        let av = SemverVersion::parse(a.version());
                        let bv = SemverVersion::parse(b.version());

                        if let (Ok(av), Ok(bv)) = (av, bv) {
                            av.partial_cmp(&bv).unwrap_or(Ordering::Equal)
                        } else {
                            Ordering::Equal
                        }
                    };

                    packages.sort_by(comp);
                } else {
                    packages.sort_by(|a, b| a.version().partial_cmp(b.version()).unwrap_or(Ordering::Equal));
                }
                packages.pop().into_iter().collect::<Vec<_>>()
            } else {
                packages
            };

            debug!("Listing packages in frontend");
            frontend.list_packages(packages)
        },

        ("problems", Some(mtch)) => {
            debug!("Subcommand: 'problems'");

            let repo = mtch.value_of("repo");
            let maintainer = mtch.value_of("maintainer");

            trace!("repo       = {:?}", repo);
            trace!("maintainer = {:?}", maintainer);

            let problems = {
                debug!("Finding problems...");
                let iter = match (repo, maintainer) {
                    (Some(r), None) => backend.problems_for_repo(<