# Copyright 2024 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
from typing import Optional
import os
import socket
import sys

from opentelemetry import trace as otel_trace_api
from opentelemetry.sdk import (
    resources as otel_resources,
    trace as otel_trace_sdk,
)
from opentelemetry.sdk.trace import export as otel_export

from . import config
from . import clearcut_span_exporter
from . import detector

DEFAULT_BANNER = """
===============================================================================
To help improve the quality of this product, we collect usage data and
stacktraces from googlers. This includes a uuid generated weekly to identify
invocation from the same users as well as metrics as described in
go/chrome-infra-telemetry-readme. You may choose to opt out of this collection
at any time by setting the flag `enabled = False` under [trace] section in
{config_file}
or by executing from your depot_tools checkout:

vpython3 third_party/depot_tools/infra_lib/telemetry --disable

This notice will be displayed {run_count} more times.
===============================================================================
"""

# This does not include Googlers' physical machines/laptops
_GOOGLE_HOSTNAME_SUFFIX = ('.google.com', '.googler.com', '.googlers.com')

# The version keeps track of telemetry changes.
_TELEMETRY_VERSION = '3'


def get_host_name(fully_qualified: bool = False) -> str:
    """Return hostname of current machine, with domain if |fully_qualified|."""
    hostname = socket.gethostname()
    try:
        hostname = socket.gethostbyaddr(hostname)[0]
    except (socket.gaierror, socket.herror) as e:
        if sys.platform.startswith('linux'):
            print(
                'please check your /etc/hosts file; resolving your hostname'
                f' ({hostname}) failed: {e}',
                file=sys.stderr)

    if fully_qualified:
        return hostname
    return hostname.partition('.')[0]


def is_google_host() -> bool:
    """Checks if the code is running on google host."""

    hostname = get_host_name(fully_qualified=True)
    return hostname.endswith(_GOOGLE_HOSTNAME_SUFFIX)


def initialize(service_name,
               notice=DEFAULT_BANNER,
               cfg_file=config.DEFAULT_CONFIG_FILE):
    if 'SWARMING_BOT_ID' in os.environ:
        return

    cfg = config.Config(cfg_file)
    if cfg.trace_config.disabled():
        return

    bot_enabled = (cfg.trace_config.has_enabled()
                   and cfg.trace_config.enabled_reason == 'BOT_USER')
    if not is_google_host() and not bot_enabled:
        return

    if not cfg.trace_config.has_enabled():
        if cfg.root_config.notice_countdown > -1:
            print(notice.format(run_count=cfg.root_config.notice_countdown,
                                config_file=cfg_file),
                  file=sys.stderr)
            cfg.root_config.update(
                notice_countdown=cfg.root_config.notice_countdown - 1)
        else:
            cfg.trace_config.update(enabled=True, reason='AUTO')

        cfg.flush()

    default_resource = otel_resources.Resource.create({
        otel_resources.SERVICE_NAME: service_name,
        'telemetry.version': _TELEMETRY_VERSION,
        'user.id': cfg.trace_config.user_uuid(),
    })

    detected_resource = otel_resources.get_aggregated_resources([
        otel_resources.ProcessResourceDetector(),
        otel_resources.OTELResourceDetector(),
        detector.ProcessDetector(),
        detector.SystemDetector(),
    ])

    resource = detected_resource.merge(default_resource)
    trace_provider = otel_trace_sdk.TracerProvider(resource=resource)
    otel_trace_api.set_tracer_provider(trace_provider)
    trace_provider.add_span_processor(
        otel_export.BatchSpanProcessor(
            # Replace with ConsoleSpanExporter() to debug spans on the console
            clearcut_span_exporter.ClearcutSpanExporter()))


def get_tracer(name: str, version: Optional[str] = None):
    return otel_trace_api.get_tracer(name, version)


def opted_in(cfg_file=config.DEFAULT_CONFIG_FILE):
    """Get if the user is opted-in

    Unlike initialize which continues when not explicitly opted out this will
    return if the user is opted in, either by user or automatically after the
    banner display period.
    """
    cfg = config.Config(cfg_file)
    if cfg.trace_config.disabled():
        return False

    bot_enabled = (cfg.trace_config.has_enabled()
                   and cfg.trace_config.enabled_reason == 'BOT_USER')
    if not is_google_host() and not bot_enabled:
        return False

    cfg = config.Config(cfg_file)
    return cfg.trace_config.enabled