import ProtomuxRPC from "protomux-rpc";
// @ts-ignore
import c from "compact-encoding";
import b4a from "b4a";

const ID = b4a.from("lumeweb");

export default class RPC extends ProtomuxRPC {
  private _ready: Promise<void>;

  constructor(stream: any, options = {}) {
    options = {
      ...{
        id: ID,
        valueEncoding: c.json,
      },
      ...options,
    };
    super(stream, options);

    this._ready = new Promise((resolve, reject) => {
      stream.on("open", onopen);
      stream.on("destroy", ondestroy);

      function onopen(handshake) {
        removeListener();
        resolve(handshake);
      }

      function ondestroy() {
        removeListener();
        reject(new Error("Client could not connect"));
      }

      function removeListener() {
        stream.off("open", onopen);
        stream.off("destroy", ondestroy);
      }
    });

    stream.setKeepAlive(5000);
  }

  async request(method: any, value: any | string = "", options = {}) {
    return super.request(method, value, options);
  }

  get ready(): Promise<void> {
    return this._ready;
  }
}