# 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 import constants
from pylib.local.device import local_device_environment
try:
  from pylib.remote.device import remote_device_environment
except ImportError:
  remote_device_environment = None

def CreateEnvironment(args, error_func):

  if args.environment == 'local':
    if args.command not in constants.LOCAL_MACHINE_TESTS:
      return local_device_environment.LocalDeviceEnvironment(args, error_func)
    # TODO(jbudorick) Add local machine environment.
  if args.environment == 'remote_device' and remote_device_environment:
    return remote_device_environment.RemoteDeviceEnvironment(args,
                                                             error_func)
  error_func('Unable to create %s environment.' % args.environment)