import {MAX_REQUEST_BLOCKS} from "@lodestar/params";
import {allForks, phase0, ssz} from "@lodestar/types";
import {ContextBytesType, Encoding, ProtocolDefinition, ProtocolDefinitionGenerator} from "../types.js";

/* eslint-disable @typescript-eslint/naming-convention */
const BeaconBlocksByRangeCommon: Pick<
  ProtocolDefinition<phase0.BeaconBlocksByRangeRequest, allForks.SignedBeaconBlock>,
  "method" | "encoding" | "requestType" | "renderRequestBody" | "inboundRateLimits"
> = {
  method: "beacon_blocks_by_range",
  encoding: Encoding.SSZ_SNAPPY,
  requestType: () => ssz.phase0.BeaconBlocksByRangeRequest,
  renderRequestBody: (req) => `${req.startSlot},${req.step},${req.count}`,
  inboundRateLimits: {
    // Rationale: https://github.com/sigp/lighthouse/blob/bf533c8e42cc73c35730e285c21df8add0195369/beacon_node/lighthouse_network/src/rpc/mod.rs#L118-L130
    byPeer: {quota: MAX_REQUEST_BLOCKS, quotaTimeMs: 10_000},
    getRequestCount: (req) => req.count,
  },
};

// eslint-disable-next-line @typescript-eslint/naming-convention
export const BeaconBlocksByRange: ProtocolDefinitionGenerator<
  phase0.BeaconBlocksByRangeRequest,
  allForks.SignedBeaconBlock
> = (_modules, handler) => {
  return {
    ...BeaconBlocksByRangeCommon,
    version: 1,
    handler,
    responseType: () => ssz.phase0.SignedBeaconBlock,
    contextBytes: {type: ContextBytesType.Empty},
  };
};

// eslint-disable-next-line @typescript-eslint/naming-convention
export const BeaconBlocksByRangeV2: ProtocolDefinitionGenerator<
  phase0.BeaconBlocksByRangeRequest,
  allForks.SignedBeaconBlock
> = (modules, handler) => {
  return {
    ...BeaconBlocksByRangeCommon,
    version: 2,
    handler,
    responseType: (forkName) => ssz[forkName].SignedBeaconBlock,
    contextBytes: {
      type: ContextBytesType.ForkDigest,
      forkDigestContext: modules.config,
      forkFromResponse: (block) => modules.config.getForkName(block.message.slot),
    },
  };
};