From 95a48166045e924d84b25d4bbbce5f404769d352 Mon Sep 17 00:00:00 2001
From: Matthias Beyer <mail@beyermatthias.de>
Date: Tue, 24 Apr 2018 22:47:46 +0200
Subject: Implement imag-view argument to put seperator between shown entries

---
 bin/core/imag-view/src/main.rs | 47 +++++++++++++++++++++++++++++++++++++-----
 bin/core/imag-view/src/ui.rs   | 10 +++++++++
 2 files changed, 52 insertions(+), 5 deletions(-)

(limited to 'bin')

diff --git a/bin/core/imag-view/src/main.rs b/bin/core/imag-view/src/main.rs
index bc9e83b4..5cafb007 100644
--- a/bin/core/imag-view/src/main.rs
+++ b/bin/core/imag-view/src/main.rs
@@ -61,6 +61,8 @@ use libimagrt::runtime::Runtime;
 use libimagerror::str::ErrFromStr;
 use libimagerror::trace::MapErrTrace;
 use libimagerror::iter::TraceIterator;
+use libimagerror::io::ToExitCode;
+use libimagerror::exit::ExitUnwrap;
 use libimagentryview::builtin::stdout::StdoutViewer;
 use libimagentryview::builtin::md::MarkdownViewer;
 use libimagentryview::viewer::Viewer;
@@ -185,10 +187,33 @@ fn main() {
         let out         = rt.stdout();
         let mut outlock = out.lock();
 
+        let basesep = if rt.cli().occurrences_of("seperator") != 0 { // checker for default value
+            rt.cli().value_of("seperator").map(String::from)
+        } else {
+            None
+        };
+
+        let mut sep_width = 80; // base width, automatically overridden by wrap width
+
+        // Helper to build the seperator with a base string `sep` and a `width`
+        let build_seperator = |sep: String, width: usize| -> String {
+            sep.repeat(width / sep.len())
+        };
+
         if rt.cli().is_present("compile-md") {
-            let viewer = MarkdownViewer::new(&rt);
-            for entry in iter {
-                viewer.view_entry(&entry, &mut outlock).map_err_trace_exit_unwrap(1);
+            let viewer    = MarkdownViewer::new(&rt);
+            let seperator = basesep.map(|s| build_seperator(s, sep_width));
+
+            for (n, entry) in iter.enumerate() {
+                if n != 0 {
+                    seperator
+                        .as_ref()
+                        .map(|s| writeln!(outlock, "{}", s).to_exit_code().unwrap_or_exit());
+                }
+
+                viewer
+                    .view_entry(&entry, &mut outlock)
+                    .map_err_trace_exit_unwrap(1);
             }
         } else {
             let mut viewer = StdoutViewer::new(view_header, !hide_content);
@@ -202,11 +227,23 @@ fn main() {
                     ::std::process::exit(1)
                 });
 
+                // Copying this value over, so that the seperator has the right len as well
+                sep_width = width;
+
                 viewer.wrap_at(width);
             }
 
-            for entry in iter {
-                viewer.view_entry(&entry, &mut outlock).map_err_trace_exit_unwrap(1);
+            let seperator = basesep.map(|s| build_seperator(s, sep_width));
+            for (n, entry) in iter.enumerate() {
+                if n != 0 {
+                    seperator
+                        .as_ref()
+                        .map(|s| writeln!(outlock, "{}", s).to_exit_code().unwrap_or_exit());
+                }
+
+                viewer
+                    .view_entry(&entry, &mut outlock)
+                    .map_err_trace_exit_unwrap(1);
             }
         }
     }
diff --git a/bin/core/imag-view/src/ui.rs b/bin/core/imag-view/src/ui.rs
index 157a701f..c97bd3f2 100644
--- a/bin/core/imag-view/src/ui.rs
+++ b/bin/core/imag-view/src/ui.rs
@@ -73,6 +73,16 @@ pub fn build_ui<'a>(app: App<'a, 'a>) -> App<'a, 'a> {
             .conflicts_with("not-view-content")
             .conflicts_with("autowrap")) // markdown viewer does not support wrapping
 
+        .arg(Arg::with_name("seperator")
+            .long("seperate")
+            .short("s")
+            .required(false)
+            .takes_value(true)
+            .value_name("SEPCHR")
+            .default_value("-")
+            .help("Do seperate entries with a string if viewing multiple entries"))
+
+
         .arg(Arg::with_name("in")
             .long("in")
             .takes_value(true)
-- 
cgit v1.2.3