"""
OpenObserve Provider is a class that allows to install webhooks in OpenObserve.
"""

import dataclasses
import json
import logging
import uuid
from pathlib import Path
from typing import List
from urllib.parse import urlencode, urljoin

import pydantic
import requests

from keep.api.models.alert import AlertDto, AlertSeverity
from keep.contextmanager.contextmanager import ContextManager
from keep.providers.base.base_provider import BaseProvider
from keep.providers.models.provider_config import ProviderConfig, ProviderScope
from keep.validation.fields import UrlPort


class ResourceAlreadyExists(Exception):
    def __init__(self, *args):
        super().__init__(*args)


@pydantic.dataclasses.dataclass
class OpenobserveProviderAuthConfig:
    """
    OpenObserve authentication configuration.
    """

    openObserveUsername: str = dataclasses.field(
        metadata={
            "required": True,
            "description": "OpenObserve Username",
            "hint": "Your Username",
        },
    )
    openObservePassword: str = dataclasses.field(
        metadata={
            "required": True,
            "description": "Password",
            "hint": "Password associated with your account",
            "sensitive": True,
        },
    )
    openObserveHost: pydantic.AnyHttpUrl = dataclasses.field(
        metadata={
            "required": True,
            "description": "OpenObserve host url",
            "hint": "e.g. http://localhost",
            "validation": "any_http_url"
        },
    )

    openObservePort: UrlPort = dataclasses.field(
        metadata={
            "required": True,
            "description": "OpenObserve Port",
            "hint": "e.g. 5080",
            "validation": "port"
        },
    )
    organisationID: str = dataclasses.field(
        metadata={
            "required": True,
            "description": "OpenObserve organisationID",
            "hint": "default",
        },
    )


class OpenobserveProvider(BaseProvider):
    """Install Webhooks and receive alerts from OpenObserve."""

    PROVIDER_DISPLAY_NAME = "OpenObserve"
    PROVIDER_CATEGORY = ["Monitoring"]
    PROVIDER_SCOPES = [
        ProviderScope(
            name="authenticated",
            description="User is Authorized",
            mandatory=True,
            mandatory_for_webhook=True,
            alias="Rules Reader",
        ),
    ]

    SEVERITIES_MAP = {
        "ERROR": AlertSeverity.CRITICAL,
        "WARN": AlertSeverity.WARNING,
        "INFO": AlertSeverity.INFO,
    }

    def __init__(
        self, context_manager: ContextManager, provider_id: str, config: ProviderConfig
    ):
        super().__init__(context_manager, provider_id, config)

    def dispose(self):
        """
        Dispose the provider.
        """
        pass

    def validate_config(self):
        """
        Validates required configuration for OpenObserve provider.
        """
        if self.is_installed or self.is_provisioned:
            host = self.config.authentication['openObserveHost']
            if not (host.startswith("http://") or host.startswith("https://")):
                scheme = "http://" if ("localhost" in host or "127.0.0.1" in host) else "https://"
                self.config.authentication['openObserveHost'] = scheme + host

        self.authentication_config = OpenobserveProviderAuthConfig(
            **self.config.authentication
        )

    def __get_url(self, paths: List[str] = [], query_params: dict = None, **kwargs):
        """
        Helper method to build the url for OpenObserve api requests.

        Example:

        paths = ["issue", "createmeta"]
        query_params = {"projectKeys": "key1"}
        url = __get_url("test", paths, query_params)

        # url = https://baseballxyz.saas.openobserve.com/rest/api/2/issue/createmeta?projectKeys=key1
        """

        url = urljoin(
            f"{self.authentication_config.openObserveHost}:{self.authentication_config.openObservePort}",
            "/".join(str(path) for path in paths),
        )

        # add query params
        if query_params:
            url = f"{url}?{urlencode(query_params)}"

        return url

    def validate_scopes(self) -> dict[str, bool | str]:
        authenticated = False
        self.logger.info("Validating OpenObserve Scopes")
        try:
            response = requests.post(
                url=self.__get_url(
                    paths=[
                        "auth/login",
                    ]
                ),
                json={
                    "name": self.authentication_config.openObserveUsername,
                    "password": self.authentication_config.openObservePassword,
                },
                timeout=10,
            )
        except Exception as e:
            self.logger.error(
                "Error while validating scopes for OpenObserve",
                extra=e,
            )
            return {"authenticated": str(e)}
        print(
            self.__get_url(
                paths=[
                    "auth/login",
                ]
            )
        )
        if response.ok:
            response = response.json()
            authenticated = response["status"]
        else:
            self.logger.error(
                "Error while validating scopes for OpenObserve",
                extra={"status_code": response.status_code, "error": response.text},
            )

        return {"authenticated": authenticated}

    def __get_auth(self) -> tuple[str, str]:
        return (
            self.authentication_config.openObserveUsername,
            self.authentication_config.openObservePassword,
        )

    def __update_alert_template(self, data):
        res = requests.put(
            url=self.__get_url(
                paths=[
                    f"api/{self.authentication_config.organisationID}/alerts/templates/KeepAlertTemplate"
                ]
            ),
            json=data,
            auth=self.__get_auth(),
        )
        if res.ok:
            res = res.json()
            if res["code"] == 200:
                self.logger.info("Alert template Updated Successfully")
            else:

                self.logger.error(
                    "Failed to update Alert Template",
                    extra={"code": res["code"], "error": res["message"]},
                )
        else:
            self.logger.error(
                "Error while updating Alert Template",
                extra={"status_code": res.status_code, "error": res.text},
            )

    def __create_alert_template(self):

        # This is the template used for creating the alert template in openobserve
        template = open(rf"{Path(__file__).parent}/alerttemplate.json", "rt")
        data = template.read()
        try:
            res = requests.post(
                self.__get_url(
                    paths=[
                        f"api/{self.authentication_config.organisationID}/alerts/templates"
                    ]
                ),
                json={"body": data, "isDefault": False, "name": "KeepAlertTemplate"},
                auth=self.__get_auth(),
            )
            res = res.json()
            if res["code"] == 200:
                self.logger.info("Alert template Successfully Created")

            elif "already exists" in res["message"]:
                self.logger.info(
                    "Alert template creation failed as it already exists",
                    extra={"code": res["code"], "error": res["message"]},
                )
                self.logger.info(
                    "Attempting to Update Alert Template with latest data..."
                )
                self.__update_alert_template(
                    data={"body": data, "isDefault": False, "name": "KeepAlertTemplate"}
                )
            else:
                self.logger.error(
                    "Alert template creation failed",
                    extra={"code": res["code"], "error": res["message"]},
                )

        except Exception as e:
            self.logger.error(
                "Error While making alert Template",
                extra=e,
            )

    def __update_destination(self, keep_api_url: str, api_key: str, data):
        res = requests.put(
            json=data,
            url=self.__get_url(
                paths=[
                    f"api/{self.authentication_config.organisationID}/alerts/destinations/KeepDestination"
                ]
            ),
            auth=self.__get_auth(),
        )
        if res.ok:
            self.logger.info("Destination Successfully Updated")
        else:
            self.logger.error(
                "Error while updating destination",
                extra={"code": res.status_code, "error": res.text},
            )

    def __create_destination(self, keep_api_url: str, api_key: str):
        data = {
            "headers": {
                "X-API-KEY": api_key,
            },
            "method": "post",
            "name": "KeepDestination",
            "template": "KeepAlertTemplate",
            "url": keep_api_url,
        }

        response = requests.post(
            url=self.__get_url(
                paths=[
                    f"api/{self.authentication_config.organisationID}/alerts/destinations"
                ]
            ),
            auth=self.__get_auth(),
            json=data,
        )
        # if response.ok:
        res = response.json()
        if res["code"] == 200:
            self.logger.info("Destination Successfully Created")
        elif "already exists" in res["message"]:
            self.logger.info("Destination creation failed as it already exists")
            self.logger.info("Attempting to Update Destination...")
            self.__update_destination(
                keep_api_url=keep_api_url, api_key=api_key, data=data
            )
        else:
            self.logger.error(
                "Destination creation failed",
                extra={"code": res["code"], "error": res["message"]},
            )

    def __get_all_stream_names(self) -> list[str]:
        names = []
        response = requests.get(
            url=self.__get_url(
                paths=[f"api/{self.authentication_config.organisationID}/streams"]
            ),
            auth=self.__get_auth(),
        )
        res = response.json()
        for stream in res["list"]:
            names.append(stream["name"])
        return names

    def __get_and_update_actions(self):
        response = requests.get(
            url=self.__get_url(
                paths=[f"api/{self.authentication_config.organisationID}/alerts"]
            ),
            auth=self.__get_auth(),
        )
        res = response.json()
        for alert in res["list"]:
            alert_stream = alert["stream_name"]
            alert_name = alert["name"]
            if "KeepDestination" not in alert["destinations"]:
                alert["destinations"].append("KeepDestination")
            self.logger.info(f"Updating Alert: {alert_name} in Stream: {alert_stream}")
            update_response = requests.put(
                url=self.__get_url(
                    paths=[f"api/default/{alert_stream}/alerts/{alert_name}"]
                ),
                auth=self.__get_auth(),
                json=alert,
            )
            update_res = update_response.json()
            if update_res["code"] == 200:
                self.logger.info(
                    f"Updated Alert: {alert_name} in Stream: {alert_stream}",
                    extra={"code": update_res["code"], "error": update_res["message"]},
                )
            else:
                self.logger.error(
                    f"Error while updating Alert: {alert_name} in Stream: {alert_stream}",
                    extra={"code": update_res["code"], "error": update_res["message"]},
                )

    def setup_webhook(
        self, tenant_id: str, keep_api_url: str, api_key: str, setup_alerts: bool = True
    ):
        try:
            self.__create_alert_template()
        except Exception as e:
            self.logger.error("Error while creating Alert Template", extra=e)
        self.__create_destination(keep_api_url=keep_api_url, api_key=api_key)
        try:
            self.__get_and_update_actions()
        except Exception as e:
            self.logger.error("Error while updating Alerts", extra=e)
        self.logger.info("Webhook created")

    @staticmethod
    def _format_alert(
        event: dict, provider_instance: "BaseProvider" = None
    ) -> AlertDto | List[AlertDto]:
        logger = logging.getLogger(__name__)
        alert_name = event.pop("alert_name", "")
        # openoboserve does not provide severity
        severity = AlertSeverity.WARNING
        # Mapping 'stream_name' to 'environment'
        environment = event.pop("stream_name", "")
        # Mapping 'alert_start_time' to 'startedAt'
        startedAt = event.pop("alert_start_time", "")
        # Mapping 'alert_start_time' to 'startedAt'
        lastReceived = event.pop("alert_start_time", "")
        # Mapping 'alert_type' to 'description'
        description = event.pop("alert_type", "")

        alert_url = event.pop("alert_url", "")

        org_name = event.pop("org_name", "")
        # Our only way to distinguish between non aggregated alert and aggregated alerts is the alert_agg_value
        if "alert_agg_value" in event and (
            len(event["alert_agg_value"].split(","))
            == int(event.get("alert_count", -1))
            or len(event["alert_agg_value"].split(",")) == 1
        ):
            logger.info("Formatting openobserve aggregated alert")
            rows = event.pop("rows", "")
            if not rows:
                logger.exception(
                    "Rows not found in the aggregated alert event",
                    extra={"event": event},
                )
                raise ValueError("Rows not found in the aggregated alert event")
            alerts = []
            number_of_rows = event.pop("alert_count", "")
            rows = rows.split("\n")
            agg_values = event.pop("alert_agg_value", "").split(",")
            # if there is only one value, repeat it for all rows
            if len(agg_values) == 1:
                logger.info("Only one value found, repeating it for all rows")
                agg_values = [agg_values[0]] * int(number_of_rows)
            # trim
            agg_values = [agg_value.strip() for agg_value in agg_values]
            for i in range(int(number_of_rows)):
                try:
                    logger.info(
                        "Formatting aggregated alert",
                        extra={"row": rows[i]},
                    )
                    row = rows[i]
                    value = agg_values[i]
                    # try to parse value as a number since its metric
                    try:
                        value = float(value)
                    except ValueError:
                        pass
                    try:
                        row_data = json.loads(row)
                    except json.JSONDecodeError:
                        try:
                            row_data = json.loads(row.replace("'", '"'))
                        except json.JSONDecodeError:
                            logger.exception(f"Failed to parse row: {row}")
                            continue
                    row_name = row_data.pop("name", "")
                    if row_name:
                        row_data['row_name'] = row_name
                    group_by_keys = list(row_data.keys())
                    logger.info(
                        "Formatting aggregated alert with group by keys",
                        extra={
                            "group_by_keys": group_by_keys,
                        },
                    )

                    alert_id = str(uuid.uuid4())

                    # we already take the value from the agg_value
                    event.pop("value", "")
                    # if the group_by_key is already in the event, remove it
                    #   since we are adding it to the alert_dto
                    for group_by_key in group_by_keys:
                        event.pop(group_by_key, "")

                    alert_dto = AlertDto(
                        id=f"{alert_id}",
                        name=f"{alert_name}: {row_name}" if row_name else f"{alert_name}",
                        severity=severity,
                        environment=environment,
                        startedAt=startedAt,
                        lastReceived=lastReceived,
                        description=description,
                        row_data=row_data,
                        source=["openobserve"],
                        org_name=org_name,
                        value=value,
                        alert_url=alert_url,  # I'm not putting on URL since sometimes it doesn't return full URL so pydantic will throw an error
                        **event,
                        **row_data
                    )
                    # calculate the fingerprint based on name + group_by_value
                    alert_dto.fingerprint = OpenobserveProvider.get_alert_fingerprint(
                        alert_dto, fingerprint_fields=["name", *group_by_keys]
                    )
                    logger.info(
                        "Formatted openobserve aggregated alert",
                        extra={"fingerprint": alert_dto.fingerprint},
                    )
                    alerts.append(alert_dto)
                except json.JSONDecodeError:
                    logger.error(f"Failed to parse row: {row}")
            return alerts
        # else, one alert, one row, old calculation
        else:
            alert_id = str(uuid.uuid4())
            labels = {
                "url": event.pop("alert_url", ""),
                "alert_period": event.pop("alert_period", ""),
                "alert_operator": event.pop("alert_operator", ""),
                "alert_threshold": event.pop("alert_threshold", ""),
                "alert_count": event.pop("alert_count", ""),
                "alert_agg_value": event.pop("alert_agg_value", ""),
                "alert_end_time": event.pop("alert_end_time", ""),
            }
            alert_dto = AlertDto(
                id=alert_id,
                name=alert_name,
                severity=severity,
                environment=environment,
                startedAt=startedAt,
                lastReceived=lastReceived,
                description=description,
                labels=labels,
                source=["openobserve"],
                org_name=org_name,
                alert_url=alert_url,  # I'm not putting on URL since sometimes it doesn't return full URL so pydantic will throw an error
                **event,  # any other fields
            )
            # calculate fingerprint based on name + environment + event keys (e.g. host)
            fingerprint_fields = ["name", "environment", *event.keys()]
            # remove 'value' as its too dynamic
            try:
                fingerprint_fields.remove("value")
            except ValueError:
                pass
            logger.info(
                "Calculating fingerprint fields",
                extra={"fingerprint_fields": fingerprint_fields},
            )

            # sort the fields to ensure the fingerprint is consistent
            # for e.g. host1, host2 is the same as host2, host1
            for field in fingerprint_fields:
                try:
                    field_attr = getattr(alert_dto, field)
                    if "," not in field_attr:
                        continue
                    # sort it lexographically
                    logger.info(
                        "Sorting field attributes",
                        extra={"field": field, "field_attr": field_attr},
                    )
                    sorted_field_attr = sorted(field_attr.replace(" ", "").split(","))
                    sorted_field_attr = ", ".join(sorted_field_attr)
                    logger.info(
                        "Sorted field attributes",
                        extra={"field": field, "sorted_field_attr": sorted_field_attr},
                    )
                    # set the attr
                    setattr(alert_dto, field, sorted_field_attr)
                except AttributeError:
                    pass
                except Exception as e:
                    logger.error(
                        "Error while sorting field attributes",
                        extra={"field": field, "error": e},
                    )

            alert_dto.fingerprint = OpenobserveProvider.get_alert_fingerprint(
                alert_dto, fingerprint_fields=fingerprint_fields
            )
            logger.info(
                "Formatted openobserve alert",
                extra={"fingerprint": alert_dto.fingerprint},
            )
            return alert_dto