/*
 * Copyright 2016-present Open Networking Laboratory
 *
 * 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.
 */

/*
 ONOS GUI -- Topology Devices Module.
 Module that holds the devices for a region
 */

(function () {
    'use strict';

    var Collection, Model;

    var remappedDeviceTypes = {
        switch: 'm_switch',
        virtual: 'cord'
    };

    function createDeviceCollection(data, region) {

        var DeviceCollection = Collection.extend({
            model: Model
        });

        var devices = [];
        data.forEach(function (deviceLayer) {
            deviceLayer.forEach(function (device) {
                devices.push(device);
            });
        });

        return new DeviceCollection(devices);
    }

    angular.module('ovTopo2')
    .factory('Topo2DeviceService',
        ['Topo2Collection', 'Topo2NodeModel', 'Topo2DeviceDetailsPanel',
            function (_c_, _nm_, detailsPanel) {

                Collection = _c_;

                Model = _nm_.extend({
                    initialize: function () {
                        this.super = this.constructor.__super__;
                        this.super.initialize.apply(this, arguments);
                    },
                    events: {
                        'click': 'onClick'
                    },
                    onChange: function () {

                        // Update class names when the model changes
                        if (this.el) {
                            this.el.attr('class', this.svgClassName());
                        }
                    },
                    nodeType: 'device',
                    icon: function () {
                        var type = this.get('type');
                        return remappedDeviceTypes[type] || type || 'unknown';
                    },
                    onClick: function () {

                        var ev = d3.event;

                        if (ev.shiftKey) {
                            // TODO: Multi-Select Details Panel
                            this.set('selected', true);
                        } else {

                            var s = Boolean(this.get('selected'));
                            // Clear all selected Items
                            _.each(this.collection.models, function (m) {
                                m.set('selected', false);
                            });

                            this.set('selected', !s);
                        }

                        var selected = this.collection.filter(function (m) {
                            return m.get('selected');
                        });

                        if (_.isArray(selected) && selected.length > 0) {
                            if (selected.length === 1) {
                                var model = selected[0],
                                    id = model.get('id'),
                                    nodeType = model.get('nodeType');
                                detailsPanel.updateDetails(id, nodeType);
                                detailsPanel.show();
                            } else {
                                // Multi Panel
                                detailsPanel.showMulti(selected);
                            }
                        } else {
                            detailsPanel.hide();
                        }
                    },
                    onExit: function () {
                        var node = this.el;
                        node.select('use')
                            .style('opacity', 0.5)
                            .transition()
                            .duration(800)
                            .style('opacity', 0);

                        node.selectAll('rect')
                            .style('stroke-fill', '#555')
                            .style('fill', '#888')
                            .style('opacity', 0.5);
                    }
                });

                return {
                    createDeviceCollection: createDeviceCollection
                };
            }
        ]);

})();
