ipfs-grpc-client/src/utils/load-services.js

A client library for the IPFS gRPC API

import protocol from 'ipfs-grpc-protocol'
import protobuf from 'protobufjs/light.js'

const { Service } = protobuf

const CONVERSION_OPTS = {
  keepCase: false,
  longs: String, // long.js is required
  enums: String,
  defaults: false,
  oneofs: true
}

/**
 * Converts protobufjs service definitions into the format expected
 * by @improbable-eng/grpc-web.  This is to let us use the same
 * service definition on both the server and the client.
 */
export function loadServices () {
  // @ts-expect-error - recent protobufjs release changed the types
  const root = protobuf.Root.fromJSON(protocol)
  /** @type {Record<string, any>} */
  const output = {}

  Object
    // @ts-expect-error
    .keys(root.nested.ipfs)
    // @ts-expect-error
    .filter(key => root.nested.ipfs[key] instanceof Service)
    // @ts-expect-error
    .map(key => root.nested.ipfs[key])
    .forEach(service => {
      /** @type {Record<string, any>} */
      const serviceDef = {}

      output[service.name] = serviceDef

      Object.keys(service.methods)
        .forEach(methodName => {
          const method = service.methods[methodName].resolve()

          serviceDef[methodName] = {
            service: {
              serviceName: `ipfs.${service.name}`
            },
            methodName,
            requestStream: method.requestStream,
            responseStream: method.responseStream,
            requestType: {
              /**
               * @param {any} obj
               */
              serializeBinary: (obj) => {
                const message = method.resolvedRequestType.fromObject(obj)
                return method.resolvedRequestType.encode(message).finish()
              },
              /**
               * @param {Uint8Array} buf
               */
              deserializeBinary: (buf) => {
                const message = method.resolvedRequestType.decode(buf)
                const obj = method.resolvedRequestType.toObject(message, CONVERSION_OPTS)

                Object.defineProperty(obj, 'toObject', {
                  enumerable: false,
                  configurable: false,
                  value: () => obj
                })

                return obj
              }
            },
            responseType: {
              /**
               * @param {any} obj
               */
              serializeBinary: (obj) => {
                const message = method.resolvedResponseType.fromObject(obj)
                return method.resolvedResponseType.encode(message).finish()
              },
              /**
               * @param {Uint8Array} buf
               */
              deserializeBinary: (buf) => {
                const message = method.resolvedResponseType.decode(buf)
                const obj = method.resolvedResponseType.toObject(message, CONVERSION_OPTS)

                Object.defineProperty(obj, 'toObject', {
                  enumerable: false,
                  configurable: false,
                  value: () => obj
                })

                return obj
              }
            }
          }
        })
    })

  return output
}