from . import get_hw_model_as_number


def configure(env, features_wanted, defines, sources):
    features_available = []
    hw_model = get_hw_model_as_number('T2B1')
    hw_revision = 3
    board = 'trezor_r_v3.h'
    display = "ug-2828tswig01.c"

    defines += [f'TREZOR_BOARD=\\"boards/{board}\\"', ]
    defines += [f'HW_MODEL={hw_model}', ]
    defines += [f'HW_REVISION={hw_revision}', ]
    sources += [f'embed/trezorhal/displays/{display}', ]

    if "input" in features_wanted:
        sources += ['embed/trezorhal/button.c']
        features_available.append("button")

    if "rgb_led" in features_wanted:
        sources += ['embed/trezorhal/rgb_led.c']
        features_available.append("rgb_led")

    if "sbu" in features_wanted:
        sources += ['embed/trezorhal/sbu.c', ]
        features_available.append("sbu")

    env.get('ENV')['TREZOR_BOARD'] = board

    return features_available