openreplay/api/chalicelib/core/issues.py

from chalicelib.utils import pg_client, helper

ISSUE_TYPES = ['click_rage', 'dead_click', 'excessive_scrolling', 'bad_request', 'missing_resource', 'memory', 'cpu',
               'slow_resource', 'slow_page_load', 'crash', 'ml_cpu', 'ml_memory', 'ml_dead_click', 'ml_click_rage',
               'ml_mouse_thrashing', 'ml_excessive_scrolling', 'ml_slow_resources', 'custom', 'js_exception',
               'custom_event_error', 'js_error']
ORDER_QUERY = """\
(CASE   WHEN type = 'js_exception' THEN 0
        WHEN type = 'bad_request' THEN 1
        WHEN type = 'missing_resource' THEN 2
        WHEN type = 'click_rage' THEN 3
        WHEN type = 'dead_click' THEN 4
        WHEN type = 'memory' THEN 5
        WHEN type = 'cpu' THEN 6
        WHEN type = 'crash' THEN 7
        ELSE -1 END)::INTEGER 
"""
NAME_QUERY = """\
(CASE   WHEN type = 'js_exception' THEN 'Errors'
        WHEN type = 'bad_request' THEN 'Bad Requests'
        WHEN type = 'missing_resource' THEN 'Missing Images'
        WHEN type = 'click_rage' THEN 'Click Rage'
        WHEN type = 'dead_click' THEN 'Dead Clicks'
        WHEN type = 'memory' THEN 'High Memory'
        WHEN type = 'cpu' THEN 'High CPU'
        WHEN type = 'crash' THEN 'Crashes'
        ELSE type::text END)::text 
"""


def get(project_id, issue_id):
    with pg_client.PostgresClient() as cur:
        query = cur.mogrify(
            """\
            SELECT
                *
            FROM public.issues
            WHERE project_id = %(project_id)s
                AND issue_id = %(issue_id)s;""",
            {"project_id": project_id, "issue_id": issue_id}
        )
        cur.execute(query=query)
        data = cur.fetchone()
        if data is not None:
            data["title"] = helper.get_issue_title(data["type"])
    return helper.dict_to_camel_case(data)


def get_by_session_id(session_id, project_id, issue_type=None):
    with pg_client.PostgresClient() as cur:
        cur.execute(
            cur.mogrify(f"""\
                    SELECT *
                    FROM events_common.issues
                             INNER JOIN public.issues USING (issue_id)
                    WHERE session_id = %(session_id)s 
                        AND project_id= %(project_id)s
                        {"AND type = %(type)s" if issue_type is not None else ""}
                    ORDER BY timestamp;""",
                        {"session_id": session_id, "project_id": project_id, "type": issue_type})
        )
        return helper.list_to_camel_case(cur.fetchall())


def get_types_by_project(project_id):
    with pg_client.PostgresClient() as cur:
        cur.execute(
            cur.mogrify(f"""SELECT type,
                               {ORDER_QUERY}>=0 AS visible,
                               {ORDER_QUERY} AS order,
                               {NAME_QUERY} AS name
                            FROM (SELECT DISTINCT type
                                  FROM public.issues
                                  WHERE project_id = %(project_id)s) AS types
                            ORDER BY "order";""", {"project_id": project_id}))
        return helper.list_to_camel_case(cur.fetchall())


def get_all_types():
    return [
        {
            "type": "js_exception",
            "visible": True,
            "order": 0,
            "name": "Errors"
        },
        {
            "type": "bad_request",
            "visible": True,
            "order": 1,
            "name": "Bad Requests"
        },
        {
            "type": "missing_resource",
            "visible": True,
            "order": 2,
            "name": "Missing Images"
        },
        {
            "type": "click_rage",
            "visible": True,
            "order": 3,
            "name": "Click Rage"
        },
        {
            "type": "dead_click",
            "visible": True,
            "order": 4,
            "name": "Dead Clicks"
        },
        {
            "type": "memory",
            "visible": True,
            "order": 5,
            "name": "High Memory"
        },
        {
            "type": "cpu",
            "visible": True,
            "order": 6,
            "name": "High CPU"
        },
        {
            "type": "crash",
            "visible": True,
            "order": 7,
            "name": "Crashes"
        }
    ]