/**
 * Copyright 2013 IBM Corp.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 **/

var RED = require(process.env.NODE_RED_HOME+"/red/red");
var reconnectTime = RED.settings.socketReconnectTime||10000;
var socketTimeout = RED.settings.socketTimeout||null;
var net = require('net');

var connectionPool = {};

function TcpIn(n) {
    RED.nodes.createNode(this,n);
    this.host = n.host;
    this.port = n.port * 1;
    this.topic = n.topic;
    this.stream = (!n.datamode||n.datamode=='stream'); /* stream,single*/
    this.datatype = n.datatype||'buffer'; /* buffer,utf8,base64 */
    this.newline = (n.newline||"").replace("\\n","\n").replace("\\r","\r");
    this.base64 = n.base64;
    this.server = (typeof n.server == 'boolean')?n.server:(n.server == "server");
    this.closing = false;
    var node = this;

    if (!node.server) {
        var buffer = null;
        var client;
        var reconnectTimeout;
        function setupTcpClient() {
            node.log("connecting to "+node.host+":"+node.port);
            var id = (1+Math.random()*4294967295).toString(16);
            client = net.connect(node.port, node.host, function() {
                buffer = (node.datatype == 'buffer')? new Buffer(0):"";
                node.log("connected to "+node.host+":"+node.port);
            });
            connectionPool[id] = client;

            client.on('data', function (data) {
                if (node.datatype != 'buffer') {
                    data = data.toString(node.datatype);
                }
                if (node.stream) {
                    if ((node.datatype) === "utf8" && node.newline != "") {
                        buffer = buffer+data;
                        var parts = buffer.split(node.newline);
                        for (var i = 0;i<parts.length-1;i+=1) {
                            var msg = {topic:node.topic, payload:parts[i]};
                            msg._session = {type:"tcp",id:id};
                            node.send(msg);
                        }
                        buffer = parts[parts.length-1];
                    } else {
                        var msg = {topic:node.topic, payload:data};
                        msg._session = {type:"tcp",id:id};
                        node.send(msg);
                    }
                } else {
                    if ((typeof data) === "string") {
                        buffer = buffer+data;
                    } else {
                        buffer = Buffer.concat([buffer,data],buffer.length+data.length);
                    }
                }
            });
            client.on('end', function() {
                if (!node.stream || (node.datatype == "utf8" && node.newline != "" && buffer.length > 0)) {
                    var msg = {topic:node.topic,payload:buffer};
                    msg._session = {type:"tcp",id:id};
                    node.send(msg);
                    buffer = null;
                }
            });
            client.on('close', function() {
                delete connectionPool[id];
                node.log("connection lost to "+node.host+":"+node.port);
                if (!node.closing) {
                    reconnectTimeout = setTimeout(setupTcpClient, reconnectTime);
                }
            });
            client.on('error', function(err) {
                    node.log(err);
            });
        }
        setupTcpClient();

        this.on('close', function() {
            this.closing = true;
            client.end();
            clearTimeout(reconnectTimeout);
        });
    } else {
        var server = net.createServer(function (socket) {
            if (socketTimeout !== null) { socket.setTimeout(socketTimeout); }
            var id = (1+Math.random()*4294967295).toString(16);
            connectionPool[id] = socket;

            var buffer = (node.datatype == 'buffer')? new Buffer(0):"";
            socket.on('data', function (data) {
                if (node.datatype != 'buffer') {
                    data = data.toString(node.datatype);
                }

                if (node.stream) {
                    if ((typeof data) === "string" && node.newline != "") {
                        buffer = buffer+data;
                        var parts = buffer.split(node.newline);
                        for (var i = 0;i<parts.length-1;i+=1) {
                            var msg = {topic:node.topic, payload:parts[i],ip:socket.remoteAddress,port:socket.remotePort};
                            msg._session = {type:"tcp",id:id};
                            node.send(msg);
                        }
                        buffer = parts[parts.length-1];
                    } else {
                        var msg = {topic:node.topic, payload:data};
                        msg._session = {type:"tcp",id:id};
                        node.send(msg);
                    }
                } else {
                    if ((typeof data) === "string") {
                        buffer = buffer+data;
                    } else {
                        buffer = Buffer.concat([buffer,data],buffer.length+data.length);
                    }
                }
            });
            socket.on('end', function() {
                if (!node.stream || (node.datatype == "utf8" && node.newline != "" && buffer.length > 0)) {
                    var msg = {topic:node.topic,payload:buffer};
                    msg._session = {type:"tcp",id:id};
                    node.send(msg);
                    buffer = null;
                }
            });
            socket.on('timeout', function() {
                node.log('timeout closed socket port '+node.port);
                socket.end();
            });
            socket.on('close', function() {
                delete connectionPool[id];
            });
            socket.on('error',function(err) {
                node.log(err);
            });
        });
        server.listen(node.port);
        node.log('listening on port '+node.port);

        this.on('close', function() {
            this.closing = true;
            server.close();
            node.log('stopped listening on port '+node.port);
        });
    }

}
RED.nodes.registerType("tcp in",TcpIn);

function TcpOut(n) {
    RED.nodes.createNode(this,n);
    this.host = n.host;
    this.port = n.port * 1;
    this.base64 = n.base64;
    this.beserver = n.beserver;
    this.name = n.name;
    this.closing = false;
    var node = this;

    if (!node.beserver||node.beserver=="client") {
        var reconnectTimeout;
        var client = null;
        var connected = false;

        function setupTcpClient() {
            node.log("connecting to "+node.host+":"+node.port);
            client = net.connect(node.port, node.host, function() {
                connected = true;
                node.log("connected to "+node.host+":"+node.port);
            });
            client.on('error', function (err) {
                node.log('error : '+err);
            });
            client.on('end', function (err) {
            });
            client.on('close', function() {
                node.log("connection lost to "+node.host+":"+node.port);
                connected = false;
                client.destroy();
                if (!node.closing) {
                    reconnectTimeout = setTimeout(setupTcpClient,reconnectTime);
                }
            });
        }
        setupTcpClient();

        node.on("input", function(msg) {
            if (connected && msg.payload != null) {
                if (Buffer.isBuffer(msg.payload)) {
                    client.write(msg.payload);
                } else if (typeof msg.payload === "string" && node.base64) {
                    client.write(new Buffer(msg.payload,'base64'));
                } else {
                    client.write(new Buffer(""+msg.payload));
                }
            }
        });

        node.on("close", function() {
            this.closing = true;
            client.end();
            clearTimeout(reconnectTimeout);
        });

    } else if (node.beserver == "reply") {
        node.on("input",function(msg) {
            if (msg._session && msg._session.type == "tcp") {
                var client = connectionPool[msg._session.id];
                if (client) {
                    if (Buffer.isBuffer(msg.payload)) {
                        client.write(msg.payload);
                    } else if (typeof msg.payload === "string" && node.base64) {
                        client.write(new Buffer(msg.payload,'base64'));
                    } else {
                        client.write(new Buffer(""+msg.payload));
                    }
                }
            }
        });
    } else {
        var connectedSockets = [];
        var server = net.createServer(function (socket) {
            if (socketTimeout !== null) { socket.setTimeout(socketTimeout); }
            var remoteDetails = socket.remoteAddress+":"+socket.remotePort;
            node.log("connection from "+remoteDetails);
            connectedSockets.push(socket);
            socket.on('timeout', function() {
                node.log('timeout closed socket port '+node.port);
                socket.end();
            });
            socket.on('close',function() {
                node.log("connection closed from "+remoteDetails);
                connectedSockets.splice(connectedSockets.indexOf(socket),1);
            });
            socket.on('error',function() {
                node.log("socket error from "+remoteDetails);
                connectedSockets.splice(connectedSockets.indexOf(socket),1);
            });
        });
        node.on("input", function(msg) {
            if (msg.payload != null) {
                var buffer;
                if (Buffer.isBuffer(msg.payload)) {
                    buffer = msg.payload;
                } else if (typeof msg.payload === "string" && node.base64) {
                    buffer = new Buffer(msg.payload,'base64');
                } else {
                    buffer = new Buffer(""+msg.payload);
                }
                for (var i = 0; i<connectedSockets.length;i+=1) {
                    connectedSockets[i].write(buffer);
                }
            }
        });

        server.listen(node.port);
        node.log('listening on port '+node.port);

        node.on('close', function() {
            server.close();
            node.log('stopped listening on port '+node.port);
        });
    }
}

RED.nodes.registerType("tcp out",TcpOut);