/*
 *   Copyright 2003-2004 The Apache Software Foundation.
 *
 *   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.
 *
 * @author Rangika Mendis (rangika@opensource.lk)
 * @author Nithyakala Thangarajah (nithya@opensource.lk)
 * @author Roshan Weerasuriya (roshan@opensource.lk, roshan@jkcsworld.com)
 */

// !!! This include file must be first thing in file !!!
#include "../platforms/PlatformAutoSense.hpp"

#include "SoapAttachment.hpp"
#include "SoapSerializer.h"

#include "../common/AxisTrace.h"

AXIS_CPP_NAMESPACE_START


SoapAttachment::
SoapAttachment(ContentIdSet *pContentIdSet)
{    
    m_AttachmentHeaders = new SoapAttachmentHeaders();
    m_AttachmentBody = NULL;
    m_binaryBody = NULL;
    m_pContentIdSet = pContentIdSet;
    if (NULL!=m_pContentIdSet)
        m_AttachmentHeaders->addHeader(AXIS_CONTENT_ID, pContentIdSet->generateId());

    // Set up some sensible default values for mandatory headers.
    m_AttachmentHeaders->addHeader(AXIS_CONTENT_TYPE,"text/plain");
    m_AttachmentHeaders->addHeader(AXIS_CONTENT_TRANSFER_ENCODING,"base64");
}

SoapAttachment::
~SoapAttachment()
{
    delete m_AttachmentHeaders;
    // client code generation does not assume this object will 
    // assume ownership of m_AttachmentBody!
    // Need to revisit, maybe add boolean that can be 
    // set by caller to indicate whether ownership is 
    // passed to instances of this class.
}

void SoapAttachment::
addHeader(const char* pchName, const char* pchValue)
{
    // Registering the contentid here ensures that it is unique within the mime message.
    if (NULL!=m_pContentIdSet && 0==strcmp(AXIS_CONTENT_ID,pchName)) 
        m_pContentIdSet->registerId(pchValue);
    m_AttachmentHeaders->addHeader(pchName, pchValue);
}

void SoapAttachment::
addBody(xsd__base64Binary* objBody)
{
    iEncodingStyle = AXIS_BASE64;
    m_AttachmentBody = objBody;
}

void SoapAttachment::
addBody(char* pchBinaryBody)
{ 
    delete m_binaryBody;
    m_binaryBody = NULL;
    if (!pchBinaryBody)
        return;
    iEncodingStyle = AXIS_BINARY;
    m_binaryBody = new char[strlen(pchBinaryBody) + 1];
    strcpy(m_binaryBody,pchBinaryBody);
}

void SoapAttachment::
serialize(SoapSerializer &pSZ)
{
    /* Serialize the Attachment Headers */
    pSZ.serialize("\r\n", NULL);
    m_AttachmentHeaders->serialize(pSZ);

    /* Serialize the Attachment Body */
    if (iEncodingStyle == AXIS_BASE64)
    {
        if (m_AttachmentBody) 
        {
            pSZ.serialize("\r\n", NULL);
            pSZ.serializeAsChardata(m_AttachmentBody, XSD_BASE64BINARY);
        }
    } 
    else if (iEncodingStyle == AXIS_BINARY)
    {
        if (m_binaryBody)
        {
            pSZ.serialize("\r\n", NULL);
            string attachBody = m_binaryBody;
           
            long boundary = attachBody.find("------=_Part_0_6349096.11054186923",0);
            if (boundary != -1)
            {
                string attachment = attachBody.substr(0,(boundary-1));
                char* binary = new char[attachment.length() + 1];
                strcpy(binary, attachment.c_str()); 
                delete m_binaryBody;       
                m_binaryBody = binary;            
                pSZ.serialize(m_binaryBody, NULL);
            }
            else              
                pSZ.serialize(m_binaryBody, NULL);
        }
    }

    pSZ.serialize("\r\n", NULL);
}

xsd__base64Binary* SoapAttachment::
getBody()
{
    return m_AttachmentBody;
}

const char* SoapAttachment::
getHeader(const char *pchName)
{
    return m_AttachmentHeaders->getHeader(pchName);
}

const char* SoapAttachment::
getAttachmentId()
{
   return getHeader(AXIS_CONTENT_ID);
}

AXIS_CPP_NAMESPACE_END