bash-kv 1.0.0

A simple bash kv store
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

use clap::{Arg, Command};
use std::error::Error;

fn main() -> Result<(), Box<dyn Error>> {
    let matches = get_matches();
    let data_dir = dirs::data_dir().expect("Could not find data directory");
    let store = sled::open(data_dir.join("kv.db"))?;

    match matches.subcommand() {
        Some(("set", submatches)) => {
            let key = submatches.value_of("key").expect("No key specified");
            let val: String = submatches
                .values_of("val")
                .expect("No value specified")
                .collect::<Vec<_>>()
                .join(" ");
            store.insert(key, val.as_bytes())?;
        }

        Some(("get", submatches)) => {
            let key = submatches.value_of("key").expect("No key specified");
            if let Some(val) = store.get(key)? {
                let s = String::from_utf8(val.to_vec())?;
                println!("{}", s);
            } else {
                eprintln!("key not found");
                std::process::exit(1);
            }
        }

        Some(("del", submatches)) => {
            let key = submatches.value_of("key").expect("No key specified");
            store.remove(key)?;
        }

        Some(("list", _submatches)) => {
            for row in store.iter() {
                let (key, val) = row.expect("Could not read row");
                println!(
                    "{}\t{}",
                    String::from_utf8(key.to_vec())?,
                    String::from_utf8(val.to_vec())?
                );
            }
        }
        _ => unreachable!(),
    }

    Ok(())
}

fn get_matches() -> clap::ArgMatches {
    Command::new("kv")
        .about("A simple key-value store")
        .subcommand(
            Command::new("set")
                .about("Sets a key/value pair")
                .arg(
                    Arg::new("key")
                        .help("key to set")
                        .value_name("KEY")
                        .required(true),
                )
                .arg(
                    Arg::new("val")
                        .help("value to set")
                        .value_name("VAL")
                        .multiple_values(true)
                        .required(true),
                ),
        )
        .subcommand(
            Command::new("get")
                .about("Gets a value for a given key")
                .arg(
                    Arg::new("key")
                        .help("key to fetch")
                        .value_name("KEY")
                        .required(true),
                ),
        )
        .subcommand(
            Command::new("del")
                .about("Deletes a value for a given key")
                .arg(
                    Arg::new("key")
                        .help("key to delete")
                        .value_name("KEY")
                        .required(true),
                ),
        )
        .subcommand(Command::new("list").about("List all keys and values separated by tabs"))
        .subcommand_required(true)
        .get_matches()
}