emotiv-ts/src/services/data-stream.service.ts

A Typescript library that wraps the Cortex API functionalities to communicate with Emotiv headsets

import type {w3cwebsocket} from 'websocket';
import {AuthenticationService} from "./authentication.service";
import {Requests} from "../enums/internal/emotiv.requests";
import {Subscribe} from "../models/requests/datasubscription/subscribe";
import type {DataStream} from "../enums/data-stream.enum";
import {SessionService} from "./session.service";
import {Unsubscribe} from "../models/requests/datasubscription/unsubscribe";

export class DataStreamService {
    socket: w3cwebsocket;

    constructor(socket: w3cwebsocket) {
        this.socket = socket
    }

    subscribe(streams: DataStream[], action: (data: any) => void) {
        let authToken = AuthenticationService.getAuthToken();
        let sessionId = SessionService.getSessionId();

        let subRequest = new Subscribe(authToken, sessionId, streams);

        this.socket.send(JSON.stringify(subRequest))
        this.socket.onmessage = (message) => {
            try {
                let data = JSON.parse(message.data as unknown as string);

                console.debug("SubRequest response:", data)

                action(data)
            } catch (error) {
                console.error(error);
            }
        }
    }

    unsubscribe(streams: DataStream[], action: (data: any) => void) {
        let authToken = AuthenticationService.getAuthToken();
        let sessionId = SessionService.getSessionId();

        let unsubRequest = new Unsubscribe(authToken, sessionId, streams);

        this.socket.send(JSON.stringify(unsubRequest))
        this.socket.onmessage = (message) => {
            try {
                let data = JSON.parse(message.data as unknown as string);

                console.debug("UnsubRequest response:", data)

                action(data)
            } catch (error) {
                console.error(error);
            }
        }
    }
}