# Copyright 2017 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import logging
import os

from pylib import constants
from .expensive_line_transformer import ExpensiveLineTransformer
from .expensive_line_transformer import ExpensiveLineTransformerPool

_MINIMUM_TIMEOUT = 10.0
_PER_LINE_TIMEOUT = 0.005  # Should be able to process 200 lines per second.
_PROCESS_START_TIMEOUT = 20.0
_MAX_RESTARTS = 4  # Should be plenty unless tool is crashing on start-up.
_POOL_SIZE = 4
_PASSTHROUH_ON_FAILURE = False


class Deobfuscator(ExpensiveLineTransformer):
    def __init__(self, mapping_path):
        min_timeout = _MINIMUM_TIMEOUT
        for k in os.environ:
            if k.startswith('DRONE') or k.startswith('SKYLAB'):
                logging.warning(
                    'Likely running in a CPU-limited context due to the presence of '
                    'the %s env var. Setting min timeout for Deobfuscator to 10x.',
                    k,
                )
                # TODO(crbug.com/421235960): Remove this if the CPU constraints are
                # removed.
                min_timeout = 10 * _MINIMUM_TIMEOUT
                break
        super().__init__(_PROCESS_START_TIMEOUT, min_timeout, _PER_LINE_TIMEOUT)
        script_path = os.path.join(
            constants.DIR_SOURCE_ROOT,
            'build',
            'android',
            'stacktrace',
            'java_deobfuscate.py',
        )
        self._command = [script_path, mapping_path]
        self.start()

    @property
    def name(self):
        return "deobfuscator"

    @property
    def command(self):
        return self._command


class DeobfuscatorPool(ExpensiveLineTransformerPool):
    def __init__(self, mapping_path):
        # As of Sep 2017, each instance requires about 500MB of RAM, as measured by:
        # /usr/bin/time -v build/android/stacktrace/java_deobfuscate.py \
        #     out/Release/apks/ChromePublic.apk.mapping
        self.mapping_path = mapping_path
        super().__init__(_MAX_RESTARTS, _POOL_SIZE, _PASSTHROUH_ON_FAILURE)

    @property
    def name(self):
        return "deobfuscator-pool"

    def CreateTransformer(self):
        return Deobfuscator(self.mapping_path)