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

from pylib.base import environment
from pylib.device import adb_wrapper
from pylib.device import device_errors
from pylib.device import device_utils
from pylib.utils import parallelizer


class LocalDeviceEnvironment(environment.Environment):

  def __init__(self, args, _error_func):
    super(LocalDeviceEnvironment, self).__init__()
    self._device = args.test_device
    self._devices = []
    self._max_tries = 1 + args.num_retries
    self._tool_name = args.tool

  #override
  def SetUp(self):
    # TODO(jbudorick): This can be refined to support filters etc.
    available_devices = adb_wrapper.AdbWrapper.GetDevices()
    if not available_devices:
      raise device_errors.NoDevicesError
    if self._device:
      if self._device not in available_devices:
        raise device_errors.DeviceUnreachableError(
            'Could not find device %r' % self._device)
      self._devices = [device_utils.DeviceUtils(self._device)]
    else:
      self._devices = [
          device_utils.DeviceUtils(s)
          for s in available_devices]

  @property
  def devices(self):
    return self._devices

  @property
  def parallel_devices(self):
    return parallelizer.SyncParallelizer(self._devices)

  @property
  def max_tries(self):
    return self._max_tries

  @property
  def tool(self):
    return self._tool_name

  #override
  def TearDown(self):
    pass