# Copyright 2018, The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Metrics class."""
from atest import constants
from atest.metrics import metrics_base


class AtestStartEvent(metrics_base.MetricsBase):
  """Create Atest start event and send to clearcut.

  Usage:
      metrics.AtestStartEvent(
          command_line='example_atest_command',
          test_references=['example_test_reference'],
          cwd='example/working/dir',
          os='example_os')
  """

  _EVENT_NAME = 'atest_start_event'
  command_line = constants.INTERNAL
  test_references = constants.INTERNAL
  cwd = constants.INTERNAL
  os = constants.INTERNAL
  source_root = constants.INTERNAL
  hostname = constants.INTERNAL


class AtestExitEvent(metrics_base.MetricsBase):
  """Create Atest exit event and send to clearcut.

  Usage:
      metrics.AtestExitEvent(
          duration=metrics_utils.convert_duration(end-start),
          exit_code=0,
          stacktrace='some_trace',
          logs='some_logs')
  """

  _EVENT_NAME = 'atest_exit_event'
  duration = constants.EXTERNAL
  exit_code = constants.EXTERNAL
  stacktrace = constants.INTERNAL
  logs = constants.INTERNAL


class FindTestFinishEvent(metrics_base.MetricsBase):
  """Create find test finish event and send to clearcut.

  Occurs after a SINGLE test reference has been resolved to a test or
  not found.

  Usage:
      metrics.FindTestFinishEvent(
          duration=metrics_utils.convert_duration(end-start),
          success=true,
          test_reference='hello_world_test',
          test_finders=['example_test_reference', 'ref2'],
          test_info="test_name: hello_world_test -
              test_runner:AtestTradefedTestRunner -
              build_targets:
                  set(['MODULES-IN-platform_testing-tests-example-native']) -
              data:{'rel_config':
                  'platform_testing/tests/example/native/AndroidTest.xml',
                  'filter': frozenset([])} -
              suite:None - module_class: ['NATIVE_TESTS'] -
              install_locations:set(['device', 'host'])")
  """

  _EVENT_NAME = 'find_test_finish_event'
  duration = constants.EXTERNAL
  success = constants.EXTERNAL
  test_reference = constants.INTERNAL
  test_finders = constants.INTERNAL
  test_info = constants.INTERNAL


class BuildFinishEvent(metrics_base.MetricsBase):
  """Create build finish event and send to clearcut.

  Occurs after the build finishes, either successfully or not.

  Usage:
      metrics.BuildFinishEvent(
          duration=metrics_utils.convert_duration(end-start),
          success=true,
          targets=['target1', 'target2'])
  """

  _EVENT_NAME = 'build_finish_event'
  duration = constants.EXTERNAL
  success = constants.EXTERNAL
  targets = constants.INTERNAL


class RunnerFinishEvent(metrics_base.MetricsBase):
  """Create run finish event and send to clearcut.

  Occurs when a single test runner has completed.

  Usage:
      metrics.RunnerFinishEvent(
          duration=metrics_utils.convert_duration(end-start),
          success=true,
          runner_name='AtestTradefedTestRunner'
          test=[{name:'hello_world_test', result:0, stacktrace:''},
                {name:'test2', result:1, stacktrace:'xxx'}])
  """

  _EVENT_NAME = 'runner_finish_event'
  duration = constants.EXTERNAL
  success = constants.EXTERNAL
  runner_name = constants.EXTERNAL
  test = constants.INTERNAL


class RunTestsFinishEvent(metrics_base.MetricsBase):
  """Create run tests finish event and send to clearcut.

  Occurs after all test runners and tests have finished.

  Usage:
      metrics.RunTestsFinishEvent(
          duration=metrics_utils.convert_duration(end-start))
  """

  _EVENT_NAME = 'run_tests_finish_event'
  duration = constants.EXTERNAL


class LocalDetectEvent(metrics_base.MetricsBase):
  """Create local detection event and send it to clearcut.

  Usage:
      metrics.LocalDetectEvent(
          detect_type=0,
          result=0)
      detect_type: a value from atest_enum.DetectType.
      result: the value corresponding to the result of the detected event.
  """

  _EVENT_NAME = 'local_detect_event'
  detect_type = constants.EXTERNAL
  result = constants.EXTERNAL


def get_run_id() -> str:
  """Returns the unique run id set for the current invocation."""
  return metrics_base.MetricsBase.get_run_id()


def is_internal_user() -> bool:
  """Returns whether the user is a google internal user."""
  return metrics_base.get_user_type() == metrics_base.INTERNAL_USER