15427c96c0
- Change file mode from 644 to 755 for all project files - Add chain_id parameter to get_balance RPC endpoint with default "ait-devnet" - Rename Miner.extra_meta_data to extra_metadata for consistency
143 lines
6.1 KiB
Python
Executable File
143 lines
6.1 KiB
Python
Executable File
with open("/home/oib/windsurf/aitbc/apps/blockchain-node/src/aitbc_chain/rpc/router.py", "r") as f:
|
|
content = f.read()
|
|
|
|
# Update Account endpoint
|
|
content = content.replace(
|
|
""" account = session.get(Account, address)
|
|
if account is None:
|
|
raise HTTPException(status_code=404, detail="Account not found")
|
|
|
|
# Get transaction counts
|
|
sent_count = session.exec(select(func.count()).select_from(Transaction).where(Transaction.sender == address)).one()
|
|
received_count = session.exec(select(func.count()).select_from(Transaction).where(Transaction.recipient == address)).one()""",
|
|
""" account = session.exec(select(Account).where(Account.address == address)).first()
|
|
if account is None:
|
|
raise HTTPException(status_code=404, detail="Account not found")
|
|
|
|
# Get transaction counts
|
|
sent_count = session.exec(select(func.count()).select_from(Transaction).where(Transaction.sender == address)).one()
|
|
received_count = session.exec(select(func.count()).select_from(Transaction).where(Transaction.recipient == address)).one()"""
|
|
)
|
|
|
|
# Replace all hardcoded cfg.chain_id with a query parameter or path parameter where applicable
|
|
content = content.replace(
|
|
"""@router.get("/head", summary="Get the current chain head block")
|
|
async def get_head() -> Dict[str, Any]:""",
|
|
"""@router.get("/head", summary="Get the current chain head block")
|
|
async def get_head(chain_id: str = "ait-devnet") -> Dict[str, Any]:"""
|
|
)
|
|
|
|
content = content.replace(
|
|
"""select(Block).order_by(Block.height.desc()).limit(1)""",
|
|
"""select(Block).where(Block.chain_id == chain_id).order_by(Block.height.desc()).limit(1)"""
|
|
)
|
|
|
|
content = content.replace(
|
|
"""@router.get("/blocks/{height_or_hash}", summary="Get a block by height or hash")
|
|
async def get_block(height_or_hash: str) -> Dict[str, Any]:""",
|
|
"""@router.get("/blocks/{height_or_hash}", summary="Get a block by height or hash")
|
|
async def get_block(height_or_hash: str, chain_id: str = "ait-devnet") -> Dict[str, Any]:"""
|
|
)
|
|
|
|
content = content.replace(
|
|
"""query = select(Block).where(Block.height == height)""",
|
|
"""query = select(Block).where(Block.chain_id == chain_id).where(Block.height == height)"""
|
|
)
|
|
|
|
content = content.replace(
|
|
"""query = select(Block).where(Block.hash == height_or_hash)""",
|
|
"""query = select(Block).where(Block.chain_id == chain_id).where(Block.hash == height_or_hash)"""
|
|
)
|
|
|
|
content = content.replace(
|
|
""" txs = session.exec(select(Transaction).where(Transaction.block_height == block.height)).all()""",
|
|
""" txs = session.exec(select(Transaction).where(Transaction.chain_id == chain_id).where(Transaction.block_height == block.height)).all()"""
|
|
)
|
|
|
|
content = content.replace(
|
|
""" receipts = session.exec(select(Receipt).where(Receipt.block_height == block.height)).all()""",
|
|
""" receipts = session.exec(select(Receipt).where(Receipt.chain_id == chain_id).where(Receipt.block_height == block.height)).all()"""
|
|
)
|
|
|
|
content = content.replace(
|
|
"""@router.get("/transactions/{tx_hash}", summary="Get a transaction by hash")
|
|
async def get_transaction(tx_hash: str) -> Dict[str, Any]:""",
|
|
"""@router.get("/transactions/{tx_hash}", summary="Get a transaction by hash")
|
|
async def get_transaction(tx_hash: str, chain_id: str = "ait-devnet") -> Dict[str, Any]:"""
|
|
)
|
|
|
|
content = content.replace(
|
|
""" tx = session.exec(select(Transaction).where(Transaction.tx_hash == tx_hash)).first()""",
|
|
""" tx = session.exec(select(Transaction).where(Transaction.chain_id == chain_id).where(Transaction.tx_hash == tx_hash)).first()"""
|
|
)
|
|
|
|
content = content.replace(
|
|
""" # If not in block, check mempool
|
|
if tx is None:
|
|
mempool_txs = get_mempool().list_transactions()""",
|
|
""" # If not in block, check mempool
|
|
if tx is None:
|
|
mempool_txs = get_mempool().list_transactions(chain_id)"""
|
|
)
|
|
|
|
content = content.replace(
|
|
"""@router.get("/mempool", summary="Get current mempool transactions")
|
|
async def get_mempool_txs() -> List[Dict[str, Any]]:""",
|
|
"""@router.get("/mempool", summary="Get current mempool transactions")
|
|
async def get_mempool_txs(chain_id: str = "ait-devnet") -> List[Dict[str, Any]]:"""
|
|
)
|
|
|
|
content = content.replace(
|
|
""" txs = get_mempool().list_transactions()""",
|
|
""" txs = get_mempool().list_transactions(chain_id)"""
|
|
)
|
|
|
|
content = content.replace(
|
|
"""@router.get("/metrics", summary="Get node metrics")
|
|
async def get_metrics() -> PlainTextResponse:""",
|
|
"""@router.get("/chains", summary="Get supported chains")
|
|
async def get_chains() -> List[str]:
|
|
from ..config import settings as cfg
|
|
return [c.strip() for c in cfg.supported_chains.split(",")]
|
|
|
|
@router.get("/metrics", summary="Get node metrics")
|
|
async def get_metrics() -> PlainTextResponse:"""
|
|
)
|
|
|
|
content = content.replace(
|
|
"""async def import_block(request: ImportBlockRequest) -> Dict[str, Any]:""",
|
|
"""async def import_block(request: ImportBlockRequest, chain_id: str = "ait-devnet") -> Dict[str, Any]:"""
|
|
)
|
|
|
|
content = content.replace(
|
|
""" sync = ChainSync(
|
|
session_factory=session_scope,
|
|
chain_id=cfg.chain_id,
|
|
max_reorg_depth=cfg.max_reorg_depth,
|
|
validator=validator,
|
|
validate_signatures=cfg.sync_validate_signatures,
|
|
)""",
|
|
""" sync = ChainSync(
|
|
session_factory=session_scope,
|
|
chain_id=chain_id,
|
|
max_reorg_depth=cfg.max_reorg_depth,
|
|
validator=validator,
|
|
validate_signatures=cfg.sync_validate_signatures,
|
|
)"""
|
|
)
|
|
|
|
content = content.replace(
|
|
"""@router.get("/syncStatus", summary="Get chain sync status")
|
|
async def sync_status() -> Dict[str, Any]:""",
|
|
"""@router.get("/syncStatus", summary="Get chain sync status")
|
|
async def sync_status(chain_id: str = "ait-devnet") -> Dict[str, Any]:"""
|
|
)
|
|
|
|
content = content.replace(
|
|
""" sync = ChainSync(session_factory=session_scope, chain_id=cfg.chain_id)""",
|
|
""" sync = ChainSync(session_factory=session_scope, chain_id=chain_id)"""
|
|
)
|
|
|
|
with open("/home/oib/windsurf/aitbc/apps/blockchain-node/src/aitbc_chain/rpc/router.py", "w") as f:
|
|
f.write(content)
|