index.js

import EventEmitter from 'eventemitter3';
import * as transports from './transports/index.js';

export function rawr({ transport, timeout = 0, handlers = {}, methods, idGenerator }) {
  let callId = 0;
  methods = methods || handlers || {}; // backwards compat
  const pendingCalls = {};
  const methodHandlers = {};
  const notificationEvents = new EventEmitter();
  notificationEvents.on = notificationEvents.on.bind(notificationEvents);

  transport.on('rpc', (msg) => {
    if (msg.id) {
      // handle an RPC request
      if (msg.params && methodHandlers[msg.method]) {
        methodHandlers[msg.method](msg);
        return;
      }
      // handle an RPC result
      const promise = pendingCalls[msg.id];
      if (promise) {
        if (promise.timeoutId) {
          clearTimeout(promise.timeoutId);
        }
        delete pendingCalls[msg.id];
        if (msg.error) {
          promise.reject(msg.error);
        }
        return promise.resolve(msg.result);
      }
      return;
    }
    // handle a notification
    msg.params.unshift(msg.method);
    notificationEvents.emit(...msg.params);
  });

  function addHandler(methodName, handler) {
    methodHandlers[methodName] = (msg) => {
      Promise.resolve()
        .then(() => {
          return handler.apply(this, msg.params);
        })
        .then((result) => {
          transport.send({
            id: msg.id,
            result
          });
        })
        .catch((error) => {
          const serializedError = { message: error.message };
          if (error.code) {
            serializedError.code = error.code;
          }
          transport.send({
            id: msg.id,
            error: serializedError
          });
        });
    };
  }

  Object.keys(methods).forEach((m) => {
    addHandler(m, methods[m]);
  });

  function sendMessage(method, params, config) {
    const id = idGenerator ? idGenerator() : ++callId;
    const msg = {
      jsonrpc: '2.0',
      method,
      params,
      id
    };

    let timeoutId;
    if (config.timeout || timeout) {
      timeoutId = setTimeout(() => {
        if (pendingCalls[id]) {
          const err = new Error('RPC timeout');
          err.code = 504;
          pendingCalls[id].reject(err);
          delete pendingCalls[id];
        }
      }, config.timeout || timeout);
    }

    const response = new Promise((resolve, reject) => {
      pendingCalls[id] = { resolve, reject, timeoutId };
    });

    transport.send(msg, config);

    return response;
  }

  const methodsProxy = new Proxy({}, {
    get: (target, name) => {
      return (...args) => {
        return sendMessage(name, args, {});
      };
    }
  });

  const configurableMethodsProxy = new Proxy({}, {
    get: (target, name) => {
      return (...args) => {
        let config;
        if (args.length) {
          const testArg = args.pop();
          if (testArg && typeof testArg === 'object' && !Array.isArray(testArg)) {
            config = testArg;
          } else {
            args.push(testArg);
          }
        }
        return sendMessage(name, args, config || {});
      };
    }
  });


  const notifiers = new Proxy({}, {
    get: (target, name) => {
      return (...args) => {
        const msg = {
          jsonrpc: '2.0',
          method: name,
          params: args
        };
        transport.send(msg);
      };
    }
  });

  const configurableNotifiersProxy = new Proxy({}, {
    get: (target, name) => {
      return (...args) => {
        let config;
        if (args.length) {
          const testArg = args.pop();
          if (testArg && typeof testArg === 'object' && !Array.isArray(testArg)) {
            config = testArg;
          } else {
            args.push(testArg);
          }
        }
        const msg = {
          jsonrpc: '2.0',
          method: name,
          params: args
        };
        transport.send(msg, config || {});
      };
    }
  });

  const notifications = new Proxy({}, {
    get: (target, name) => {
      return (callback) => {
        notificationEvents.on(name.substring(2), (...args) => {
          return callback.apply(callback, args);
        });
      };
    }
  });

  return {
    methods: methodsProxy,
    methodsExt: configurableMethodsProxy,
    addHandler,
    notifications,
    notifiers,
    notifiersExt: configurableNotifiersProxy,
    transport,
  };
}

// Attach transports to rawr for convenience
rawr.transports = transports;

// Default export for backwards compatibility
export default rawr;

// Named exports
export { transports };