/*
 * 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 <axiom_soap_message.h>
#include <axiom_document.h>
#include <axiom_output.h>
#include "_axiom_soap_envelope.h"

/*************************** impl struct **************************************/

typedef struct axiom_soap_message_impl_t
{
    axiom_soap_message_t soap_message;

    axiom_document_t *om_doc;

    axiom_soap_envelope_t *soap_envelope;

}
axiom_soap_message_impl_t;

/*************************** Macro ********************************************/

#define AXIS2_INTF_TO_IMPL(message) ((axiom_soap_message_impl_t*)message)

/******************************************************************************/

axis2_status_t AXIS2_CALL
axiom_soap_message_free(axiom_soap_message_t *message,
        const axis2_env_t *env);

axiom_soap_envelope_t* AXIS2_CALL
axiom_soap_message_get_soap_envelope(axiom_soap_message_t *message,
        const axis2_env_t *env);


axis2_status_t AXIS2_CALL
axiom_soap_message_serialize(axiom_soap_message_t *message,
        const axis2_env_t *env,
        axiom_output_t *om_output);

/************************** function implementations **************************/

AXIS2_EXTERN axiom_soap_message_t* AXIS2_CALL
axiom_soap_message_create(const axis2_env_t *env,
        axiom_soap_builder_t *builder,
        axiom_document_t *om_doc)
{
    axiom_soap_message_impl_t *soap_message_impl = NULL;
    AXIS2_ENV_CHECK(env, NULL);
    AXIS2_PARAM_CHECK(env->error, om_doc, NULL);
    AXIS2_PARAM_CHECK(env->error, builder, NULL);

    soap_message_impl = (axiom_soap_message_impl_t *)AXIS2_MALLOC(env->allocator,
            sizeof(axiom_soap_message_impl_t));
    if (!soap_message_impl)
    {
        AXIS2_ERROR_SET(env->error, AXIS2_ERROR_NO_MEMORY, AXIS2_FAILURE);
        return NULL;
    }

    soap_message_impl->soap_message.ops = NULL;
    soap_message_impl->om_doc = NULL;
    soap_message_impl->soap_envelope = NULL;

    soap_message_impl->soap_message.ops = (axiom_soap_message_ops_t*)AXIS2_MALLOC(env->allocator,
            sizeof(axiom_soap_message_ops_t));

    if (!(soap_message_impl->soap_message.ops))
    {
        AXIS2_ERROR_SET(env->error, AXIS2_ERROR_NO_MEMORY, AXIS2_FAILURE);
        AXIS2_FREE(env->allocator, soap_message_impl);
        soap_message_impl = NULL;
        return NULL;
    }

    soap_message_impl->om_doc = om_doc;

    soap_message_impl->soap_message.ops->free_fn =
        axiom_soap_message_free;
    soap_message_impl->soap_message.ops->serialize =
        axiom_soap_message_serialize;
    soap_message_impl->soap_message.ops->get_soap_envelope =
        axiom_soap_message_get_soap_envelope;


    return &(soap_message_impl->soap_message);
}

axis2_status_t AXIS2_CALL
axiom_soap_message_free(axiom_soap_message_t *message,
        const axis2_env_t *env)
{
    axiom_soap_message_impl_t *soap_message_impl = NULL;
    AXIS2_ENV_CHECK(env, AXIS2_FAILURE);
    soap_message_impl = AXIS2_INTF_TO_IMPL(message);
    if (soap_message_impl->om_doc)
    {
        AXIOM_DOCUMENT_FREE(soap_message_impl->om_doc, env);
        soap_message_impl->om_doc = NULL;
    }
    if (soap_message_impl->soap_envelope)
    {
        AXIOM_SOAP_ENVELOPE_FREE(soap_message_impl->soap_envelope, env);
        soap_message_impl->soap_envelope = NULL;
    }
    if (soap_message_impl->soap_message.ops)
    {
        AXIS2_FREE(env->allocator, soap_message_impl->soap_message.ops);
        soap_message_impl->soap_message.ops = NULL;
    }
    AXIS2_FREE(env->allocator, soap_message_impl);
    soap_message_impl = NULL;
    return AXIS2_SUCCESS;
}

axiom_soap_envelope_t* AXIS2_CALL
axiom_soap_message_get_soap_envelope(axiom_soap_message_t *message,
        const axis2_env_t *env)
{
    axiom_soap_message_impl_t* soap_message_impl = NULL;
    AXIS2_ENV_CHECK(env, NULL);
    soap_message_impl = AXIS2_INTF_TO_IMPL(message);
    if (soap_message_impl->soap_envelope)
    {
        return soap_message_impl->soap_envelope;
    }
    else
    {
        soap_message_impl->soap_envelope = axiom_soap_envelope_create_null(env);

        return soap_message_impl->soap_envelope;
    }
}

axis2_status_t AXIS2_CALL
axiom_soap_message_serialize(axiom_soap_message_t *message,
        const axis2_env_t *env,
        axiom_output_t *om_output)
{
    axiom_soap_message_impl_t *soap_message_impl = NULL;
    AXIS2_ENV_CHECK(env, AXIS2_FAILURE);
    soap_message_impl = AXIS2_INTF_TO_IMPL(message);
    return AXIOM_DOCUMENT_SERIALIZE(soap_message_impl->om_doc, env, om_output);
}