wh_analytics.py

# Wahoo! Results - https://github.com/JohnStrunk/wahoo-results
# Copyright (C) 2020 - John D. Strunk
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.

"""Application usage analytics."""

import locale
import platform
import socket
import time
from pprint import pprint
from typing import Any, Dict, Optional, Tuple

import ipinfo
import ipinfo.exceptions
import requests
import sentry_sdk
from segment import analytics

import autotest
import version
from model import Model

_CONTEXT: Dict[str, Any] = {}


def application_start(model: Model, screen_size: Tuple[int, int]) -> None:
    """Event for application startup.

    :param model: Application model object
    :param screen_size: Screen size in pixels
    """
    analytics.write_key = version.SEGMENT_WRITE_KEY
    analytics.send = model.analytics.get()
    global _CONTEXT  # noqa: PLW0603
    _CONTEXT = {
        "context": _setup_context(screen_size),
        "race_count": 0,
        "race_count_with_names": 0,
        "session_start": time.time(),
        "user_id": model.client_id.get(),
    }

    if autotest.TESTING:  # Don't send analytics during testing
        return

    analytics.identify(
        user_id=_CONTEXT["user_id"],
        context=_CONTEXT["context"],
        traits=_CONTEXT["context"]["traits"],
    )
    _send_event("Scoreboard started")


def application_stop(model: Model) -> None:
    """Event for application shutdown.

    :param model: Application model object
    """
    _send_event(
        "Scoreboard stopped",
        {
            "runtime": time.time() - _CONTEXT["session_start"],
            "race_count": _CONTEXT["race_count"],
            "race_count_with_names": _CONTEXT["race_count_with_names"],
            "lane_count": model.num_lanes.get(),
            "time_threshold": model.time_threshold.get(),
            "min_times": model.min_times.get(),
            "bg_image": model.image_bg.get() != "",
            "normal_font": model.font_normal.get(),
            "time_font": model.font_time.get(),
        },
    )
    analytics.shutdown()


def results_received(has_names: bool, chromecasts: int) -> None:
    """Event for race results.

    :param has_names: True if names are included in the results
    :param chromecasts: Number of Chromecast devices currently enabled
    """
    _CONTEXT["race_count"] += 1
    if has_names:
        _CONTEXT["race_count_with_names"] += 1
    _send_event(
        "Results received", {"has_names": has_names, "chromecast_count": chromecasts}
    )


def documentation_link() -> None:
    """Follow link to online docs."""
    _send_event("Documentation click")


def update_link() -> None:
    """Follow link to download latest version."""
    _send_event("DownloadUpdate click")


def set_cts_directory(changed: bool) -> None:
    """Set CTS start list directory.

    :param changed: True if the directory was changed
    """
    _send_event(
        "Browse CTS directory",
        {
            "directory_changed": changed,
        },
    )


def wrote_dolphin_csv(num_events: int) -> None:
    """Dolphin CSV event list written.

    :param num_events: Number of events in the CSV file
    """
    _send_event(
        "Write Dolphin CSV",
        {
            "event_count": num_events,
        },
    )


def set_result_directory(changed: bool) -> None:
    """Set directory to watch for do4 files.

    :param changed: True if the directory was changed
    """
    _send_event(
        "Browse D04 directory",
        {
            "directory_changed": changed,
        },
    )


def cc_toggle(enable: bool) -> None:
    """Enable/disable a Chromecast.

    :param enable: True to enable, False to disable
    """
    _send_event(
        "Set Chromecast state",
        {
            "enable": enable,
        },
    )


def _send_event(name: str, kvparams: Optional[Dict[str, Any]] = None) -> None:
    with sentry_sdk.start_span(op="analytics", description="Process analytics event"):
        if "user_id" not in _CONTEXT:
            return
        if autotest.TESTING:  # Don't send analytics during testing
            return
        if kvparams is None:
            kvparams = {}
        analytics.track(
            _CONTEXT["user_id"], name, kvparams, context=_CONTEXT["context"]
        )
        if analytics.write_key == "unknown":  # dev environment
            print(f"Event: {name}")
            pprint(kvparams)


def _setup_context(screen_size: Tuple[int, int]) -> Dict[str, Any]:
    uname = platform.uname()
    # https://segment.com/docs/connections/spec/identify/#traits
    traits: Dict[str, Any] = {}
    if hasattr(socket, "gethostname"):
        traits["name"] = socket.gethostname()

    # https://segment.com/docs/connections/spec/common/#context
    context: Dict[str, Any] = {
        "app": {
            "version": version.WAHOO_RESULTS_VERSION,
        },
        "locale": locale.getlocale()[0],
        "os": {
            "name": uname.system,
            "version": uname.version,
        },
        "screen": {
            "height": screen_size[1],
            "width": screen_size[0],
        },
        "traits": traits,
    }

    try:
        iphandler = ipinfo.getHandler(version.IPINFO_TOKEN)
        ipdetails = iphandler.getDetails()

        traits["address"] = {
            "city": ipdetails.city,
            "state": ipdetails.region,
            "country": ipdetails.country_name,
            "postalCode": ipdetails.postal,
        }
        context["ip"] = ipdetails.ip
        context["location"] = {
            "city": ipdetails.city,
            "region": ipdetails.region,
            "country": ipdetails.country_name,
            "latitude": ipdetails.latitude,
            "longitude": ipdetails.longitude,
        }
        context["timezone"] = ipdetails.timezone
    except AttributeError:  # Tried to get a non-existant mapping from `ipdetails`
        pass
    except requests.HTTPError:  # General HTTP error
        pass
    except requests.JSONDecodeError:  # Invalid JSON returned
        pass
    except requests.Timeout:  # ConnectTimeout or ReadTimeout
        pass
    except ipinfo.exceptions.RequestQuotaExceededError:  # Over quota limit
        pass

    print(context)
    return context