/******************************************************************
*  $Id: soap-env.c,v 1.10 2004/11/02 23:09:26 snowdrop Exp $
*
* CSOAP Project:  A SOAP client/server library in C
* Copyright (C) 2003  Ferhat Ayaz
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
* Boston, MA  02111-1307, USA.
* 
* Email: ayaz@jprogrammer.net
******************************************************************/
#include <libcsoap/soap-env.h>
#include <stdarg.h>
#include <string.h>

#ifdef WIN32
#define USE_XMLSTRING
#endif

#ifdef USE_XMLSTRING
#include <libxml/xmlstring.h>
#endif

static char *soap_env_ns = "http://schemas.xmlsoap.org/soap/envelope/";
static char *soap_env_enc = "http://schemas.xmlsoap.org/soap/encoding/";
static char *soap_xsi_ns = "http://www.w3.org/1999/XMLSchema-instance";
static char *soap_xsd_ns = "http://www.w3.org/1999/XMLSchema";

/*
Parameters:
1- soap_env_ns
2- soap_env_enc
3- xsi_ns
4- xsd_ns
3- method name
4- uri
5- method name(again)
*/
#define _SOAP_MSG_TEMPLATE_ \
	"<SOAP-ENV:Envelope xmlns:SOAP-ENV=\"%s\" SOAP-ENV:encoding=\"%s\"" \
	" xmlns:xsi=\"%s\"" \
	" xmlns:xsd=\"%s\">" \
	" <SOAP-ENV:Body>"\
	"  <m:%s xmlns:m=\"%s\">"\
	"  </m:%s>" \
	" </SOAP-ENV:Body>"\
	"</SOAP-ENV:Envelope>"


/*
Parameters:
1- soap_env_ns
2- soap_env_enc
3- xsi_ns
4- xsd_ns
3- method name
4- uri
5- method name(again)
*/
#define _SOAP_MSG_TEMPLATE_EMPTY_TARGET_ \
	"<SOAP-ENV:Envelope xmlns:SOAP-ENV=\"%s\" SOAP-ENV:encoding=\"%s\"" \
	" xmlns:xsi=\"%s\"" \
	" xmlns:xsd=\"%s\">" \
	" <SOAP-ENV:Body>"\
	"  <%s xmlns=\"%s\">"\
	"  </%s>" \
	" </SOAP-ENV:Body>"\
	"</SOAP-ENV:Envelope>"


/* ---------------------------------------------------------------------------- */
/*     XML Serializers (implemented at the and of this document)              */
/* ---------------------------------------------------------------------------- */
struct XmlNodeHolder
{
  xmlNodePtr node;
};

static
  void xmlbuilder_start_element (const xmlChar * element_name, int attr_count,
				 xmlChar ** keys, xmlChar ** values,
				 void *userData);

static
  void xmlbuilder_characters (const xmlChar * element_name,
			      const xmlChar * chars, void *userData);

static
  void xmlbuilder_end_element (const xmlChar * element_name, void *userData);

/* ---------------------------------------------------------------------------- */



herror_t
soap_env_new_from_doc (xmlDocPtr doc, SoapEnv **out)
{
  xmlNodePtr node;
  SoapEnv *env;

  if (doc == NULL)
  {
    log_error1 ("Can not create xml document!");
    return herror_new("soap_env_new_from_doc", 
		GENERAL_INVALID_PARAM, "XML Document (xmlDocPtr) is NULL");
  }

  node = xmlDocGetRootElement (doc);
  if (node == NULL)
  {
    log_error1 ("xml document is empty!");
    return herror_new("soap_env_new_from_doc", 
		XML_ERROR_EMPTY_DOCUMENT, "XML Document is empty!");
  }

  env = (SoapEnv *) malloc (sizeof (SoapEnv));

  /* set root */
  env->root = node;

  /* set method root
     set call->cur (current node) to <method>.
     xpath: //Envelope/Body/
   */
  node = soap_xml_get_children (env->root);/* Check for NULL ! */
  env->cur = soap_xml_get_children (node); /* Check for NULL ! */

  *out = env;
  return H_OK;
}




herror_t
soap_env_new_from_buffer (const char *buffer, SoapEnv **out)
{
  xmlDocPtr doc;
  herror_t err;

  if (buffer == NULL)
     return herror_new("soap_env_new_from_buffer", 
		GENERAL_INVALID_PARAM, "buffer (first param) is NULL");

  doc = xmlParseDoc (BAD_CAST buffer);
  if (doc == NULL)
     return herror_new("soap_env_new_from_buffer", 
		XML_ERROR_PARSE, "Can not parse xml");

  err = soap_env_new_from_doc (doc, out);
  if (err != H_OK)  {
    xmlFreeDoc (doc);
  }

  return err;
}


herror_t
soap_env_new_with_fault (fault_code_t faultcode,
			 const char *faultstring,
			 const char *faultactor, const char *detail, SoapEnv **out)
{
  xmlDocPtr doc;
  herror_t err;
	
  doc = soap_fault_build (faultcode, faultstring, faultactor, detail);
  if (doc == NULL)
    return herror_new("soap_env_new_with_fault", 
		XML_ERROR_PARSE, "Can not parse fault xml");

  err = soap_env_new_from_doc (doc, out);
  if (err != H_OK)  {
    xmlFreeDoc (doc);
  }

  return err;
}


herror_t
soap_env_new_with_response (SoapEnv * request, SoapEnv **out)
{
  char urn[100];
  char methodname[150];
  char methodname2[150];

  if (request == NULL)
  {
     return herror_new("soap_env_new_with_response", 
		GENERAL_INVALID_PARAM, "request (first param) is NULL");
  }

  if (request->root == NULL)
  {
     return herror_new("soap_env_new_with_response", 
		GENERAL_INVALID_PARAM, "request (first param) has no xml structure");
   }

  if (!soap_env_find_methodname (request, methodname))
  {
     return herror_new("soap_env_new_with_response", 
		GENERAL_INVALID_PARAM, "Method name '%s' not found in request", SAVE_STR(methodname));
  }

  if (!soap_env_find_urn (request, urn))
  {

    /* here we have no chance to find out the namespace */
    /* try to continue without namespace (urn) */
    urn[0] = '\0';
  }

  sprintf (methodname2, "%sResponse", methodname);
  return soap_env_new_with_method (urn, methodname2, out);
}


herror_t
soap_env_new_with_method (const char *urn, const char *method, SoapEnv **out)
{
  xmlDocPtr env;
  xmlChar buffer[1054];


  log_verbose2 ("URN = '%s'", urn);
  log_verbose2 ("Method = '%s'", method);

  if (!strcmp(urn, ""))
  {
#ifdef USE_XMLSTRING
    xmlStrPrintf (buffer, 1054, BAD_CAST _SOAP_MSG_TEMPLATE_EMPTY_TARGET_,
  		soap_env_ns, soap_env_enc, soap_xsi_ns,
  		soap_xsd_ns, BAD_CAST method, BAD_CAST urn, BAD_CAST method);
#else
    sprintf(buffer, _SOAP_MSG_TEMPLATE_EMPTY_TARGET_,
  		soap_env_ns, soap_env_enc, soap_xsi_ns,
  		soap_xsd_ns, method, urn, method);
#endif
  }
  else
  {
#ifdef USE_XMLSTRING
    xmlStrPrintf (buffer, 1054, BAD_CAST _SOAP_MSG_TEMPLATE_,
  		soap_env_ns, soap_env_enc, soap_xsi_ns,
  		soap_xsd_ns, BAD_CAST method, BAD_CAST urn, BAD_CAST method);
#else  
    sprintf(buffer, _SOAP_MSG_TEMPLATE_,
  		soap_env_ns, soap_env_enc, soap_xsi_ns,
  		soap_xsd_ns, method, urn, method);
#endif

  }
  
  env = xmlParseDoc (buffer);
  if (!env)
    return herror_new("soap_env_new_with_method", 
		XML_ERROR_PARSE, "Can not parse xml");

  return soap_env_new_from_doc (env, out);

}



static 
int _soap_env_xml_io_read(void* ctx, char *buffer, int len)
{
	int readed;
	http_input_stream_t *in = (http_input_stream_t*)ctx;
	if(!http_input_stream_is_ready(in))
		return 0;
  
	readed = http_input_stream_read(in, buffer, len);
	if (readed == -1)
		return 0;
	return readed;
}

static
int _soap_env_xml_io_close(void *ctx)
{
  /* do nothing */
  return 0;
}


herror_t
soap_env_new_from_stream(http_input_stream_t *in, SoapEnv **out)
{
  xmlDocPtr doc;
  herror_t err;
 
  doc = xmlReadIO(_soap_env_xml_io_read,
    _soap_env_xml_io_close,  in, "", NULL, 0);

  if (in->err != H_OK) return in->err;
  if (doc == NULL) return herror_new("soap_env_new_from_stream", 
	  XML_ERROR_PARSE, "Trying to parse not valid xml");
  err = soap_env_new_from_doc (doc, out);
  return err;
}


xmlNodePtr
soap_env_add_item (SoapEnv * call, const char *type,
		   const char *name, const char *value)
{

  xmlNodePtr newnode;

  newnode = xmlNewTextChild (call->cur, NULL, BAD_CAST name, BAD_CAST value);

  if (newnode == NULL)
  {
    log_error1 ("Can not create new xml node");
    return NULL;
  }

  if (type)
  {
    if (!xmlNewProp (newnode, BAD_CAST "xsi:type", BAD_CAST type))
    {
      log_error1 ("Can not create new xml attribute");
      return NULL;
    }
  }

  return newnode;
}



xmlNodePtr
soap_env_add_itemf (SoapEnv * call, const char *type,
		    const char *name, const char *format, ...)
{

  va_list ap;
  char buffer[1054];


  va_start (ap, format);
  vsprintf (buffer, format, ap);
  va_end (ap);

  return soap_env_add_item (call, type, name, buffer);
}



xmlNodePtr 
soap_env_add_attachment(SoapEnv* call, const char *name, const char *href)
{
  xmlNodePtr newnode;

  newnode = xmlNewTextChild (call->cur, NULL, BAD_CAST name, BAD_CAST "");

  if (newnode == NULL)
  {
    log_error1 ("Can not create new xml node");
    return NULL;
  }

  if (href)
  {
    if (!xmlNewProp (newnode, BAD_CAST "href", BAD_CAST href))
    {
      log_error1 ("Can not create new xml attribute");
      return NULL;
    }
  }

  return newnode;
}

void
soap_env_add_custom (SoapEnv * call, void *obj, XmlSerializerCallback cb,
		     const char *type, const char *name)
{
  struct XmlNodeHolder holder;

  holder.node = soap_env_get_method (call);

  cb (obj, BAD_CAST name,
      xmlbuilder_start_element,
      xmlbuilder_characters, xmlbuilder_end_element, &holder);

}


xmlNodePtr
soap_env_push_item (SoapEnv * call, const char *type, const char *name)
{

  xmlNodePtr node;

  node = soap_env_add_item (call, type, name, "");

  if (node)
  {
    call->cur = node;
  }

  return node;
}


void
soap_env_pop_item (SoapEnv * call)
{
  call->cur = call->cur->parent;
}

void
soap_env_free (SoapEnv * env)
{
  if (env)
  {
    if (env->root)
    {
      xmlFreeDoc (env->root->doc);
    }
    free (env);
  }
}


xmlNodePtr
soap_env_get_body (SoapEnv * env)
{
  xmlNodePtr node;

  if (env == NULL)
  {
    log_error1 ("env object is NULL");
    return NULL;
  }

  if (env->root == NULL)
  {
    log_error1 ("env has no xml");
    return NULL;
  }

  node = soap_xml_get_children (env->root);

  while (node != NULL)
  {
    if (!xmlStrcmp (node->name, BAD_CAST "Body"))
      return node;
    node = soap_xml_get_next (node);
  }

  log_error1 ("Node Body tag found!");
  return NULL;
}


xmlNodePtr
soap_env_get_fault(SoapEnv * env)
{
  xmlNodePtr node;

  
  node = soap_env_get_body(env);

  if (!node) return NULL;

  while (node != NULL)
  {
    if (!xmlStrcmp (node->name, BAD_CAST "Fault"))
      return node;
    node = soap_xml_get_next (node);
  }

/*  log_warn1 ("Node Fault tag found!");*/
  return NULL;
}


xmlNodePtr
soap_env_get_method (SoapEnv * env)
{

  xmlNodePtr body;

  body = soap_env_get_body (env);
  if (body == NULL)
  {
    log_verbose1 ("body is NULL");
    return NULL;
  }

  /* mehtod is the first child */
  return soap_xml_get_children (body);

}


xmlNodePtr
_soap_env_get_body (SoapEnv * env)
{
  xmlNodePtr body;
  xmlNodeSetPtr nodeset;
  xmlXPathObjectPtr xpathobj;

  if (env == NULL)
  {
    log_error1 ("env object is NULL");
    return NULL;
  }

  if (env->root == NULL)
  {
    log_error1 ("env has no xml");
    return NULL;
  }

  /* 
     find <Body> tag find out namespace
     xpath: //Envelope/Body/
   */
  xpathobj = soap_xpath_eval (env->root->doc, "//Envelope/Body");

  if (!xpathobj)
  {
    log_error1 ("No Body (xpathobj)!");
    return NULL;
  }

  nodeset = xpathobj->nodesetval;
  if (!nodeset)
  {
    log_error1 ("No Body (nodeset)!");
    xmlXPathFreeObject (xpathobj);
    return NULL;
  }

  if (nodeset->nodeNr < 1)
  {
    log_error1 ("No Body (nodeNr)!");
    xmlXPathFreeObject (xpathobj);
    return NULL;
  }

  body = nodeset->nodeTab[0];	/* body is <Body> */
  xmlXPathFreeObject (xpathobj);
  return body;

}


int
soap_env_find_urn (SoapEnv * env, char *urn)
{
  xmlNsPtr ns;
  xmlNodePtr body, node;

  body = soap_env_get_body (env);
  if (body == NULL)
  {
    log_verbose1 ("body is NULL");
    return 0;
  }

  /* node is the first child */
  node = soap_xml_get_children (body);

  if (node == NULL)
  {
    log_error1 ("No namespace found");
    return 0;
  }

  if (node->ns && node->ns->prefix)
  {
    ns = xmlSearchNs (body->doc, node, node->ns->prefix);
    if (ns != NULL)
    {
      strcpy (urn, (char *) ns->href);
      return 1;			/* namespace found! */
    }
  }
  else
  { 
    strcpy(urn,"");
    log_warn1("No namespace found");
    return 1;
  }

  return 0;
}


int
soap_env_find_methodname (SoapEnv * env, char *method)
{
  xmlNodePtr body, node;

  body = soap_env_get_body (env);
  if (body == NULL)
    return 0;

  node = soap_xml_get_children (body);	/* node is the first child */

  if (node == NULL)
  {
    log_error1 ("No method found");
    return 0;
  }

  if (node->name == NULL)
  {
    log_error1 ("No methodname found");
    return 0;

  }

  strcpy (method, (const char *) node->name);

  return 1;
}



/* ------------------------------------------------------------------ */
/*        XML serializers                                            */
/* ------------------------------------------------------------------ */


static void
xmlbuilder_start_element (const xmlChar * element_name, int attr_count,
			  xmlChar ** keys, xmlChar ** values, void *userData)
{
  struct XmlNodeHolder *holder = (struct XmlNodeHolder *) userData;
  xmlNodePtr parent = NULL;

  if (holder == NULL)
    return;
  parent = holder->node;
  if (parent == NULL)
    return;

  holder->node = xmlNewChild (parent, NULL, element_name, NULL);

}

static void
xmlbuilder_characters (const xmlChar * element_name, const xmlChar * chars,
		       void *userData)
{
  struct XmlNodeHolder *holder = (struct XmlNodeHolder *) userData;
  xmlNodePtr parent = NULL;

  if (holder == NULL)
    return;
  parent = holder->node;
  if (parent == NULL)
    return;

  xmlNewTextChild (parent, NULL, element_name, chars);
}

static void
xmlbuilder_end_element (const xmlChar * element_name, void *userData)
{

  struct XmlNodeHolder *holder = (struct XmlNodeHolder *) userData;
  xmlNodePtr parent = NULL;

  if (holder == NULL)
    return;
  parent = holder->node;
  if (parent == NULL)
    return;

  holder->node = parent->parent;
}