export more object in index

Author: jianglibo

index.js

@@ -1,11 +1,17 @@
 var TreeLine = require('./lib/tree-line');
 var AsciiTree = require('./lib/ascii-tree');
+var AsciiTreeBuilder = require('./lib/asciitree-builder');
+var ConvertorBuilder = require('./lib/convertor-builder');
 var Convertor = require('./lib/convertor');
 var LineUtil = require('./lib/line-util');
+var BytesLine = require('./lib/bytes-line');
 var Char = require('./lib/char');
 
 module.exports.TreeLine = TreeLine;
 module.exports.AsciiTree = AsciiTree;
 module.exports.Convertor = Convertor;
 module.exports.LineUtil = LineUtil;
+module.exports.BytesLine = BytesLine;
+module.exports.AsciiTreeBuilder = AsciiTreeBuilder;
+module.exports.ConvertorBuilder = ConvertorBuilder;
 module.exports.Char = Char;

lib/ascii-tree.js

@@ -16,14 +16,24 @@ function AsciiTree(lines, leadingCharCode, enc) {
   this.topLine = null;
   this.enc = enc;
 }
+/**
+ * @function toBufferLines
+ * @return {Buffer[]} - converted Buffers.
+ */
+AsciiTree.prototype.toBufferArray = function() {
+  var enc = this.enc;
+  return this.toBytesLineArray().map(function(it) {
+    return it.toBuffer();
+  });
+};
 /**
  * @function toStringLines
  * @return {string[]} - converted lines.
  */
-AsciiTree.prototype.toStringLines = function() {
+AsciiTree.prototype.toStringArray = function() {
   var enc = this.enc;
-  return this.toBytesLines().map(function(it) {
-    return enc ? it.toBuffer().toString(enc) : it.toBuffer().toString();
+  return this.toBufferArray().map(function(it) {
+    return enc ? it.toString(enc) : it.toString();
   });
 };
 
@@ -32,22 +42,16 @@ AsciiTree.prototype.toStringLines = function() {
  * @return {string} - converted string.
  */
 AsciiTree.prototype.toString = function() {
-  var enc = this.enc,
-    bytesArray,
-    buf;
-
-  bytesArray = this.toBytesLines().reduce(function(val, it) {
-    return val.concat(it.content, it.separator);
-  }, []);
-  buf = new Buffer(bytesArray);
-  return enc ? buf.toString(enc) : buf.toString();
+  return this.toStringArray().reduce(function(val, it) {
+    return val + it;
+  }, "");
 };
 
 /**
- * @function toBytesLines
+ * @function toBytesLineArray
  * @return {BytesLine[]} - return array of BytesLine.
  */
-AsciiTree.prototype.toBytesLines = function() {
+AsciiTree.prototype.toBytesLineArray = function() {
   return this.topLine.toLines();
 };
 

lib/convertor.js

@@ -1,5 +1,6 @@
 var BytesLine = require('./bytes-line');
 var AsciiTree = require('./ascii-tree');
+var os = require('os');
 
 module.exports = Convertor;
 
@@ -58,7 +59,7 @@ Convertor.prototype.convert = function() {
       if (this.prepend) {
         this.mixedLines.push(this.prepend);
       }
-      this.mixedLines.push(new AsciiTree(lineArray).convert().toBytesLines());
+      this.mixedLines.push(new AsciiTree(lineArray).convert().toBytesLineArray());
       if (this.append) {
         this.mixedLines.push(this.append);
       }

lib/tree-stream.js

@@ -10,7 +10,7 @@ function TreeStream() {
     if (BytesLine.isBytesLine(bytesLines)) {
       this.push(bytesLines);
     } else {
-      new AsciiTree(bytesLines,null, enc).convert().toBytesLines().forEach(function(it) {
+      new AsciiTree(bytesLines,null, enc).convert().toBytesLineArray().forEach(function(it) {
         this.push(it);
       }, this);
     }

package.json

@@ -1,14 +1,14 @@
 {
   "name": "my-ascii-tree",
-  "version": "1.0.8",
+  "version": "1.0.11",
   "description": "convert indented line block to an ascii directory tree",
   "main": "index.js",
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1"
   },
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/jianglibo/dtree-converter.git"
+    "url": "git+https://github.com/jianglibo/ascii-tree.git"
   },
   "keywords": [
     "ascii",

test/ascii-tree.js

@@ -4,6 +4,7 @@ var assert = require('assert');
 var BytesLine = require('../lib/bytes-line');
 var AsciiTree = require('../lib/ascii-tree');
 var LineUtil = require('../lib/line-util');
+var os = require('os');
 
 describe('AsciiTree', function() {
   describe('#constructor', function() {
@@ -12,5 +13,10 @@ describe('AsciiTree', function() {
       assert.equal(null, tree.leadingCharCode);
       assert.equal("└── hello", tree.toString());
     });
+
+    it('should handle eol', function() {
+      var eol = os.EOL;
+      assert.equal('string', typeof eol);
+    });
   });
 });

test/index.js

@@ -2,7 +2,10 @@ var assert = require('assert');
 var AsciiTree = require('../index').AsciiTree;
 var Convertor = require('../index').Convertor;
 var TreeLine = require('../index').TreeLine;
+var BytesLine = require('../index').BytesLine;
 var LineUtil = require('../index').LineUtil;
+var AsciiTreeBuilder = require('../index').AsciiTreeBuilder;
+var ConvertorBuilder = require('../index').ConvertorBuilder;
 var Char = require('../index').Char;
 
 
@@ -14,6 +17,9 @@ describe('required', function() {
       assert(TreeLine, "Line should imported.");
       assert(LineUtil, "LineUtil should imported.");
       assert(Char, "Char should imported.");
+      assert(BytesLine, "BytesLine should imported.");
+      assert(AsciiTreeBuilder, "AsciiTreeBuilder should imported.");
+      assert(ConvertorBuilder, "ConvertorBuilder should imported.");
     });
   });
 });