# 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 = .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)