"""Features module."""

from abacus_common_logic.utils.features import is_feature_enabled
from flask import g

from royalties.constants.features import (
    ABACUS_AUTO_GENERATE_ADJUSTMENTS_FLOWTHROUGH,
    ABACUS_CONTRACT_PAGE_UPDATE_RUN_CONTROLLER,
    ABACUS_STATEMENT_PERIOD_QUERY_OPTIMIZATION,
)


def is_statement_period_query_optimization_ff_enabled():
    """Get feature state."""
    return is_feature_enabled(
        ABACUS_STATEMENT_PERIOD_QUERY_OPTIMIZATION, g.request_context
    )


def is_abacus_contract_page_update_run_controller_ff_enabled():
    """Get feature state."""
    return is_feature_enabled(
        ABACUS_CONTRACT_PAGE_UPDATE_RUN_CONTROLLER, g.request_context
    )


def is_abacus_auto_generate_adjustments_flowthrough_ff_enabled():
    """Get feature state."""
    return is_feature_enabled(
        ABACUS_AUTO_GENERATE_ADJUSTMENTS_FLOWTHROUGH, g.request_context
    )