#include "graphbuilderadapter.h"

namespace YAML
{
  int GraphBuilderAdapter::ContainerFrame::sequenceMarker;
  
  void GraphBuilderAdapter::OnNull(const Mark& mark, anchor_t anchor)
  {
    void *pParent = GetCurrentParent();
    void *pNode = m_builder.NewNull(mark, pParent);
    RegisterAnchor(anchor, pNode);
    
    DispositionNode(pNode);
  }
  
  void GraphBuilderAdapter::OnAlias(const Mark& mark, anchor_t anchor)
  {
    void *pReffedNode = m_anchors.Get(anchor);
    DispositionNode(m_builder.AnchorReference(mark, pReffedNode));
  }
  
  void GraphBuilderAdapter::OnScalar(const Mark& mark, const std::string& tag, anchor_t anchor, const std::string& value)
  {
    void *pParent = GetCurrentParent();
    void *pNode = m_builder.NewScalar(mark, tag, pParent, value);
    RegisterAnchor(anchor, pNode);
    
    DispositionNode(pNode);
  }
  
  void GraphBuilderAdapter::OnSequenceStart(const Mark& mark, const std::string& tag, anchor_t anchor)
  {
    void *pNode = m_builder.NewSequence(mark, tag, GetCurrentParent());
    m_containers.push(ContainerFrame(pNode));
    RegisterAnchor(anchor, pNode);
  }
  
  void GraphBuilderAdapter::OnSequenceEnd()
  {
    void *pSequence = m_containers.top().pContainer;
    m_containers.pop();
    
    DispositionNode(pSequence);
  }
  
  void GraphBuilderAdapter::OnMapStart(const Mark& mark, const std::string& tag, anchor_t anchor)
  {
    void *pNode = m_builder.NewMap(mark, tag, GetCurrentParent());
    m_containers.push(ContainerFrame(pNode, m_pKeyNode));
    m_pKeyNode = NULL;
    RegisterAnchor(anchor, pNode);
  }
  
  void GraphBuilderAdapter::OnMapEnd()
  {
    void *pMap = m_containers.top().pContainer;
    m_pKeyNode = m_containers.top().pPrevKeyNode;
    m_containers.pop();
    DispositionNode(pMap);
  }
  
  void *GraphBuilderAdapter::GetCurrentParent() const
  {
    if (m_containers.empty()) {
      return NULL;
    }
    return m_containers.top().pContainer;
  }
  
  void GraphBuilderAdapter::RegisterAnchor(anchor_t anchor, void *pNode)
  {
    if (anchor) {
      m_anchors.Register(anchor, pNode);
    }
  }
  
  void GraphBuilderAdapter::DispositionNode(void *pNode)
  {
    if (m_containers.empty()) {
      m_pRootNode = pNode;
      return;
    }
    
    void *pContainer = m_containers.top().pContainer;
    if (m_containers.top().isMap()) {
      if (m_pKeyNode) {
        m_builder.AssignInMap(pContainer, m_pKeyNode, pNode);
        m_pKeyNode = NULL;
      } else {
        m_pKeyNode = pNode;
      }
    } else {
      m_builder.AppendToSequence(pContainer, pNode);
    }
  }
}