diff options
-rw-r--r-- | lib/etatsbasen.js | 32 |
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 + ) + ) ) ) ), |