/*
 * Copyright 2016 Red Hat Inc.
 *
 * 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.
 */
'use strict';

var log = require("./log.js").logger();
var util = require('util');

var MAX_OUTSTANDING = process.env.MAX_OUTSTANDING_QDR_REQUESTS || 250;

var Router = function (connection, router, agent) {
    if (router) {
        this.target = agent;
        this.connection = router.connection;
        this.counter = router.counter;
        this.handlers = router.handlers;
        this.requests = router.requests;
        this.address = router.address;
        this.sender = router.sender;
        this.tracking = router.tracking;
    } else {
        this.target = '$management';
        this.sender = connection.open_sender();
        this.sender.on('sendable', this._send_pending_requests.bind(this));
        this.connection = connection;
        this.counter = 0;
        this.handlers = {};
        this.requests = [];
        connection.open_receiver({source:{dynamic:true}});
        connection.on('message', this.incoming.bind(this));
        connection.on('connection_close', this.closed.bind(this));
        this.tracking = {
            sent: 0,
            recv: 0,
            outstanding: 0,
            unexpected_responses: 0
        };
        this.last_status = undefined;
        var self = this;
        var interval = process.env.STATUS_LOG_INTERVAL || 300000;
        this.timer = setInterval(function () {
            self.health_check();
        }, interval);
    }
    this.connection.on('receiver_open', this.ready.bind(this));
    this.connection.on('disconnected', this.disconnected.bind(this));
    this.connection.on('sender_error', this.on_sender_error.bind(this));
};

Router.prototype.get_id = function () {
    if (this.name) {
        return this.name + ':' + this.connection.container_id;
    } else {
        return this.connection.container_id;
    }
};

Router.prototype.log_info = function () {
    log.info('[%s] qdr handlers: %d, requests pending: %d, requests sent: %d, responses received: %d, unexpected response: %d, ready: %s',
             this.get_id(), Object.keys(this.handlers).length, this.requests.length,
             this.tracking.sent, this.tracking.recv, this.tracking.unexpected_responses, (this.address !== undefined));
};

Router.prototype.health_check = function () {
    var current = this._get_status();
    var last = this.last_status;
    if (this.last_status === undefined) {
        this.last_status = current;
    } else {
        var ready_state;
        if (current.ready) {
            if (last.ready) {
                ready_state = 'remains active';
            } else {
                ready_state = 'transitioned to active';
            }
        } else {
            if (last.ready) {
                ready_state = 'transitioned to inactive';
            } else {
                ready_state = 'remains inactive';
            }
        }

        var messages = [];
        var still_outstanding = current.outstanding.filter(function (id) { return last.outstanding.indexOf(id) >=0; });
        if (still_outstanding.length > 0) {
            messages.push(util.format('still waiting for responses for %j', still_outstanding));
        }

        if (current.unsent > last.unsent && current.sent === last.sent) {
            messages.push('sending of requests appears to be blocked');
        }

        if (current.unexpected_responses > last.unexpected_responses) {
            messages.push(util.format('%d unexpected responses since last status', current.unexpected_responses - last.unexpected_responses));
        }

        if (messages.length) {
            log.warn('[%s] %s (%s)', this.get_id(), messages.join(', '), ready_state);
        } else if (current.sent !== last.sent || current.recv != last.recv) {
            log.info('[%s] %s sent %d requests and received %d responses since last status; %d outstanding requests issued', this.get_id(),
                     ready_state, current.sent - last.sent, current.recv - last.recv, current.outstanding);
        } else {
            log.info('[%s] %s no requests/responses since last status', this.get_id(), ready_state);
        }
        last = current;
    }
};

Router.prototype._get_status = function () {
    return  {
        sent: this.tracking.sent,
        recv: this.tracking.recv,
        unexpected_responses: this.tracking.unexpected_responses,
        unsent: this.requests.length,
        outstanding: Object.keys(this.handlers),
        ready: (this.address !== undefined)
    };
};

Router.prototype.closed = function (context) {
    if (context.connection.error) {
        log.error('[%s] ERROR: router closed connection with %s', this.get_id(), context.connection.error.description);
    }
    log.info('[%s] router closed ', this.get_id(), this.target);
    this.address = undefined;
    this._abort_requests('closed');
};

Router.prototype._abort_requests = function (error) {
    log.info('[%s] aborting pending requests: %s', this.get_id(), error);
    for (var h in this.handlers) {
        this.handlers[h](error);
        delete this.handlers[h];
    }
    while (this.requests.length > 0) { this.requests.shift(); };
}

Router.prototype.on_sender_error = function (context) {
    log.info('[%s] sender error %s', this.get_id(), error);
};

Router.prototype.disconnected = function (context) {
    this.address = undefined;
    this._abort_requests('disconnected');
}

Router.prototype.ready = function (context) {
    log.info('[%s] router ready', this.get_id());
    this.address = context.receiver.source.address;
    this._send_pending_requests();
};

function create_record(names, values) {
    var record = {};
    for (var j = 0; j < names.length; j++) {
        record[names[j]] = values[j];
    }
    return record;
}

function extract_records(body) {
    return body.results ? body.results.map(create_record.bind(null, body.attributeNames)) : body;
}

function as_handler(resolve, reject) {
    return function (context) {
        var message = context.message === undefined ? context : context.message;
        if (message.application_properties) {
            if (message.application_properties.statusCode >= 200 && message.application_properties.statusCode < 300) {
                if (message.body) resolve(extract_records(message.body));
                else resolve({code:message.application_properties.statusCode, description:message.application_properties.statusDescription});
            } else {
                reject({code:message.application_properties.statusCode, description:message.application_properties.statusDescription});
            }
        } else {
            reject(message.toString());
        }
    };
}

Router.prototype._sendable = function () {
    return this.sender.sendable() && this.tracking.outstanding < MAX_OUTSTANDING;
}

Router.prototype._send_pending_requests = function () {
    if (this.address === undefined) return false;

    var i = 0;
    while (i < this.requests.length && this._sendable()) {
        this._send_request(this.requests[i++]);
    }
    this.requests.splice(0, i);
    return this.requests.length === 0 && this._sendable();
}

Router.prototype._send_request = function (request) {
    request.reply_to = this.address;
    this.sender.send(request);
    this.tracking.sent++;
    this.tracking.outstanding++;
    log.debug('sent: %j', request);
}

Router.prototype.request = function (operation, properties, body) {
    var id = this.target + this.counter.toString();
    this.counter++;
    var req = {correlation_id:id, to:this.target};
    req.application_properties = properties || {};
    req.application_properties.operation = operation;
    req.body = body;

    if (this._send_pending_requests()) {
        this._send_request(req);
    } else {
        this.requests.push(req);
    }
    var handlers = this.handlers;
    return new Promise(function (resolve, reject) {
        handlers[id] = as_handler(resolve, reject);
    });
}

Router.prototype.query = function (type, options) {
    return this.request('QUERY', {entityType:type}, options || {attributeNames:[]});
};

Router.prototype.create_entity = function (type, name, attributes) {
    return this.request('CREATE', {'type':type, 'name':name}, attributes || {});
};

Router.prototype.delete_entity = function (type, name) {
    return this.request('DELETE', {'type':type, 'name':name}, {});
};

Router.prototype.get_mgmt_nodes = function () {
    return this.request('GET-MGMT-NODES', {}, {});
};

Router.prototype._get_all_routers = function () {
    var self = this;
    return this.get_mgmt_nodes().then(
        function (agents) {
            return agents.map(function (agent) { return new Router(undefined, self, agent); });
        }
    );
};

Router.prototype.get_all_routers = function (current) {
    if (current === undefined || current.length === 0) {
        return this._get_all_routers();
    } else {
        var self = this;
        return this.get_mgmt_nodes().then(
            function (results) {
                var agents = {};
                results.forEach(function (name) { agents[name] = true; });
                var routers = [];
                for (var i = 0; i < current.length; i++) {
                    if (agents[current[i].target] !== undefined) {
                        delete agents[current[i].target];
                        routers.push(current[i]);
                    }
                }
                return routers.concat( Object.keys(agents).map(function (agent) { return new Router(undefined, self, agent); }) );
            }
        );
    }
};

Router.prototype.incoming = function (context) {
    log.debug('recv: %j', context.message);
    this.tracking.recv++
    this.tracking.outstanding--;
    var message = context.message;
    var handler = this.handlers[message.correlation_id];
    if (handler) {
        delete this.handlers[message.correlation_id];
        handler(context);
    } else {
        this.tracking.unexpected_responses++;
        log.warn('WARNING: unexpected response: ' + message.correlation_id + ' [' + JSON.stringify(message) + ']');
    }
    this._send_pending_requests();
};

Router.prototype.close = function () {
    if (this.connection) this.connection.close();
    if (this.timer) clearInterval(this.timer);
}

function add_resource_type (name, typename, plural) {
    var resource_type = typename || name;
    Router.prototype['create_' + name] = function (o) {
        return this.create_entity(resource_type, o.name, o);
    };
    Router.prototype['delete_' + name] = function (o) {
        return this.delete_entity(resource_type, o.name);
    };
    var plural_name = plural || name + 's';
    Router.prototype['get_' + plural_name] = function (options) {
        return this.query(resource_type, options);
    };

}

function add_queryable_type (name, typename) {
    var resource_type = typename || name;
    Router.prototype['get_' + name] = function (options) {
        return this.query(resource_type, options);
    };
}

add_resource_type('connector');
add_resource_type('listener');
add_resource_type('address', 'org.apache.qpid.dispatch.router.config.address', 'addresses');
add_resource_type('link_route', 'org.apache.qpid.dispatch.router.config.linkRoute');

add_queryable_type('connections', 'org.apache.qpid.dispatch.connection');
add_queryable_type('links', 'org.apache.qpid.dispatch.router.link');
add_queryable_type('address_stats', 'org.apache.qpid.dispatch.router.address');

var amqp = require('rhea').create_container();
module.exports.Router = Router;
module.exports.connect = function (options) {
    return new Router(amqp.connect(options));
}
