babe_block_header.hpp

Go to the documentation of this file.

#ifndef KAGOME_BABE_BLOCK_HEADER_HPP
#define KAGOME_BABE_BLOCK_HEADER_HPP

#include "consensus/babe/common.hpp"
#include "consensus/babe/types/slot.hpp"
#include "crypto/sr25519_types.hpp"
#include "primitives/authority.hpp"
#include "primitives/common.hpp"

#include <scale/scale.hpp>

namespace kagome::consensus {
  struct BabeBlockHeader {
    SlotType slot_assignment_type{};

    primitives::AuthorityIndex authority_index;

    BabeSlotNumber slot_number;

    crypto::VRFOutput vrf_output{};

    SlotType slotType() const {
      return slot_assignment_type;
    }

    bool needVRFCheck() const {
      return slot_assignment_type == SlotType::Primary
             or slot_assignment_type == SlotType::SecondaryVRF;
    }

    bool needVRFWithThresholdCheck() const {
      return slot_assignment_type == SlotType::Primary;
    }

    bool isProducedInSecondarySlot() const {
      return slot_assignment_type == SlotType::SecondaryPlain
             or slot_assignment_type == SlotType::SecondaryVRF;
    }

    friend inline ::scale::ScaleEncoderStream &operator<<(
        ::scale::ScaleEncoderStream &s, const BabeBlockHeader &bh) {
      s << bh.slot_assignment_type << bh.authority_index << bh.slot_number;
      if (bh.needVRFCheck()) {
        s << bh.vrf_output;
      }
      return s;
    }

    friend inline ::scale::ScaleDecoderStream &operator>>(
        ::scale::ScaleDecoderStream &s, BabeBlockHeader &bh) {
      s >> bh.slot_assignment_type >> bh.authority_index >> bh.slot_number;
      if (bh.needVRFCheck()) {
        s >> bh.vrf_output;
      }
      return s;
    }
  };
}  // namespace kagome::consensus

#endif  // KAGOME_BABE_BLOCK_HEADER_HPP