Package org.jboss.xb.binding.sunday.marshalling

Source Code of org.jboss.xb.binding.sunday.marshalling.MarshallerImpl$MarshallingContextImpl

/*
  * JBoss, Home of Professional Open Source
  * Copyright 2005, JBoss Inc., and individual contributors as indicated
  * by the @authors tag. See the copyright.txt in the distribution for a
  * full listing of individual contributors.
  *
  * This is free software; you can redistribute it and/or modify it
  * under the terms of the GNU Lesser General Public License as
  * published by the Free Software Foundation; either version 2.1 of
  * the License, or (at your option) any later version.
  *
  * This software 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. See the GNU
  * Lesser General Public License for more details.
  *
  * You should have received a copy of the GNU Lesser General Public
  * License along with this software; if not, write to the Free
  * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
  * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
  */
package org.jboss.xb.binding.sunday.marshalling;

import java.io.IOException;
import java.io.Reader;
import java.io.Writer;
import java.lang.reflect.Array;
import java.lang.reflect.Method;
import java.util.AbstractList;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import javax.xml.namespace.QName;
import javax.xml.parsers.ParserConfigurationException;
import org.jboss.util.Classes;
import org.jboss.xb.binding.AbstractMarshaller;
import org.jboss.xb.binding.AttributesImpl;
import org.jboss.xb.binding.Constants;
import org.jboss.xb.binding.Content;
import org.jboss.xb.binding.ContentWriter;
import org.jboss.xb.binding.JBossXBRuntimeException;
import org.jboss.xb.binding.ObjectLocalMarshaller;
import org.jboss.xb.binding.ObjectModelProvider;
import org.jboss.xb.binding.SimpleTypeBindings;
import org.jboss.xb.binding.Util;
import org.jboss.xb.binding.NamespaceRegistry;
import org.jboss.xb.binding.introspection.FieldInfo;
import org.jboss.xb.binding.metadata.CharactersMetaData;
import org.jboss.xb.binding.metadata.PropertyMetaData;
import org.jboss.xb.binding.sunday.unmarshalling.*;
import org.jboss.xb.binding.sunday.xop.XOPMarshaller;
import org.jboss.xb.binding.sunday.xop.XOPObject;
import org.xml.sax.ContentHandler;
import org.xml.sax.SAXException;

/**
* @author <a href="mailto:alex@jboss.org">Alexey Loubyansky</a>
* @version <tt>$Revision: 2366 $</tt>
*/
public class MarshallerImpl
    extends AbstractMarshaller
{
   private Stack stack = new StackImpl();

   private Object root;

   /**
    * Whether NULL values should be ignored or marshalled as xsi:nil='1'
    */
   private boolean supportNil = true;

   private boolean ignoreUnresolvedWildcard;

   private QName rootTypeQName;

   private SchemaBindingResolver schemaResolver;

   private SchemaBinding schema;

   private MarshallingContextImpl ctx = new MarshallingContextImpl();

   public boolean isIgnoreUnresolvedWildcard()
   {
      return ignoreUnresolvedWildcard;
   }

   public void setIgnoreUnresolvedWildcard(boolean ignoreUnresolvedWildcard)
   {
      this.ignoreUnresolvedWildcard = ignoreUnresolvedWildcard;
   }

   public SchemaBindingResolver getSchemaResolver()
   {
      return schemaResolver;
   }

   public void setSchemaResolver(SchemaBindingResolver schemaResolver)
   {
      this.schemaResolver = schemaResolver;
   }

   public QName getRootTypeQName()
   {
      return rootTypeQName;
   }

   public void setRootTypeQName(QName rootTypeQName)
   {
      this.rootTypeQName = rootTypeQName;
   }

   public boolean isSupportNil()
   {
      return supportNil;
   }

   public void setSupportNil(boolean supportNil)
   {
      this.supportNil = supportNil;
   }

   /**
    * Adds an attribute to the top most elements.
    * First, we check whether there is a namespace associated with the passed in prefix.
    * If the prefix was not declared, an exception is thrown.
    *
    * @param prefix    the prefix of the attribute to be declared
    * @param localName local name of the attribute
    * @param type      the type of the attribute
    * @param value     the value of the attribute
    */
   public void addAttribute(String prefix, String localName, String type, String value)
   {
      // todo addAttribute(String prefix, String localName, String type, String value)
   }

   // AbstractMarshaller implementation

   public void marshal(Reader xsdReader, ObjectModelProvider provider, Object root, Writer writer)
       throws IOException, SAXException, ParserConfigurationException
   {
      SchemaBinding model = XsdBinder.bind(xsdReader, null, schemaResolver);
      marshallInternal(root, model, writer);
   }

   public void marshal(String xsdURL, ObjectModelProvider provider, Object root, Writer writer) throws IOException,
       SAXException
   {
      SchemaBinding model = XsdBinder.bind(xsdURL, schemaResolver);
      marshallInternal(root, model, writer);
   }

   public void marshal(SchemaBinding model, ObjectModelProvider provider, Object root, Writer writer)
       throws IOException,
       SAXException
   {
      marshallInternal(root, model, writer);
   }

   private void marshallInternal(Object root, SchemaBinding schema, Writer writer)
       throws IOException, SAXException
   {
      if(schema == null)
      {
         throw new JBossXBRuntimeException("XSModel is not available!");
      }

      this.schema = schema;
      this.root = root;

      content.startDocument();

      if(rootTypeQName != null)
      {
         if(rootQNames.isEmpty())
         {
            throw new JBossXBRuntimeException("If type name (" +
                rootTypeQName +
                ") for the root element is specified then the name for the root element is required!"
            );
         }
         QName rootQName = (QName)rootQNames.get(0);

         TypeBinding type = schema.getType(rootTypeQName);
         if(type == null)
         {
            throw new JBossXBRuntimeException("Global type definition is not found: " + rootTypeQName);
         }

         if(isArrayWrapper(type))
         {
            stack.push(root);
            marshalComplexType(rootQName, type, true, false);
            stack.pop();
         }
         else
         {
            ElementBinding element = new ElementBinding(schema, rootQName, type);
            ctx.particle = new ParticleBinding(element);
            marshalElementOccurence(element, root, false, true);
         }
      }
      else if(rootQNames.isEmpty())
      {
         Iterator elements = schema.getElementParticles();
         if(!elements.hasNext())
         {
            throw new JBossXBRuntimeException("The schema doesn't contain global element declarations.");
         }

         while(elements.hasNext())
         {
            ParticleBinding element = (ParticleBinding)elements.next();
            ctx.particle = element;
            marshalElementOccurence((ElementBinding) element.getTerm(), root, true, true);
         }
      }
      else
      {
         for(int i = 0; i < rootQNames.size(); ++i)
         {
            QName qName = (QName)rootQNames.get(i);
            ParticleBinding element = schema.getElementParticle(qName);
            if(element == null)
            {
               Iterator components = schema.getElements();
               String roots = "";
               for(int j = 0; components.hasNext(); ++j)
               {
                  ElementBinding xsObject = (ElementBinding)components.next();
                  if(j > 0)
                  {
                     roots += ", ";
                  }
                  roots += xsObject.getQName();
               }
               throw new IllegalStateException("Root element not found: " + qName + " among " + roots);
            }

            ctx.particle = element;
            marshalElementOccurence((ElementBinding) element.getTerm(), root, true, true);
         }
      }

      content.endDocument();

      // version & encoding
      writeXmlVersion(writer);

      ContentWriter contentWriter = new ContentWriter(writer,
          propertyIsTrueOrNotSet(org.jboss.xb.binding.Marshaller.PROP_OUTPUT_INDENTATION)
      );
      content.handleContent(contentWriter);

      if(log.isTraceEnabled())
      {
         java.io.StringWriter traceWriter = new java.io.StringWriter();
         contentWriter = new ContentWriter(traceWriter,
             propertyIsTrueOrNotSet(org.jboss.xb.binding.Marshaller.PROP_OUTPUT_INDENTATION)
         );
         content.handleContent(contentWriter);
         log.trace("marshalled:\n" + traceWriter.getBuffer().toString());
      }
   }

   private boolean marshalElementOccurence(ElementBinding element,
                                           Object value,
                                           boolean optional,
                                           boolean declareNs)
   {
      QName xsiTypeQName = null;
      TypeBinding xsiType = null;
      if(value != null)
      {
         QName typeQName = element.getType().getQName();
         xsiTypeQName = (QName)cls2TypeMap.get(value.getClass());
         // in case xsiTypeQName is not null, typeQName should also be not null
         if(xsiTypeQName != null &&
             !(typeQName.getLocalPart().equals(xsiTypeQName.getLocalPart()) &&
                 typeQName.getNamespaceURI().equals(xsiTypeQName.getNamespaceURI())
             ))
         {
            if(log.isTraceEnabled())
            {
               log.trace(value.getClass() + " is mapped to xsi:type " + xsiTypeQName);
            }

            xsiType = schema.getType(xsiTypeQName);
            if(xsiType == null)
            {
               log.warn("Class " +
                   value.getClass() +
                   " is mapped to type " +
                   xsiTypeQName +
                   " but the type is not found in schema."
               );
            }
            // todo should check derivation also, i.e. if(xsiType.derivedFrom())
         }
      }

      TermBeforeMarshallingCallback marshallingHandler = element.getBeforeMarshallingCallback();
      if(marshallingHandler != null)
      {
         value = marshallingHandler.beforeMarshalling(value, ctx);
      }
     
      stack.push(value);
      boolean marshalled = marshalElement(element, xsiType, optional, declareNs);
      stack.pop();

      return marshalled;
   }

   private boolean marshalElement(ElementBinding element, TypeBinding xsiType, boolean optional, boolean declareNs)
   {
      Object value = stack.peek();
      boolean nillable = element.isNillable();
      boolean result = value != null || value == null && (optional || nillable);
      boolean trace = log.isTraceEnabled() && result;
      if(trace)
      {
         log.trace("started element " + element.getQName());
      }

      if(value != null)
      {
         boolean declareXsiType = xsiType != null;
         marshalElementType(element.getQName(),
             declareXsiType ? xsiType : element.getType(),
             declareNs,
             declareXsiType
         );
      }
      else if(nillable)
      {
         writeNillable(element.getQName(), nillable);
      }

      if(trace)
      {
         log.trace("finished element " + element.getQName());
      }

      return result;
   }

   private void marshalElementType(QName elementQName,
                                   TypeBinding type,
                                   boolean declareNs,
                                   boolean declareXsiType)
   {
      String elementNs = elementQName.getNamespaceURI();
      String elementLocal = elementQName.getLocalPart();

      XOPMarshaller xopMarshaller = schema.getXopMarshaller();
      if(xopMarshaller == null)
      {
         xopMarshaller = type.getXopMarshaller();
      }
     
      if(xopMarshaller != null && isXopOptimizable(type))
      {

         if(xopMarshaller.isXOPPackage())
         {
            // XOPMarshaller callback will create the attachment part
            Object o = stack.peek();
            String cid = xopMarshaller.addMtomAttachment(new XOPObject(o), elementNs, elementLocal);

            // Create the xopInclude element from CID and exit
            AttributesImpl attrs = null;
            String prefix = getPrefix(elementNs);
            boolean genPrefix = prefix == null && elementNs != null && elementNs.length() > 0;
            if(genPrefix)
            {
               prefix = "ns_" + elementLocal;
               attrs = new AttributesImpl(1);
               declareNs(attrs, prefix, elementNs);
            }

            String qName = prefixLocalName(prefix, elementLocal);
            content.startElement(elementNs, elementLocal, qName, attrs);

            AttributesImpl xopAttrs = new AttributesImpl(2);
            xopAttrs.add(Constants.NS_XML_SCHEMA, "xop", "xmlns:xop", "CDATA", Constants.NS_XOP_INCLUDE);
            xopAttrs.add(null, "href", "href", "CDATA", cid);

            content.startElement(Constants.NS_XOP_INCLUDE, "Include", "xop:Include", xopAttrs);
            content.endElement(Constants.NS_XOP_INCLUDE, "Include", "xop:Include");

            content.endElement(elementNs, elementLocal, qName);

            // Skip further processing
            return;

         }
         else
         {
            // XOPMarshaller did not process the object
            // In this case we try to marshall the corresponding simple type
            if(!type.isSimple())
            {
               if(type.hasOnlyXmlMimeAttributes()) // TODO: what's the purpose of this? It's xopOptimizable anyway...
               {
                  if(log.isTraceEnabled())
                  {
                     log.trace(
                         "XML MIME attributes of type " + type.getQName() +
                             " are ignored, the value is marshalled as " + type.getSimpleType().getQName()
                     );
                  }

                  type = type.getSimpleType();
               }
            }         
         }

      }

      // If we reach this point then either it wasn't a XOP element at all
      // or the XOPMarshaller did not process it.

      if(type.isSimple())
      {
         marshalSimpleType(elementQName, type, declareNs, declareXsiType);
      }
      else
      {
         marshalComplexType(elementQName, type, declareNs, declareXsiType);
      }
   }

   private void marshalSimpleType(QName elementQName,
                                  TypeBinding type,
                                  boolean declareNs,
                                  boolean declareXsiType)
   {
      ctx.attrs = null;
      if((declareNs || declareXsiType) && nsRegistry.size() > 0)
      {
         if(ctx.attrs == null)
         {
            ctx.attrs = new AttributesImpl(nsRegistry.size() + 1);
         }
         declareNs(ctx.attrs);
      }

      String elementNs = elementQName.getNamespaceURI();
      String elementLocal = elementQName.getLocalPart();

      String prefix = getPrefix(elementNs);
      boolean genPrefix = prefix == null && elementNs != null && elementNs.length() > 0;
      if(genPrefix)
      {
         prefix = "ns_" + elementLocal;
         if(ctx.attrs == null)
         {
            ctx.attrs = new AttributesImpl(1);
         }
         declareNs(ctx.attrs, prefix, elementNs);
      }

      if(declareXsiType)
      {
         declareXsiType(type.getQName(), ctx.attrs);
      }

      String typeName = type.getQName() == null ? null : type.getQName().getLocalPart();
      if(ctx.attrs == null && SimpleTypeBindings.XS_QNAME_NAME.equals(typeName) ||
          SimpleTypeBindings.XS_NOTATION_NAME.equals(typeName) ||
          type.getItemType() != null &&
              (SimpleTypeBindings.XS_QNAME_NAME.equals(type.getItemType().getQName().getLocalPart()) ||
                  SimpleTypeBindings.XS_NOTATION_NAME.equals(type.getItemType().getQName().getLocalPart())
              )
          )
      {
         ctx.attrs = new AttributesImpl(5);
      }

      Object value = stack.peek();
      String marshalled = marshalCharacters(elementNs, prefix, type, value);

      String qName = prefixLocalName(prefix, elementLocal);
      content.startElement(elementNs, elementLocal, qName, ctx.attrs);
      content.characters(marshalled.toCharArray(), 0, marshalled.length());
      content.endElement(elementNs, elementLocal, qName);
   }

   private void marshalComplexType(QName elementQName,
                                   TypeBinding type,
                                   boolean declareNs,
                                   boolean declareXsiType)
   {
      Collection attrBindings = type.getAttributes();
      int attrsTotal = declareNs || declareXsiType ? nsRegistry.size() + attrBindings.size() + 1: attrBindings.size();
      ctx.attrs = attrsTotal > 0 ? new AttributesImpl(attrsTotal) : null;

      if(declareNs && nsRegistry.size() > 0)
      {
         declareNs(ctx.attrs);
      }

      String generatedPrefix = null;
      if(declareXsiType)
      {
         generatedPrefix = declareXsiType(type.getQName(), ctx.attrs);
         if(generatedPrefix != null)
         {
            String typeNsWithGeneratedPrefix = type.getQName().getNamespaceURI();
            declareNs(ctx.attrs, generatedPrefix, typeNsWithGeneratedPrefix);
            declareNamespace(generatedPrefix, typeNsWithGeneratedPrefix);
         }
      }

      String elementNs = elementQName.getNamespaceURI();
      String elementLocal = elementQName.getLocalPart();

      String prefix = getPrefix(elementNs);
      boolean genPrefix = prefix == null && elementNs != null && elementNs.length() > 0;
      if(genPrefix)
      {
         // todo: it's possible that the generated prefix already mapped. this should be fixed
         prefix = "ns_" + elementLocal;
         declareNamespace(prefix, elementNs);
         if(ctx.attrs == null)
         {
            ctx.attrs = new AttributesImpl(1);
         }
         declareNs(ctx.attrs, prefix, elementNs);
      }

      if(!attrBindings.isEmpty())
      {
         for(Iterator i = attrBindings.iterator(); i.hasNext();)
         {
            AttributeBinding attrBinding = (AttributeBinding)i.next();
            QName attrQName = attrBinding.getQName();

            if(Constants.QNAME_XMIME_CONTENTTYPE.equals(attrQName))
            {
               continue;
            }

            ctx.attr = attrBinding;
            AttributeMarshaller marshaller = attrBinding.getMarshaller();
            String marshalledAttr = marshaller.marshal(ctx);

            if(marshalledAttr != null)
            {
               if(ctx.attrs == null)
               {
                  ctx.attrs = new AttributesImpl(5);
               }

               String attrNs = attrQName.getNamespaceURI();
               String attrLocal = attrQName.getLocalPart();
               String attrPrefix = null;
               if(attrNs != null)
               {
                  attrPrefix = getPrefix(attrNs);
                  if(attrPrefix == null && attrNs != null && attrNs.length() > 0)
                  {
                     attrPrefix = "ns_" + attrLocal;
                     declareNs(ctx.attrs, attrPrefix, attrNs);
                  }
               }

               String prefixedName = prefixLocalName(attrPrefix, attrLocal);
               ctx.attrs.add(attrNs, attrLocal, prefixedName, "CDATA", marshalledAttr);
            }
         }
         ctx.attr = null;
      }

      String characters = null;
      TypeBinding simpleType = type.getSimpleType();
      if(simpleType != null && !Constants.QNAME_ANYTYPE.equals(type.getQName()))
      {
         String fieldName = ctx.getSimpleContentProperty();
         CharactersMetaData charactersMetaData = type.getCharactersMetaData();
         PropertyMetaData propertyMetaData = charactersMetaData == null ? null : charactersMetaData.getProperty();
         if(propertyMetaData != null)
         {
            fieldName = propertyMetaData.getName();
         }

         if(fieldName != null)
         {
            boolean ignoreUnresolvedFieldOrClass = type.getSchemaBinding().isIgnoreUnresolvedFieldOrClass();
            Object o = stack.peek();
            Object value = getElementValue(o, fieldName, ignoreUnresolvedFieldOrClass);
            if(value != null)
            {
               String typeName = simpleType.getQName().getLocalPart();
               if(ctx.attrs == null && (SimpleTypeBindings.XS_QNAME_NAME.equals(typeName) ||
                   SimpleTypeBindings.XS_NOTATION_NAME.equals(typeName) ||
                   simpleType.getItemType() != null &&
                       (SimpleTypeBindings.XS_QNAME_NAME.equals(simpleType.getItemType().getQName().getLocalPart()) ||
                           SimpleTypeBindings.XS_NOTATION_NAME.equals(simpleType.getItemType().getQName().getLocalPart())
                       )
               )
                   )
               {
                  ctx.attrs = new AttributesImpl(5);
               }

               characters = marshalCharacters(elementNs, prefix, simpleType, value);
            }
         }
      }

      String qName = prefixLocalName(prefix, elementLocal);
      content.startElement(elementNs, elementLocal, qName, ctx.attrs);

      ParticleBinding particle = type.getParticle();
      if(particle != null)
      {
         marshalParticle(particle, false);
      }

      if(characters != null)
      {
         content.characters(characters.toCharArray(), 0, characters.length());
      }
      content.endElement(elementNs, elementLocal, qName);

      ctx.attrs = null;

      if(genPrefix)
      {
         removePrefixMapping(prefix);
      }

      if(generatedPrefix != null)
      {
         removePrefixMapping(generatedPrefix);
      }
   }

   private boolean marshalParticle(ParticleBinding particle, boolean declareNs)
   {
      boolean marshalled;
      TermBinding term = particle.getTerm();
      Object o;
      Iterator i;
     
      ParticleBinding ctxParticle = ctx.particle;
      ctx.particle = particle;
     
      if(term.isModelGroup())
      {
         ModelGroupBinding modelGroup = (ModelGroupBinding)term;
         if(modelGroup.isSkip() || stack.isEmpty())
         {
            marshalled = marshalModelGroup(modelGroup, declareNs);
         }
         else
         {
            PropertyMetaData propertyMetaData = modelGroup.getPropertyMetaData();
            if(propertyMetaData == null)
            {
               throw new JBossXBRuntimeException(
                   "Currently, property binding metadata must be available for a model group to be marshalled!"
               );
            }

            o = getChildren(stack.peek(), propertyMetaData.getName(),
                modelGroup.getSchema().isIgnoreUnresolvedFieldOrClass()
            );

            TermBeforeMarshallingCallback marshallingHandler = modelGroup.getBeforeMarshallingCallback();

            i = o != null && isRepeatable(particle) ? getIterator(o) : null;
            if(i != null)
            {
               marshalled = true;
               while(i.hasNext() && marshalled)
               {
                  Object value = i.next();

                  if(marshallingHandler != null)
                  {
                     value = marshallingHandler.beforeMarshalling(value, ctx);
                  }

                  stack.push(value);
                  marshalled = marshalModelGroup(modelGroup, declareNs);
                  stack.pop();
               }
            }
            else
            {
               if(marshallingHandler != null)
               {
                  o = marshallingHandler.beforeMarshalling(o, ctx);
               }

               stack.push(o);
               marshalled = marshalModelGroup(modelGroup, declareNs);
               stack.pop();
            }
         }
      }
      else if(term.isWildcard())
      {
         o = stack.peek();

         boolean popWildcardValue = false;
         ObjectLocalMarshaller marshaller = null;
         FieldToWildcardMapping mapping = (FieldToWildcardMapping)field2WildcardMap.get(o.getClass());
         if(mapping != null)
         {
            marshaller = mapping.marshaller;
            o = mapping.fieldInfo.getValue(o);
            stack.push(o);
            popWildcardValue = true;
         }

         TermBeforeMarshallingCallback marshallingHandler = term.getBeforeMarshallingCallback();
        
         i = o != null && isRepeatable(particle) ? getIterator(o) : null;
         if(i != null)
         {
            marshalled = true;
            while(i.hasNext() && marshalled)
            {
               Object value = i.next();

               if(marshallingHandler != null)
               {
                  value = marshallingHandler.beforeMarshalling(value, ctx);
               }

               marshalled = marshalWildcardOccurence(particle, marshaller, value, declareNs);
            }
         }
         else
         {
            if(marshallingHandler != null)
            {
               o = marshallingHandler.beforeMarshalling(o, ctx);
            }

            marshalled = marshalWildcardOccurence(particle, marshaller, o, declareNs);
         }

         if(popWildcardValue)
         {
            stack.pop();
         }
      }
      else
      {
         ElementBinding element = (ElementBinding)term;
         SchemaBinding schema = element.getSchema();
         o = getElementValue(element, schema.isIgnoreLowLine(), schema.isIgnoreUnresolvedFieldOrClass());

         i = o != null && isRepeatable(particle) ? getIterator(o) : null;
         if(i != null)
         {
            marshalled = true;
            while(i.hasNext() && marshalled)
            {
               Object value = i.next();
               marshalled = marshalElementOccurence(element, value, particle.getMinOccurs() == 0, declareNs);
            }
         }
         else
         {
            marshalled = marshalElementOccurence(element, o, particle.getMinOccurs() == 0, declareNs);
         }
      }
     
      ctx.particle = ctxParticle;
      return marshalled;
   }

   private boolean marshalWildcardOccurence(ParticleBinding particle,
                                            ObjectLocalMarshaller marshaller,
                                            Object value,
                                            boolean declareNs)
   {
      boolean marshalled = true;
      if(marshaller != null)
      {
         marshaller.marshal(ctx, value);
      }
      else if(value != null)
      {
         stack.push(value);
         marshalled = marshalWildcard(particle, declareNs);
         stack.pop();
      }
      return marshalled;
   }

   private boolean marshalWildcard(ParticleBinding particle, boolean declareNs)
   {
      WildcardBinding wildcard = (WildcardBinding)particle.getTerm();
      Object o = stack.peek();
      ClassMapping mapping = getClassMapping(o.getClass());
      if(mapping == null)
      {
         // todo: YAH (yet another hack)
         QName autoType = SimpleTypeBindings.typeQName(o.getClass());
         if(autoType != null)
         {
            String marshalled = SimpleTypeBindings.marshal(autoType.getLocalPart(), o, null);
            content.characters(marshalled.toCharArray(), 0, marshalled.length());
            return true;
         }
         else
         {
            ObjectLocalMarshaller marshaller = wildcard.getUnresolvedMarshaller();
            if(marshaller != null)
            {
               marshaller.marshal(ctx, o);
               return true;
            }
           
            String msg = "Failed to marshal wildcard: neither class mapping was found for "
               + o.getClass() + "@" + o.hashCode()
               + " (toString: " + o
               + ") nor marshaller for unresolved classes was setup.";
            if(ignoreUnresolvedWildcard)
            {
               log.warn(msg);
               return true;
            }
            else
            {
               throw new JBossXBRuntimeException(msg);
            }
         }
      }

      Object parentRoot = this.root;
      Stack parentStack = this.stack;
      SchemaBinding parentSchema = this.schema;

      this.root = o;
      this.stack = new StackImpl();
      this.schema = mapping.schemaUrl == null ? this.schema : XsdBinder.bind(mapping.schemaUrl, schemaResolver);

      boolean marshalled;
      if(mapping.elementName != null)
      {
         ParticleBinding element = schema.getElementParticle(mapping.elementName);
         if(element == null)
         {
            throw new JBossXBRuntimeException("Element " + mapping.elementName + " is not declared in the schema.");
         }

         ParticleBinding ctxParticle = ctx.particle;
         ctx.particle = element;
         marshalled = marshalElementOccurence((ElementBinding) element.getTerm(), root, particle.getMinOccurs() == 0, declareNs);
         ctx.particle = ctxParticle;
      }
      else if(mapping.typeName != null)
      {
         TypeBinding typeDef = schema.getType(mapping.typeName);
         if(typeDef == null)
         {
            throw new JBossXBRuntimeException("Type " +
                mapping.typeName +
                " is not defined in the schema."
            );
         }

         if(wildcard.getQName() == null)
         {
            throw new JBossXBRuntimeException("Expected the wildcard to have a non-null QName.");
         }

         ElementBinding element = new ElementBinding(schema, wildcard.getQName(), typeDef);
         ParticleBinding ctxParticle = ctx.particle;
         ctx.particle = new ParticleBinding(element);
         marshalled = marshalElementOccurence(element, root, particle.getMinOccurs() == 0, declareNs);
         ctx.particle = ctxParticle;
      }
      else
      {
         throw new JBossXBRuntimeException("Class mapping for " +
             mapping.cls +
             " is associated with neither global element name nor global type name."
         );
      }

      this.root = parentRoot;
      this.stack = parentStack;
      this.schema = parentSchema;

      return marshalled;
   }

   private boolean marshalModelGroup(ModelGroupBinding modelGroup, boolean declareNs)
   {
      boolean marshalled;
      if(modelGroup instanceof AllBinding)
      {
         marshalled = marshalModelGroupAll(modelGroup.getParticles(), declareNs);
      }
      else if(modelGroup instanceof ChoiceBinding)
      {
         marshalled = marshalModelGroupChoice(modelGroup.getParticles(), declareNs);
      }
      else
      {
         marshalled = marshalModelGroupSequence(modelGroup, declareNs);
      }
      return marshalled;
   }

   private boolean marshalModelGroupAll(Collection particles, boolean declareNs)
   {
      boolean marshalled = false;
      for(Iterator i = particles.iterator(); i.hasNext();)
      {
         ParticleBinding particle = (ParticleBinding)i.next();
         marshalled |= marshalParticle(particle, declareNs);
      }
      return marshalled;
   }

   private boolean marshalModelGroupChoice(Collection particles, boolean declareNs)
   {
      boolean marshalled = false;
      Content mainContent = this.content;
      for(Iterator i = particles.iterator(); i.hasNext() && !marshalled;)
      {
         ParticleBinding particle = (ParticleBinding)i.next();
         this.content = new Content();
         marshalled = marshalParticle(particle, declareNs);
      }

      if(marshalled)
      {
         mainContent.append(this.content);
      }
      this.content = mainContent;

      return marshalled;
   }

   private boolean marshalModelGroupSequence(ModelGroupBinding sequence, boolean declareNs)
   {
      // if sequence is bound to a collection,
      // we assume the iterator over the collection is in sync with the particle iterator
      Iterator valueIterator = null;
      if(!sequence.isSkip() && !stack.isEmpty())
      {
         Object o = stack.peek();
         if(o != null && (Collection.class.isAssignableFrom(o.getClass()) || o.getClass().isArray()))
         {
            valueIterator = getIterator(o);
         }
      }

      boolean marshalled = true;
      for(Iterator i = sequence.getParticles().iterator(); i.hasNext();)
      {
         if(valueIterator != null)
         {
            Object o = valueIterator.hasNext() ? valueIterator.next() : null;
            stack.push(o);
         }

         ParticleBinding particle = (ParticleBinding)i.next();
         marshalled &= marshalParticle(particle, declareNs);

         if(valueIterator != null)
         {
            stack.pop();
         }
      }
      return marshalled;
   }

   private String marshalCharacters(String elementUri,
                                    String elementPrefix,
                                    TypeBinding simpleType,
                                    Object value)
   {
      String marshalled;
      QName simpleTypeQName = simpleType.getQName();
      if(simpleType.getItemType() != null)
      {
         TypeBinding itemType = simpleType.getItemType();
         if(Constants.NS_XML_SCHEMA.equals(itemType.getQName().getNamespaceURI()))
         {
            List list;
            if(value instanceof List)
            {
               list = (List)value;
            }
            else if(value.getClass().isArray())
            {
               list = asList(value);
            }
            else
            {
               // todo: qname are also not yet supported
               throw new JBossXBRuntimeException(
                   "Expected value for list type is an array or " + List.class.getName() + " but got: " + value
               );
            }

            marshalled = SimpleTypeBindings.marshalList(itemType.getQName().getLocalPart(), list, null);
         }
         else
         {
            throw new JBossXBRuntimeException("Marshalling of list types with item types not from " +
                Constants.NS_XML_SCHEMA + " is not supported."
            );
         }
      }
      else if(simpleTypeQName != null && Constants.NS_XML_SCHEMA.equals(simpleTypeQName.getNamespaceURI()))
      {
         String typeName = simpleTypeQName.getLocalPart();

         String prefix = null;
         boolean removePrefix = false;
         if(SimpleTypeBindings.XS_QNAME_NAME.equals(typeName) ||
             SimpleTypeBindings.XS_NOTATION_NAME.equals(typeName))
         {
            QName qNameValue = (QName)value;
            if(qNameValue.getNamespaceURI() != null && qNameValue.getNamespaceURI().length() > 0)
            {
               prefix = nsRegistry.getPrefix(qNameValue.getNamespaceURI());
               if(prefix == null)
               {
                  prefix = qNameValue.getPrefix();
                  if(prefix == null || prefix.length() == 0)
                  {
                     prefix = qNameValue.getLocalPart() + "_ns";
                  }
                  nsRegistry.addPrefixMapping(prefix, qNameValue.getNamespaceURI());
                  ctx.declareNamespace(prefix, qNameValue.getNamespaceURI());

                  removePrefix = true;
               }
            }
         }

         marshalled = SimpleTypeBindings.marshal(typeName, value, nsRegistry);

         if(removePrefix)
         {
            nsRegistry.removePrefixMapping(prefix);
         }
      }
      // todo: this is a quick fix for boolean pattern (0|1 or true|false) should be refactored
      else if(simpleType.getLexicalPattern() != null &&
          simpleType.getBaseType() != null &&
          Constants.QNAME_BOOLEAN.equals(simpleType.getBaseType().getQName()))
      {
         String item = (String)simpleType.getLexicalPattern().get(0);
         if(item.indexOf('0') != -1 && item.indexOf('1') != -1)
         {
            marshalled = ((Boolean)value).booleanValue() ? "1" : "0";
         }
         else
         {
            marshalled = ((Boolean)value).booleanValue() ? "true" : "false";
         }
      }
      else
      {
         if(simpleType.getLexicalEnumeration() != null)
         {
            Method getValue;
            try
            {
               getValue = value.getClass().getMethod("value", null);
            }
            catch(NoSuchMethodException e)
            {
               try
               {
                  getValue = value.getClass().getMethod("getValue", null);
               }
               catch(NoSuchMethodException e1)
               {
                  throw new JBossXBRuntimeException("Failed to find neither value() nor getValue() in " +
                      value.getClass() +
                      " which is bound to enumeration type " + simpleTypeQName
                  );
               }
            }

            try
            {
               value = getValue.invoke(value, null);
            }
            catch(Exception e)
            {
               throw new JBossXBRuntimeException(
                   "Failed to invoke getValue() on " + value + " to get the enumeration value", e
               );
            }
         }

         marshalled = marshalCharacters(elementUri,
             elementPrefix,
             simpleType.getBaseType(),
             value
         );
      }
      return marshalled;
   }

   private void writeNillable(QName elementQName, boolean nillable)
   {
      if(!supportNil)
      {
         return;
      }

      if(!nillable)
      {
         throw new JBossXBRuntimeException("Failed to marshal " +
             elementQName +
             ": Java value is null but the element is not nillable."
         );
      }

      String elementNs = elementQName.getNamespaceURI();
      String elementLocal = elementQName.getLocalPart();

      AttributesImpl attrs;
      String prefix = getPrefix(elementNs);
      if(prefix == null && elementNs != null && elementNs.length() > 0)
      {
         prefix = "ns_" + elementLocal;
         attrs = new AttributesImpl(2);
         declareNs(attrs, prefix, elementNs);
      }
      else
      {
         attrs = new AttributesImpl(1);
      }

      String xsiPrefix = getPrefix(Constants.NS_XML_SCHEMA_INSTANCE);
      if(xsiPrefix == null)
      {
         xsiPrefix = "xsi";
         declareNs(attrs, "xsi", Constants.NS_XML_SCHEMA_INSTANCE);
      }

      String nilQName = xsiPrefix + ":nil";
      attrs.add(Constants.NS_XML_SCHEMA_INSTANCE, "nil", nilQName, null, "1");

      String qName = prefixLocalName(prefix, elementLocal);
      content.startElement(elementNs, elementLocal, qName, attrs);
      content.endElement(elementNs, elementLocal, qName);
   }

   private Object getElementValue(ElementBinding element,
                                  boolean ignoreLowLine,
                                  boolean ignoreNotFoundField)
   {
      Object value;
      Object peeked = stack.peek();
      if(peeked == null)
      {
         value = null;
      }
      else if(peeked instanceof Collection || peeked.getClass().isArray())
      {
         value = peeked;
      }
      else
      {
         String fieldName = null;
         PropertyMetaData propertyMetaData = element.getPropertyMetaData();
         if(propertyMetaData != null)
         {
            fieldName = propertyMetaData.getName();
         }

         if(fieldName == null)
         {
            fieldName = Util.xmlNameToFieldName(element.getQName().getLocalPart(), ignoreLowLine);
         }

         value = getChildren(peeked, fieldName, ignoreNotFoundField);
         if(value == null)
         {
            value = getElementValue(peeked, fieldName, ignoreNotFoundField);
         }
      }
      return value;
   }

   private static boolean isArrayWrapper(TypeBinding type)
   {
      boolean is = false;
      if(!type.isSimple())
      {
         ParticleBinding particle = type.getParticle();
         if(particle != null)
         {
            is = particle.getMaxOccursUnbounded() || particle.getMaxOccurs() > 1;
         }
      }
      return is;
   }

   private Iterator getIterator(Object value)
   {
      Iterator i = null;
      if(value instanceof Collection)
      {
         i = ((Collection)value).iterator();
      }
      else if(value.getClass().isArray())
      {
         final Object arr = value;
         i = new Iterator()
         {
            private int curInd = 0;
            private int length = Array.getLength(arr);

            public boolean hasNext()
            {
               return curInd < length;
            }

            public Object next()
            {
               return Array.get(arr, curInd++);
            }

            public void remove()
            {
               throw new UnsupportedOperationException("remove is not implemented.");
            }
         };
      }
      else if(value instanceof Iterator)
      {
         i = (Iterator)value;
      }
      else
      {
         //throw new JBossXBRuntimeException("Unexpected type for children: " + value.getClass());
      }
      return i;
   }

   private static Object getChildren(Object o, String fieldName, boolean ignoreNotFoundField)
   {
      Object children = null;
      if(!writeAsValue(o.getClass()))
      {
         children = getJavaValue(fieldName, o, true, ignoreNotFoundField);
      }
      return children;
   }

   private static Object getJavaValue(String fieldName,
                                      Object o,
                                      boolean forComplexType,
                                      boolean ignoreNotFoundField)
   {
      FieldInfo fieldInfo = FieldInfo.getFieldInfo(o.getClass(), fieldName, !ignoreNotFoundField);
      Object value = null;
      if(fieldInfo != null && (!forComplexType || forComplexType && !writeAsValue(fieldInfo.getType())))
      {
         value = fieldInfo.getValue(o);
      }
      return value;
   }

   private static Object getElementValue(Object o, String fieldName, boolean ignoreNotFoundField)
   {
      Object value;
      if(writeAsValue(o.getClass()))
      {
         value = o;
      }
      else
      {
         value = getJavaValue(fieldName, o, false, ignoreNotFoundField);
      }

      return value;
   }

   private static boolean writeAsValue(final Class type)
   {
      return Classes.isPrimitive(type) ||
          type == String.class ||
          type == java.util.Date.class ||
          type == java.math.BigDecimal.class ||
          type == java.math.BigInteger.class;
   }

   private static boolean isRepeatable(ParticleBinding particle)
   {
      return particle.getMaxOccursUnbounded() || particle.getMaxOccurs() > 1 || particle.getMinOccurs() > 1;
   }

   private static final List asList(final Object arr)
   {
      return new AbstractList()
      {
         private final Object array = arr;

         public Object get(int index)
         {
            return Array.get(array, index);
         }

         public int size()
         {
            return Array.getLength(array);
         }
      };
   }

   private static boolean isXopOptimizable(TypeBinding type)
   {
      while(type != null)
      {
         if(Constants.QNAME_BASE64BINARY.equals(type.getQName()))
         {
            return true;
         }
         type = type.getBaseType();
      }
      return false;
   }

   private class MarshallingContextImpl implements MarshallingContext
   {
      private ContentHandler ch;
      private AttributeBinding attr;
      private ParticleBinding particle;

      private AttributesImpl attrs;

      public boolean isAttributeRequired()
      {
         throw new UnsupportedOperationException();
      }

      public boolean isTypeComplex()
      {
         throw new UnsupportedOperationException();
      }

      public String getSimpleContentProperty()
      {
         return schema.getSimpleContentProperty();
      }

      public ContentHandler getContentHandler()
      {
         if(ch == null)
         {
            ch = new ContentHandlerAdaptor();
         }
         return ch;
      }

      public SchemaBinding getSchemaBinding()
      {
         return schema;
      }

      public AttributeBinding getAttributeBinding()
      {
         return attr;
      }

      public String getPrefix(String ns)
      {
         return MarshallerImpl.this.getPrefix(ns);
      }

      public void declareNamespace(String prefix, String ns)
      {
         declareNs(attrs, prefix, ns);
         nsRegistry.addPrefixMapping(prefix, ns);
      }

      public NamespaceRegistry getNamespaceContext()
      {
         return nsRegistry;
      }

      public Object peek()
      {
         return stack.isEmpty() ? null : stack.peek();
      }

      public ParticleBinding getParticleBinding()
      {
         return particle;
      }
   }
}
TOP

Related Classes of org.jboss.xb.binding.sunday.marshalling.MarshallerImpl$MarshallingContextImpl

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.