import {ForkSeq, MIN_SEED_LOOKAHEAD, SLOTS_PER_EPOCH} from "@lodestar/params";
import {ssz} from "@lodestar/types";
import {CachedBeaconStateFulu, EpochTransitionCache} from "../types.js";
import {FLAG_UNSLASHED} from "../util/attesterStatus.js";
import {computeEpochShuffling} from "../util/epochShuffling.js";
import {computeProposerIndices} from "../util/seed.js";

/**
 * This function updates the `proposer_lookahead` field in the beacon state
 * by shifting out proposer indices from the earliest epoch and appending new
 * proposer indices for the latest epoch. With `MIN_SEED_LOOKAHEAD` set to `1`,
 * this means that at the start of epoch `N`, the proposer lookahead for epoch
 * `N+1` will be computed and included in the beacon state's lookahead.
 */
export function processProposerLookahead(
  fork: ForkSeq,
  state: CachedBeaconStateFulu,
  cache: EpochTransitionCache
): void {
  // Shift out proposers in the first epoch
  const remainingProposerLookahead = state.proposerLookahead.getAll().slice(SLOTS_PER_EPOCH);

  // Fill in the last epoch with new proposer indices
  const epoch = state.epochCtx.epoch + MIN_SEED_LOOKAHEAD + 1;

  const shuffling = computeEpochShuffling(state, cache.nextShufflingActiveIndices, epoch);
  // Save shuffling to cache so afterProcessEpoch can reuse it instead of recomputing
  cache.nextShuffling = shuffling;

  const activeIndices =
    fork >= ForkSeq.gloas
      ? // Exclude slashed validators from proposing (EIP-8045)
        shuffling.activeIndices.filter((index) => (cache.flags[index] & FLAG_UNSLASHED) !== 0)
      : shuffling.activeIndices;

  const lastEpochProposerLookahead = computeProposerIndices(fork, state, {activeIndices}, epoch);

  state.proposerLookahead = ssz.fulu.ProposerLookahead.toViewDU([
    ...remainingProposerLookahead,
    ...lastEpochProposerLookahead,
  ]);
}