mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2024-09-21 08:31:52 +03:00
7fa1861def
* wallet fixes * Don't show false positive synched * Code cleanup and lint * Fixes * Revert issue * Fix reorg issue * WIP wallet node * More wallet refactor * More wallet changes * More * Fix subscription bugs * Fix some tests * Fix pool tests * More tweaks * Lint and small issues * call update_ui at the correct points * Small changes * new peak queue * Fix peer height issue * Rollback more for safety, and tweak logging * Small WSM style fixes * Change fork point in long_sync * More fixes with real world testing * Fix reversed filter * Fix function name * Fix coin store bug properly * Raise CancelledError * Fix rollback issue * Lint * Small fix * Fix CAT issue * Fix test trades race condition * Fix test trades race condition * Try to reduce flakiness * Test coin store and fix additional method * Improve flakiness
164 lines
6.1 KiB
Python
164 lines
6.1 KiB
Python
from chia.protocols import full_node_protocol, introducer_protocol, wallet_protocol
|
|
from chia.server.outbound_message import NodeType
|
|
from chia.server.ws_connection import WSChiaConnection
|
|
from chia.types.mempool_inclusion_status import MempoolInclusionStatus
|
|
from chia.util.api_decorators import api_request, peer_required, execute_task
|
|
from chia.util.errors import Err
|
|
from chia.wallet.wallet_node import WalletNode
|
|
|
|
|
|
class WalletNodeAPI:
|
|
wallet_node: WalletNode
|
|
|
|
def __init__(self, wallet_node) -> None:
|
|
self.wallet_node = wallet_node
|
|
|
|
@property
|
|
def log(self):
|
|
return self.wallet_node.log
|
|
|
|
@property
|
|
def api_ready(self):
|
|
return self.wallet_node.logged_in
|
|
|
|
@peer_required
|
|
@api_request
|
|
async def respond_removals(self, response: wallet_protocol.RespondRemovals, peer: WSChiaConnection):
|
|
pass
|
|
|
|
async def reject_removals_request(self, response: wallet_protocol.RejectRemovalsRequest, peer: WSChiaConnection):
|
|
"""
|
|
The full node has rejected our request for removals.
|
|
"""
|
|
pass
|
|
|
|
@api_request
|
|
async def reject_additions_request(self, response: wallet_protocol.RejectAdditionsRequest):
|
|
"""
|
|
The full node has rejected our request for additions.
|
|
"""
|
|
pass
|
|
|
|
@execute_task
|
|
@peer_required
|
|
@api_request
|
|
async def new_peak_wallet(self, peak: wallet_protocol.NewPeakWallet, peer: WSChiaConnection):
|
|
"""
|
|
The full node sent as a new peak
|
|
"""
|
|
self.wallet_node.node_peaks[peer.peer_node_id] = (peak.height, peak.header_hash)
|
|
await self.wallet_node.new_peak_queue.new_peak_wallet(peak, peer)
|
|
|
|
@api_request
|
|
async def reject_block_header(self, response: wallet_protocol.RejectHeaderRequest):
|
|
"""
|
|
The full node has rejected our request for a header.
|
|
"""
|
|
pass
|
|
|
|
@api_request
|
|
async def respond_block_header(self, response: wallet_protocol.RespondBlockHeader):
|
|
pass
|
|
|
|
@peer_required
|
|
@api_request
|
|
async def respond_additions(self, response: wallet_protocol.RespondAdditions, peer: WSChiaConnection):
|
|
pass
|
|
|
|
@api_request
|
|
async def respond_proof_of_weight(self, response: full_node_protocol.RespondProofOfWeight):
|
|
pass
|
|
|
|
@peer_required
|
|
@api_request
|
|
async def transaction_ack(self, ack: wallet_protocol.TransactionAck, peer: WSChiaConnection):
|
|
"""
|
|
This is an ack for our previous SendTransaction call. This removes the transaction from
|
|
the send queue if we have sent it to enough nodes.
|
|
"""
|
|
assert peer.peer_node_id is not None
|
|
name = peer.peer_node_id.hex()
|
|
status = MempoolInclusionStatus(ack.status)
|
|
if self.wallet_node.wallet_state_manager is None:
|
|
return None
|
|
if status == MempoolInclusionStatus.SUCCESS:
|
|
self.wallet_node.log.info(f"SpendBundle has been received and accepted to mempool by the FullNode. {ack}")
|
|
elif status == MempoolInclusionStatus.PENDING:
|
|
self.wallet_node.log.info(f"SpendBundle has been received (and is pending) by the FullNode. {ack}")
|
|
else:
|
|
if not self.wallet_node.is_trusted(peer) and ack.error == Err.NO_TRANSACTIONS_WHILE_SYNCING.name:
|
|
self.wallet_node.log.info(f"Peer {peer.get_peer_info()} is not synced, closing connection")
|
|
await peer.close()
|
|
return
|
|
self.wallet_node.log.warning(f"SpendBundle has been rejected by the FullNode. {ack}")
|
|
if ack.error is not None:
|
|
await self.wallet_node.wallet_state_manager.remove_from_queue(ack.txid, name, status, Err[ack.error])
|
|
else:
|
|
await self.wallet_node.wallet_state_manager.remove_from_queue(ack.txid, name, status, None)
|
|
|
|
@peer_required
|
|
@api_request
|
|
async def respond_peers_introducer(
|
|
self, request: introducer_protocol.RespondPeersIntroducer, peer: WSChiaConnection
|
|
):
|
|
if self.wallet_node.wallet_peers is not None:
|
|
await self.wallet_node.wallet_peers.respond_peers(request, peer.get_peer_info(), False)
|
|
|
|
if peer is not None and peer.connection_type is NodeType.INTRODUCER:
|
|
await peer.close()
|
|
|
|
@peer_required
|
|
@api_request
|
|
async def respond_peers(self, request: full_node_protocol.RespondPeers, peer: WSChiaConnection):
|
|
if self.wallet_node.wallet_peers is None:
|
|
return None
|
|
|
|
self.log.info(f"Wallet received {len(request.peer_list)} peers.")
|
|
await self.wallet_node.wallet_peers.respond_peers(request, peer.get_peer_info(), True)
|
|
|
|
return None
|
|
|
|
@api_request
|
|
async def respond_puzzle_solution(self, request: wallet_protocol.RespondPuzzleSolution):
|
|
if self.wallet_node.wallet_state_manager is None:
|
|
return None
|
|
await self.wallet_node.wallet_state_manager.puzzle_solution_received(request)
|
|
|
|
@api_request
|
|
async def reject_puzzle_solution(self, request: wallet_protocol.RejectPuzzleSolution):
|
|
self.log.warning(f"Reject puzzle solution: {request}")
|
|
|
|
@api_request
|
|
async def respond_header_blocks(self, request: wallet_protocol.RespondHeaderBlocks):
|
|
pass
|
|
|
|
@api_request
|
|
async def reject_header_blocks(self, request: wallet_protocol.RejectHeaderBlocks):
|
|
self.log.warning(f"Reject header blocks: {request}")
|
|
|
|
@execute_task
|
|
@peer_required
|
|
@api_request
|
|
async def coin_state_update(self, request: wallet_protocol.CoinStateUpdate, peer: WSChiaConnection):
|
|
await self.wallet_node.new_peak_queue.full_node_state_updated(request, peer)
|
|
|
|
@api_request
|
|
async def respond_to_ph_update(self, request: wallet_protocol.RespondToPhUpdates):
|
|
pass
|
|
|
|
@api_request
|
|
async def respond_to_coin_update(self, request: wallet_protocol.RespondToCoinUpdates):
|
|
pass
|
|
|
|
@api_request
|
|
async def respond_children(self, request: wallet_protocol.RespondChildren):
|
|
pass
|
|
|
|
@api_request
|
|
async def respond_ses_hashes(self, request: wallet_protocol.RespondSESInfo):
|
|
pass
|
|
|
|
@api_request
|
|
async def respond_blocks(self, request: full_node_protocol.RespondBlocks) -> None:
|
|
pass
|