trezor-firmware/core/src/trezor/wire/thp/channel_manager.py

from typing import TYPE_CHECKING

from storage import cache_thp
from trezor import utils

from . import ChannelState, interface_manager
from .channel import Channel

if TYPE_CHECKING:
    from trezorio import WireInterface  # pyright:ignore[reportMissingImports]


def create_new_channel(iface: WireInterface, buffer: utils.BufferType) -> "Channel":
    channel_cache = cache_thp.get_new_unauthenticated_channel(
        interface_manager.encode_iface(iface)
    )
    r = Channel(channel_cache)
    r.set_buffer(buffer)
    r.set_channel_state(ChannelState.TH1)
    return r


def load_cached_channels(buffer: utils.BufferType) -> dict[int, Channel]:  # TODO
    channels: dict[int, Channel] = {}
    cached_channels = cache_thp.get_all_allocated_channels()
    for c in cached_channels:
        channels[int.from_bytes(c.channel_id, "big")] = Channel(c)
    for c in channels.values():
        c.set_buffer(buffer)
    return channels