react-relay-network-modern/lib/RelayRequestBatch.js.flow

Network Layer for React Relay and Express (Batch Queries, AuthToken, Logging, Retry)

/* @flow */

import type { FetchOpts, Variables } from './definition';
import type RelayRequest from './RelayRequest';
import RRNLError from './RRNLError';

export type Requests = RelayRequest[];

export default class RelayRequestBatch {
  fetchOpts: $Shape<FetchOpts>;
  requests: Requests;

  constructor(requests: Requests) {
    this.requests = requests;
    this.fetchOpts = {
      method: 'POST',
      headers: {},
      body: this.prepareBody(),
    };
  }

  setFetchOption(name: string, value: mixed) {
    this.fetchOpts[name] = value;
  }

  setFetchOptions(opts: Object) {
    this.fetchOpts = {
      ...this.fetchOpts,
      ...opts,
    };
  }

  getBody(): string {
    if (!this.fetchOpts.body) {
      this.fetchOpts.body = this.prepareBody();
    }
    return (this.fetchOpts.body: any) || '';
  }

  prepareBody(): string {
    return `[${this.requests.map((r) => r.getBody()).join(',')}]`;
  }

  getIds(): string[] {
    return this.requests.map((r) => r.getID());
  }

  getID(): string {
    return `BATCH_REQUEST:${this.getIds().join(':')}`;
  }

  isMutation() {
    return false;
  }

  isFormData() {
    return false;
  }

  clone(): RelayRequestBatch {
    // $FlowFixMe
    const newRequest = Object.assign(Object.create(Object.getPrototypeOf(this)), this);
    newRequest.fetchOpts = { ...this.fetchOpts };
    newRequest.fetchOpts.headers = { ...this.fetchOpts.headers };
    return (newRequest: any);
  }

  getVariables(): Variables {
    throw new RRNLError('Batch request does not have variables.');
  }

  getQueryString(): string {
    return this.prepareBody();
  }
}