import os.path
import random
import time

import libcxx.test.executor

from libcxx.android import adb
from lit.util import executeCommand  # pylint: disable=import-error


class AdbExecutor(libcxx.test.executor.RemoteExecutor):
    def __init__(self, device_dir, serial=None):
        # TODO(danalbert): Should factor out the shared pieces of SSHExecutor
        # so we don't have this ugly parent constructor...
        super(AdbExecutor, self).__init__()
        self.device_dir = device_dir
        self.serial = serial
        self.local_run = executeCommand

    def _remote_temp(self, is_dir):
        # Android didn't have mktemp until M :(
        # Just use a random number generator and hope for no collisions. Should
        # be very unlikely for a 64-bit number.
        test_dir_name = 'test.{}'.format(random.randrange(2 ** 64))
        return os.path.join(self.device_dir, test_dir_name)

    def _copy_in_file(self, src, dst):  # pylint: disable=no-self-use
        adb.push(src, dst)

    def _execute_command_remote(self, cmd, remote_work_dir='.', env=None):
        adb_cmd = ['adb', 'shell']
        if self.serial:
            adb_cmd.extend(['-s', self.serial])

        delimiter = 'x'
        probe_cmd = ' '.join(cmd) + '; echo {}$?'.format(delimiter)

        env_cmd = []
        if env is not None:
            env_cmd = ['%s=%s' % (k, v) for k, v in env.items()]

        remote_cmd = ' '.join(env_cmd + [probe_cmd])
        if remote_work_dir != '.':
            remote_cmd = 'cd {} && {}'.format(remote_work_dir, remote_cmd)

        adb_cmd.append(remote_cmd)

        # Tests will commonly fail with ETXTBSY. Possibly related to this bug:
        # https://code.google.com/p/android/issues/detail?id=65857. Work around
        # it by just waiting a second and then retrying.
        for _ in range(10):
            out, err, exit_code = self.local_run(adb_cmd)
            if 'Text file busy' in out or 'text busy' in out:
                time.sleep(1)
            else:
                out, delim, rc_str = out.rpartition(delimiter)
                if delim == '':
                    continue

                out = out.strip()
                try:
                    exit_code = int(rc_str)
                    break
                except ValueError:
                    continue
        return adb_cmd, out, err, exit_code