index.ts

import { useCallback, useEffect, useMemo, useRef, useState, } from 'react';

import { createCommander, } from './Commander';
import { createDeviceLink, } from './DeviceLink';
import { createDeviceStore, } from './DeviceStore';

type onStream = (data : Int16Array) => void;

export const useDevice = (url : string, initialState : Record<string, any>, onStream? : onStream, heartbeatInterval? : number, webSocket? : WebSocket) => {
    const store = useMemo(() => createDeviceStore(initialState), [ initialState, ]);

    const [ status, setStatus, ] = useState<'connected' | 'disconnected' | 'error'>('disconnected');

    const linkRef      = useRef<ReturnType<typeof createDeviceLink> | null>(null);
    const commanderRef = useRef<ReturnType<typeof createCommander> | null>(null);

    const connect = useCallback(() => {
        linkRef.current = createDeviceLink({
            url,
            onStatus  : (s, e) => {
                setStatus(s);

                if (s === 'error' && e) console.error(e);
            },
            onMessage : data => store.update(data),
            onStream  : data => {
                onStream?.(data);
            },
            webSocket,
        });

        commanderRef.current = createCommander(linkRef.current, heartbeatInterval);

        const unsubscribe = store.subscribe(store.update);

        return () => {
            unsubscribe();

            commanderRef.current?.dispose();
        };
    }, [ url, onStream, heartbeatInterval, store, webSocket, ]);

    useEffect(() => {
         const interval = setInterval(() => {
            if (status === 'disconnected') connect();
        }, 5000);

         connect();

        return () => clearInterval(interval);
    }, [ status, connect, ]);

    return {
        status,
        store,
        execute : ({
            type,
            payload,
        } : {
            type     : string,
            payload? : Record<string, any>,
        }) => commanderRef.current?.sendCommand({
            type,
            payload,
        }),
        stream  : (data : Uint8Array) => linkRef.current?.stream(data),
    };
};