Message.h

Go to the documentation of this file.

/*
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 *
 *     http://aws.amazon.com/apache2.0/
 *
 * or in the "license" file accompanying this file. This file 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.
 */

#ifndef ALEXA_CLIENT_SDK_LIBIPCSERVERSAMPLEAPP_INCLUDE_IPCSERVERSAMPLEAPP_MESSAGES_MESSAGE_H_
#define ALEXA_CLIENT_SDK_LIBIPCSERVERSAMPLEAPP_INCLUDE_IPCSERVERSAMPLEAPP_MESSAGES_MESSAGE_H_

#include <string>

#include <rapidjson/document.h>
#include <rapidjson/writer.h>
#include <rapidjson/stringbuffer.h>

#include "MessageInterface.h"

const char MSG_PAYLOAD_TAG[] = "payload";

const char MSG_WINDOWID_TAG[] = "windowId";

const char MSG_ENABLED_TAG[] = "enabled";

const char MSG_TOKEN_TAG[] = "token";

const char MSG_STATE_TAG[] = "state";

const std::string EMPTY_JSON("{}");

namespace alexaClientSDK {
namespace sampleApplications {
namespace ipcServerSampleApp {
namespace messages {

class Message : public MessageInterface {
private:
    rapidjson::Document mParsedDocument;

public:
    Message(const std::string& nameSpace, const int version, const std::string& name) :
            MessageInterface(version, nameSpace, name),
            mParsedDocument(&alloc()) {
    }

    Message& addMember(const std::string& name, const std::string& value) {
        mDocument.AddMember(
            rapidjson::Value(name.c_str(), mDocument.GetAllocator()).Move(),
            rapidjson::Value(value.c_str(), mDocument.GetAllocator()).Move(),
            mDocument.GetAllocator());
        return *this;
    }

    Message& addMember(const std::string& name, unsigned value) {
        mDocument.AddMember(
            rapidjson::Value(name.c_str(), mDocument.GetAllocator()).Move(), value, mDocument.GetAllocator());
        return *this;
    }

    Message& addMemberInPayload(const std::string& name, const std::string& value) {
        mPayload.AddMember(
            rapidjson::Value(name.c_str(), mDocument.GetAllocator()).Move(),
            rapidjson::Value(value.c_str(), mDocument.GetAllocator()).Move(),
            mDocument.GetAllocator());
        return *this;
    }

    Message& addIntegerInPayload(const std::string& name, const int value) {
        mPayload.AddMember(
            rapidjson::Value(name.c_str(), mDocument.GetAllocator()).Move(), value, mDocument.GetAllocator());
        return *this;
    }

    Message& addMemberInPayload(const std::string& name, unsigned value) {
        mPayload.AddMember(
            rapidjson::Value(name.c_str(), mDocument.GetAllocator()).Move(), value, mDocument.GetAllocator());
        return *this;
    }

    Message& setEnabledInPayload(bool enabled) {
        addMemberInPayload(MSG_ENABLED_TAG, enabled);
        return *this;
    }

    Message& setStateInPayload(const std::string& state) {
        mPayload.AddMember(
            MSG_STATE_TAG, rapidjson::Value(state.c_str(), mDocument.GetAllocator()).Move(), mDocument.GetAllocator());
        return *this;
    }

    Message& setTokenInPayload(const std::string& token) {
        mPayload.AddMember(MSG_TOKEN_TAG, token, mDocument.GetAllocator());
        return *this;
    }

    Message& setTokenInPayload(unsigned token) {
        mPayload.AddMember(MSG_TOKEN_TAG, token, mDocument.GetAllocator());
        return *this;
    }

    Message& setParsedPayload(const std::string& payload) {
        mDocument.AddMember(MSG_PAYLOAD_TAG, mParsedDocument.Parse(payload), mDocument.GetAllocator());
        return *this;
    }

    Message& setParsedPayloadInPayload(const std::string& payload) {
        mPayload.AddMember(MSG_PAYLOAD_TAG, mParsedDocument.Parse(payload), mDocument.GetAllocator());
        return *this;
    }

    Message& setWindowIdInPayload(const std::string& windowId) {
        mPayload.AddMember(MSG_WINDOWID_TAG, windowId, mDocument.GetAllocator());
        return *this;
    }

    Message& addPayload() {
        mDocument.AddMember(MSG_PAYLOAD_TAG, mPayload, mDocument.GetAllocator());
        return *this;
    }

    Message& setPayload(rapidjson::Value&& payload) {
        mDocument.AddMember(MSG_PAYLOAD_TAG, std::move(payload), mDocument.GetAllocator());
        return *this;
    }

    auto alloc() -> decltype(mDocument.GetAllocator()) {
        return mDocument.GetAllocator();
    };

    std::string get() {
        rapidjson::StringBuffer buffer;
        buffer.Clear();
        rapidjson::Writer<
            rapidjson::StringBuffer,
            rapidjson::UTF8<>,
            rapidjson::UTF8<>,
            rapidjson::CrtAllocator,
            rapidjson::kWriteNanAndInfFlag>
            writer(buffer);
        if (!mDocument.Accept(writer)) {
            return EMPTY_JSON;
        }
        return std::string(buffer.GetString(), buffer.GetSize());
    }

    rapidjson::Value&& getValue() {
        return std::move(mDocument);
    };
};
}  // namespace messages
}  // namespace ipcServerSampleApp
}  // namespace sampleApplications
}  // namespace alexaClientSDK

#endif  // ALEXA_CLIENT_SDK_LIBIPCSERVERSAMPLEAPP_INCLUDE_IPCSERVERSAMPLEAPP_MESSAGES_MESSAGE_H_