ChatHandler.h

 
#pragma once
#include <aws/core/client/AWSError.h>
#include <aws/core/utils/HashingUtils.h>
#include <aws/core/utils/event/EventStreamHandler.h>
#include <aws/qbusiness/QBusinessErrors.h>
#include <aws/qbusiness/QBusiness_EXPORTS.h>
#include <aws/qbusiness/model/ActionReviewEvent.h>
#include <aws/qbusiness/model/AuthChallengeRequestEvent.h>
#include <aws/qbusiness/model/ChatInitialResponse.h>
#include <aws/qbusiness/model/FailedAttachmentEvent.h>
#include <aws/qbusiness/model/MetadataEvent.h>
#include <aws/qbusiness/model/TextOutputEvent.h>
 
namespace Aws {
namespace QBusiness {
namespace Model {
enum class ChatEventType {
  INITIAL_RESPONSE,
  TEXTEVENT,
  METADATAEVENT,
  ACTIONREVIEWEVENT,
  FAILEDATTACHMENTEVENT,
  AUTHCHALLENGEREQUESTEVENT,
  UNKNOWN
};
 
class ChatHandler : public Aws::Utils::Event::EventStreamHandler {
  typedef std::function<void(const ChatInitialResponse&)> ChatInitialResponseCallback;
  typedef std::function<void(const ChatInitialResponse&, const Utils::Event::InitialResponseType)> ChatInitialResponseCallbackEx;
  typedef std::function<void(const TextOutputEvent&)> TextOutputEventCallback;
  typedef std::function<void(const MetadataEvent&)> MetadataEventCallback;
  typedef std::function<void(const ActionReviewEvent&)> ActionReviewEventCallback;
  typedef std::function<void(const FailedAttachmentEvent&)> FailedAttachmentEventCallback;
  typedef std::function<void(const AuthChallengeRequestEvent&)> AuthChallengeRequestEventCallback;
  typedef std::function<void(const Aws::Client::AWSError<QBusinessErrors>& error)> ErrorCallback;
 
 public:
  AWS_QBUSINESS_API ChatHandler();
  AWS_QBUSINESS_API ChatHandler& operator=(const ChatHandler&) = default;
  AWS_QBUSINESS_API ChatHandler(const ChatHandler&) = default;
 
  AWS_QBUSINESS_API virtual void OnEvent() override;
 
 
  inline void SetInitialResponseCallbackEx(const ChatInitialResponseCallbackEx& callback) { m_onInitialResponse = callback; }
  inline void SetInitialResponseCallback(const ChatInitialResponseCallback& noArgCallback) {
    m_onInitialResponse = [noArgCallback](const ChatInitialResponse& rs, const Utils::Event::InitialResponseType) {
      return noArgCallback(rs);
    };
  }
  inline void SetTextOutputEventCallback(const TextOutputEventCallback& callback) { m_onTextOutputEvent = callback; }
  inline void SetMetadataEventCallback(const MetadataEventCallback& callback) { m_onMetadataEvent = callback; }
  inline void SetActionReviewEventCallback(const ActionReviewEventCallback& callback) { m_onActionReviewEvent = callback; }
  inline void SetFailedAttachmentEventCallback(const FailedAttachmentEventCallback& callback) { m_onFailedAttachmentEvent = callback; }
  inline void SetAuthChallengeRequestEventCallback(const AuthChallengeRequestEventCallback& callback) {
    m_onAuthChallengeRequestEvent = callback;
  }
  inline void SetOnErrorCallback(const ErrorCallback& callback) { m_onError = callback; }
 
  inline ChatInitialResponseCallbackEx& GetInitialResponseCallbackEx() { return m_onInitialResponse; }
 
 private:
  AWS_QBUSINESS_API void HandleEventInMessage();
  AWS_QBUSINESS_API void HandleErrorInMessage();
  AWS_QBUSINESS_API void MarshallError(const Aws::String& errorCode, const Aws::String& errorMessage);
 
  ChatInitialResponseCallbackEx m_onInitialResponse;
  TextOutputEventCallback m_onTextOutputEvent;
  MetadataEventCallback m_onMetadataEvent;
  ActionReviewEventCallback m_onActionReviewEvent;
  FailedAttachmentEventCallback m_onFailedAttachmentEvent;
  AuthChallengeRequestEventCallback m_onAuthChallengeRequestEvent;
  ErrorCallback m_onError;
};
 
namespace ChatEventMapper {
AWS_QBUSINESS_API ChatEventType GetChatEventTypeForName(const Aws::String& name);
 
AWS_QBUSINESS_API Aws::String GetNameForChatEventType(ChatEventType value);
}  // namespace ChatEventMapper
}  // namespace Model
}  // namespace QBusiness
}  // namespace Aws