nttgin-BGPalerter/connectors/connectorRIS.js

import WebSocket from "ws";
import Connector from "./connector";

export default class ConnectorRIS extends Connector{

    constructor(params, env) {
        super(params, env);
        this.ws = null;
    }

    connect = () =>
        new Promise((resolve, reject) => {
            try {

                this.ws = new WebSocket(this.params.url);
                this.ws.on('message', this.message);
                this.ws.on('close', this.error);
                this.ws.on('open', () => {
                    resolve(true);
                    this.connected();
                });

            } catch(error) {
                resolve(false);
            }
        });


    subscribe = (input) =>
        new Promise((resolve, reject) => {
            try {
                this.ws.send(JSON.stringify({
                    type: "ris_subscribe",
                    data: this.params
                }));
                resolve(true);
            } catch(error) {
                resolve(false);
            }
        });


    transform = (message) => {
        message = message.data;
        const components = [];
        const announcements = message["announcements"] || [];
        const withdrawals = message["withdrawals"] || [];
        const peer = message["peer"];
        const path = message["path"];

        for (let announcement of announcements){
            const nextHop = announcement["next_hop"];
            const prefixes = announcement["prefixes"] || [];

            for (let prefix of prefixes){
                components.push({
                    type: "announcement",
                    prefix,
                    peer,
                    path,
                    originAs: path[path.length - 1],
                    nextHop
                })
            }
        }

        for (let prefix of withdrawals){
            components.push({
                type: "withdrawal",
                prefix,
                peer

            })
        }

        return components;
    };

}