From 0d4718c3f38f09aaad4bad0e19ee82c032d9040f Mon Sep 17 00:00:00 2001 From: Adrian Malacoda Date: Sat, 12 May 2018 00:11:44 -0500 Subject: [PATCH] add a helper trait that extends Table for more convenient value retrieval --- src/helpers/config.rs | 11 +++++++++++ src/helpers/mod.rs | 1 + 2 files changed, 12 insertions(+) create mode 100644 src/helpers/config.rs diff --git a/src/helpers/config.rs b/src/helpers/config.rs new file mode 100644 index 0000000..1dd685c --- /dev/null +++ b/src/helpers/config.rs @@ -0,0 +1,11 @@ +use toml::value::Table; + +pub trait Config { + fn get_string (&self, key: &str) -> Option<&str>; +} + +impl Config for Table { + fn get_string (&self, key: &str) -> Option<&str> { + self.get(key).and_then(|value| value.as_str()) + } +} diff --git a/src/helpers/mod.rs b/src/helpers/mod.rs index 9fe7961..6d982ac 100644 --- a/src/helpers/mod.rs +++ b/src/helpers/mod.rs @@ -1 +1,2 @@ pub mod command; +pub mod config;