# Code generated by Lark OpenAPI.

from typing import Any, Optional, Union, Dict, List, Set, IO, Callable, Type
from lark_oapi.core.construct import init
from .order_field import OrderField


class PageRequest(object):
    _types = {
        "page_type": int,
        "limit": int,
        "order_type": int,
        "offset": int,
        "cursor": str,
        "order_fields": List[OrderField],
        "cursor_field": str,
    }

    def __init__(self, d=None):
        self.page_type: Optional[int] = None
        self.limit: Optional[int] = None
        self.order_type: Optional[int] = None
        self.offset: Optional[int] = None
        self.cursor: Optional[str] = None
        self.order_fields: Optional[List[OrderField]] = None
        self.cursor_field: Optional[str] = None
        init(self, d, self._types)

    @staticmethod
    def builder() -> "PageRequestBuilder":
        return PageRequestBuilder()


class PageRequestBuilder(object):
    def __init__(self) -> None:
        self._page_request = PageRequest()

    def page_type(self, page_type: int) -> "PageRequestBuilder":
        self._page_request.page_type = page_type
        return self

    def limit(self, limit: int) -> "PageRequestBuilder":
        self._page_request.limit = limit
        return self

    def order_type(self, order_type: int) -> "PageRequestBuilder":
        self._page_request.order_type = order_type
        return self

    def offset(self, offset: int) -> "PageRequestBuilder":
        self._page_request.offset = offset
        return self

    def cursor(self, cursor: str) -> "PageRequestBuilder":
        self._page_request.cursor = cursor
        return self

    def order_fields(self, order_fields: List[OrderField]) -> "PageRequestBuilder":
        self._page_request.order_fields = order_fields
        return self

    def cursor_field(self, cursor_field: str) -> "PageRequestBuilder":
        self._page_request.cursor_field = cursor_field
        return self

    def build(self) -> "PageRequest":
        return self._page_request