#
# (c) 2024, Yegor Yakubovich, yegoryakubovich.com, personal@yegoryakybovich.com
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#


from mybody_api_client.routes.admin.days.meals import AdminDayMealRoute
from mybody_api_client.routes.admin.days.trainings import AdminDayTrainingRoute
from mybody_api_client.utils import BaseRoute, RequestTypes


class AdminDayRoute(BaseRoute):
    _prefix = '/days'

    meals = AdminDayMealRoute()
    training = AdminDayTrainingRoute()

    async def create(
            self,
            account_service_id: int,
            date: str,
            water_amount: int,
    ):
        return await self._request(
            type_=RequestTypes.POST,
            prefix='/create',
            parameters={
                'account_service_id': account_service_id,
                'date': date,
                'water_amount': water_amount,
            },
            response_key='id',
        )

    async def update(
            self,
            id_: int,
            water_amount: int = None,
            water_intake: int = None,
    ):
        return await self._request(
            type_=RequestTypes.POST,
            prefix='/update',
            parameters={
                'id': id_,
                'water_amount': water_amount,
                'water_intake': water_intake,
            },
        )

    async def delete(
            self,
            id_: int,
    ):
        return await self._request(
            type_=RequestTypes.POST,
            prefix='/delete',
            parameters={
                'id': id_,
            },
        )

    async def duplicate(
            self,
            id_: int,
            date: str,
    ):
        return await self._request(
            type_=RequestTypes.POST,
            prefix='/duplicate',
            parameters={
                'id': id_,
                'date': date,
            },
            response_key='id',
        )

    async def get(
            self,
            id_: int,
    ):
        return await self._request(
            type_=RequestTypes.GET,
            prefix='/get',
            parameters={
                'id': id_,
            },
            response_key='day',
        )

    async def get_by_date(
            self,
            account_service_id: int,
            date: str,
    ):
        return await self._request(
            type_=RequestTypes.GET,
            prefix='/by-date/get',
            parameters={
                'account_service_id': account_service_id,
                'date': date,
            },
            response_key='day',
        )

    async def get_list(
            self,
            account_service_id: int,
    ):
        return await self._request(
            type_=RequestTypes.GET,
            prefix='/list/get',
            parameters={
                'account_service_id': account_service_id,
            },
            response_key='days',
        )

