/*
* Copyright (c) 1999 - 2001 Nokia Corporation and/or its subsidiary(-ies).
* All rights reserved.
* This component and the accompanying materials are made available
* under the terms of the License "Eclipse Public License v1.0"
* which accompanies this distribution, and is available
* at the URL "http://www.eclipse.org/legal/epl-v10.html".
*
* Initial Contributors:
* Nokia Corporation - initial contribution.
*
* Contributors:
*
* Description: 
*
*/


#include "nw_hed_domtree.h"
#include "nw_tinydom_utils.h"
#include "nw_object_exceptions.h"
#include "nwx_statuscodeconvert.h"
//#include "Nw_string_string.h"

struct NW_HED_DomTree_s {
  Parser_t          domParser;
  NW_Buffer_t*      domBuffer;
};

/* 
 * 
 *
 *
 */

static TBrowserStatusCode
NW_HED_Write_WBXML_StrTbl(Parser_t *p, void* WBXMLEncStrTbl)
{
     TBrowserStatusCode status = KBrsrSuccess;
     NW_Uint32 stringTableByteCount = 0;
     NW_Encoder_StringTableIterator_t strTableIterator;
     NW_Status_t s = NW_STAT_FAILURE;
     NW_Uint8* pLiteralBuf;
     NW_String_t pLiteralStr;
     NW_Uint32 docTableIndex;
     NW_Encoder_StringTable_t* strTable = NULL;
     NW_WBXML_Document_t* doc = &(p->document);
     
     
         

     strTable = (NW_Encoder_StringTable_t*) WBXMLEncStrTbl;

     stringTableByteCount = NW_Encoder_StringTable_getTotalBytes(strTable);
     
     
     if(stringTableByteCount > 0)
     {
        /*The following function always return success so why bother to check? */

        NW_Encoder_StringTable_StringTableIterateInit(WBXMLEncStrTbl,&strTableIterator);


        s =  NW_Encoder_StringTable_StringTableIterateNext
                    (&strTableIterator,&stringTableByteCount,&pLiteralBuf);

        while(s == NW_STAT_WBXML_ITERATE_MORE)
        {

         pLiteralStr.length = stringTableByteCount;
         pLiteralStr.storage = pLiteralBuf;

         status = StatusCodeConvert( NW_WBXML_Document_putTableString(
                                      doc,&pLiteralStr,&docTableIndex) );

         if(status != KBrsrSuccess)
         {
          break;
         }

           //Just a sanity check

         /*
           if(WBXMLwriterPrevIndex != docTableIndex)
           {
            status = KBrsrFailure;
            break;
           }
         
             WBXMLwriterPrevIndex += pLiteralStr.length;
        */
        

           s =  NW_Encoder_StringTable_StringTableIterateNext
                    (&strTableIterator,&stringTableByteCount,&pLiteralBuf) ;

        }//end while

     }//end if(stringTableByteCount > 0)

     //Delete the tree

     NW_Encoder_StringTable_delete(WBXMLEncStrTbl);
     WBXMLEncStrTbl = NULL;
 return status;
}//end NW_HED_Write_WBXML_StrTbl()

/* ------------------------------------------------------------------------- */
NW_HED_DomTree_t* NW_HED_DomTree_New (NW_Buffer_t* domBuffer,
                                      NW_Http_CharSet_t charset,
                                      NW_Uint32 defaultPublicId,
                                      void* WBXMLEncStrTbl)
{
  NW_HED_DomTree_t* domTree;

  NW_ASSERT(domBuffer != NULL);
  NW_REQUIRED_PARAM(charset);

  NW_TRY(status) {
    /* create the dom-tree */
    domTree = (NW_HED_DomTree_t*) NW_Mem_Malloc(sizeof(NW_HED_DomTree_t));
    NW_THROW_OOM_ON_NULL(domTree, status);
    NW_Mem_memset(domTree, 0, sizeof(NW_HED_DomTree_t));

    /* init the parser */
    status = StatusCodeConvert(NW_TinyDom_ParserInitialize(&domTree->domParser, NULL, 
                                         0, defaultPublicId));
    NW_THROW_ON_ERROR(status);

    /*If the WBXML encoder table is present then write these contents to
     * extension table.
     */

    if(WBXMLEncStrTbl)
    {
     status = NW_HED_Write_WBXML_StrTbl(&domTree->domParser, WBXMLEncStrTbl);
     NW_THROW_ON_ERROR(status);
    }//end if(WBXMLEncStrTbl)

    /* build the dom-tree */
    status = StatusCodeConvert(NW_TinyDom_MakeDOMTree(&domTree->domParser, 
                                    domBuffer->data, domBuffer->length, NULL));
    NW_THROW_ON_ERROR(status);
    domTree->domBuffer = domBuffer; 
  }
  NW_CATCH(status) {
    if (domTree != NULL) {
      NW_TinyDom_ParserDelete(&domTree->domParser);
    }
    NW_Mem_Free(domTree);
    NW_Buffer_Free(domBuffer);
    domTree = NULL;
  }
  NW_FINALLY {
    return domTree;
  }
  NW_END_TRY
}

/* ------------------------------------------------------------------------- */
TBrowserStatusCode NW_HED_DomTree_Append (NW_HED_DomTree_t** ppDomTree,
                                      NW_Buffer_t* domBuffer,
                                      NW_Http_CharSet_t charset,
                                      NW_Uint32 defaultPublicId,
                                      NW_Uint32 cp_count,
                                      NW_Int32 lastValid,
                                      void* WBXMLEncStrTbl)
{
  NW_HED_DomTree_t* domTree;
  
  NW_ASSERT(domBuffer != NULL);
  NW_ASSERT(ppDomTree != NULL);
  NW_REQUIRED_PARAM(charset);

  domTree = *ppDomTree;

  NW_TRY(status) {
    if (domTree == NULL)
    {
      /* create the dom-tree */
      domTree = (NW_HED_DomTree_t*) NW_Mem_Malloc(sizeof(NW_HED_DomTree_t));
      NW_THROW_OOM_ON_NULL(domTree, status);
      NW_Mem_memset(domTree, 0, sizeof(NW_HED_DomTree_t));
      
      /* init the parser */
      status = StatusCodeConvert(NW_TinyDom_ParserInitialize(&domTree->domParser, NULL, 
        0, defaultPublicId));
      NW_THROW_ON_ERROR(status);
      
    }

    /*If the WBXML encoder table is present then write these contents to
     * extension table.
     */

    if(WBXMLEncStrTbl)
    {
     status = NW_HED_Write_WBXML_StrTbl(&domTree->domParser,WBXMLEncStrTbl);
     NW_THROW_ON_ERROR(status);
    }//end if(WBXMLEncStrTbl)

    /* build the dom-tree */

    status = StatusCodeConvert(NW_TinyDom_AppendDOMTree(&domTree->domParser, 
                                    domBuffer->data, domBuffer->length, cp_count, lastValid, NULL));
    NW_THROW_ON_ERROR(status);
    if (domTree->domBuffer != domBuffer)
        {
        NW_Buffer_Free(domTree->domBuffer);
        }
    domTree->domBuffer = domBuffer;
    *ppDomTree = domTree;
  }
  NW_CATCH(status) {
    if (domTree != NULL) {
      NW_TinyDom_ParserDelete(&domTree->domParser);
    }
    NW_Mem_Free(domTree);
    NW_Buffer_Free(domBuffer);
    domTree = NULL;
    *ppDomTree = domTree;
  }
  NW_FINALLY {
    return status;
  }
  NW_END_TRY
}


/* ------------------------------------------------------------------------- */
void NW_HED_DomTree_Delete (NW_HED_DomTree_t* thisObj)
{
  if (thisObj != NULL) {
    NW_TinyDom_ParserDelete(&thisObj->domParser);

    if (thisObj->domBuffer != NULL) {
      NW_Buffer_Free (thisObj->domBuffer);
    }

    NW_Mem_Free (thisObj);
  }
}

/* ------------------------------------------------------------------------- */
NW_DOM_Node_t* NW_HED_DomTree_GetRootNode (const NW_HED_DomTree_t* thisObj)
{
  NW_ASSERT (thisObj != NULL);

  return thisObj->domParser.tinyParser.dom_tree->root_node;
}

/* ------------------------------------------------------------------------- */
NW_Uint32 NW_HED_DomTree_GetPublicId (const NW_HED_DomTree_t* thisObj)
{
  NW_ASSERT (thisObj != NULL);

  return NW_DOM_DocumentNode_getPublicIdAsNumber (
      NW_HED_DomTree_GetRootNode (thisObj));
}

/* ------------------------------------------------------------------------- */
NW_Uint8 NW_HED_DomTree_GetVersion (const NW_HED_DomTree_t* thisObj)
{
  NW_ASSERT (thisObj != NULL);

  return NW_DOM_DocumentNode_getVersion (
      NW_HED_DomTree_GetRootNode (thisObj));
}

/* ------------------------------------------------------------------------- */
NW_Uint32 NW_HED_DomTree_GetCharEncoding (const NW_HED_DomTree_t* thisObj)
{
  NW_ASSERT (thisObj != NULL);

  return NW_DOM_DocumentNode_getCharacterEncoding (
      NW_HED_DomTree_GetRootNode (thisObj));
}

/* ------------------------------------------------------------------------- */
NW_Buffer_t* NW_HED_DomTree_GetDomBuffer (const NW_HED_DomTree_t* thisObj)
{
  NW_ASSERT (thisObj != NULL);

  return thisObj->domBuffer;
}

/* ------------------------------------------------------------------------- */
NW_Buffer_t* NW_HED_DomTree_OrphanDomBuffer (NW_HED_DomTree_t* thisObj)
{
  NW_Buffer_t* buff;

  NW_ASSERT (thisObj != NULL);

  buff = thisObj->domBuffer;
  thisObj->domBuffer = NULL;

  return buff;
}