# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
from typing import Any, Union
from .imports import FileImport
from .lro_operation import LROOperationBase
from .paging_operation import PagingOperationBase
from .response import LROPagingResponse, Response


class LROPagingOperation(LROOperationBase[LROPagingResponse], PagingOperationBase[LROPagingResponse]):
    @property
    def success_status_codes(self) -> list[Union[int, str, list[int]]]:
        """The list of all successfull status code."""
        return [200]

    @property
    def operation_type(self) -> str:
        return "lropaging"

    def cls_type_annotation(self, *, async_mode: bool, **kwargs: Any) -> str:
        return f"ClsType[{Response.type_annotation(self.responses[0], async_mode=async_mode, **kwargs)}]"

    def imports(self, async_mode: bool, **kwargs: Any) -> FileImport:
        lro_imports = LROOperationBase.imports(self, async_mode, **kwargs)
        paging_imports = PagingOperationBase.imports(self, async_mode, **kwargs)

        file_import = lro_imports
        file_import.merge(paging_imports)
        return file_import