support closing the standard streams on node.js

ghcjs · May 22, 2015 · 4f2af22 · 4f2af22
1 parent fbddb25
commit 4f2af22
Showing 1 changed file with 23 additions and 4 deletions.
diff --git a/pkg/base.js b/pkg/base.js
@@ -456,7 +456,8 @@ function h$unlockFile(fd) {
 
 
 // engine-dependent setup
-var h$base_readStdin, h$base_writeStderr, h$base_writeStdout;
+var h$base_readStdin , h$base_writeStderr, h$base_writeStdout;
+var h$base_closeStdin = null, h$base_closeStderr = null, h$base_closeStdout = null;
 var h$base_readFile,  h$base_writeFile,   h$base_closeFile;
 #ifndef GHCJS_BROWSER
 var h$base_stdin_waiting = new h$Queue();
@@ -515,6 +516,12 @@ if(h$isNode) {
         h$base_process_stdin();
     }
 
+    h$base_closeStdin = function(fd, fdo, c) {
+        TRACE_IO("close stdin");
+        process.stdin.close();
+        c();
+    }
+
     h$base_writeFile = function(fd, fdo, buf, buf_offset, n, c) {
         var pos = typeof fdo.pos === 'number' ? fdo.pos : null;
         TRACE_IO("base_writeFile: " + fd + " " + pos + " " + buf_offset + " " + n);
@@ -541,11 +548,23 @@ if(h$isNode) {
         h$base_writeFile(1, fdo, buf, buf_offset, n, c);
     }
 
+    h$base_closeStdout = function(fd, fdo, c) {
+        TRACE_IO("close stdout");
+        process.stdout.close();
+        c();
+    }
+
     h$base_writeStderr = function(fd, fdo, buf, buf_offset, n, c) {
         TRACE_IO("write stderr");
         h$base_writeFile(2, fdo, buf, buf_offset, n, c);
     }
 
+    h$base_closeStderr = function(fd, fdo, c) {
+        TRACE_IO("close stderr");
+        process.stderr.close();
+        c();
+    }
+
     process.stdin.on('readable', h$base_process_stdin);
     process.stdin.on('end', function() { h$base_stdin_eof = true; h$base_process_stdin(); });
 
@@ -606,9 +625,9 @@ if(h$isNode) {
 }
 #endif
 
-var h$base_stdin_fd  = { read:  h$base_readStdin };
-var h$base_stdout_fd = { write: h$base_writeStdout };
-var h$base_stderr_fd = { write: h$base_writeStderr };
+var h$base_stdin_fd  = { read:  h$base_readStdin,   close: h$base_closeStdin  };
+var h$base_stdout_fd = { write: h$base_writeStdout, close: h$base_closeStdout };
+var h$base_stderr_fd = { write: h$base_writeStderr, close: h$base_closeStderr };
 
 var h$base_fdN = -1; // negative file descriptors are 'virtual'
 var h$base_fds = [h$base_stdin_fd, h$base_stdout_fd, h$base_stderr_fd];