path: root/cpp/bindings/qpid/dotnet/src/Message.cpp

/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements.  See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership.  The ASF licenses this file
* to you 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.
*/

#include <windows.h>
#include <msclr\lock.h>
#include <oletx2xa.h>
#include <typeinfo.h>
#include <string>
#include <limits>
#include <iostream>
#include <stdlib.h>

#include "qpid/messaging/Message.h"
#include "qpid/types/Variant.h"

#include "QpidMarshal.h"
#include "Address.h"
#include "Duration.h"
#include "Message.h"
#include "QpidTypeCheck.h"
#include "QpidException.h"
#include "TypeTranslator.h"

namespace Org {
namespace Apache {
namespace Qpid {
namespace Messaging {

    /// <summary>
    /// Message is a managed wrapper for a ::qpid::messaging::Message
    /// </summary>

    // Create empty message
    Message::Message() :
        messagep(new ::qpid::messaging::Message(QpidMarshal::ToNative("")))
    {
    }

    // Create from string
    Message::Message(System::String ^ theStr) :
        messagep(new ::qpid::messaging::Message(QpidMarshal::ToNative(theStr)))
    {
    }

    // Create from object
    Message::Message(System::Object ^ theValue) :
        messagep(new ::qpid::messaging::Message(QpidMarshal::ToNative("")))
    {
        if (QpidTypeCheck::ObjectIsMap(theValue))
        {
            // Create a mapped message using given dictionary

            // Allocate a map
            ::qpid::types::Variant::Map newMap;

            // Add the map variables to the map
            TypeTranslator::ManagedToNative((QpidMap ^)theValue, newMap);

            // Set message content type
            messagep->setContentType("ampq/map");

            // Insert the map into the message
            ::qpid::messaging::encode(newMap, *messagep, QpidMarshal::ToNative("amqp/map"));
        }
        else if (QpidTypeCheck::ObjectIsList(theValue))
        {
            // Create a list message using given list

            // Allocate a list
            ::qpid::types::Variant::List newList;

            // Add the list variables to the list
            TypeTranslator::ManagedToNative((QpidList ^)theValue, newList);

            // Set message content type
            messagep->setContentType("ampq/list");

            // Insert the list into the message
            ::qpid::messaging::encode(newList, *messagep, QpidMarshal::ToNative("amqp/list"));
        }
        else
        {
            // Create a binary string message
            messagep->setContent(QpidMarshal::ToNative(theValue->ToString()));
        }
    }


	// Create from bytes
	Message::Message(array<System::Byte> ^ bytes)
	{
		pin_ptr<unsigned char> pBytes = &bytes[0];
		messagep = new ::qpid::messaging::Message((char *)pBytes, bytes->Length);
	}

    // Create from byte array slice
	Message::Message(array<System::Byte> ^ bytes, int offset, int size)
	{
        if ((offset + size) > bytes->Length)
			throw gcnew QpidException("Message::Message Create from byte array slice: buffer length exceeded");

		pin_ptr<unsigned char> pBytes = &bytes[offset];
		messagep = new ::qpid::messaging::Message((char *)pBytes, size);
	}


    // Create from received message
    Message::Message(::qpid::messaging::Message * msgp) :
        messagep(msgp)
    {
    }


    // Destructor
    Message::~Message()
    {
        Cleanup();
    }


    // Finalizer
    Message::!Message()
    {
        Cleanup();
    }

    // Copy constructor
    // TODO: prevent copy
    Message::Message(const Message % rhs)
    {
        messagep      = rhs.messagep;
    }

    // Destroys kept object
    // TODO: add lock
    void Message::Cleanup()
    {
        if (NULL != messagep)
        {
            delete messagep;
            messagep = NULL;
        }
    }


    //
    // ReplyTo
    //
    void Message::SetReplyTo(Address ^ address)
    {
        messagep->setReplyTo(*(address->NativeAddress));
    }

    Address ^ Message::GetReplyTo()
    {
        const ::qpid::messaging::Address & addrp =
            messagep->::qpid::messaging::Message::getReplyTo();

        return gcnew Address(const_cast<::qpid::messaging::Address *>(&addrp));
    }


    //
    // Subject
    //
    void Message::SetSubject(System::String ^ subject)
    {
        messagep->setSubject(QpidMarshal::ToNative(subject));
    }
    
    System::String ^ Message::GetSubject()
    {
        return gcnew String(messagep->getSubject().c_str());
    }
    

    //
    // ContentType
    //
    void Message::SetContentType(System::String ^ ct)
    {
        messagep->setContentType(QpidMarshal::ToNative(ct));
    }
    
	System::String ^ Message::GetContentType()
    {
		return gcnew String(messagep->::qpid::messaging::Message::getContentType().c_str());
    }
    
    
    //
    // MessageId
    //
    void Message::SetMessageId(System::String ^ messageId)
    {
        messagep->setMessageId(QpidMarshal::ToNative(messageId));
    }
    
    System::String ^ Message::GetMessageId()
    {
        return gcnew String(messagep->getMessageId().c_str());
    }
    
    
    //
    // UserId
    //
    void Message::SetUserId(System::String ^ uId)
    {
        messagep->setUserId(QpidMarshal::ToNative(uId));
    }
    
    System::String ^ Message::GetUserId()
    {
        return gcnew String(messagep->getUserId().c_str());
    }
    
    
    //
    // CorrelationId
    //
    void Message::SetCorrelationId(System::String ^ correlationId)
    {
        messagep->setCorrelationId(QpidMarshal::ToNative(correlationId));
    }
    
    System::String ^ Message::GetCorrelationId()
    {
        return gcnew String(messagep->getCorrelationId().c_str());
    }
    

    //
    // Priority
    //
    void Message::SetPriority(unsigned char priority)
    {
        messagep->setPriority(priority);
    }
    
    unsigned char Message::GetPriority()
    {
        return messagep->getPriority();
    }
    

    //
    // Ttl
    //
    void Message::SetTtl(Duration ^ ttl)
    {
        ::qpid::messaging::Duration dur(ttl->Milliseconds);

        messagep->setTtl(dur);
    }
    
    Duration ^ Message::GetTtl()
    {
        Duration ^ dur = gcnew Duration(messagep->getTtl().getMilliseconds());

        return dur;
    }

    void Message::SetDurable(bool durable)
    {
        messagep->setDurable(durable);
    }
    
    bool Message::GetDurable()
    {
        return messagep->getDurable();
    }


    bool Message::GetRedelivered()
    {
        return messagep->getRedelivered();
    }

    void Message::SetRedelivered(bool redelivered)
    {
        messagep->setRedelivered(redelivered);
    }

	// Properties
    System::Collections::Generic::Dictionary<
            System::String^, System::Object^> ^ Message::GetProperties()
    {
        ::qpid::types::Variant::Map map;

        map = messagep->getProperties();

        System::Collections::Generic::Dictionary<
            System::String^, System::Object^> ^ dict =
            gcnew System::Collections::Generic::Dictionary<
                      System::String^, System::Object^> ;

        TypeTranslator::NativeToManaged(map, dict);

        return dict;
    }


	void Message::SetProperty(System::String ^ name, System::Object ^ value)
	{
        ::qpid::types::Variant entryValue;
        TypeTranslator::ManagedToNativeObject(value, entryValue);

		messagep->getProperties()[QpidMarshal::ToNative(name)] = entryValue;
	}


	void Message::SetProperties(System::Collections::Generic::Dictionary<
            System::String^, System::Object^> ^ properties)
	{
		for each (System::Collections::Generic::KeyValuePair
			     <System::String^, System::Object^> kvp in properties)
        {
			SetProperty(kvp.Key, kvp.Value);
		}
	}


	
	// Content
	void Message::SetContent(System::String ^ content)
    {
        messagep->setContent(QpidMarshal::ToNative(content));
    }


    System::String ^ Message::GetContent()
    {
        return gcnew String(messagep->getContent().c_str());
    }


    //
    // User wants to extract a Dictionary from the message
    //
    void Message::GetContent(System::Collections::Generic::Dictionary<
                                System::String^, 
                                System::Object^> ^ dict)
    {
        // Extract the message map from the message
        ::qpid::types::Variant::Map map;
        
        ::qpid::messaging::decode(*messagep, map, QpidMarshal::ToNative("amqp/map"));

        TypeTranslator::NativeToManaged(map, dict);
    }


    //
    // User wants to extract a list from the message
    //
    void Message::GetContent(System::Collections::ObjectModel::Collection<
                        System::Object^> ^ list)
    {
        // allocate a native messaging::List
        ::qpid::types::Variant::List nativeList;
        
        // Extract the list from the message in native format
        ::qpid::messaging::decode(*messagep, nativeList, QpidMarshal::ToNative("amqp/list"));

        // translate native list into user's managed list
        TypeTranslator::NativeToManaged(nativeList, list);
    }

    //
    // Return message content to raw byte array.
    // On entry message size must not be zero and
	// caller's byte array must be equal to message size.
    //
    void Message::GetRaw(array<System::Byte> ^ arr)
    {
        System::UInt32 size = messagep->getContentSize();
     
        if (0 == size)
            throw gcnew QpidException("Message::GetRaw - message size is zero");

        if (arr->Length != size)
            throw gcnew QpidException("Message::GetRaw - receive buffer is wrong size");

        const char * pMsgSrc = messagep->getContentPtr();
		pin_ptr<unsigned char> pArr = &arr[0];
		memcpy(pArr, pMsgSrc, size);
    }


    System::UInt64 Message::GetContentSize()
    {
        return messagep->getContentSize();
    }


	System::String ^ Message::MapAsString(System::Collections::Generic::Dictionary<
					           System::String^, System::Object^> ^ dict)
    {
		System::String ^ leading = "";
		System::Text::StringBuilder ^ sb = gcnew System::Text::StringBuilder("{");

		for each (System::Collections::Generic::KeyValuePair
			     <System::String^, System::Object^> kvp in dict)
        {
            sb->Append(leading);
            leading = ", ";

			if (QpidTypeCheck::ObjectIsMap(kvp.Value))
            {
				sb->AppendFormat(
					"{0}={1}", 
					kvp.Key,
					MapAsString((System::Collections::Generic::Dictionary<System::String^, System::Object^> ^)kvp.Value));
            }
			else if (QpidTypeCheck::ObjectIsList(kvp.Value))
            {
                sb->AppendFormat(
					"{0}={1}", 
					kvp.Key,
					ListAsString((System::Collections::ObjectModel::Collection<
							System::Object^> ^)kvp.Value));
            }
            else
                sb->AppendFormat("{0}={1}", kvp.Key, kvp.Value);
        }
		sb->Append("}");

		System::String ^ result = gcnew System::String(sb->ToString());
		return result;
    }

    /// <summary>
    /// A function to display a ampq/list message packaged as a List.
    /// </summary>
    /// <param name="list">The AMQP list</param>
	System::String ^ Message::ListAsString(System::Collections::ObjectModel::Collection<System::Object^> ^ list)
    {
		System::String ^ leading = "";
		System::Text::StringBuilder ^ sb = gcnew System::Text::StringBuilder("[");

		for each (System::Object ^ obj in list)
        {
            sb->Append(leading);
            leading = ", ";

			if (QpidTypeCheck::ObjectIsMap(obj))
            {
                sb->Append(MapAsString((System::Collections::Generic::Dictionary<
                                System::String^, System::Object^> ^)obj));
            }
			else if (QpidTypeCheck::ObjectIsList(obj))
            {
                sb->Append(ListAsString((System::Collections::ObjectModel::Collection<
                                System::Object^> ^)obj));
            }
            else
                sb->Append(obj->ToString());
        }
        sb->Append("]");

		System::String ^ result = gcnew System::String(sb->ToString());
		return result;
    }

	System::String ^ Message::AsString(System::Object ^ obj)
	{
		if (QpidTypeCheck::ObjectIsMap(obj))
			return MapAsString((System::Collections::Generic::Dictionary<
                                System::String^, System::Object^> ^)obj);
		else if (QpidTypeCheck::ObjectIsList(obj))
			return ListAsString((System::Collections::ObjectModel::Collection<
                                System::Object^> ^)obj);
		else
			return obj->ToString();
	}
}}}}