pueue/client/main.rs

use ::anyhow::Result;
use ::simplelog::{Config, LevelFilter, SimpleLogger};
use ::structopt::StructOpt;

use ::pueue::message::Message;
use ::pueue::settings::Settings;

pub mod cli;
pub mod client;
pub mod edit;
pub mod message;
pub mod output;
pub mod output_helper;

use crate::cli::{Opt, SubCommand};
use crate::client::Client;
use crate::message::get_message_from_opt;
use crate::output::follow_task_logs;

#[async_std::main]
async fn main() -> Result<()> {
    let settings = Settings::new()?;
    let save_result = settings.save();
    if save_result.is_err() {
        println!("Failed saving config file.");
        println!("{:?}", save_result.err());
    }

    // Parse commandline options.
    let opt = Opt::from_args();

    // Set the verbosity level for the client app.
    if opt.verbose >= 3 {
        SimpleLogger::init(LevelFilter::Debug, Config::default())?;
    } else if opt.verbose == 2 {
        SimpleLogger::init(LevelFilter::Info, Config::default())?;
    } else if opt.verbose == 1 {
        SimpleLogger::init(LevelFilter::Warn, Config::default())?;
    } else if opt.verbose == 0 {
        SimpleLogger::init(LevelFilter::Error, Config::default())?;
    }

    if let SubCommand::Completions {
        shell,
        output_directory,
    } = &opt.cmd
    {
        let mut clap = Opt::clap();
        clap.gen_completions("pueue", shell.clone(), output_directory);
        return Ok(());
    }

    // Create the message that should be sent to the daemon
    // depending on the given commandline options.
    let message = get_message_from_opt(&opt, &settings)?;

    // Some special command handling.
    // Simple log output follows for local logs don't need any communication with the daemon.
    // Thereby we handle this separately over here.
    if let Message::StreamRequest(message) = &message {
        if settings.client.read_local_logs {
            let pueue_directory = settings.daemon.pueue_directory.clone();
            follow_task_logs(pueue_directory, message.task_id, message.err);
            return Ok(());
        }
    }

    let mut client = Client::new(settings, message, opt)?;
    client.run().await?;

    Ok(())
}