wscript

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from __future__ import print_function

import os
import shutil
import subprocess
import urllib2
import tarfile
import hashlib


gdb_url = "http://ftp.gnu.org/gnu/gdb/gdb-7.11.tar.gz"
gdb_archive_length = 34526368
gdb_archive_sha256 = "9382f5534aa0754169e1e09b5f1a3b77d1fa8c59c1e57617e0" \
                     "6af37cb29c669a"


def hash(path):
    digest = hashlib.sha256()
    with open(path, "rb") as file:
        buf = file.read(4096)
        while len(buf) > 0:
            digest.update(buf)
            buf = file.read(4096)
    return digest.hexdigest()


def build_gdb(ctx):
    try:
        build_dir = ctx.out_dir
    except AttributeError as e:
        build_dir = Context.out_dir
    if not build_dir:
        ctx.fatal("Could not determine output dir - has configure been run?")

    gdb_build_dir = os.path.join(build_dir, "gdb-build")
    gdb_extract_dir = os.path.join(build_dir, "gdb-source")
    gdb_src_dir = os.path.join(gdb_extract_dir, "gdb-7.11")
    gdb_src_zip = os.path.join(build_dir, "gdb-7.11.tar.gz")
    if (not os.path.isfile(gdb_src_zip) or
            hash(gdb_src_zip) != gdb_archive_sha256):
        print("Downloading GDB 7.11...")

        gdb = urllib2.urlopen(gdb_url)
        if gdb.getcode() != 200:
            raise BaseException("GDB could not be downloaded, error code {}".
                                format(gdb.getcode()))

        info = gdb.info()
        length = info["Content-Length"]
        total_read = 0

        with open(gdb_src_zip, 'wb') as archive:
            while True:
                data = gdb.read(16384)
                if len(data) < 1:
                    break
                else:
                    total_read += len(data)
                    archive.write(data)
                    print("\r{:0.2f} %".format((total_read / float(length)) *
                                               100.0), end="")
        print("\n")

    gdb_bin_path = os.path.join(gdb_build_dir, "gdb")
    if os.path.isfile(gdb_bin_path):
        print("%s exists, assuming GDB is built" % gdb_bin_path)
    else:
        print("Extracting GDB...")

        with tarfile.open(gdb_src_zip) as archive:
            archive.extractall(gdb_extract_dir)

        cwd = os.getcwd()
        os.chdir(gdb_src_dir)

        print("Compiling and installing GDB...")

        result = subprocess.call("./configure --prefix={0} --bindir={0}"
                             " --with-python"
                             .format(gdb_build_dir), shell=True)
        try:
            if result == 0:
                result == subprocess.call("make -j4", shell=True)

                if result != 0:
                    raise BaseException("GDB could no be compiled")
                else:
                    result == subprocess.call("make install", shell=True)

                    if result != 0:
                        raise BaseException("GDB could not be installed")
            else:
                raise BaseException("GDB could not be configured")
        except Exception as exc:
            print(exc.message)
            shutil.rmtree(gdb_extract_dir, ignore_errors=True)
            shutil.rmtree(gdb_build_dir, ignore_errors=True)

        os.chdir(cwd)


def patch_lldb(conf):
    exit_status = subprocess.call(["./util/lldb_patch.sh", conf.options.lldb],
                                  stdout=subprocess.PIPE,
                                  stderr=subprocess.PIPE)

    if exit_status == 0:
        print("LLDB patched")
    else:
        print("LLDB not found")


def options(opt):
    opt.add_option("-l", "--lldb", default="", action="store",
                   help="version of LLDB to use")

    opt.load("python")
    opt.load("compiler_cxx")


def configure(conf):
    conf.load("python")
    conf.load("compiler_cxx")

    patch_lldb(conf)

    conf.check_python_version((2, 7, 0))
    if conf.options.lldb:
        conf.check_python_module("lldb")
    conf.check_python_module("enum")
    conf.check_python_module("gi.repository.Gtk")
    conf.check_python_module("gi.repository.Gdk")
    conf.check_python_module("gi.repository.GtkSource")
    conf.check_python_module("cairo")
    conf.check_python_module("numpy")
    conf.check_python_module("matplotlib")
    conf.check_python_module("clang.cindex")


def build(ctx):
    ctx.recurse("debugger")
    ctx.recurse("examples")
    build_gdb(ctx)


def download(ctx):
    apt_args = ["make", "g++", "gawk", "libtool", "texinfo", "clang-3.6", "python-dev",
                "python-matplotlib", "python-enum34", "python-clang-3.6",
                "python-gi", "python-gobject", "python-gi-cairo",
                "gir1.2-gtk-3.0", "gir1.2-gtksource-3.0"]
    subprocess.check_call(["sudo", "apt-get", "install"] + apt_args)


def clean(ctx):
    import shutil
    shutil.rmtree("./docs", True)
    shutil.rmtree("./build", True)
    subprocess.call(["find", ".", "-name", "*.pyc", "-delete"])


def docs(ctx):
    try:
        import epydoc
        subprocess.call(["epydoc", "epydoc", "-v", "--config", "epydoc"])
    except ImportError:
        raise ImportError("Couldn't not import package epydoc,"
                          "do you have it installed?")