/* myxml.c
 * APR-UTIL の XML API を使うサンプル
 * ・getElementsByTagName と getElementById 
 */

#include "myxml.h"

/*
 * apr_xml_elem* の配列のイテレーション
 * func は配列内の apr_xml_elem* を処理するコールバック関数
 * void func (void *func_ctx, apr_xml_elem * elem);
 */

void my_apr_xml_elem_array_do(
  apr_array_header_t *arr
  , void *func
  , void *func_ctx
) {
  int i=0;
  void (*f)(void *,apr_xml_elem*) = func;
  if (!arr || !f) {
    return;
  }
  
  for (i=0; i<arr->nelts; i++) {
    apr_xml_elem *elem = ((apr_xml_elem **)arr->elts)[i];
    f(func_ctx, elem);
  }
}

/*
 * エレメントのＣＤＡＴＡを返す関数
 * 返り値：
 *   エレメントのＣＤＡＴＡの文字列
 * メモ：
 * n->first_cdata.first から next メンバ変数をたどって text メンバ変数の
 * 文字列をコンカテネートする。
 */

char * concat_cdata(
  apr_xml_elem *n    /* IN: 対象エレメント */
  , apr_pool_t *pool /* IN: メモリプール */
) {
  char *buf = NULL;
  apr_text *x = n->first_cdata.first;
  for (; x; x=x->next) {
    if (x->text) {
      if (!buf) {
        buf = apr_pstrdup(pool, x->text);
      } else {
        buf = apr_pstrcat(pool, buf, x->text, NULL);
      }
    }
  }
  return buf;
}

/*
 * エレメントから、指定した名前のアトリビュートの値を検索する関数
 */

char const * getAttribute(
  apr_xml_elem *n /* IN: 対象エレメント */
  , char *aname   /* IN: 検索アトリビュート名 */
) {
  apr_xml_attr *attr = NULL;
  if (NULL == n) {
    return NULL;
  }
  
  for (attr=n->attr; attr; attr=attr->next) {
    if (!strcmp(attr->name, aname))
      return attr->value;
  }

  return NULL;
}

/* getElementsByTagName の補助関数
 * 戻り値:
 *   検索タグ名にマッチしたエレメントを保持する配列
 */
 
static apr_array_header_t *_getElementsByTagName(
  apr_array_header_t *arr /* IN/OUT: マッチしたエレメントを保持する配列 */
  , apr_xml_elem *n       /* IN: 対象エレメント */
  , char *tagname         /* IN: 検索タグ名 */
) {

  if (!strcmp(n->name, tagname)) {
    *(const apr_xml_elem **)apr_array_push(arr) = n;
  }
  if (n->first_child) {
    _getElementsByTagName(arr, n->first_child, tagname);
  }
  if (n->next) {
    _getElementsByTagName(arr, n->next, tagname);
  }
  return arr;
}

/* 指定エレメント配下の、検索タグ名にマッチするエレメント群を検索する関数
 * 戻り値:
 *   検索タグ名にマッチしたエレメントを保持する配列
 *   子エレメントが存在しない場合はヌル。
 */

apr_array_header_t *getElementsByTagNameFromElement(
  apr_pool_t *pool   /* IN: メモリプール */
  , apr_xml_elem *n  /* IN: 対象エレメント */
  , char *tagname    /* IN: 検索タグ名 */
) {
  apr_array_header_t *arr=NULL;
  
  if (NULL == tagname || NULL == n || NULL == n->first_child) {
    return NULL;
  }
  
  arr = apr_array_make(pool, 1, sizeof(apr_xml_elem *));
  return _getElementsByTagName(arr, n->first_child, tagname);
}

/* XMLドキュメント内の検索タグ名にマッチしたエレメント群を検索する関数
 * 戻り値:
 * 検索タグ名にマッチしたエレメントを保持する配列
 */

apr_array_header_t *getElementsByTagName(
  apr_pool_t *pool   /* IN: メモリプール */
  , apr_xml_doc *doc /* IN: 対象XMLドキュメント */
  , char *tagname    /* IN: 検索タグ名 */
) {
  apr_array_header_t *arr=NULL;
  
  if (NULL == doc || NULL == tagname || NULL == doc->root) {
    return NULL;
  }
  
  arr = apr_array_make(pool, 1, sizeof(apr_xml_elem *));
  return _getElementsByTagName(arr, doc->root, tagname);
}

/* getElementById の補助関数
 * 戻り値:
 * 検索タグ名にマッチしたエレメント
 */

apr_xml_elem * _getElementById (
  apr_xml_elem *n /* IN: 対象エレメント */
  , char *id      /* IN: 検索ID */
) {
  apr_xml_elem *r=NULL;
  apr_xml_attr *a = n->attr;
  for (; a; a = a->next) {
    if (!strcmp(a->value, id)) {
      return n;
    }
  }
  if (n->first_child && (r = _getElementById(n->first_child, id))) {
    return r;
  }
  if (n->next) {
    return _getElementById(n->next, id);
  }
  
  return NULL;
}

/* XMLドキュメント内の検索IDにマッチしたエレメントを検索する関数
 * 戻り値:
 *   検索IDにマッチしたエレメント
 */

apr_xml_elem * getElementById (
  apr_xml_doc *doc  /* IN: 対象XMLドキュメント */
  , char *id        /* IN: 検索ID */
) {
  if (NULL == doc || NULL == id || NULL == doc->root) {
    return NULL;
  }
  
  return _getElementById (doc->root, id);
}

/* printElement[Rec] の補助関数。アトリビュートの表示
 * 戻り値:
 *   なし
 */
static void _printAttr (
  apr_file_t *out      /* IN: 出力先ファイルポインタ */
  , apr_xml_attr *attr /* IN: 表示対象アトリビュート */
) {
  for (; attr; attr=attr->next) {
    apr_file_printf(out, "[%s]=[%s]", attr->name, attr->value);
  }
}

/*
 * 空白文字の検査
 */

static int _is_spaces(const char *s) {
  char *x=(char *)s;
  if (NULL) {
    return 1;
  }
  for(;*x;x++) {
    if (*x!='\t' && *x!='\r' && *x!='\n' && *x!=' ') {
      return 0;
    }
  }
  return 1;
}

/* printElement[Rec] の補助関数。CDATAの表示
 * 戻り値:
 *   なし
 */

static void _printCDATA(
  apr_file_t *out   /* IN: 出力先ファイルポインタ */
  , apr_text *cdata /* IN: 表示対象CDATA */
) {
  for (; cdata; cdata=cdata->next) {
    if (! _is_spaces(cdata->text))
      apr_file_printf(out, "%s", cdata->text);
  }
}

/* エレメントの表示。指定されたエレメントのタグ名、アトリビュート、
 * 直下のCDATAの表示のみ。NS 非対応。
 * 戻り値:
 *   なし
 * メモ：
 *   apr_xml_to_text 関数が既に用意されている。どちらかというと、apr_xml_elem
 *   の構造の勉強用。
 */

void printElement (
  apr_file_t *out      /* IN: 出力先ファイルポインタ */
  , apr_xml_elem *elem /* IN: 表示対象エレメント */
) {
  if (NULL == elem || NULL == out) {
    return;
  }
  apr_file_printf(out, "<%s", elem->name);
  if (elem->attr) {
    _printAttr(out, elem->attr);
  }
  if (elem->first_cdata.first) {
    apr_file_printf(out, ">");
    _printCDATA(out, elem->first_cdata.first);
    apr_file_printf(out, "</%s>", elem->name);
  } else {
    apr_file_printf(out, "/>");
  }
}

/* エレメントの表示。指定されたエレメントのタグ名、アトリビュート、
 * 直下のCDATAの表示に加え、子エレメント(child)と続くエレメントも再帰的に表示。
 * 戻り値:
 *   なし
 * メモ：
 *   apr_xml_to_text 関数が既に用意されている。どちらかというと、apr_xml_elem
 *   の構造の勉強用。
 */
void printElementRec (
  apr_file_t *out      /* IN: 出力先ファイルポインタ */
  , apr_xml_elem *elem /* IN: 表示対象エレメント */
) {
  if (NULL == elem || NULL == out) {
    return;
  }
  apr_file_printf(out, "<%s", elem->name);

  if (elem->attr) {
    _printAttr(out, elem->attr);
  }
  if (APR_XML_ELEM_IS_EMPTY(elem)) {
    apr_file_printf(out, "/>");
  } else {
    apr_file_printf(out, ">");
    if (elem->first_cdata.first) {
      _printCDATA(out, elem->first_cdata.first);
    }
    if (elem->first_child) {
      printElementRec(out, elem->first_child);
    }
    apr_file_printf(out, "</%s>", elem->name);
  }
  if (elem->following_cdata.first) {
    _printCDATA(out, elem->following_cdata.first);
  }
  if (elem->next) {
    printElementRec(out, elem->next);
  }
}