bare-tty 2.0.1 → 2.1.1

package/binding.c

@@ -463,6 +463,51 @@ bare_tty_set_mode (js_env_t *env, js_callback_info_t *info) {
   return NULL;
 }
 
+static js_value_t *
+bare_tty_get_window_size (js_env_t *env, js_callback_info_t *info) {
+  int err;
+
+  size_t argc = 1;
+  js_value_t *argv[1];
+
+  err = js_get_callback_info(env, info, &argc, argv, NULL, NULL);
+  assert(err == 0);
+
+  assert(argc == 1);
+
+  bare_tty_t *tty;
+  err = js_get_arraybuffer_info(env, argv[0], (void **) &tty, NULL);
+  assert(err == 0);
+
+  int width, height;
+  err = uv_tty_get_winsize(&tty->handle, &width, &height);
+
+  if (err < 0) {
+    js_throw_error(env, uv_err_name(err), uv_strerror(err));
+    return NULL;
+  }
+
+  js_value_t *result;
+  err = js_create_array_with_length(env, 2, &result);
+  assert(err == 0);
+
+  js_value_t *value;
+
+  err = js_create_int32(env, width, &value);
+  assert(err == 0);
+
+  err = js_set_element(env, result, 0, value);
+  assert(err == 0);
+
+  err = js_create_int32(env, height, &value);
+  assert(err == 0);
+
+  err = js_set_element(env, result, 1, value);
+  assert(err == 0);
+
+  return result;
+}
+
 static js_value_t *
 init (js_env_t *env, js_value_t *exports) {
   {
@@ -500,6 +545,11 @@ init (js_env_t *env, js_value_t *exports) {
     js_create_function(env, "setMode", -1, bare_tty_set_mode, NULL, &fn);
     js_set_named_property(env, exports, "setMode", fn);
   }
+  {
+    js_value_t *fn;
+    js_create_function(env, "getWindowSize", -1, bare_tty_get_window_size, NULL, &fn);
+    js_set_named_property(env, exports, "getWindowSize", fn);
+  }
   {
     js_value_t *val;
     js_create_uint32(env, UV_TTY_MODE_NORMAL, &val);

package/index.js

@@ -25,6 +25,8 @@ exports.ReadStream = class TTYReadStream extends Readable {
       this._onread,
       this._onclose
     )
+
+    TTYReadStream._streams.add(this)
   }
 
   get isTTY () {
@@ -46,6 +48,7 @@ exports.ReadStream = class TTYReadStream extends Readable {
   _destroy (cb) {
     this._pendingDestroy = cb
     binding.close(this._handle)
+    TTYReadStream._streams.delete(this)
   }
 
   _continueDestroy () {
@@ -80,6 +83,8 @@ exports.ReadStream = class TTYReadStream extends Readable {
     this._handle = null
     this._continueDestroy()
   }
+
+  static _streams = new Set()
 }
 
 exports.WriteStream = class TTYWriteStream extends Writable {
@@ -96,12 +101,18 @@ exports.WriteStream = class TTYWriteStream extends Writable {
       noop,
       this._onclose
     )
+
+    TTYWriteStream._streams.add(this)
   }
 
   get isTTY () {
     return true
   }
 
+  getWindowSize () {
+    return binding.getWindowSize(this._handle)
+  }
+
   _writev (datas, cb) {
     this._pendingWrite = cb
     binding.writev(this._handle, datas)
@@ -115,6 +126,7 @@ exports.WriteStream = class TTYWriteStream extends Writable {
   _destroy (cb) {
     this._pendingDestroy = cb
     binding.close(this._handle)
+    TTYWriteStream._streams.delete(this)
   }
 
   _continueWrite (err) {
@@ -171,6 +183,8 @@ exports.WriteStream = class TTYWriteStream extends Writable {
     this._handle = null
     this._continueDestroy()
   }
+
+  static _streams = new Set()
 }
 
 exports.constants = {
@@ -179,6 +193,17 @@ exports.constants = {
   MODE_IO: binding.MODE_IO || 0
 }
 
+process
+  .on('exit', () => {
+    for (const stream of exports.ReadStream._streams) {
+      stream.destroy()
+    }
+
+    for (const stream of exports.WriteStream._streams) {
+      stream.destroy()
+    }
+  })
+
 const empty = Buffer.alloc(0)
 
 function noop () {}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bare-tty",
-  "version": "2.0.1",
+  "version": "2.1.1",
   "description": "Native TTY streams for JavaScript",
   "main": "index.js",
   "files": [