Synnefo

// Copyright 2011 GRNET S.A. All rights reserved.

// 

// Redistribution and use in source and binary forms, with or

// without modification, are permitted provided that the following

// conditions are met:

// 

//   1. Redistributions of source code must retain the above

//      copyright notice, this list of conditions and the following

//      disclaimer.

// 

//   2. Redistributions in binary form must reproduce the above

//      copyright notice, this list of conditions and the following

//      disclaimer in the documentation and/or other materials

//      provided with the distribution.

// 

// THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

// OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

// WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

// USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

// AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

// LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

// ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

// POSSIBILITY OF SUCH DAMAGE.

// 

// The views and conclusions contained in the software and

// documentation are those of the authors and should not be

// interpreted as representing official policies, either expressed

// or implied, of GRNET S.A.

// 

;(function(root){

    // root

    var root = root;

    // setup namepsaces

    var snf = root.synnefo = root.synnefo || {};

    var models = snf.models = snf.models || {}

    var storage = snf.storage = snf.storage || {};

    var util = snf.util = snf.util || {};

    // shortcuts

    var bb = root.Backbone;

    var slice = Array.prototype.slice

    // logging

    var logger = new snf.logging.logger("SNF-MODELS");

    var debug = _.bind(logger.debug, logger);

    // get url helper

    var getUrl = function(baseurl) {

        var baseurl = baseurl || snf.config.api_urls[this.api_type];

        return baseurl + "/" + this.path;

    }

    // i18n

    BUILDING_MESSAGES = window.BUILDING_MESSAGES || {'INIT': 'init', 'COPY': '{0}, {1}, {2}', 'FINAL': 'final'};

    // Base object for all our models

    models.Model = bb.Model.extend({

        sync: snf.api.sync,

        api: snf.api,

        api_type: 'compute',

        has_status: false,

        initialize: function() {

            if (this.has_status) {

                this.bind("change:status", this.handle_remove);

                this.handle_remove();

            }

            this.api_call = _.bind(this.api.call, this);

            models.Model.__super__.initialize.apply(this, arguments);

        },

        handle_remove: function() {

            if (this.get("status") == 'DELETED') {

                if (this.collection) {

                    try { this.clear_pending_action();} catch (err) {};

                    try { this.reset_pending_actions();} catch (err) {};

                    try { this.stop_stats_update();} catch (err) {};

                    this.collection.remove(this.id);

                }

            }

        },

        // custom set method to allow submodels to use

        // set_<attr> methods for handling the value of each

        // attribute and overriding the default set method

        // for specific parameters

        set: function(params, options) {

            _.each(params, _.bind(function(value, key){

                if (this["set_" + key]) {

                    params[key] = this["set_" + key](value);

                }

            }, this))

            var ret = bb.Model.prototype.set.call(this, params, options);

            return ret;

        },

        url: function(options) {

            return getUrl.call(this, this.base_url) + "/" + this.id;

        },

        api_path: function(options) {

            return this.path + "/" + this.id;

        },

        parse: function(resp, xhr) {

            return resp.server;

        },

        remove: function() {

            this.api_call(this.api_path(), "delete");

        },

        changedKeys: function() {

            return _.keys(this.changedAttributes() || {});

        },

        hasOnlyChange: function(keys) {

            var ret = false;

            _.each(keys, _.bind(function(key) {

                if (this.changedKeys().length == 1 && this.changedKeys().indexOf(key) > -1) { ret = true};

            }, this));

            return ret;

        }

    })

    // Base object for all our model collections

    models.Collection = bb.Collection.extend({

        sync: snf.api.sync,

        api: snf.api,

        api_type: 'compute',

        supportIncUpdates: true,

        initialize: function() {

            models.Collection.__super__.initialize.apply(this, arguments);

            this.api_call = _.bind(this.api.call, this);

        },

        url: function(options, method) {

            return getUrl.call(this, this.base_url) + (

                    options.details || this.details && method != 'create' ? '/detail' : '');

        },

        fetch: function(options) {

            if (!options) { options = {} };

            // default to update

            if (!this.noUpdate) {

                if (options.update === undefined) { options.update = true };

                if (!options.removeMissing && options.refresh) { options.removeMissing = true };

            } else {

                if (options.refresh === undefined) {

                    options.refresh = true;

                }

            }

            // custom event foreach fetch

            return bb.Collection.prototype.fetch.call(this, options)

        },

        create: function(model, options) {

            var coll = this;

            options || (options = {});

            model = this._prepareModel(model, options);

            if (!model) return false;

            var success = options.success;

            options.success = function(nextModel, resp, xhr) {

                if (success) success(nextModel, resp, xhr);

            };

            model.save(null, options);

            return model;

        },

        get_fetcher: function(interval, increase, fast, increase_after_calls, max, initial_call, params) {

            var fetch_params = params || {};

            var handler_options = {};

            fetch_params.skips_timeouts = true;

            handler_options.interval = interval;

            handler_options.increase = increase;

            handler_options.fast = fast;

            handler_options.increase_after_calls = increase_after_calls;

            handler_options.max= max;

            handler_options.id = "collection id";

            var last_ajax = undefined;

            var callback = _.bind(function() {

                // clone to avoid referenced objects

                var params = _.clone(fetch_params);

                updater._ajax = last_ajax;

                // wait for previous request to finish

                if (last_ajax && last_ajax.readyState < 4 && last_ajax.statusText != "timeout") {

                    // opera readystate for 304 responses is 0

                    if (!($.browser.opera && last_ajax.readyState == 0 && last_ajax.status == 304)) {

                        return;

                    }

                }

                last_ajax = this.fetch(params);

            }, this);

            handler_options.callback = callback;

            var updater = new snf.api.updateHandler(_.clone(_.extend(handler_options, fetch_params)));

            snf.api.bind("call", _.throttle(_.bind(function(){ updater.faster(true)}, this)), 1000);

            return updater;

        }

    });

    // Image model

    models.Image = models.Model.extend({

        path: 'images',

        get_size: function() {

            return parseInt(this.get('metadata') ? this.get('metadata').values.size : -1)

        },

        get_meta: function(key) {

            if (this.get('metadata') && this.get('metadata').values && this.get('metadata').values[key]) {

                return _.escape(this.get('metadata').values[key]);

            }

            return undefined;

        },

        get_owner: function() {

            return this.get('owner') || _.keys(synnefo.config.system_images_owners)[0];

        },

        display_owner: function() {

            var owner = this.get_owner();

            if (_.include(_.keys(synnefo.config.system_images_owners), owner)) {

                return synnefo.config.system_images_owners[owner];

            } else {

                return owner;

            }

        },

        get_readable_size: function() {

            return this.get_size() > 0 ? util.readablizeBytes(this.get_size() * 1024 * 1024) : "unknown";

        },

        get_os: function() {

            return this.get("OS");

        },

        get_created_user: function() {

            return synnefo.config.os_created_users[this.get_os()] || "root";

        },

        get_sort_order: function() {

            return parseInt(this.get('metadata') ? this.get('metadata').values.sortorder : -1)

        },

        get_vm: function() {

            var vm_id = this.get("serverRef");

            var vm = undefined;

            vm = storage.vms.get(vm_id);

            return vm;

        },

        is_public: function() {

            return this.get('is_public') || true;

        },

        ssh_keys_path: function() {

            prepend = '';

            if (this.get_created_user() != 'root') {

                prepend = '/home'

            }

            return '{1}/{0}/.ssh/authorized_keys'.format(this.get_created_user(), prepend);

        },

        _supports_ssh: function() {

            if (synnefo.config.support_ssh_os_list.indexOf(this.get_os()) > -1) {

                return true;

            }

            return false;

        },

        supports: function(feature) {

            if (feature == "ssh") {

                return this._supports_ssh()

            }

            return false;

        },

        personality_data_for_keys: function(keys) {

            contents = '';

            _.each(keys, function(key){

                contents = contents + key.get("content") + "\n"

            });

            contents = $.base64.encode(contents);

            return {

                path: this.ssh_keys_path(),

                contents: contents

            }

        }

    });

    // Flavor model

    models.Flavor = models.Model.extend({

        path: 'flavors',

        details_string: function() {

            return "{0} CPU, {1}MB, {2}GB".format(this.get('cpu'), this.get('ram'), this.get('disk'));

        },

        get_disk_size: function() {

            return parseInt(this.get("disk") * 1000)

        },

        get_disk_template_info: function() {

            var info = snf.config.flavors_disk_templates_info[this.get("disk_template")];

            if (!info) {

                info = { name: this.get("disk_template"), description:'' };

            }

            return info

        }

    });

    //network vms list helper

    var NetworkVMSList = function() {

        this.initialize = function() {

            this.vms = [];

            this.pending = [];

            this.pending_for_removal = [];

        }

        this.add_pending_for_remove = function(vm_id) {

            if (this.pending_for_removal.indexOf(vm_id) == -1) {

                this.pending_for_removal.push(vm_id);

            }

            if (this.pending_for_removal.length) {

                this.trigger("pending:remove:add");

            }

        },

        this.add_pending = function(vm_id) {

            if (this.pending.indexOf(vm_id) == -1) {

                this.pending[this.pending.length] = vm_id;

            }

            if (this.pending.length) {

                this.trigger("pending:add");

            }

        }

        this.check_pending = function() {

            var len = this.pending.length;

            var args = [this.pending];

            this.pending = _.difference(this.pending, this.vms);

            if (len != this.pending.length) {

                if (this.pending.length == 0) {

                    this.trigger("pending:clear");

                }

            }

            var len = this.pending_for_removal.length;

            this.pending_for_removal = _.intersection(this.pending_for_removal, this.vms);

            if (this.pending_for_removal.length == 0) {

                this.trigger("pending:remove:clear");

            }

        }

        this.add = function(vm_id) {

            if (this.vms.indexOf(vm_id) == -1) {

                this.vms[this.vms.length] = vm_id;

                this.trigger("network:connect", vm_id);

                this.check_pending();

                return true;

            }

        }

        this.remove = function(vm_id) {

            if (this.vms.indexOf(vm_id) > -1) {

                this.vms = _.without(this.vms, vm_id);

                this.trigger("network:disconnect", vm_id);

                this.check_pending();

                return true;

            }

        }

        this.get = function() {

            return this.vms;

        }

        this.list = function() {

            return storage.vms.filter(_.bind(function(vm){

                return this.vms.indexOf(vm.id) > -1;

            }, this))

        }

        this.initialize();

    };

    _.extend(NetworkVMSList.prototype, bb.Events);

    // vm networks list helper

    var VMNetworksList = function() {

        this.initialize = function() {

            this.networks = {};

            this.network_ids = [];

        }

        this.add = function(net_id, data) {

            if (!this.networks[net_id]) {

                this.networks[net_id] = data || {};

                this.network_ids[this.network_ids.length] = net_id;

                this.trigger("network:connect", net_id);

                return true;

            }

        }

        this.remove = function(net_id) {

            if (this.networks[net_id]) {

                delete this.networks[net_id];

                this.network_ids = _.without(this.network_ids, net_id);

                this.trigger("network:disconnect", net_id);

                return true;

            }

            return false;

        }

        this.get = function() {

            return this.networks;

        }

        this.list = function() {

            return storage.networks.filter(_.bind(function(net){

                return this.network_ids.indexOf(net.id) > -1;

            }, this))

        }

        this.initialize();

    };

    _.extend(VMNetworksList.prototype, bb.Events);

    models.ParamsList = function(){this.initialize.apply(this, arguments)};

    _.extend(models.ParamsList.prototype, bb.Events, {

        initialize: function(parent, param_name) {

            this.parent = parent;

            this.actions = {};

            this.param_name = param_name;

            this.length = 0;

        },

        has_action: function(action) {

            return this.actions[action] ? true : false;

        },

        _parse_params: function(arguments) {

            if (arguments.length <= 1) {

                return [];

            }

            var args = _.toArray(arguments);

            return args.splice(1);

        },

        contains: function(action, params) {

            params = this._parse_params(arguments);

            var has_action = this.has_action(action);

            if (!has_action) { return false };

            var paramsEqual = false;

            _.each(this.actions[action], function(action_params) {

                if (_.isEqual(action_params, params)) {

                    paramsEqual = true;

                }

            });

            return paramsEqual;

        },

        is_empty: function() {

            return _.isEmpty(this.actions);

        },

        add: function(action, params) {

            params = this._parse_params(arguments);

            if (this.contains.apply(this, arguments)) { return this };

            var isnew = false

            if (!this.has_action(action)) {

                this.actions[action] = [];

                isnew = true;

            };

            this.actions[action].push(params);

            this.parent.trigger("change:" + this.param_name, this.parent, this);

            if (isnew) {

                this.trigger("add", action, params);

            } else {

                this.trigger("change", action, params);

            }

            return this;

        },

        remove_all: function(action) {

            if (this.has_action(action)) {

                delete this.actions[action];

                this.parent.trigger("change:" + this.param_name, this.parent, this);

                this.trigger("remove", action);

            }

            return this;

        },

        reset: function() {

            this.actions = {};

            this.parent.trigger("change:" + this.param_name, this.parent, this);

            this.trigger("reset");

            this.trigger("remove");

        },

        remove: function(action, params) {

            params = this._parse_params(arguments);

            if (!this.has_action(action)) { return this };

            var index = -1;

            _.each(this.actions[action], _.bind(function(action_params) {

                if (_.isEqual(action_params, params)) {

                    index = this.actions[action].indexOf(action_params);

                }

            }, this));

            if (index > -1) {

                this.actions[action].splice(index, 1);

                if (_.isEmpty(this.actions[action])) {

                    delete this.actions[action];

                }

                this.parent.trigger("change:" + this.param_name, this.parent, this);

                this.trigger("remove", action, params);

            }

        }

    });

    // Image model

    models.Network = models.Model.extend({

        path: 'networks',

        has_status: true,

        initialize: function() {

            this.vms = new NetworkVMSList();

            this.vms.bind("pending:add", _.bind(this.handle_pending_connections, this, "add"));

            this.vms.bind("pending:clear", _.bind(this.handle_pending_connections, this, "clear"));

            this.vms.bind("pending:remove:add", _.bind(this.handle_pending_connections, this, "add"));

            this.vms.bind("pending:remove:clear", _.bind(this.handle_pending_connections, this, "clear"));

            var ret = models.Network.__super__.initialize.apply(this, arguments);

            storage.vms.bind("change:linked_to_nets", _.bind(this.update_connections, this, "vm:change"));

            storage.vms.bind("add", _.bind(this.update_connections, this, "add"));

            storage.vms.bind("remove", _.bind(this.update_connections, this, "remove"));

            storage.vms.bind("reset", _.bind(this.update_connections, this, "reset"));

            this.bind("change:linked_to", _.bind(this.update_connections, this, "net:change"));

            this.update_connections();

            this.update_state();

            this.set({"actions": new models.ParamsList(this, "actions")});

            return ret;

        },

        toJSON: function() {

            var attrs = _.clone(this.attributes);

            attrs.actions = _.clone(this.get("actions").actions);

            return attrs;

        },

        update_state: function() {

            if (this.vms.pending.length) {

                this.set({state: "CONNECTING"});

                return

            }

            if (this.vms.pending_for_removal.length) {

                this.set({state: "DISCONNECTING"});

                return

            }   

            var firewalling = false;

            _.each(this.vms.get(), _.bind(function(vm_id){

                var vm = storage.vms.get(vm_id);

                if (!vm) { return };

                if (!_.isEmpty(vm.pending_firewalls)) {

                    this.set({state:"FIREWALLING"});

                    firewalling = true;

                    return false;

                }

            },this));

            if (firewalling) { return };

            this.set({state:"NORMAL"});

        },

        handle_pending_connections: function(action) {

            this.update_state();

        },

        // handle vm/network connections

        update_connections: function(action, model) {

            // vm removed disconnect vm from network

            if (action == "remove") {

                var removed_from_net = this.vms.remove(model.id);

                var removed_from_vm = model.networks.remove(this.id);

                if (removed_from_net) {this.trigger("vm:disconnect", model, this); this.change()};

                if (removed_from_vm) {model.trigger("network:disconnect", this, model); this.change()};

                return;

            }

            // update links for all vms

            var links = this.get("linked_to");

            storage.vms.each(_.bind(function(vm) {

                var vm_links = vm.get("linked_to") || [];

                if (vm_links.indexOf(this.id) > -1) {

                    // vm has connection to current network

                    if (links.indexOf(vm.id) > -1) {

                        // and network has connection to vm, so try

                        // to append it

                        var add_to_net = this.vms.add(vm.id);

                        var index = _.indexOf(vm_links, this.id);

                        var add_to_vm = vm.networks.add(this.id, vm.get("linked_to_nets")[index]);

                        // call only if connection did not existed

                        if (add_to_net) {this.trigger("vm:connect", vm, this); this.change()};

                        if (add_to_vm) {vm.trigger("network:connect", this, vm); vm.change()};

                    } else {

                        // no connection, try to remove it

                        var removed_from_net = this.vms.remove(vm.id);

                        var removed_from_vm = vm.networks.remove(this.id);

                        if (removed_from_net) {this.trigger("vm:disconnect", vm, this); this.change()};

                        if (removed_from_vm) {vm.trigger("network:disconnect", this, vm); vm.change()};

                    }

                } else {

                    // vm has no connection to current network, try to remove it

                    var removed_from_net = this.vms.remove(vm.id);

                    var removed_from_vm = vm.networks.remove(this.id);

                    if (removed_from_net) {this.trigger("vm:disconnect", vm, this); this.change()};

                    if (removed_from_vm) {vm.trigger("network:disconnect", this, vm); vm.change()};

                }

            },this));

        },

        is_public: function() {

            return this.id == "public";

        },

        contains_vm: function(vm) {

            var net_vm_exists = this.vms.get().indexOf(vm.id) > -1;

            var vm_net_exists = vm.is_connected_to(this);

            return net_vm_exists && vm_net_exists;

        },

        call: function(action, params, success, error) {

            if (action == "destroy") {

                this.set({state:"DESTROY"});

                this.get("actions").remove("destroy");

                this.remove(_.bind(function(){

                    success();

                }, this), error);

            }

            if (action == "disconnect") {

                _.each(params, _.bind(function(vm_id) {

                    var vm = snf.storage.vms.get(vm_id);

                    this.get("actions").remove("disconnect", vm_id);

                    if (vm) {

                        this.remove_vm(vm, success, error);

                    }

                }, this));

            }

        },

        add_vm: function (vm, callback, error, options) {

            var payload = {add:{serverRef:"" + vm.id}};

            payload._options = options || {};

            return this.api_call(this.api_path() + "/action", "create", 

                                 payload,

                                 _.bind(function(){

                                     this.vms.add_pending(vm.id);

                                     if (callback) {callback()}

                                 },this), error);

        },

        remove_vm: function (vm, callback, error, options) {

            var payload = {remove:{serverRef:"" + vm.id}};

            payload._options = options || {};

            return this.api_call(this.api_path() + "/action", "create", 

                                 {remove:{serverRef:"" + vm.id}},

                                 _.bind(function(){

                                     this.vms.add_pending_for_remove(vm.id);

                                     if (callback) {callback()}

                                 },this), error);

        },

        rename: function(name, callback) {

            return this.api_call(this.api_path(), "update", {

                network:{name:name}, 

                _options:{

                    critical: false, 

                    error_params:{

                        title: "Network action failed",

                        ns: "Networks",

                        extra_details: {"Network id": this.id}

                    }

                }}, callback);

        },

        get_connectable_vms: function() {

            var servers = this.vms.list();

            return storage.vms.filter(function(vm){

                return servers.indexOf(vm) == -1 && !vm.in_error_state();

            })

        },

        state_message: function() {

            if (this.get("state") == "NORMAL" && this.is_public()) {

                return "Public network";

            }

            return models.Network.STATES[this.get("state")];

        },

        in_progress: function() {

            return models.Network.STATES_TRANSITIONS[this.get("state")] != undefined;

        },

        do_all_pending_actions: function(success, error) {

            var destroy = this.get("actions").has_action("destroy");

            _.each(this.get("actions").actions, _.bind(function(params, action) {

                _.each(params, _.bind(function(with_params) {

                    this.call(action, with_params, success, error);

                }, this));

            }, this));

        }

    });

    models.Network.STATES = {

        'NORMAL': 'Private network',

        'CONNECTING': 'Connecting...',

        'DISCONNECTING': 'Disconnecting...',

        'FIREWALLING': 'Firewall update...',

        'DESTROY': 'Destroying...'

    }

    models.Network.STATES_TRANSITIONS = {

        'CONNECTING': ['NORMAL'],

        'DISCONNECTING': ['NORMAL'],

        'FIREWALLING': ['NORMAL']

    }

    // Virtualmachine model

    models.VM = models.Model.extend({

        path: 'servers',

        has_status: true,

        initialize: function(params) {

            this.networks = new VMNetworksList();