/*
 * Copyright (c) 1998-2011 Caucho Technology -- all rights reserved
 *
 * This file is part of Resin(R) Open Source
 *
 * Each copy or derived work must preserve the copyright notice and this
 * notice unmodified.
 *
 * Resin Open Source is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * Resin Open Source is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE, or any warranty
 * of NON-INFRINGEMENT.  See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Resin Open Source; if not, write to the
 *
 *   Free Software Foundation, Inc.
 *   59 Temple Place, Suite 330
 *   Boston, MA 02111-1307  USA
 *
 * @author Emil Ong
 */

package com.caucho.jaxb.mapping;

import com.caucho.jaxb.BinderImpl;
import com.caucho.jaxb.JAXBContextImpl;
import com.caucho.jaxb.NodeIterator;
import com.caucho.jaxb.accessor.Accessor;
import com.caucho.jaxb.skeleton.ClassSkeleton;
import com.caucho.util.L10N;
import com.caucho.xml.stream.StaxUtil;

import org.w3c.dom.Node;

import static javax.xml.XMLConstants.*;

import javax.xml.bind.JAXBElement;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;
import javax.xml.bind.Unmarshaller;
import javax.xml.bind.UnmarshalException;

import javax.xml.bind.annotation.XmlAnyAttribute;

import javax.xml.namespace.QName;

import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
import javax.xml.stream.XMLStreamWriter;

import java.io.IOException;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

public class AnyAttributeMapping extends XmlMapping {
  private static final L10N L = new L10N(AnyAttributeMapping.class);

  public AnyAttributeMapping(JAXBContextImpl context, Accessor accessor)
    throws JAXBException
  {
    super(context, accessor);

    if (! Map.class.isAssignableFrom(accessor.getType()))
      throw new JAXBException(L.l("Fields or properties annotated with @XmlAnyAttribute must be Maps"));
  }

  // output methods

  public void write(Marshaller m, XMLStreamWriter out, Object obj)
    throws IOException, XMLStreamException, JAXBException
  {
    Map map = (Map) _accessor.get(obj);

    if (map != null) {
      for (Map.Entry entry : (Set<Map.Entry>) map.entrySet()) {
        QName name = (QName) entry.getKey();
        String value = (String) entry.getValue();

        StaxUtil.writeAttribute(out, name, value);
      }
    }
  }

  public void write(Marshaller m, XMLStreamWriter out,
                    Object obj, ArrayList<XmlMapping> attributes)
    throws IOException, JAXBException, XMLStreamException
  {
    throw new UnsupportedOperationException();
  }

  // input methods

  public void readAttribute(XMLStreamReader in, int i, Object parent)
    throws IOException, XMLStreamException, JAXBException
  {
    Map map = (Map) _accessor.get(parent);

    if (map == null) {
      map = new HashMap(); // XXX other map types
      _accessor.set(parent, map);
    }

    map.put(in.getAttributeName(i), in.getAttributeValue(i));

  }

  public void read(Unmarshaller u, XMLStreamReader in, 
                   Object parent, ClassSkeleton attributed)
    throws IOException, XMLStreamException, JAXBException
  {
    throw new UnsupportedOperationException();
  }

  public void read(Unmarshaller u, XMLStreamReader in, Object parent)
    throws IOException, XMLStreamException, JAXBException
  {
    throw new UnsupportedOperationException();
  }

  public void bindFrom(BinderImpl binder, NodeIterator node, Object obj)
    throws IOException, JAXBException
  {
    throw new UnsupportedOperationException();
  }

  public QName getQName(Object obj)
    throws JAXBException
  {
    throw new UnsupportedOperationException();
  }

  public void generateSchema(XMLStreamWriter out)
    throws JAXBException, XMLStreamException
  {
    out.writeStartElement(XML_SCHEMA_PREFIX, "anyAttribute", 
                          W3C_XML_SCHEMA_NS_URI);
    out.writeAttribute("namespace", "##other");
    out.writeAttribute("processContents", "skip");
  }
}