Trim whitespaces in names

1 files changed, 28 insertions, 4 deletions

diff --git a/lib/etatsbasen.js b/lib/etatsbasen.js
index c31d9c9..ef966a7 100644
--- a/lib/etatsbasen.js
+++ b/lib/etatsbasen.js
@@ -90,6 +90,28 @@ function renameHeader(data) {
   return data;
 }
 
+function trimString(str) {
+  var newStr = str.replace(/^\s\s*/, '').replace(/\s\s*$/, '');
+  return newStr;
+}
+
+function trim(data) {
+  if (! data) {
+    throw new TypeError('Missing `csvdata` argument');
+  }
+
+  data.forEach(function(item, i) {
+    if (i > 0) {
+      item[2] = trimString(item[2]);
+      item[3] = trimString(item[3]);
+      item[4] = trimString(item[4]);
+    }
+  });
+
+  return data;
+}
+
+
 function removeColumns(csvdata, headerFilter) {
   var headernameToIndex = {};
   if (! csvdata) {
@@ -220,10 +242,12 @@ exports.printCSV = function(cb, options) {
                       removeColumns(
                         addURL(
                           addTags(
-                            renameHeader(
-                              filter(data,
-                                     options.categories
-                                    )
+                            trim(
+                              renameHeader(
+                                filter(data,
+                                       options.categories
+                                      )
+                              )
                             )
                           )
                         ),