yuna 0.3.1

Naive terminal variables manager.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125

mod cli;
mod config;
mod serde;

use anyhow::{Context, Result};
use clap::Parser;
use comfy_table::{Attribute, Cell, Color, Table, modifiers::UTF8_ROUND_CORNERS};
use std::borrow::Cow;
use std::io::{Error, ErrorKind};

use crate::cli::Args;
use crate::config::{
  delete_value, interpolate, read_configuration_files, read_current_configuration_file, set_value,
  write_current_configuration_file,
};

const TABLE_BORDERS: &str = "││──├─┼┤│    ┬┴┬┴┌┐└┘";

fn show_value(opts: &Args, name: &str) -> Result<(), Error> {
  let values = read_configuration_files(opts)?;

  if let Some(value) = values.get(name) {
    let output = match opts.raw {
      true => Cow::Borrowed(&value.0),
      false => Cow::Owned(interpolate(&values, &value.0)?),
    };
    println!("{}", output);
  }

  Ok(())
}

fn list_values(opts: &Args) -> Result<(), Error> {
  let values = read_configuration_files(opts)?;

  if values.is_empty() {
    println!("No variables found.");
    return Ok(());
  }

  // Build and show the table
  let mut table = Table::new();
  table
    .load_preset(TABLE_BORDERS)
    .apply_modifier(UTF8_ROUND_CORNERS)
    .set_header(vec![
      Cell::new("Name").add_attribute(Attribute::Bold),
      Cell::new("Value").add_attribute(Attribute::Bold),
      Cell::new("Raw").add_attribute(Attribute::Bold),
      Cell::new("Source ").add_attribute(Attribute::Bold),
    ]);

  let cwd_str = opts.cwd.display().to_string();

  for (name, (value, source)) in values.iter() {
    let parent = source
      .parent()
      .ok_or_else(|| Error::new(ErrorKind::InvalidInput, "Invalid source"))?;

    let source = if parent == opts.cwd {
      source
        .file_name()
        .ok_or_else(|| Error::new(ErrorKind::InvalidInput, "Invalid source"))?
        .to_string_lossy()
    } else {
      Cow::Owned(
        source
          .display()
          .to_string()
          .replace(&cwd_str, ".")
          .replace(&opts.home, "~"),
      )
    };

    let interpolated = match opts.raw {
      true => Cow::Borrowed(value),
      false => Cow::Owned(interpolate(&values, value)?),
    };

    table.add_row(vec![
      Cell::new(name).fg(Color::Cyan).add_attribute(Attribute::Bold),
      Cell::new(value).fg(Color::Green).add_attribute(Attribute::Bold),
      Cell::new(interpolated.as_ref()),
      Cell::new(source).fg(Color::Grey),
    ]);
  }

  println!("{}", table);

  Ok(())
}

fn main() -> Result<()> {
  let mut opts = match Args::try_parse() {
    Ok(opts) => opts,
    Err(e) => match e.kind() {
      clap::error::ErrorKind::DisplayVersion => {
        println!("{}", env!("CARGO_PKG_VERSION"));
        return Ok(());
      }
      _ => {
        e.exit();
      }
    },
  };

  opts.cwd = std::env::current_dir()?;
  opts.home = std::env::var("HOME").or_else(|_| std::env::var("USERPROFILE"))?; // Linux or Windows

  if opts.delete {
    let mut file = read_current_configuration_file(&opts).context("Cannot read configuration file")?;
    delete_value(&opts, &mut file).context("Cannot delete value")?;
    write_current_configuration_file(&opts, &file).context("Cannot write configuration file")?;
  } else if !opts.value.is_empty() {
    let mut file = read_current_configuration_file(&opts).context("Cannot read configuration file")?;
    set_value(&opts, &mut file).context("Cannot set value")?;
    write_current_configuration_file(&opts, &file).context("Cannot write configuration file")?;
  } else if let Some(ref name) = opts.name {
    show_value(&opts, name).context("Cannot read configuration files")?;
  } else {
    list_values(&opts).context("Cannot read configuration files")?;
  }

  Ok(())
}