#!/usr/bin/env python

import sys

from PySide2 import QtCore, QtWidgets

import gemini

class GViewport(QtWidgets.QTextBrowser):

    hoverUrlChanged = QtCore.Signal(str)
    resourceLoadedStatus = QtCore.Signal(str)

    def __init__(self, address_bar):
        self.address_bar = address_bar
        self._hover_url = None
        QtWidgets.QTextBrowser.__init__(self)

    def mouseMoveEvent(self, event):
        cur = self.cursorForPosition(event.localPos().toPoint())
        hover_url = cur.charFormat().anchorHref()
        if hover_url:
            hover_url = gemini.urljoin(self.address_bar.text(), hover_url)
        if hover_url != self._hover_url:
            self._hover_url = hover_url
            self.hoverUrlChanged.emit(self._hover_url)
        return super().mouseMoveEvent(event)

    def loadResource(self, type_, url):
        gem = gemini.get(url.toString())
        if 'body' in gem:
            html = gemini.gem2html(gem['body'])
        else:
            html = '<h1>{} {}</h1>'.format(gem['status'], gem['meta'])
        self.resourceLoadedStatus.emit('{status} {meta}'.format(status=gem['status'], meta=gem['meta']))
        return html

    def setRawSource(self):
        return self.setSource(QtCore.QUrl(self.address_bar.text()))

class GUrlBar(QtWidgets.QLineEdit):

    def __init__(self):
        QtWidgets.QLineEdit.__init__(self)

    def setUrl(self, url):
        return self.setText(url.toDisplayString())

class GBrowser(QtWidgets.QMainWindow):

    def __init__(self):
        QtWidgets.QMainWindow.__init__(self)
        self.init_ui()

    def init_ui(self):
        request_status = QtWidgets.QLabel()
        self.statusBar().addWidget(request_status)

        back = QtWidgets.QPushButton("back")
        forward = QtWidgets.QPushButton("forward")
        address = GUrlBar()
        toolbar = QtWidgets.QToolBar()
        toolbar.setObjectName('navigationToolbar')
        toolbar.addWidget(back)
        toolbar.addWidget(forward)
        toolbar.addWidget(address)
        browser = GViewport(address)

        back.clicked.connect(browser.backward)
        forward.clicked.connect(browser.forward)
        browser.sourceChanged.connect(address.setUrl)
        address.returnPressed.connect(browser.setRawSource)
        browser.hoverUrlChanged.connect(self.set_status_url)
        browser.resourceLoadedStatus.connect(request_status.setText)

        self.addToolBar(toolbar)
        self.setCentralWidget(browser)

        browser.setSource(QtCore.QUrl('gemini://gemini.circumlunar.space/'))

        settings = QtCore.QSettings("xcolour.net", "GeminiBrowser")
        self.restoreGeometry(settings.value("geometry"))
        self.restoreState(settings.value("windowState"))
        self.setWindowTitle('Gemini Browser')
        self.show()

    def set_status_url(self, url):
        if url:
            self.statusBar().showMessage(url)
        else:
            self.statusBar().clearMessage()

    def closeEvent(self, event):
        settings = QtCore.QSettings("xcolour.net", "GeminiBrowser")
        settings.setValue("geometry", self.saveGeometry())
        settings.setValue("windowState", self.saveState())
        super().closeEvent(event)

app = QtWidgets.QApplication(sys.argv)
ex = GBrowser()
sys.exit(app.exec_())