electrum-bitcoinprivate/plugins/audio_modem.py

from electrum.plugins import BasePlugin, hook
from electrum_gui.qt.util import WaitingDialog, EnterButton
from electrum.util import print_msg
from electrum.i18n import _

from PyQt4.QtGui import *
from PyQt4.QtCore import *

import traceback
import zlib
import json
from io import BytesIO

try:
    import amodem
    print_msg('Audio MODEM is enabled.')
except ImportError:
    amodem = None
    print_msg('Audio MODEM is not found.')


class Plugin(BasePlugin):

    def __init__(self, config, name):
        BasePlugin.__init__(self, config, name)
        if self.is_available():
            self.modem_config = amodem.config.slowest()

    def fullname(self):
        return 'Audio MODEM'

    def description(self):
        return ('Provides support for air-gapped transaction signing.\n\n'
                'Requires http://github.com/romanz/amodem/')

    def is_available(self):
        return amodem is not None

    is_enabled = is_available

    def requires_settings(self):
        return True

    def settings_widget(self, window):
        return EnterButton(_('Settings'), self.settings_dialog)

    def settings_dialog(self):
        d = QDialog()
        d.setWindowTitle("Settings")

        layout = QGridLayout(d)
        layout.addWidget(QLabel(_('Bit rate [kbps]: ')), 0, 0)

        bitrates = list(sorted(amodem.config.bitrates.keys()))

        def _index_changed(index):
            bitrate = bitrates[index]
            self.modem_config = amodem.config.bitrates[bitrate]

        combo = QComboBox()
        combo.addItems(map(str, bitrates))
        combo.currentIndexChanged.connect(_index_changed)
        layout.addWidget(combo, 0, 1)

        ok_button = QPushButton(_("OK"))
        ok_button.clicked.connect(d.accept)
        layout.addWidget(ok_button, 1, 1)

        return bool(d.exec_())

    @hook
    def transaction_dialog(self, dialog):
        b = QPushButton()
        b.setIcon(QIcon(":icons/speaker.png"))

        def handler():
            blob = json.dumps(dialog.tx.as_dict())
            self.sender = self._send(parent=dialog, blob=blob)
            self.sender.start()
        b.clicked.connect(handler)
        dialog.buttons.insertWidget(1, b)

    @hook
    def scan_text_edit(self, parent):
        def handler():
            self.receiver = self._recv(parent=parent)
            self.receiver.start()
        button = add_button(parent=parent, icon_name=':icons/microphone.png')
        button.clicked.connect(handler)

    @hook
    def show_text_edit(self, parent):
        def handler():
            blob = str(parent.toPlainText())
            self.sender = self._send(parent=parent, blob=blob)
            self.sender.start()
        button = add_button(parent=parent, icon_name=':icons/speaker.png')
        button.clicked.connect(handler)

    def _send(self, parent, blob):
        def sender_thread():
            try:
                audio_interface = amodem.audio.Interface(self.modem_config)
                src = BytesIO(blob)
                dst = audio_interface.player()
                amodem.send.main(config=self.modem_config, src=src, dst=dst)
            except Exception:
                traceback.print_exc()

        print_msg('Sending:', repr(blob))
        blob = zlib.compress(blob)

        kbps = self.modem_config.modem_bps / 1e3
        msg = 'Sending to Audio MODEM ({0:.1f} kbps)...'.format(kbps)
        return WaitingDialog(parent=parent, message=msg, run_task=sender_thread)

    def _recv(self, parent):
        def receiver_thread():
            try:
                audio_interface = amodem.audio.Interface(self.modem_config)
                src = audio_interface.recorder()
                dst = BytesIO()
                amodem.recv.main(config=self.modem_config, src=src, dst=dst)
                return dst.getvalue()
            except Exception:
                traceback.print_exc()

        def on_success(blob):
            if blob:
                blob = zlib.decompress(blob)
                print_msg('Received:', repr(blob))
                parent.setText(blob)

        kbps = self.modem_config.modem_bps / 1e3
        msg = 'Receiving from Audio MODEM ({0:.1f} kbps)...'.format(kbps)
        return WaitingDialog(parent=parent, message=msg,
                             run_task=receiver_thread, on_success=on_success)


def add_button(parent, icon_name):
    audio_button = QToolButton(parent)
    audio_button.setIcon(QIcon(icon_name))
    audio_button.setStyleSheet("QToolButton { border: none; padding: 0px; }")
    audio_button.setVisible(True)

    parent_resizeEvent = parent.resizeEvent

    def resizeEvent(e):
        result = parent_resizeEvent(e)
        qr_button = parent.button
        left = qr_button.geometry().left() - audio_button.sizeHint().width()
        top = qr_button.geometry().top()
        audio_button.move(left, top)
        return result

    parent.resizeEvent = resizeEvent
    return audio_button