Add segment structure and logic

1 files changed, 14 insertions, 5 deletions

diff --git a/src/main.rs b/src/main.rs
index 8a6cd5fd7..ab7e04e56 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,9 +1,19 @@
 #[macro_use]
 extern crate clap;
+extern crate ansi_term;
+
+mod modules;
+mod print;
+
+use ansi_term::Style;
 use clap::App;
-use std::io;
 
-mod char;
+pub struct Segment {
+    style: Style,
+    value: String,
+    prefix: Option<Box<Segment>>,
+    suffix: Option<Box<Segment>>,
+}
 
 fn main() {
     App::new("Starship")
@@ -12,9 +22,8 @@ fn main() {
         .version(crate_version!())
         // pull the authors from Cargo.toml
         .author(crate_authors!())
+        .after_help("https://github.com/matchai/starship")
         .get_matches();
 
-    prompt::char();
-    // let stdout = io::stdout();
-    // let mut handle = io::BufWriter::new(stdout);
+    print::prompt();
 }