# Copyright 2019 gRPC authors.
#
# 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.


cdef class CallbackFailureHandler:
    cdef str _core_function_name
    cdef object _error_details
    cdef object _exception_type

    cdef handle(self, object future)


cdef struct CallbackContext:
    # C struct to store callback context in the form of pointers.
    #    
    #   Attributes:
    #     functor: A grpc_experimental_completion_queue_functor represents the
    #       callback function in the only way Core understands.
    #     waiter: An asyncio.Future object that fulfills when the callback is
    #       invoked by Core.
    #     failure_handler: A CallbackFailureHandler object that called when Core
    #       returns 'success == 0' state.
    #     wrapper: A self-reference to the CallbackWrapper to help life cycle
    #       management.
    grpc_experimental_completion_queue_functor functor
    cpython.PyObject *waiter
    cpython.PyObject *failure_handler
    cpython.PyObject *callback_wrapper


cdef class CallbackWrapper:
    cdef CallbackContext context
    cdef object _reference_of_future
    cdef object _reference_of_failure_handler

    @staticmethod
    cdef void functor_run(
            grpc_experimental_completion_queue_functor* functor,
            int succeed)

    cdef grpc_experimental_completion_queue_functor *c_functor(self)


cdef class CallbackCompletionQueue:
    cdef grpc_completion_queue *_cq
    cdef object _shutdown_completed  # asyncio.Future
    cdef CallbackWrapper _wrapper

    cdef grpc_completion_queue* c_ptr(self)


cdef class GrpcCallWrapper:
    cdef grpc_call* call
