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

"""System metrics."""

from __future__ import absolute_import
from __future__ import print_function

import collections
import platform
import sys

from infra_libs import ts_mon

_os_name_metric = ts_mon.StringMetric(
    'proc/os/name',
    description='OS name on the machine')

_os_version_metric = ts_mon.StringMetric(
    'proc/os/version',
    description='OS version on the machine')

_os_arch_metric = ts_mon.StringMetric(
    'proc/os/arch',
    description='OS architecture on this machine')

_python_arch_metric = ts_mon.StringMetric(
    'proc/python/arch',
    description='python userland architecture on this machine')


def collect_os_info():
  os_info = _get_osinfo()
  _os_name_metric.set(os_info.name)
  _os_version_metric.set(os_info.version)
  _os_arch_metric.set(platform.machine())
  _python_arch_metric.set(_get_python_arch())


class OSInfo(collections.namedtuple('OSInfo', 'name,version')):
  """namedtuple representing OS info (all fields lowercased)."""

  def __new__(cls, name, version):
    return super(OSInfo, cls).__new__(cls, name.lower(), version.lower())


def _get_osinfo():
  """Get OS name and version.

  Returns:
    OSInfo instance
  """
  os_name = platform.system().lower()
  if 'windows' in os_name:
    return _get_windows_osinfo()
  elif 'linux' in os_name:
    return _get_linux_osinfo()
  elif _is_mac():
    # On mac platform.system() reports 'darwin'.
    #
    # TODO(ayatane): I'm not sure how true the above comment is, but I
    # have no reason to remove it nor change the existing logic right
    # now.
    return _get_mac_osinfo()
  else:
    return OSInfo(name='', version='')


def _get_windows_osinfo():
  os_name = 'windows'
  # release will be something like '7', 'vista', or 'xp'
  os_version = platform.release()
  return OSInfo(name=os_name, version=os_version)


def _get_linux_osinfo():
  # will return something like ('Ubuntu', '14.04', 'trusty')
  os_name, os_version, _ = platform.dist()
  return OSInfo(name=os_name, version=os_version)


def _get_mac_osinfo():
  return OSInfo(name='mac', version=_get_mac_version())


def _is_mac():
  """Return whether the current system is a Mac."""
  return bool(_get_mac_version())


def _get_mac_version():
  """Get Mac system version.

  Returns:
    Version string, which is empty if not a valid Mac system.
  """
  return platform.mac_ver()[0]


def _get_python_arch():
  if sys.maxsize > 2**32:
    return '64'
  else:
    return '32'