#!/usr/bin/python -u

import os
import string
import random
import shutil
import subprocess
import sys
import time
import tempfile

from devbot import environ
from devbot import config


def run(command):
    environ.setup()

    args = [command, "--home-dir", config.home_dir]

    resolution = config.get_pref("RESOLUTION")
    if resolution:
        args.extend(["--resolution", resolution])

    output = config.get_pref("OUTPUT")
    if output:
        args.extend(["--output", output])

    stdout = open(config.log_path, 'a')
    stderr = stdout

    os.dup2(stdout.fileno(), sys.stdout.fileno())
    os.dup2(stderr.fileno(), sys.stderr.fileno())

    os.execlp(args[0], *args)


def run_test(command, test_path):
    environ.setup()

    temp_dir = tempfile.mkdtemp("sugar-build-test")
    display_path = os.path.join(temp_dir, "display")

    args = [command,
            "--display-path", display_path,
            "--virtual"]

    command_process = subprocess.Popen(args, stdout=subprocess.PIPE)

    while True:
        if not os.path.exists(display_path):
            time.sleep(1)
        else:
            break

    with open(display_path) as f:
        os.environ["DISPLAY"] = f.read()

    os.unlink(display_path)
    os.rmdir(temp_dir)

    try:
        subprocess.check_call(["python", "-u", test_path])
        result = True
    except subprocess.CalledProcessError:
        result = False

    command_process.terminate()

    return result


def collect_logs(source_path):
    logs = {}
    for filename in os.listdir(source_path):
        if filename.endswith(".log"):
            path = os.path.join(source_path, filename)
            with open(path) as f:
                logs[filename] = f.read()

    with open(config.log_path, "w") as f:
        for filename, log in logs.items():
            f.write("\n===== %s =====\n\n%s" % (filename, log))


def _get_random_id():
    return ''.join(random.choice(string.letters) for i in xrange(8))