const Ip4Prefix = '10.0.0.1/24';
const Ip6Prefix = '2001::/64';


class NodeHelper {
    constructor() {
        this.displays = {
            // default router
            0: {
                name: 'node',
                display: 'Default'
            },
            // switch
            4: {
                name: 'switch',
                display: 'Switch'
            },
            // hub
            5: {
                name: 'hub',
                display: 'Hub'
            },
            // wlan
            6: {
                name: 'wlan',
                display: 'WLAN'
            },
            12: {
                name: 'ptp',
                display: 'PTP'
            }
        };

        this.icons = {
            router: 'static/router.svg',
            host: 'static/host.gif',
            PC: 'static/pc.gif',
            mdr: 'static/mdr.svg',
            switch: 'static/lanswitch.svg',
            hub: 'static/hub.svg',
            wlan: 'static/wlan.gif'
        };

        this.defaultNode = 0;
        this.ptpNode = 12;
    }

    isNetworkNode(node) {
        return [4, 5, 6, 12].includes(node.type);
    }

    getDisplay(nodeType) {
        return this.displays[nodeType];
    }

    getIcon(node) {
        let iconName = this.getDisplay(node.type).name;
        if (node.type === 0) {
            iconName = node.model;
        }
        return this.icons[iconName];
    }
}
const CoreNodeHelper = new NodeHelper();

class CoreNode {
    constructor(id, type, name, x, y) {
        this.id = id;
        this.type = type;
        this.name = name;
        this.model = null;
        this.canvas = null;
        this.icon = null;
        this.opaque = null;
        this.services = [];
        this.x = x;
        this.y = y;
        this.lat = null;
        this.lon = null;
        this.alt = null;
        this.emulation_id = null;
        this.emulation_server = null;
        this.interfaces = {};
    }

    getNetworkNode() {
        const icon = CoreNodeHelper.getIcon(this);

        return {
            id: this.id,
            x: this.x,
            y: this.y,
            label: this.name,
            coreNode: this,
            //color: '#FFF',
            shape: 'image',
            image: icon
        };
    }

    json() {
        return {
            id: this.id,
            type: this.type,
            name: this.name,
            model: this.model,
            x: this.x,
            y: this.y,
            lat: this.lat,
            lon: this.lon,
            alt: this.alt
        }
    }
}

class CoreNetwork {
    constructor(elementId, coreRest) {
        this.coreRest = coreRest;
        this.nodeType = 0;
        this.nodeModel = 'router';
        this.nodeId = 0;
        this.container = document.getElementById(elementId);
        this.nodes = new vis.DataSet();
        this.edges = new vis.DataSet();
        this.links = {};
        this.networkData = {
            nodes: this.nodes,
            edges: this.edges
        };
        this.networkOptions = {
            height: '95%',
            physics: false,
            interaction: {
                selectConnectedEdges: false
            },
            edges: {
                shadow: true,
                width: 3,
                smooth: false,
                color: {
                    color: '#000000'
                }
            },
            nodes: {
                shadow: true
            }
        };
        this.network = new vis.Network(this.container, this.networkData, this.networkOptions);
        this.network.on('doubleClick', this.addNode.bind(this));
        this.edges.on('add', this.addEdge.bind(this));
        this.nodesEnabled = false;
    }

    enableNodeCreation(enabled) {
        console.log('node created enabled: ', enabled);
        this.nodesEnabled = enabled;
    }

    getCoreNodes() {
        const coreNodes = [];
        for (let node of this.nodes.get()) {
            coreNodes.push(node.coreNode.json());
        }
        return coreNodes;
    }

    addCoreNode(node) {
        const position = node.position;
        const coreNode = new CoreNode(node.id, node.type, node.name, position.x, position.y);
        coreNode.model = node.model;
        this.nodes.add(coreNode.getNetworkNode());
    }

    nextNodeId() {
        this.nodeId += 1;
        return this.nodeId;
    }

    joinedSessions(nodes) {
        const self = this;
        const nodeIds = [0];

        for (let node of nodes) {
            if (node.type === CoreNodeHelper.ptpNode) {
                continue;
            }

            nodeIds.push(node.id);
            this.addCoreNode(node);
        }

        for (let node of nodes) {
            if (!CoreNodeHelper.isNetworkNode(node)) {
                continue;
            }

            this.coreRest.getLinks(node.id)
                .then(function(response) {
                    console.log('link response: ', response);
                    for (let linkData of response.links) {
                        self.createEdgeFromLink(linkData);
                    }
                })
                .catch(function(err) {
                    console.log('get link error: ', err);
                });
        }

        if (nodes.length) {
            this.nodeId = Math.max.apply(Math, nodeIds) || 0;
        } else {
            this.nodeId = 0;
        }
    }

    createEdgeFromLink(linkData) {
        const fromNode = this.nodes.get(linkData.node1_id).coreNode;
        const toNode = this.nodes.get(linkData.node2_id).coreNode;
        const linkId = `${fromNode.id}-${toNode.id}`;

        let interfaceOne = null;
        if (linkData.interface1_id !== null) {
            interfaceOne = {
                id: linkData.interface1_id,
                ip4: linkData.interface1_ip4,
                ip4mask: linkData.interface1_ip4_mask,
                ip6: linkData.interface1_ip6,
                ip6mask: linkData.interface1_ip6_mask
            };
            fromNode.interfaces[linkData.interface1_id] = interfaceOne;
        }

        let interfaceTwo = null;
        if (linkData.interface2_id !== null) {
            interfaceTwo = {
                id: linkData.interface2_id,
                ip4: linkData.interface2_ip4,
                ip4mask: linkData.interface2_ip4_mask,
                ip6: linkData.interface2_ip6,
                ip6mask: linkData.interface2_ip6_mask
            };
            toNode.interfaces[linkData.interface2_id] = interfaceTwo;
        }

        this.links[linkId] = {
            node_one: fromNode.id,
            node_two: toNode.id,
            interface_one: interfaceOne,
            interface_two: interfaceTwo
        };

        const edge = {
            id: linkId,
            from: fromNode.id,
            to: toNode.id,
            recreated: true,
            label: 'from: name\nto: name',
            title: 'from: name\nto: name'
        };
        this.edges.add(edge);
    }

    async start() {
        const nodes = coreNetwork.getCoreNodes();
        for (let node of nodes) {
            const response = await coreRest.createNode(node);
            console.log('created node: ', response);
        }

        for (let linkId in this.links) {
            const link = this.links[linkId];
            const response = await coreRest.createLink(link);
            console.log('created link: ', response);
        }

        return await coreRest.setSessionState(SessionStates.instantiation);
    }

    addNode(properties) {
        if (!this.nodesEnabled) {
            console.log('node creation disabled');
            return;
        }

        console.log('add node event: ', properties);
        if (properties.nodes.length !== 0) {
            return;
        }

        const {x, y} = properties.pointer.canvas;
        const nodeId = this.nextNodeId();
        const nodeDisplay = CoreNodeHelper.getDisplay(this.nodeType);
        const name = `${nodeDisplay.name}${nodeId}`;
        const coreNode = new CoreNode(nodeId, this.nodeType, name, x, y);
        coreNode.model = this.nodeModel;
        this.nodes.add(coreNode.getNetworkNode());
        console.log('added node: ', coreNode.getNetworkNode());
    }

    addEdge(_, properties) {
        const edgeId = properties.items[0];
        const edge = this.edges.get(edgeId);
        if (edge.recreated) {
            console.log('ignoring recreated edge');
            return;
        }

        console.log('added edge: ', edgeId, edge);
        if (edge.from === edge.to) {
            console.log('removing cyclic edge');
            this.edges.remove(edge.id);
            setTimeout(() => this.network.addEdgeMode(), 250);
            return;
        }

        const fromNode = this.nodes.get(edge.from).coreNode;
        const toNode = this.nodes.get(edge.to).coreNode;

        this.addEdgeLink(edge, fromNode, toNode)
            .then(function() {
                console.log('create edge link success!');
            })
            .catch(function(err) {
                console.log('create link error: ', err);
            });

        setTimeout(() => this.network.addEdgeMode(), 250);
    }

    async addEdgeLink(edge, fromNode, toNode) {
        const linkId = `${fromNode.id}-${toNode.id}`;
        let interfaceOne = null;
        if (fromNode.type === CoreNodeHelper.defaultNode) {
            const fromIps = await this.coreRest.getNodeIps(fromNode.id, Ip4Prefix, Ip6Prefix);
            console.log('from ips: ', fromIps);
            const interfaceOneId = Object.keys(fromNode.interfaces).length;
            interfaceOne = {
                id: interfaceOneId,
                ip4: fromIps.ip4,
                ip4mask: fromIps.ip4mask,
                ip6: fromIps.ip6,
                ip6mask: fromIps.ip6mask
            };
            fromNode.interfaces[interfaceOneId] = interfaceOne;
        }

        let interfaceTwo = null;
        if (toNode.type === CoreNodeHelper.defaultNode) {
            const toIps = await this.coreRest.getNodeIps(toNode.id, Ip4Prefix, Ip6Prefix);
            console.log('to ips: ', toIps);
            const interfaceTwoId = Object.keys(toNode.interfaces).length;
            interfaceTwo = {
                id: interfaceTwoId,
                ip4: toIps.ip4,
                ip4mask: toIps.ip4mask,
                ip6: toIps.ip6,
                ip6mask: toIps.ip6mask
            };
            toNode.interfaces[interfaceTwoId] = interfaceTwo;
        }

        this.links[linkId] = {
            node_one: fromNode.id,
            node_two: toNode.id,
            interface_one: interfaceOne,
            interface_two: interfaceTwo
        };

        edge.id = linkId;
        edge.label = 'from: name\nto: name';
        edge.title = 'from: name\nto: name';
        this.edges.update(edge);
    }

    linkMode(enabled) {
        console.log('link mode:', enabled);
        if (enabled) {
            this.network.addEdgeMode();
        } else {
            this.network.disableEditMode();
        }
    }

    setNodeMode(nodeType, model) {
        this.nodeType = nodeType;
        this.nodeModel = model || null;
    }
}