#include <string.h>
#include <sqlcli1.h>
#include <postgres_ext.h>
#include "db2_fdw.h"

/** global variables */

/** external variables */

/** external prototypes */
extern void*     db2Alloc             (size_t size);
extern void      db2Debug1            (const char* message, ...);

/** local prototypes */
char*            db2CopyText          (const char* string, int size, int quote);

/** db2CopyText
 *   Returns a palloc'ed string containing a (possibly quoted) copy of "string".
 *   If the string starts with "(" and ends with ")", no quoting will take place
 *   even if "quote" is true.
 */
char* db2CopyText (const char* string, int size, int quote) {
  int      resultsize = (quote ? size + 2 : size);
  register int i; 
  register int j = -1;
  char*    result;

  db2Debug1("> db2CopyText(string: '%s', size: %d, quote: %d)",string,size,quote);
  /* if "string" is parenthized, return a copy */
  if (string[0] == '(' && string[size - 1] == ')') {
    result = db2Alloc (size + 1);
    memcpy (result, string, size);
    result[size] = '\0';
    return result;
  }

  if (quote) {
    for (i = 0; i < size; ++i) {
      if (string[i] == '"')
        ++resultsize;
    }
  }

  result = db2Alloc (resultsize + 1);
  if (quote)
    result[++j] = '"';
  for (i = 0; i < size; ++i) {
    result[++j] = string[i];
    if (quote && string[i] == '"')
      result[++j] = '"';
  }
  if (quote)
    result[++j] = '"';
  result[j + 1] = '\0';

  db2Debug1("< db2CopyText - result: %s",result);
  return result;
}