electrum/gui/qt/lightning_invoice_list.py

# -*- coding: utf-8 -*-
import base64
import binascii
from PyQt5 import QtCore, QtWidgets
from collections import OrderedDict
import logging
from electrum.lightning import lightningCall
from .qrcodewidget import QRDialog

mapping = {0: "r_hash", 1: "pay_req", 2: "settled"}
revMapp = {"r_hash": 0, "pay_req": 1, "settled": 2}
datatable = OrderedDict([])
idx = 0

class MyTableRow(QtWidgets.QTreeWidgetItem):
    def __init__(self, di):
        if "settled" not in di:
            di["settled"] = False
        strs = [str(di[mapping[key]]) for key in range(len(mapping))]
        print(strs)
        super(MyTableRow, self).__init__(strs)
        assert isinstance(di, dict)
        self.di = di
    def __getitem__(self, idx):
        return self.di[idx]
    def __setitem__(self, idx, val):
        self.di[idx] = val
        try:
            self.setData(revMapp[idx], QtCore.Qt.DisplayRole, '{0}'.format(val))
        except KeyError:
            logging.warning("Lightning Invoice field %s unknown", idx)
    def __str__(self):
        return str(self.di)

def addInvoiceRow(new):
    made = MyTableRow(new)
    datatable[new["r_hash"]] = made
    datatable.move_to_end(new["r_hash"], last=False)
    return made

def clickHandler(numInput, treeView, lightningRpc):
    amt = numInput.value()
    if amt < 1:
        print("value too small")
        return
    print("creating invoice with value {}".format(amt))
    global idx
    #obj = {
    #    "r_hash": binascii.hexlify((int.from_bytes(bytearray.fromhex("9500edb0994b7bc23349193486b25c82097045db641f35fa988c0e849acdec29"), "big")+idx).to_bytes(byteorder="big", length=32)).decode("ascii"),
    #    "pay_req": "lntb81920n1pdf258s" + str(idx),
    #    "settled": False
    #}
    #treeView.insertTopLevelItem(0, addInvoiceRow(obj))
    idx += 1
    lightningCall(lightningRpc, "addinvoice")("--amt=" + str(amt))

class LightningInvoiceList(QtWidgets.QWidget):
    def create_menu(self, position):
        menu = QtWidgets.QMenu()
        pay_req = self._tv.currentItem()["pay_req"]
        cb = QtWidgets.QApplication.instance().clipboard()
        def copy():
            print(pay_req)
            cb.setText(pay_req)
        def qr():
            d = QRDialog(pay_req, self, "Lightning invoice")
            d.exec_()
        menu.addAction("Copy payment request", copy)
        menu.addAction("Show payment request as QR code", qr)
        menu.exec_(self._tv.viewport().mapToGlobal(position))
    def lightningWorkerHandler(self, sourceClassName, obj):
        new = {}
        for k, v in obj.items():
            try:
                v = binascii.hexlify(base64.b64decode(v)).decode("ascii")
            except:
                pass
            new[k] = v
        try:
            obj = datatable[new["r_hash"]]
        except KeyError:
            print("lightning payment invoice r_hash {} unknown!".format(new["r_hash"]))
        else:
            for k, v in new.items():
                try:
                    if obj[k] != v: obj[k] = v
                except KeyError:
                    obj[k] = v
    def lightningRpcHandler(self, methodName, obj):
        if methodName != "addinvoice":
            print("ignoring reply {} to {}".format(obj, methodName))
            return
        self._tv.insertTopLevelItem(0, addInvoiceRow(obj))
        
    def __init__(self, parent, lightningWorker, lightningRpc):
        QtWidgets.QWidget.__init__(self, parent)

        lightningWorker.subscribe(self.lightningWorkerHandler)
        lightningRpc.subscribe(self.lightningRpcHandler)

        self._tv=QtWidgets.QTreeWidget(self)
        self._tv.setHeaderLabels([mapping[i] for i in range(len(mapping))])
        self._tv.setColumnCount(len(mapping))
        self._tv.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
        self._tv.customContextMenuRequested.connect(self.create_menu)

        class SatoshiCountSpinBox(QtWidgets.QSpinBox):
            def keyPressEvent(self2, e):
                super(SatoshiCountSpinBox, self2).keyPressEvent(e)
                if QtCore.Qt.Key_Return == e.key():
                    clickHandler(self2, self._tv, lightningRpc)

        numInput = SatoshiCountSpinBox(self)

        button = QtWidgets.QPushButton('Add invoice', self)
        button.clicked.connect(lambda: clickHandler(numInput, self._tv, lightningRpc))

        l=QtWidgets.QVBoxLayout(self)
        h=QtWidgets.QGridLayout(self)
        h.addWidget(numInput, 0, 0)
        h.addWidget(button, 0, 1)
        #h.addItem(QtWidgets.QSpacerItem(100, 200, QtWidgets.QSizePolicy.Preferred, QtWidgets.QSizePolicy.Preferred), 0, 2)
        #h.setSizePolicy(
        h.setColumnStretch(0, 1)
        h.setColumnStretch(1, 1)
        h.setColumnStretch(2, 2)
        l.addLayout(h)
        l.addWidget(self._tv)

        self.resize(2500,1000)

def tick():
  key = "9500edb0994b7bc23349193486b25c82097045db641f35fa988c0e849acdec29"
  if not key in datatable:
      return
  row = datatable[key]
  row["settled"] = not row["settled"]
  print("data changed")

if __name__=="__main__":
    from sys import argv, exit

    a=QtWidgets.QApplication(argv)

    w=LightningInvoiceList()
    w.show()
    w.raise_()

    timer = QtCore.QTimer()
    timer.timeout.connect(tick)
    timer.start(1000)
    exit(a.exec_())