#!/usr/bin/python3
# vim: ft=python fileencoding=utf-8 sts=4 sw=4 et:

# Copyright 2014 Florian Bruhin (The Compiler) <mail@qutebrowser.org>
#
# This file is part of qutebrowser.
#
# qutebrowser is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# qutebrowser is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with qutebrowser.  If not, see <http://www.gnu.org/licenses/>.

"""Tester for Qt segfaults with different harfbuzz engines."""

import os
import signal
import sys
import subprocess

import colorama as col


SCRIPT = """
import sys

from PyQt5.QtCore import QUrl
from PyQt5.QtWidgets import QApplication
from PyQt5.QtWebKitWidgets import QWebView

def on_load_finished(ok):
    if ok:
        app.exit(0)
    else:
        app.exit(1)

app = QApplication([])
wv = QWebView()
wv.loadFinished.connect(on_load_finished)
wv.load(QUrl(sys.argv[1]))
#wv.show()
app.exec_()
"""


def print_ret(ret):
    """Print information about an exit status."""
    if ret == 0:
        print("{}success{}".format(col.Fore.GREEN, col.Fore.RESET))
    elif ret == -signal.SIGSEGV:
        print("{}segfault{}".format(col.Fore.RED, col.Fore.RESET))
    else:
        print("{}error {}{}".format(col.Fore.YELLOW, ret, col.Fore.RESET))
    print()


def main():
    """Main entry point."""
    retvals = []
    if len(sys.argv) < 2:
        # pages which previously caused problems
        pages = ['http://www.binpress.com/', 'http://www.20min.ch/',
                 'https://imzdl.com/', 'http://www.the-compiler.org/',
                 'http://phoronix.com', 'http://lenta.ru/']
    else:
        pages = sys.argv[1:]
    for page in pages:
        print("{}==== {} ===={}".format(col.Style.BRIGHT, page,
                                        col.Style.NORMAL))
        print("With system harfbuzz:")
        ret = subprocess.call([sys.executable, '-c', SCRIPT, page])
        print_ret(ret)
        retvals.append(ret)
        print("With QT_HARFBUZZ=old:")
        env = dict(os.environ)
        env['QT_HARFBUZZ'] = 'old'
        ret = subprocess.call([sys.executable, '-c', SCRIPT, page], env=env)
        print_ret(ret)
        retvals.append(ret)
        print("With QT_HARFBUZZ=nev:")
        env = dict(os.environ)
        env['QT_HARFBUZZ'] = 'new'
        ret = subprocess.call([sys.executable, '-c', SCRIPT, page], env=env)
        print_ret(ret)
        retvals.append(ret)
    if all(r == 0 for r in retvals):
        sys.exit(0)
    else:
        sys.exit(1)


if __name__ == '__main__':
    main()