08-27-周三_17-09-29

node_modules/dnode/index.js

@@ -0,0 +1,137 @@
+var dnode = require('./lib/dnode');
+var parseArgs = require('./lib/parse_args');
+var net = require('net');
+var util = require('util');
+var weak
+
+exports = module.exports = function (cons, opts) {
+    return new D(cons, opts);
+};
+
+exports.connect = function () {
+    var d = new D();
+    return d.connect.apply(d, arguments);
+};
+
+exports.listen = function () {
+    var d = new D();
+    return d.listen.apply(d, arguments);
+};
+
+util.inherits(D, dnode);
+function D (cons, opts) {
+    var self = this;
+    if (!opts) opts = {};
+    
+    if (opts.weak !== false && !opts.proto) {
+        if (!weak) {
+            weak = require("weak")
+        }
+        
+        opts.proto = {};
+        opts.proto.wrap = function (cb, id) {
+            var proto = this;
+            return weak(cb, function () {
+                proto.cull(id);
+            });
+        };
+        opts.proto.unwrap = function (ref, id) {
+            var cb = weak.get(ref);
+            return cb || function () {};
+        };
+    }
+    return dnode.call(self, cons, opts);
+}
+
+D.prototype.connect = function () {
+    var self = this;  
+    var params = parseArgs(arguments);
+    
+    var stream;
+    if (params.path) {
+        stream = net.connect(params.path);
+    }
+    else if (params.port) {
+        stream = net.connect(params.port, params.host);
+    }
+    else {
+        throw new Error('no port or unix path given');
+    }
+    
+    if (params.block) self.on('remote', params.block);
+    
+    stream.on('error', function (err) {
+        if (err && err.code === 'EPIPE') return; // eat EPIPEs
+        self.emit('error', err);
+    });
+    
+    self.id = randomId();
+    self.stream = stream;
+    stream.pipe(self);
+    self.pipe(stream);
+    
+    return self;
+};
+
+dnode.prototype.listen = function () {
+    var self = this;
+    
+    // just copy over the opts and cons, the rest will need to be re-created
+    var cons = self.cons, opts = self.opts;
+    self.cons = function () {};
+    self.end();
+    
+    var params = parseArgs(arguments);
+    
+    var server = net.createServer(function (stream) {
+        var d = new dnode(cons, opts);
+        do { d.id = randomId() }
+        while (server.sessions[d.id]);
+        
+        server.sessions[d.id] = d;
+        d.on('end', function () {
+            delete server.sessions[d.id];
+        });
+        
+        d.on('local', function (ref) {
+            server.emit('local', ref, d);
+        });
+        
+        d.on('remote', function (remote) {
+            server.emit('remote', remote, d);
+        });
+        
+        stream.on('error', function (err) {
+            if (err && err.code === 'EPIPE') return; // eat EPIPEs
+            d.emit('error', err);
+        });
+        
+        d.stream = stream;
+        stream.pipe(d);
+        d.pipe(stream);
+    });
+    
+    server.sessions = {};
+    
+    if (typeof params.port === 'number' && params.port >= 0) {
+        server.listen(params.port, params.host);
+    }
+    else if (params.path) {
+        server.listen(params.path);
+    }
+    else {
+        throw new Error('no port or path provided');
+    }
+    
+    if (params.block) server.on('listening', params.block);
+    
+    return server;
+};
+
+function randomId () {
+    var s = '';
+    for (var i = 0; i < 4; i++) {
+        s += Math.random().toString(16).slice(2);
+    }
+    return s;
+}