fleetmesh-node 0.2.0__py3-none-any.whl

node/commands/compute_cmds.py

@@ -0,0 +1,401 @@
+"""Compute market, order book, DVM gateway, and settlement rails."""
+from __future__ import annotations
+
+import argparse
+
+from node.config.parameter import Parameter
+from node.planes.services.compute.compartment import Compute
+from node.planes.services.dvm.gateway import DVMGateway
+from node.planes.services.settle.compartment import Settlement
+from node.planes.transport.bridge.compartment import RelayBridge
+from .adapter import CommandDispatcher
+from .base import _ask, add_command, emit, fail, load_json_arg
+
+
+# ---- 1. Relay Bridge --------------------------------------------------------
+
+def _bridge_status(r: dict) -> None:
+    print('Relay Bridge: %s' % ('enabled' if r.get('enabled') else 'disabled'))
+    print('  relays configured: %d, connected: %d' % (r.get('relays_configured', 0), r.get('relays_connected', 0)))
+    ing = r.get('ingress', {})
+    egr = r.get('egress', {})
+    print('  ingress: stored=%d, duplicates=%d, refused=%d' % (ing.get('stored', 0), ing.get('duplicates', 0), ing.get('refused', 0)))
+    print('  egress:  sent=%d, failed=%d, pending=%d' % (egr.get('sent', 0), egr.get('failed', 0), egr.get('pending', 0)))
+    relays = r.get('relays', {})
+    if relays:
+        print('\n  Relays:')
+        for url, stats in relays.items():
+            print('    %-30s  %s (in=%d, out=%d, errs=%d)' % (
+                url, 'UP' if stats.get('connected') else 'DOWN',
+                stats.get('ingress_count', 0), stats.get('egress_count', 0), stats.get('errors_count', 0)))
+
+
+def _bridge_cursor(res: list) -> None:
+    if not res:
+        print('No relay bridge checkpoints recorded.')
+        return
+    print('Bridge Checkpoints:')
+    for c in res:
+        print(f"  {c.get('relay')}: last_reconciled={c.get('last_reconciled_at', 0)} last_event={c.get('last_event_at', 0)} (events={c.get('events_reconciled', 0)})")
+
+
+def _cmd_bridge_publish(a: argparse.Namespace) -> int:
+    if not a.event:
+        fail('publish requires --event <json>')
+    targets = [a.relay] if a.relay else None
+    res = _ask(a.dir, 'bridge.publish', event=load_json_arg(a.event, '--event'), relays=targets, timeout=15)
+    return emit(a, res, lambda res: print(str(res)))
+
+
+def _cmd_bridge_add_relay(a: argparse.Namespace) -> int:
+    if not a.relay:
+        fail('add-relay requires --relay <url>')
+    res = _ask(a.dir, 'bridge.add_relay', url=a.relay, timeout=10)
+    return emit(a, res, lambda res: print(f"added {a.relay}"))
+
+
+def _cmd_bridge_remove_relay(a: argparse.Namespace) -> int:
+    if not a.relay:
+        fail('remove-relay requires --relay <url>')
+    res = _ask(a.dir, 'bridge.remove_relay', url=a.relay, timeout=10)
+    return emit(a, res, lambda res: print(f"removed {a.relay}"))
+
+
+def _cmd_bridge_sync(a: argparse.Namespace) -> int:
+    f = load_json_arg(a.filter, '--filter') if a.filter else None
+    res = _ask(a.dir, 'bridge.sync', filter=f, relay=a.relay, timeout=30)
+    return emit(a, res, lambda res: print(f"sync complete: {res.get('ingested', 0)} ingested, {res.get('events_found', 0)} found"))
+
+
+def _cmd_bridge_reconcile(a: argparse.Namespace) -> int:
+    f = load_json_arg(a.filter, '--filter') if a.filter else None
+    res = _ask(a.dir, 'bridge.reconcile', filter=f, relay=a.relay, timeout=30)
+    return emit(a, res, lambda res: print(f"reconcile complete: {res.get('stored', 0)} stored, {res.get('examined', 0)} examined from {res.get('relay')}"))
+
+
+_bridge_map = {c.key: c for c in RelayBridge.COMMANDS}
+bridge_dispatcher = CommandDispatcher('bridge', default_action='status')
+bridge_dispatcher.add_command('status', _bridge_map['show'], human=_bridge_status, timeout=10)
+bridge_dispatcher.add_command('cursor', _bridge_map['cursor'], human=_bridge_cursor, timeout=10)
+bridge_dispatcher.add_custom('publish', _cmd_bridge_publish, parameters=(
+    Parameter(key='relay', name='relay', description='relay URL for bridge action', type=str, default=''),
+    Parameter(key='event', name='event', description='event JSON string for publish', type=str, default=''),
+))
+bridge_dispatcher.add_custom('add-relay', _cmd_bridge_add_relay, parameters=(
+    Parameter(key='relay', name='relay', description='relay URL for bridge action', type=str, default=''),
+))
+bridge_dispatcher.add_custom('remove-relay', _cmd_bridge_remove_relay, parameters=(
+    Parameter(key='relay', name='relay', description='relay URL for bridge action', type=str, default=''),
+))
+bridge_dispatcher.add_custom('sync', _cmd_bridge_sync, parameters=(
+    Parameter(key='filter', name='filter', description='filter JSON string for sync or reconcile', type=str, default=''),
+    Parameter(key='relay', name='relay', description='relay URL for bridge action', type=str, default=''),
+))
+bridge_dispatcher.add_custom('reconcile', _cmd_bridge_reconcile, parameters=(
+    Parameter(key='filter', name='filter', description='filter JSON string for sync or reconcile', type=str, default=''),
+    Parameter(key='relay', name='relay', description='relay URL for bridge action', type=str, default=''),
+))
+
+
+def cmd_bridge(a: argparse.Namespace) -> int:
+    return bridge_dispatcher.dispatch(a)
+
+
+# ---- 2. Compute Market & Order Book -----------------------------------------
+
+def _compute_show(r: dict) -> None:
+    print('Compute Compartment:')
+    print('  offers published:  %d' % r.get('offers_published', 0))
+    print('  active asks:       %d' % r.get('active_asks', 0))
+    print('  order-book offers: %d' % r.get('indexed_offers', 0))
+    print('  order-book asks:   %d' % r.get('indexed_asks', 0))
+    print('  receipts held:     %d' % r.get('receipts_held', 0))
+    print('  jobs executed:     %d' % r.get('jobs_executed', 0))
+    print('  available engines: %s' % ', '.join(r.get('available_engines', [])))
+    print('  active sessions:   %d' % r.get('active_sessions', 0))
+
+
+def _compute_book(r: dict) -> None:
+    print('Compute Order Book:')
+    print('  Offers (%d):' % r.get('offers_count', 0))
+    for off in r.get('offers', []):
+        execs = ', '.join(f"{e.get('engine')}@{e.get('rate')}{e.get('unit')}" for e in off.get('executors', []))
+        print('    - %s... [%s] exp=%s' % (off.get('pubkey', '')[:16], execs, off.get('expiration', 0)))
+    print('  Asks (%d):' % r.get('asks_count', 0))
+    for ask in r.get('asks', []):
+        print('    - %s: %s bid=%s%s exp=%s' % (
+            ask.get('ask_id'), ask.get('executor'), ask.get('bid_price'), ask.get('bid_currency'), ask.get('expiration', 0)
+        ))
+
+
+def _compute_models(r: dict) -> None:
+    print('Available AI/LLM Models:')
+    print('  Default:  %s' % r.get('default_model'))
+    print('  Endpoint: %s' % r.get('endpoint'))
+    print('  Features: %s' % ', '.join(r.get('features', [])))
+    print('  Models:')
+    for m in r.get('available_models', []):
+        print('    - %s' % m)
+
+
+def _cmd_compute_offer(a: argparse.Namespace) -> int:
+    engine = getattr(a, 'engine', None) or 'wasm'
+    rate = getattr(a, 'rate', None) or '45'
+    r = _ask(a.dir, 'compute.offer', engine=engine, msat_per_mfuel=rate, timeout=10)
+    return emit(a, r, lambda r: print('Published Kind 11803 Compute Offer: %s' % r.get('id')))
+
+
+def _cmd_compute_match(a: argparse.Namespace) -> int:
+    ask_id = getattr(a, 'ask', None)
+    if not ask_id:
+        fail('match requires --ask <ask-id>')
+    r = _ask(a.dir, 'compute.match', ask=ask_id, timeout=10)
+
+    def matches(r: list) -> None:
+        print('Matches for Ask %s:' % ask_id)
+        for m in r:
+            free_str = ' [FREE COMPUTE]' if m.get('is_free') else ''
+            print('  - Seller %s... Engine: %s Rate: %s%s (est: %s msat)%s' % (
+                m.get('seller_pubkey', '')[:16],
+                m.get('executor'),
+                m.get('rate'),
+                m.get('rate_unit'),
+                m.get('estimated_cost_msat'),
+                free_str,
+            ))
+    return emit(a, r, matches)
+
+
+def _cmd_compute_stream(a: argparse.Namespace) -> int:
+    prompt = getattr(a, 'prompt', None)
+    if not prompt:
+        fail('stream requires --prompt <text>')
+    model = getattr(a, 'model', None)
+    kwargs = {'prompt': prompt, 'timeout': 30}
+    if model:
+        kwargs['model'] = model
+    r = _ask(a.dir, 'compute.stream', **kwargs)
+    return emit(a, r, lambda r: print(r.get('output', '')))
+
+
+_compute_map = {c.key: c for c in Compute.COMMANDS}
+compute_dispatcher = CommandDispatcher('compute', default_action='status')
+compute_dispatcher.add_command('status', _compute_map['show'], human=_compute_show, timeout=10)
+compute_dispatcher.add_command('show', _compute_map['show'], human=_compute_show, timeout=10)
+compute_dispatcher.add_command('book', _compute_map['book'], human=_compute_book, timeout=10)
+compute_dispatcher.add_command('receipts', _compute_map['receipts'], human=lambda r: print('Held Compute Receipts: %d' % len(r)), timeout=10)
+compute_dispatcher.add_command('models', _compute_map['models'], human=_compute_models, timeout=10)
+compute_dispatcher.add_command('list-models', _compute_map['models'], human=_compute_models, timeout=10)
+compute_dispatcher.add_custom('offer', _cmd_compute_offer, parameters=(
+    Parameter(key='engine', name='engine', description='executor engine (wasm, llm, bacalhau, zk-verify-groth16, zkvm)', type=str, default=''),
+    Parameter(key='rate', name='rate', description='msat per mfuel rate (0 for free compute)', type=str, default=''),
+))
+compute_dispatcher.add_custom('match', _cmd_compute_match, parameters=(
+    Parameter(key='ask', name='ask', description='ask id or event JSON for matching', type=str, default=''),
+))
+compute_dispatcher.add_custom('stream', _cmd_compute_stream, parameters=(
+    Parameter(key='model', name='model', description='AI/LLM model name', type=str, default=''),
+    Parameter(key='prompt', name='prompt', description='prompt text for LLM streaming inference', type=str, default=''),
+))
+
+
+def cmd_compute(a: argparse.Namespace) -> int:
+    return compute_dispatcher.dispatch(a)
+
+
+# ---- 3. DVM Gateway ---------------------------------------------------------
+
+def _dvm_show(r: dict) -> None:
+    print('DVM Gateway: %s' % ('enabled' if r.get('enabled') else 'disabled'))
+    print('  jobs received:  %d' % r.get('jobs_received', 0))
+    print('  jobs succeeded: %d' % r.get('jobs_succeeded', 0))
+    print('  jobs failed:    %d' % r.get('jobs_failed', 0))
+    print('  total fuel:     %d' % r.get('total_fuel_used', 0))
+    print('  refusals:       %d' % r.get('refusals', 0))
+
+
+def _dvm_result(res: dict) -> None:
+    resp = res.get('response')
+    fb = res.get('feedback')
+    print('DVM Execution Result:')
+    if resp:
+        print('  Response (Kind %s): %s' % (resp.get('kind'), resp.get('content', '')))
+    if fb:
+        print('  Feedback: %s' % (fb.get('content', '')))
+
+
+def _dvm_delegation(res: dict) -> None:
+    print('Order Book Delegation:')
+    print('  Ask ID: %s' % res.get('ask_id'))
+    print('  Matches Found: %d' % res.get('matches_found', 0))
+    for cand in res.get('candidates', []):
+        free_str = ' [FREE COMPUTE]' if cand.get('is_free') else ''
+        print('    - Seller: %s... Rate: %s%s (est: %s msat)%s' % (
+            cand.get('seller_pubkey', '')[:16],
+            cand.get('rate', 0),
+            cand.get('rate_unit', 'msat_per_mfuel'),
+            cand.get('estimated_cost_msat', 0),
+            free_str,
+        ))
+
+
+def _cmd_dvm_process(a: argparse.Namespace) -> int:
+    if not getattr(a, 'request', None):
+        fail('process requires --request <nip-90-json>')
+    res = _ask(a.dir, 'dvm.process', request=load_json_arg(a.request, '--request'), timeout=30)
+    return emit(a, res, _dvm_result)
+
+
+def _cmd_dvm_delegate(a: argparse.Namespace) -> int:
+    if not getattr(a, 'request', None):
+        fail('delegate requires --request <nip-90-json>')
+    res = _ask(a.dir, 'dvm.delegate', request=load_json_arg(a.request, '--request'), timeout=30)
+    return emit(a, res, _dvm_delegation)
+
+
+_dvm_map = {c.key: c for c in DVMGateway.COMMANDS}
+dvm_dispatcher = CommandDispatcher('dvm', default_action='status')
+dvm_dispatcher.add_command('status', _dvm_map['show'], human=_dvm_show, timeout=10)
+dvm_dispatcher.add_command('show', _dvm_map['show'], human=_dvm_show, timeout=10)
+dvm_dispatcher.add_custom('process', _cmd_dvm_process, parameters=(
+    Parameter(key='request', name='request', description='NIP-90 request event JSON for process or delegate', type=str, default=''),
+))
+dvm_dispatcher.add_custom('delegate', _cmd_dvm_delegate, parameters=(
+    Parameter(key='request', name='request', description='NIP-90 request event JSON for process or delegate', type=str, default=''),
+))
+
+
+def cmd_dvm(a: argparse.Namespace) -> int:
+    return dvm_dispatcher.dispatch(a)
+
+
+# ---- 4. Settlement Rails ----------------------------------------------------
+
+def _settle_show(r: dict) -> None:
+    print('Settlement Rails: %s' % ', '.join(r.get('rails', [])))
+    escrows = r.get('escrows', [])
+    print('Active Escrows: %d' % len(escrows))
+    for e in escrows:
+        print('  - Ask: %s... Rail: %s Amount: %d msat Status: %s' % (
+            e.get('ask_id', '')[:16],
+            e.get('rail'),
+            e.get('amount_msat', 0),
+            e.get('status'),
+        ))
+
+
+def _cmd_settle_create(a: argparse.Namespace) -> int:
+    for field in ('ask', 'buyer', 'seller', 'amount', 'hash'):
+        if not getattr(a, field, None):
+            fail(f'create requires --{field}')
+    res = _ask(a.dir, 'settle.create',
+               ask=a.ask, buyer=a.buyer, seller=a.seller,
+               amount_msat=int(a.amount), hash=a.hash,
+               rail=getattr(a, 'rail', 'lightning') or 'lightning',
+               timeout=10)
+    return emit(a, res, lambda res: print('Escrow Created: %s (%s msat on %s)' % (res.get('ask_id'), res.get('amount_msat'), res.get('rail'))))
+
+
+def _cmd_settle_hold(a: argparse.Namespace) -> int:
+    if not getattr(a, 'ask', None):
+        fail('hold requires --ask')
+    res = _ask(a.dir, 'settle.hold', ask=a.ask, proof=getattr(a, 'proof', None), timeout=10)
+    return emit(a, {'held': res}, lambda _: print('Escrow Held: %s' % res), indent=None)
+
+
+def _cmd_settle_settle(a: argparse.Namespace) -> int:
+    if not getattr(a, 'ask', None) or not getattr(a, 'preimage', None):
+        fail('settle requires --ask and --preimage')
+    res = _ask(a.dir, 'settle.settle', ask=a.ask, preimage=a.preimage, timeout=10)
+    return emit(a, {'settled': res}, lambda _: print('Escrow Settled: %s' % res), indent=None)
+
+
+def _cmd_settle_cancel(a: argparse.Namespace) -> int:
+    if not getattr(a, 'ask', None):
+        fail('cancel requires --ask')
+    res = _ask(a.dir, 'settle.cancel', ask=a.ask, reason=getattr(a, 'reason', '') or '', timeout=10)
+    return emit(a, {'cancelled': res}, lambda _: print('Escrow Cancelled: %s' % res), indent=None)
+
+
+def _cmd_settle_nwc_pay(a: argparse.Namespace) -> int:
+    if not getattr(a, 'hash', None) or not getattr(a, 'bolt11', None):
+        fail('nwc-pay requires --hash and --bolt11')
+    res = _ask(a.dir, 'settle.nwc_pay', hash=a.hash, bolt11=a.bolt11, timeout=30)
+    return emit(a, res, lambda res: print('NWC Payment Dispatched: Preimage %s' % res.get('preimage')))
+
+
+def _cmd_settle_cashu_swap(a: argparse.Namespace) -> int:
+    if not getattr(a, 'hash', None) or not getattr(a, 'outputs', None):
+        fail('cashu-swap requires --hash and --outputs (JSON string)')
+    outputs = load_json_arg(a.outputs, '--outputs')
+    res = _ask(a.dir, 'settle.cashu_swap', hash=a.hash, outputs=outputs, timeout=30)
+    return emit(a, res, lambda res: print('Cashu Swap Completed: %d fresh signatures' % len(res.get('signatures', []))))
+
+
+def _cmd_settle_cashu_melt(a: argparse.Namespace) -> int:
+    if not getattr(a, 'hash', None) or not getattr(a, 'bolt11', None):
+        fail('cashu-melt requires --hash and --bolt11')
+    res = _ask(a.dir, 'settle.cashu_melt', hash=a.hash, bolt11=a.bolt11, timeout=30)
+    return emit(a, res, lambda res: print('Cashu Melt Completed: Paid=%s Preimage=%s' % (res.get('paid'), res.get('payment_preimage'))))
+
+
+_settle_map = {c.key: c for c in Settlement.COMMANDS}
+settle_dispatcher = CommandDispatcher('settle', default_action='status')
+settle_dispatcher.add_command('status', _settle_map['show'], human=_settle_show, timeout=10)
+settle_dispatcher.add_command('show', _settle_map['show'], human=_settle_show, timeout=10)
+settle_dispatcher.add_custom('create', _cmd_settle_create, parameters=(
+    Parameter(key='ask', name='ask', description='ask id', type=str, default=''),
+    Parameter(key='buyer', name='buyer', description='buyer pubkey', type=str, default=''),
+    Parameter(key='seller', name='seller', description='seller pubkey', type=str, default=''),
+    Parameter(key='amount', name='amount', description='amount in msats', type=int, default=0),
+    Parameter(key='hash', name='hash', description='payment hash (32-byte hex)', type=str, default=''),
+    Parameter(key='rail', name='rail', description='settlement rail', type=str, default='lightning', choices=('lightning', 'cashu', 'nwc', 'mock')),
+))
+settle_dispatcher.add_custom('hold', _cmd_settle_hold, parameters=(
+    Parameter(key='ask', name='ask', description='ask id', type=str, default=''),
+    Parameter(key='proof', name='proof', description='cashu proof or secret token for hold', type=str, default=''),
+))
+settle_dispatcher.add_custom('settle', _cmd_settle_settle, parameters=(
+    Parameter(key='ask', name='ask', description='ask id', type=str, default=''),
+    Parameter(key='preimage', name='preimage', description='payment preimage (32-byte hex)', type=str, default=''),
+))
+settle_dispatcher.add_custom('cancel', _cmd_settle_cancel, parameters=(
+    Parameter(key='ask', name='ask', description='ask id', type=str, default=''),
+    Parameter(key='reason', name='reason', description='escrow cancellation reason', type=str, default=''),
+))
+settle_dispatcher.add_custom('nwc-pay', _cmd_settle_nwc_pay, parameters=(
+    Parameter(key='hash', name='hash', description='payment hash (32-byte hex)', type=str, default=''),
+    Parameter(key='bolt11', name='bolt11', description='Lightning bolt11 invoice string', type=str, default=''),
+))
+settle_dispatcher.add_custom('cashu-swap', _cmd_settle_cashu_swap, parameters=(
+    Parameter(key='hash', name='hash', description='payment hash (32-byte hex)', type=str, default=''),
+    Parameter(key='outputs', name='outputs', description='blinded outputs JSON for Cashu token swap', type=str, default=''),
+))
+settle_dispatcher.add_custom('cashu-melt', _cmd_settle_cashu_melt, parameters=(
+    Parameter(key='hash', name='hash', description='payment hash (32-byte hex)', type=str, default=''),
+    Parameter(key='bolt11', name='bolt11', description='Lightning bolt11 invoice string', type=str, default=''),
+))
+
+
+def cmd_settle(a: argparse.Namespace) -> int:
+    return settle_dispatcher.dispatch(a)
+
+
+# ---- 5. Provider Registration -----------------------------------------------
+
+class ComputeCommandProvider:
+    """Registers compute market, order book, DVM, settlement, and bridge commands."""
+
+    @classmethod
+    def register(cls, sub: argparse._SubParsersAction) -> None:
+        bg = add_command(sub, 'bridge', cmd_bridge, 'interact with the Universal Nostr Relay Bridge', js=True)
+        bridge_dispatcher.register_arguments(bg, action_choices=('status', 'publish', 'add-relay', 'remove-relay', 'sync', 'reconcile', 'cursor'))
+
+        cp = add_command(sub, 'compute', cmd_compute, 'inspect or interact with the compute market and order book', js=True)
+        compute_dispatcher.register_arguments(cp, action_choices=('status', 'show', 'book', 'offer', 'match', 'receipts', 'models', 'stream'))
+
+        dv = add_command(sub, 'dvm', cmd_dvm, 'interact with the NIP-90 Data Vending Machine Gateway', js=True)
+        dvm_dispatcher.register_arguments(dv, action_choices=('status', 'show', 'process', 'delegate'))
+
+        se = add_command(sub, 'settle', cmd_settle, 'inspect settlement rails, escrows, and execute live payments', js=True)
+        settle_dispatcher.register_arguments(se, action_choices=('status', 'show', 'create', 'hold', 'settle', 'cancel', 'nwc-pay', 'cashu-swap', 'cashu-melt'))