export const requestInChunks = async <Item = any, Response = any>(
  chunkSize: number,
  url: string,
  items: Item[],
  queryBuilder: (chunk: Item[]) => string
): Promise<Response> => {
  const firstQuery = queryBuilder(items.slice(0, 1));

  const wasmCallsPerQuery = ((firstQuery).match(/wasm {/g) || []).length;
  const correctedChunkSize = Math.floor(chunkSize / (wasmCallsPerQuery || 1));
  const totalChunks = Math.ceil(items.length / correctedChunkSize);

  const chunks = await Promise.all(
    Array.from(Array(totalChunks).keys()).map((i) => {
      const chunk = items.slice(i * correctedChunkSize, (i + 1) * correctedChunkSize);

      console.log(queryBuilder(chunk));

      return request<Response>(url, queryBuilder(chunk));
    })
  );

  return chunks.reduce((all, chunk) => ({ ...all, ...chunk }));
};

export const requestInChunks = async <Item = any, Response = any>(
  chunkSize: number,
  url: string,
  items: Item[],
  queryBuilder: (chunk: Item[]) => string
): Promise<Response> => {
  const query = queryBuilder(items);
  const { kind, loc, selections } = (parse(query).definitions[0] as OperationDefinitionNode).selectionSet;
  
  const selectionChunks = _.chunk(selections, chunkSize);
  const chunks = selectionChunks.map((chunk) => print({ kind, loc: loc!, selections: chunk ?? [] }));

  const requests = await Promise.all(chunks.map((chunk) => request<Response>(url, chunk)));

  return requests.reduce((all, chunk) => ({ ...all, ...chunk }));
};