import uuid

from app.config import TaskEnv
from app.models.report import StatusReport, ReportCountry, DonutList
from core.service.mongo_db import use_collection


class ReportRepository():
    async def project_status(self, _inp: StatusReport, type):
        if _inp.state:
            match = {f'{type}.sts': {'$exists': 1}, 'design.state': _inp.state}

        else:
            match = {f'{type}.sts': {'$exists': 1}}
        report = await use_collection(TaskEnv.CLN_PROJECT).aggregate([
            {
                "$match": match
            },
            {
                '$group': {
                    '_id': f"${type}.sts",
                    'count': {'$sum': 1}
                }
            }
        ]).to_list(None)
        return report

    async def project_status_design(self, _inp: StatusReport):
        return await self.project_status(_inp, 'design')

    async def project_status_monitor(self, _inp: StatusReport):
        return await self.project_status(_inp, 'monitor')

    async def report_country(self, _inp: ReportCountry):
        match = {}
        if _inp.state:
            match = {'design.state': _inp.state}
        if _inp.type_id:
            match.update({'design.type_id': _inp.type_id})
        if _inp.nature_id:
            match.update({'design.nature_id': _inp.nature_id})
        if _inp.monitor_type:
            match.update({'monitor.base_information.type': _inp.monitor_type})

            # count_nature = await use_collection(TaskEnv.CLN_PROJECT).count_documents(
            #     {'design.nature_id':_inp.nature_id,'design.state':_inp.state})
            # count_type = await use_collection(TaskEnv.CLN_PROJECT).count_documents({'design.type_id': _inp.type_id,'design.state':_inp.state})
            # count_monitor_type = await use_collection(TaskEnv.CLN_PROJECT).count_documents({'monitor.base_information.type': _inp.monitor_type,'design.state':_inp.state})

        report = await use_collection(TaskEnv.CLN_PROJECT).aggregate([
            {"$match": match},
            {'$group': {
                '_id': "$design.state",
                'nature_id': {'$addToSet': '$design.nature_id'},
                'type_id': {'$addToSet': '$design.type_id'},
                'monitor_type': {'$addToSet': '$monitor.base_information.type'}
            }},
            {'$project': {
                'count_nature_id': {"$size": "$nature_id"},
                'count_type_id': {"$size": "$type_id"},
                'count_monitor_type': {"$size": "$monitor_type"},
                "_id": 1
            }}
        ]).to_list(None)
        # count_nature = await use_collection(TaskEnv.CLN_PROJECT).count_documents(
        #     {'design.nature_id': _inp.nature_id})
        # count_type = await use_collection(TaskEnv.CLN_PROJECT).count_documents({'design.type_id':_inp.type_id})
        # count_monitor_type = await use_collection(TaskEnv.CLN_PROJECT).count_documents({'monitor.base_information.type': _inp.monitor_type})
        # return {'count_nature':count_nature,'count_type':count_type,'count_monitor_type':count_monitor_type}
        return report

    async def exist_project_state(self):
        report = await use_collection(TaskEnv.CLN_PROJECT).aggregate([
            {
                '$group': {
                    '_id': "$design.state",
                    'count': {'$sum': 1}
                }
            }
        ]).to_list(None)

        return report

    async def donut(self, _select: int):
        report = None
        if _select == 1:
            report = await use_collection(TaskEnv.CLN_PROJECT).aggregate([
                {"$group": {"_id": "$design.nature_id", 'count': {'$sum': 1}}}
            ]).to_list(None)
        elif _select == 2:
            report = await use_collection(TaskEnv.CLN_PROJECT).aggregate([
                {
                    "$group": {
                        "_id": {
                            "$cond": [
                                {"$lt": ["$design.building_infrastructure", 1000]},  # Check if a < 10
                                "less_than_1000",
                                {
                                    "$cond": [
                                        {"$and": [{"$gte": ["$design.building_infrastructure", 1000]},
                                                  {"$lt": ["$design.building_infrastructure", 5000]}]},
                                        # Check if 10 <= a < 20
                                        "between_1000_and_5000",
                                        {
                                            "$cond": [
                                                {"$and": [{"$gte": ["$design.building_infrastructure", 5000]},
                                                          {"$lt": ["$design.building_infrastructure", 15000]}]},
                                                "between_5000_and_15000",
                                                {
                                                    "$cond": [
                                                        {"$and": [{"$gte": ["$design.building_infrastructure", 15000]},
                                                                  {"$lt": ["$design.building_infrastructure", 30000]}]},
                                                        "between_15000_and_30000",
                                                        "greater_than_or_equal_30000"  # Otherwise, a >= 30
                                                    ]
                                                }
                                            ]
                                        }
                                    ]
                                }
                            ]
                        },
                        "total": {"$sum": 1},
                    }
                }
            ]).to_list(None)

        elif _select == 3:
            report = await use_collection(TaskEnv.CLN_PROJECT).aggregate([
                {"$group": {"_id": "$design.type_id", 'count': {'$sum': 1}}}
            ]).to_list(None)
        return report

    async def list_donut(self, _inp: DonutList):
        if _inp.type == 1:
            _filter = {'design.nature_id': int(_inp.donut_select)}
        if _inp.type == 2:
            if _inp.donut_select == "less_than_1000":
                _filter = {'design.building_infrastructure': {'$lt': 1000}}
            elif _inp.donut_select == "between_1000_and_5000":
                _filter = {'design.building_infrastructure': {'$lt': 5000, "$gte": 1000}}
            elif _inp.donut_select == "between_5000_and_15000":
                _filter = {'design.building_infrastructure': {'$lt': 15000, "$gte": 5000}}
            elif _inp.donut_select == "between_15000_and_30000":
                _filter = {'design.building_infrastructure': {'$lt': 30000, "$gte": 15000}}
            else:
                _filter = {'design.building_infrastructure': {'$gte': 30000}}
        elif _inp.type == 3:
            _filter = {'design.type_id': int(_inp.donut_select)}
        report = await use_collection(TaskEnv.CLN_PROJECT).find(_filter).to_list(None)

        return report

    async def delay_donut(self):
        report = await use_collection(TaskEnv.CLN_BILL_DELAY).aggregate([
            {'$group': {"_id": None, "total_financial": {"$sum": "$financial.total"},
                        "total_force_major": {"$sum": "$force_major.total"},
                        "total_technical": {"$sum": "$technical.total"},
                        "total_extra_new_work": {"$sum": "$extra_new_work"},
                        "total_other": {"$sum": "$other"},
                        }}
        ]).to_list(None)
        return report

    async def list_delay_donut(self, _select: str):
        _map = {'1': 'financial.total', '2': 'force_major.total', '3': 'technical.total', '4': 'extra_new_work',
                '5': 'other'}
        report = await use_collection(TaskEnv.CLN_BILL_DELAY).aggregate([
            {'$match': {_map[_select]: {'$gt': 0}}},
            {'$group': {"_id": '$project_id'}}
        ]).to_list(None)
        if report:
            get_list = []
            for i in report:
                get_list.append(i['_id'])
        return get_list