1 files changed, 76 insertions, 0 deletions
diff --git a/Flushable.js b/Flushable.js
new file mode 100644
index 0000000..0e73b6d
--- /dev/null
+++ b/Flushable.js
@@ -0,0 +1,76 @@
+const ansi = require('./ansi')
+
+module.exports = class Flushable {
+  // A writable that can be used to collect chunks of data before writing
+  // them.
+
+  constructor(writable, shouldCompress = false) {
+    this.target = writable
+
+    // Use the magical ANSI self-made compression method that probably
+    // doesn't *quite* work but should drastically decrease write size?
+    this.shouldCompress = shouldCompress
+
+    // Update these if you plan on using the ANSI compressor!
+    this.screenLines = 24
+    this.screenCols = 80
+
+    this.ended = false
+
+    this.chunks = []
+  }
+
+  write(what) {
+    this.chunks.push(what)
+  }
+
+  flush() {
+    // Don't write if we've ended.
+    if (this.ended) {
+      return
+    }
+
+    // End if the target is destroyed.
+    // Yes, this relies on the target having a destroyed property
+    // Don't worry, it'll still work if there is no destroyed property though
+    // (I think)
+    if (this.target.destroyed) {
+      this.end()
+      return
+    }
+
+    let toWrite = this.chunks.join('')
+
+    if (this.shouldCompress) {
+      toWrite = this.compress(toWrite)
+    }
+
+    try {
+      this.target.write(toWrite)
+    } catch(err) {
+      console.error('Flushable write error (ending):', err.message)
+      this.end()
+    }
+
+    this.chunks = []
+  }
+
+  end() {
+    this.ended = true
+  }
+
+  compress(toWrite) {
+    // TODO: customize screen size
+    const screen = ansi.interpret(toWrite, this.screenLines, this.screenCols)
+
+    /*
+    const pcSaved = Math.round(100 - (100 / toWrite.length * screen.length))
+    console.log(
+      '\x1b[1A' +
+      `${toWrite.length} - ${screen.length} ${pcSaved}% saved   `
+    )
+    */
+
+    return screen
+  }
+}