mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2026-08-28 18:14:19 -05:00
Matt HauffandGitHub
40db4635a8
* Enable PEP604 Ruff rules * Fix harcoded signature in test * Hack CLVMStreamable test with note to fast follow
58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Callable
|
|
|
|
from chia_rs.sized_bytes import bytes32
|
|
|
|
from chia.consensus.blockchain import StateChangeSummary
|
|
|
|
|
|
def get_hints_and_subscription_coin_ids(
|
|
state_change_summary: StateChangeSummary,
|
|
has_coin_subscription: Callable[[bytes32], bool],
|
|
has_puzzle_subscription: Callable[[bytes32], bool],
|
|
) -> tuple[list[tuple[bytes32, bytes]], list[bytes32]]:
|
|
# Precondition: all hints passed in are max 32 bytes long
|
|
# Returns the hints that we need to add to the DB, and the coin ids that need to be looked up
|
|
|
|
# Finds the coin IDs that we need to lookup in order to notify wallets of hinted transactions
|
|
hint: bytes | None
|
|
hints_to_add: list[tuple[bytes32, bytes]] = []
|
|
|
|
# Goes through additions and removals for each block and flattens to a map and a set
|
|
lookup_coin_ids: set[bytes32] = set()
|
|
|
|
def add_if_coin_subscription(coin_id: bytes32) -> None:
|
|
if has_coin_subscription(coin_id):
|
|
lookup_coin_ids.add(coin_id)
|
|
|
|
def add_if_ph_subscription(puzzle_hash: bytes32, coin_id: bytes32) -> None:
|
|
if has_puzzle_subscription(puzzle_hash):
|
|
lookup_coin_ids.add(coin_id)
|
|
|
|
for spend_id, puzzle_hash in state_change_summary.removals:
|
|
# Record all coin_ids that we are interested in, that had changes
|
|
add_if_coin_subscription(spend_id)
|
|
add_if_ph_subscription(puzzle_hash, spend_id)
|
|
|
|
for addition_coin, hint in state_change_summary.additions:
|
|
addition_coin_name = addition_coin.name()
|
|
add_if_coin_subscription(addition_coin_name)
|
|
add_if_ph_subscription(addition_coin.puzzle_hash, addition_coin_name)
|
|
if hint is None:
|
|
continue
|
|
if len(hint) == 32:
|
|
add_if_ph_subscription(bytes32(hint), addition_coin_name)
|
|
|
|
if len(hint) > 0:
|
|
assert len(hint) <= 32
|
|
hints_to_add.append((addition_coin_name, hint))
|
|
|
|
# Goes through all new reward coins
|
|
for reward_coin in state_change_summary.new_rewards:
|
|
reward_coin_name: bytes32 = reward_coin.name()
|
|
add_if_coin_subscription(reward_coin_name)
|
|
add_if_ph_subscription(reward_coin.puzzle_hash, reward_coin_name)
|
|
|
|
return hints_to_add, list(lookup_coin_ids)
|