//
// Node set functions for Mini-XML, a small XML file parsing library.
//
// https://www.msweet.org/mxml
//
// Copyright © 2003-2024 by Michael R Sweet.
//
// Licensed under Apache License v2.0.  See the file "LICENSE" for more
// information.
//

#include "mxml-private.h"


//
// 'mxmlSetCDATA()' - Set the element name of a CDATA node.
//
// The node is not changed if it (or its first child) is not a CDATA element node.
//

int					// O - 0 on success, -1 on failure
mxmlSetCDATA(mxml_node_t *node,		// I - Node to set
             const char  *data)		// I - New data string
{
  size_t	datalen;		// Length of data string
  char		*s;			// New element name


  // Range check input...
  if (node && node->type == MXML_TYPE_ELEMENT && strncmp(node->value.element.name, "![CDATA[", 8) && node->child && node->child->type == MXML_TYPE_ELEMENT && !strncmp(node->child->value.element.name, "![CDATA[", 8))
    node = node->child;

  if (!node || node->type != MXML_TYPE_ELEMENT || strncmp(node->value.element.name, "![CDATA[", 8))
  {
    mxml_error("Wrong node type.");
    return (-1);
  }
  else if (!data)
  {
    mxml_error("NULL string not allowed.");
    return (-1);
  }

  if (data == (node->value.element.name + 8))
  {
    // Don't change the value...
    return (0);
  }

  // Allocate the new value, free any old element value, and set the new value...
  datalen = strlen(data);

  if ((s = malloc(datalen + 9)) == NULL)
  {
    mxml_error("Unable to allocate memory for CDATA.");
    return (-1);
  }

  free(node->value.element.name);
  node->value.element.name = s;
  snprintf(node->value.element.name, datalen + 9, "![CDATA[%s", data);

  return (0);
}


//
// 'mxmlSetCustom()' - Set the data and destructor of a custom data node.
//
// The node is not changed if it (or its first child) is not a custom node.
//

int					// O - 0 on success, -1 on failure
mxmlSetCustom(
    mxml_node_t              *node,	// I - Node to set
    void                     *data,	// I - New data pointer
    mxml_custom_destroy_cb_t destroy)	// I - New destructor function
{
  // Range check input...
  if (node && node->type == MXML_TYPE_ELEMENT && node->child && node->child->type == MXML_TYPE_CUSTOM)
    node = node->child;

  if (!node || node->type != MXML_TYPE_CUSTOM)
  {
    mxml_error("Wrong node type.");
    return (-1);
  }

  if (data == node->value.custom.data)
  {
    node->value.custom.destroy = destroy;
    return (0);
  }

  // Free any old element value and set the new value...
  if (node->value.custom.data && node->value.custom.destroy)
    (*(node->value.custom.destroy))(node->value.custom.data);

  node->value.custom.data    = data;
  node->value.custom.destroy = destroy;

  return (0);
}


//
// 'mxmlSetElement()' - Set the name of an element node.
//
// The node is not changed if it is not an element node.
//

int					// O - 0 on success, -1 on failure
mxmlSetElement(mxml_node_t *node,	// I - Node to set
               const char  *name)	// I - New name string
{
  char *s;				// New name string


  // Range check input...
  if (!node || node->type != MXML_TYPE_ELEMENT)
  {
    mxml_error("Wrong node type.");
    return (-1);
  }
  else if (!name)
  {
    mxml_error("NULL string not allowed.");
    return (-1);
  }

  if (name == node->value.element.name)
    return (0);

  // Free any old element value and set the new value...
  if ((s = strdup(name)) == NULL)
  {
    mxml_error("Unable to allocate memory for element name.");
    return (-1);
  }

  free(node->value.element.name);
  node->value.element.name = s;

  return (0);
}


//
// 'mxmlSetInteger()' - Set the value of an integer node.
//
// The node is not changed if it (or its first child) is not an integer node.
//

int					// O - 0 on success, -1 on failure
mxmlSetInteger(mxml_node_t *node,	// I - Node to set
               int         integer)	// I - Integer value
{
  // Range check input...
  if (node && node->type == MXML_TYPE_ELEMENT && node->child && node->child->type == MXML_TYPE_INTEGER)
    node = node->child;

  if (!node || node->type != MXML_TYPE_INTEGER)
  {
    mxml_error("Wrong node type.");
    return (-1);
  }

  // Set the new value and return...
  node->value.integer = integer;

  return (0);
}


//
// 'mxmlSetOpaque()' - Set the value of an opaque node.
//
// The node is not changed if it (or its first child) is not an opaque node.
//

int					// O - 0 on success, -1 on failure
mxmlSetOpaque(mxml_node_t *node,	// I - Node to set
              const char  *opaque)	// I - Opaque string
{
  char *s;				// New opaque string


  // Range check input...
  if (node && node->type == MXML_TYPE_ELEMENT && node->child && node->child->type == MXML_TYPE_OPAQUE)
    node = node->child;

  if (!node || node->type != MXML_TYPE_OPAQUE)
  {
    mxml_error("Wrong node type.");
    return (-1);
  }
  else if (!opaque)
  {
    mxml_error("NULL string not allowed.");
    return (-1);
  }

  if (node->value.opaque == opaque)
    return (0);

  // Free any old opaque value and set the new value...
  if ((s = strdup(opaque)) == NULL)
  {
    mxml_error("Unable to allocate memory for opaque string.");
    return (-1);
  }

  free(node->value.opaque);
  node->value.opaque = s;

  return (0);
}


//
// 'mxmlSetOpaquef()' - Set the value of an opaque string node to a formatted string.
//
// The node is not changed if it (or its first child) is not an opaque node.
//

int					// O - 0 on success, -1 on failure
mxmlSetOpaquef(mxml_node_t *node,	// I - Node to set
               const char  *format,	// I - Printf-style format string
	       ...)			// I - Additional arguments as needed
{
  va_list	ap;			// Pointer to arguments
  char		buffer[16384];		// Format buffer
  char		*s;			// Temporary string


  // Range check input...
  if (node && node->type == MXML_TYPE_ELEMENT && node->child && node->child->type == MXML_TYPE_OPAQUE)
    node = node->child;

  if (!node || node->type != MXML_TYPE_OPAQUE)
  {
    mxml_error("Wrong node type.");
    return (-1);
  }
  else if (!format)
  {
    mxml_error("NULL string not allowed.");
    return (-1);
  }

  // Format the new string, free any old string value, and set the new value...
  va_start(ap, format);
  vsnprintf(buffer, sizeof(buffer), format, ap);
  va_end(ap);

  if ((s = strdup(buffer)) == NULL)
  {
    mxml_error("Unable to allocate memory for opaque string.");
    return (-1);
  }

  free(node->value.opaque);
  node->value.opaque = s;

  return (0);
}


//
// 'mxmlSetReal()' - Set the value of a real number node.
//
// The node is not changed if it (or its first child) is not a real number node.
//

int					// O - 0 on success, -1 on failure
mxmlSetReal(mxml_node_t *node,		// I - Node to set
            double      real)		// I - Real number value
{
 /*
  * Range check input...
  */

  if (node && node->type == MXML_TYPE_ELEMENT && node->child && node->child->type == MXML_TYPE_REAL)
    node = node->child;

  if (!node || node->type != MXML_TYPE_REAL)
  {
    mxml_error("Wrong node type.");
    return (-1);
  }

  // Set the new value and return...
  node->value.real = real;

  return (0);
}


//
// 'mxmlSetText()' - Set the value of a text node.
//
// The node is not changed if it (or its first child) is not a text node.
//

int					// O - 0 on success, -1 on failure
mxmlSetText(mxml_node_t *node,		// I - Node to set
            int         whitespace,	// I - 1 = leading whitespace, 0 = no whitespace
	    const char  *string)	// I - String
{
  char *s;				// New string


  // Range check input...
  if (node && node->type == MXML_TYPE_ELEMENT && node->child && node->child->type == MXML_TYPE_TEXT)
    node = node->child;

  if (!node || node->type != MXML_TYPE_TEXT)
  {
    mxml_error("Wrong node type.");
    return (-1);
  }
  else if (!string)
  {
    mxml_error("NULL string not allowed.");
    return (-1);
  }

  if (string == node->value.text.string)
  {
    node->value.text.whitespace = whitespace;
    return (0);
  }

  // Free any old string value and set the new value...
  if ((s = strdup(string)) == NULL)
  {
    mxml_error("Unable to allocate memory for text string.");
    return (-1);
  }

  free(node->value.text.string);

  node->value.text.whitespace = whitespace;
  node->value.text.string     = s;

  return (0);
}


//
// 'mxmlSetTextf()' - Set the value of a text node to a formatted string.
//
// The node is not changed if it (or its first child) is not a text node.
//

int					// O - 0 on success, -1 on failure
mxmlSetTextf(mxml_node_t *node,		// I - Node to set
             int         whitespace,	// I - 1 = leading whitespace, 0 = no whitespace
             const char  *format,	// I - Printf-style format string
	     ...)			// I - Additional arguments as needed
{
  va_list	ap;			// Pointer to arguments
  char		buffer[16384];		// Format buffer
  char		*s;			// Temporary string


  // Range check input...
  if (node && node->type == MXML_TYPE_ELEMENT && node->child && node->child->type == MXML_TYPE_TEXT)
    node = node->child;

  if (!node || node->type != MXML_TYPE_TEXT)
  {
    mxml_error("Wrong node type.");
    return (-1);
  }
  else if (!format)
  {
    mxml_error("NULL string not allowed.");
    return (-1);
  }

 /*
  * Free any old string value and set the new value...
  */

  va_start(ap, format);
  vsnprintf(buffer, sizeof(buffer), format, ap);
  va_end(ap);

  if ((s = strdup(buffer)) == NULL)
  {
    mxml_error("Unable to allocate memory for text string.");
    return (-1);
  }

  free(node->value.text.string);

  node->value.text.whitespace = whitespace;
  node->value.text.string     = s;

  return (0);
}


//
// 'mxmlSetUserData()' - Set the user data pointer for a node.
//

int					// O - 0 on success, -1 on failure
mxmlSetUserData(mxml_node_t *node,	// I - Node to set
                void        *data)	// I - User data pointer
{
  // Range check input...
  if (!node)
    return (-1);

  // Set the user data pointer and return...
  node->user_data = data;
  return (0);
}