#pragma once
/**
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * SPDX-License-Identifier: Apache-2.0.
 */

/* This file is generated. */

#include <awstest/EchoTestRpcModel.h>

#include <memory>

using namespace Aws::Eventstreamrpc;

namespace Awstest
{
    class DefaultConnectionConfig : public ConnectionConfig
    {
      public:
        DefaultConnectionConfig() noexcept;
    };

    class EchoTestRpcClient
    {
      public:
        EchoTestRpcClient(
            Aws::Crt::Io::ClientBootstrap &clientBootstrap,
            Aws::Crt::Allocator *allocator = Aws::Crt::g_allocator) noexcept;
        /**
         * Connect the client to the server
         * @param lifecycleHandler An interface that is called upon when lifecycle events relating to the connection
         * occur.
         * @param connectionConfig The configuration parameters used for establishing the connection.
         * @return An `RpcError` that can be used to check whether the connection was established.
         */
        std::future<RpcError> Connect(
            ConnectionLifecycleHandler &lifecycleHandler,
            const ConnectionConfig &connectionConfig = DefaultConnectionConfig()) noexcept;
        bool IsConnected() const noexcept { return m_connection.IsOpen(); }
        void Close() noexcept;
        void WithLaunchMode(std::launch mode) noexcept;

        std::shared_ptr<GetAllProductsOperation> NewGetAllProducts() noexcept;

        std::shared_ptr<CauseServiceErrorOperation> NewCauseServiceError() noexcept;

        std::shared_ptr<CauseStreamServiceToErrorOperation> NewCauseStreamServiceToError(
            std::shared_ptr<CauseStreamServiceToErrorStreamHandler> streamHandler) noexcept;

        std::shared_ptr<EchoStreamMessagesOperation> NewEchoStreamMessages(
            std::shared_ptr<EchoStreamMessagesStreamHandler> streamHandler) noexcept;

        std::shared_ptr<EchoMessageOperation> NewEchoMessage() noexcept;

        std::shared_ptr<GetAllCustomersOperation> NewGetAllCustomers() noexcept;

        ~EchoTestRpcClient() noexcept;

      private:
        EchoTestRpcServiceModel m_echoTestRpcServiceModel;
        ClientConnection m_connection;
        Aws::Crt::Io::ClientBootstrap &m_clientBootstrap;
        Aws::Crt::Allocator *m_allocator;
        MessageAmendment m_connectAmendment;
        std::launch m_asyncLaunchMode;
    };
} // namespace Awstest