hodlinvoice.py

gistfile1.txt

#!/usr/bin/env python3
from pyln.client import Plugin


plugin = Plugin()

# TODO/FIXME:
# - on restart, all the state in wait_on;
#      - datastore: rpcs on CLN
# - how long to hodl for?
#    build in smarts about

# dictionary: key is the payment_hash
#        value: list of htlcs for that payment_hash
wait_on = {}


@plugin.subscribe('new_block')
def new_block(...):
  # todo: check if too close and fail?
  pass

@plugin.async_hook("htlc_accepted")
def on_htlc_accepted(request, htlc, onion, plugin, **kwargs):
  if htlc['payment_hash'] in wait_on:
    hash_list = wait_on[htlc['payment_hash']]
    hash_list.append(request)
  else:
    request.set_result({'result': 'continue'})


@plugin.method("hodlinvoice")
def hodl(plugin, payment_hash):
  if payment_hash not in wait_on:
    wait_on[payment_hash] = []

  return {'hodl': True, 'payment_hash': payment_hash}


@plugin.method("inspecthodl")
def inspect():
  return {'hodled': wait_on}


@plugin.method("unhodlinvoice")
def unhodl(plugin, payment_hash):
  count = 0
  hodled = False
  if payment_hash in wait_on:
    hodled = True
    for req in wait_on[payment_hash]:
      count += 1
      req.set_result({'result': 'continue'})
    
  return {'hodl': hodled, 
          'payment_hash': payment_hash,
          'unhodled_count': count}


plugin.run()