tinymark/src/commands.rs

use url::Url;
use std::env::VarError;
use chrono::Utc;
use crate::structures::Bookmark;
use paris::*;
use crate::database;
use serde_json::json;
use std::path::PathBuf;
use std::fs::File;
use std::io::{BufReader, BufWriter};

use dialoguer::{
    Select,
    theme::ColorfulTheme,
    console::Term
};

pub fn edit(json: bool, url: &Option<Url>, path: Option<PathBuf>) {
    if json {
        println!("{}", json!({
            "status": "fail",
            "reason": "unsupported command",
        }));
        std::process::exit(exitcode::DATAERR);
    } else {
        match url {
            Some(link) => {
                println!("User selected item :\n{}", link);
            },
            None => {
                match database::get_all(json, path) {
                    Some(bookmarks) => {
                        let mut items: Vec<&Url> = Vec::new();
                        for i in &bookmarks {
                            items.push(&i.link);
                        }
                        let selection = Select::with_theme(&ColorfulTheme::default())
                            .items(&items)
                            .default(0)
                            .interact_on_opt(&Term::stderr()).unwrap();

                        match selection {
                            Some(index) => println!("User selected item :\n{}", bookmarks[index]),
                            None => println!("User did not select anything")
                        }
                    },
                    None => {
                        if json {
                            println!("{}", json!({
                                "status": "fail",
                                "reason": "an error ocurred in running your command",
                            }));
                        } else {
                            warn!("an error ocurred in running your command");
                        }
                    },
                }
            },
        }
    }
}

pub fn add(url: &Url, name: &String, description: &Option<String>, tags: &Vec<String>, json: bool, path: Option<PathBuf>) {
    let bookmark = Bookmark {
        link: url.to_owned(),
        label: name.to_string(),
        description: description.to_owned(),
        tags: tags.to_vec(),
        container: None,
        created_at: Utc::now(),
    };

    database::insert_entry(&bookmark, json, path);
    if json {
        println!("{}", serde_json::to_string(&bookmark).unwrap());
    } else {
        println!("Added new bookmark!");
        println!("{}", bookmark);
    };
}

pub fn list(json: bool, path: Option<PathBuf>) {
    match database::get_all(json, path) {
        Some(bookmarks) => {
            for i in bookmarks {
                if json {
                    println!("{}", serde_json::to_string(&i).unwrap());
                } else {
                    println!("{}", i);
                }
            }
        },
        None => {
            if json {
                println!("{}", json!({
                    "status": "fail",
                    "reason": "an error ocurred in running your command",
                }));
            } else {
                warn!("an error ocurred in running your command");
            }
        }
    }
}

pub fn env_err(json: bool, e: VarError) {
    if json {
        println!("{}", json!({
            "status": "fail",
            "reason": e.to_string(),
        }));
    } else {
        warn!("couldn't read $HOME environment variable: {}", e);
    }
}

pub fn export(file_path: PathBuf, json: bool, path: Option<PathBuf>) {
    let file = match File::create(&file_path) {
        Ok(f) => f,
        Err(e) => {
            if json {
                println!("{}", json!({
                    "status": "fail",
                    "reason": e.to_string(),
                }));
            } else {
                warn!("error opening file! {}", e);
            }
            std::process::exit(exitcode::DATAERR);
        },
    };
    let writer = BufWriter::new(file);
        
    match database::get_all(json, path) {
        Some(bookmarks) => serde_json::to_writer(writer, &bookmarks).unwrap(),
        None => std::process::exit(exitcode::IOERR),
    }
    
    if json {
        println!("{}", json!({
            "status": "success",
            "reason": format!("exported bookmarks to {}", file_path.to_str().unwrap()),
        }));
    } else {
        info!("Succesfully exported bookmarks to {}!", file_path.to_str().unwrap());
    }
}

pub fn import(file_path: PathBuf, json: bool, store_path: Option<PathBuf>) {
    let file = match File::open(&file_path) {
        Ok(f) => f,
        Err(e) => {
            if json {
                println!("{}", json!({
                    "status": "fail",
                    "reason": e.to_string(),
                }));
            } else {
                warn!("error opening file! {}", e);
            }
            std::process::exit(exitcode::DATAERR);
        }
    };
    let reader = BufReader::new(file);

    let bookmarks: Vec<Bookmark> = match serde_json::from_reader(reader) {
        Ok(contents) => contents,
        Err(e) => {
            if json {
                println!("{}", json!({
                    "status": "fail",
                    "reason": e.to_string(),
                }));
            } else {
                warn!("error serializing file! {}", e);
            }
            std::process::exit(exitcode::DATAERR);
        }
    };

    database::insert_multiple(&bookmarks, json, store_path);

    if json {
        println!("{}", json!({
            "status": "success",
            "reason": format!("imported bookmarks from {}", file_path.to_str().unwrap()),
        }));
    } else {
        info!("succesfully imported bookmarks from {}!", file_path.to_str().unwrap());
    }
}