from datetime import datetime

from flask import Blueprint, jsonify, request

from app.dao.fact_notification_status_dao import get_total_notifications_for_date_range
from app.dao.fact_processing_time_dao import (
    get_processing_time_percentage_for_date_range,
)
from app.dao.services_dao import get_live_services_with_organization
from app.errors import register_errors
from app.performance_dashboard.performance_dashboard_schema import (
    performance_dashboard_request,
)
from app.schema_validation import validate
from app.utils import utc_now

performance_dashboard_blueprint = Blueprint(
    "performance_dashboard", __name__, url_prefix="/performance-dashboard"
)

register_errors(performance_dashboard_blueprint)


@performance_dashboard_blueprint.route("")
def get_performance_dashboard():
    # All statistics are as of last night this matches the existing performance platform
    # and avoids the need to query notifications.
    if request.args:
        # Is it ok to reuse this? - should probably create a new one
        validate(request.args, performance_dashboard_request)

    # If start and end date are not set, we are expecting today's stats.
    today = str(utc_now().date())

    start_date = datetime.strptime(
        request.args.get("start_date", today),
        "%Y-%m-%d",
    ).date()
    end_date = datetime.strptime(
        request.args.get("end_date", today),
        "%Y-%m-%d",
    ).date()
    total_for_all_time = get_total_notifications_for_date_range(
        start_date=None,
        end_date=None,
    )
    total_notifications, emails, sms = transform_results_into_totals(total_for_all_time)
    totals_for_date_range = get_total_notifications_for_date_range(
        start_date=start_date,
        end_date=end_date,
    )
    processing_time_results = get_processing_time_percentage_for_date_range(
        start_date=start_date,
        end_date=end_date,
    )
    services = get_live_services_with_organization()
    stats = {
        "total_notifications": total_notifications,
        "email_notifications": emails,
        "sms_notifications": sms,
        "notifications_by_type": transform_into_notification_by_type_json(
            totals_for_date_range
        ),
        "processing_time": transform_processing_time_results_to_json(
            processing_time_results
        ),
        "live_service_count": len(services),
        "services_using_notify": transform_services_to_json(services),
    }

    return jsonify(stats)


def transform_results_into_totals(total_notifications_results):
    total_notifications = 0
    emails = 0
    sms = 0
    for x in total_notifications_results:
        total_notifications += x.emails
        total_notifications += x.sms
        emails += x.emails
        sms += x.sms
    return total_notifications, emails, sms


def transform_into_notification_by_type_json(total_notifications):
    j = []
    for x in total_notifications:
        j.append(
            {
                "date": x.local_date.strftime("%Y-%m-%d"),
                "emails": x.emails,
                "sms": x.sms,
            }
        )

    return j


def transform_processing_time_results_to_json(processing_time_results):
    j = []
    for x in processing_time_results:
        j.append({"date": x.date, "percentage_under_10_seconds": x.percentage})

    return j


def transform_services_to_json(services_results):
    j = []
    for x in services_results:
        j.append(
            {
                "service_id": x.service_id,
                "service_name": x.service_name,
                "organization_id": x.organization_id,
                "organization_name": x.organization_name,
            }
        )
    return j