From 60a696ebe96a36f62be4e5a1c56eb787c0ed44e1 Mon Sep 17 00:00:00 2001
From: Sam Hedin <sam.hedin@gmail.com>
Date: Fri, 27 Mar 2020 23:42:52 +0100
Subject: [PATCH] Rebuild indexmap with proper headers

---
 crates/nu-cli/src/commands/headers.rs | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/crates/nu-cli/src/commands/headers.rs b/crates/nu-cli/src/commands/headers.rs
index b10f419b0..91db95ca0 100644
--- a/crates/nu-cli/src/commands/headers.rs
+++ b/crates/nu-cli/src/commands/headers.rs
@@ -54,8 +54,25 @@ pub fn count(
             _ => ()
         }
 
+        let mut newrows: Vec<_> = vec![];
+        for r in rows.iter().skip(1) {
+            match &r.value {
+                UntaggedValue::Row(d) => {
+                    let mut i = 0;
+                    let mut newrow = IndexMap::new();
+
+                    for (_, v) in d.entries.iter() {
+                        newrow.insert(headers[i].clone(), v);
+                        i += 1;
+                    }
+                    newrows.push(newrow);
+                }
+                _ => panic!("huh?")
+            }
+        }
+
         let mut file =  File::create("headout").unwrap();
-        write!(file, "args: {:#?}", headers);
+        write!(file, "args: {:#?}", newrows).unwrap();
 
         yield ReturnSuccess::value(UntaggedValue::int(rows.len()).into_value(name))
     };