smom-dbis-138/contracts/mainnet-checkpoint/extensions/MetricsExtension.sol

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.20;

import {CheckpointExtensionBase} from "./CheckpointExtensionBase.sol";
import {CheckpointStorage} from "../storage/CheckpointStorage.sol";
import {CheckpointLeaf} from "../libraries/CheckpointLeaf.sol";

/// @notice On-chain counters for explorers / DefiLlama activity metrics (not TVL).
contract MetricsExtension is CheckpointExtensionBase {
    uint256 public totalBatches;
    uint256 public totalPaymentsAttested;
    uint256 public totalValueWei;

    function HOOK_BEFORE_SUBMIT() external pure override returns (uint32) {
        return 0;
    }

    function HOOK_AFTER_SUBMIT() external pure override returns (uint32) {
        return 1 << 1;
    }

    function HOOK_ON_CCIP() external pure override returns (uint32) {
        return 0;
    }

    function beforeSubmit(CheckpointStorage.CheckpointHeader calldata, bytes calldata) external pure override {}

    function afterSubmit(CheckpointStorage.CheckpointHeader calldata header, bytes calldata data) external {
        totalBatches++;
        totalPaymentsAttested += header.txCount;
        if (data.length == 0) return;
        if (data.length > 0 && data[0] == bytes1(0x02)) {
            (, CheckpointLeaf.PaymentLeafV2[] memory v2Leaves) =
                abi.decode(data, (bytes1, CheckpointLeaf.PaymentLeafV2[]));
            for (uint256 i = 0; i < v2Leaves.length; i++) {
                totalValueWei += v2Leaves[i].value;
            }
            return;
        }
        CheckpointLeaf.PaymentLeafV1[] memory leaves = abi.decode(data, (CheckpointLeaf.PaymentLeafV1[]));
        for (uint256 i = 0; i < leaves.length; i++) {
            totalValueWei += leaves[i].value;
        }
    }

    function onCCIPReceive(bytes calldata) external pure override {}
}