package uk.co.badgersinfoil.asxsd.components;

 import org.eclipse.xsd.XSDAttributeDeclaration;

 import org.eclipse.xsd.XSDAttributeUse;

 import org.eclipse.xsd.XSDConcreteComponent;

 import uk.co.badgersinfoil.asxsd.CodegenContext;

 import uk.co.badgersinfoil.asxsd.CodegenRole;

 import uk.co.badgersinfoil.asxsd.DocUtils;

 import uk.co.badgersinfoil.asxsd.MarshalBuilder;

 import uk.co.badgersinfoil.asxsd.StringUtils;

 import uk.co.badgersinfoil.asxsd.TypeBuilder;

 import uk.co.badgersinfoil.asxsd.TypeDescriptor;

 import uk.co.badgersinfoil.metaas.dom.ASClassType;

 import uk.co.badgersinfoil.metaas.dom.ASField;

 import uk.co.badgersinfoil.metaas.dom.ASIfStatement;

 import uk.co.badgersinfoil.metaas.dom.StatementContainer;

 import uk.co.badgersinfoil.metaas.dom.Visibility;

 public class AttributeUseComponent extends AbstractMappingComponent {

         public boolean willAccept(XSDConcreteComponent component) {

                 return component instanceof XSDAttributeUse;

         }

         public void generateCode(CodegenContext context,

                                  XSDConcreteComponent component)

         {

                 XSDAttributeUse attrUse = (XSDAttributeUse)component;

                 CodegenRole role = context.getCurrentRole();

                 if (role == CodegenRole.UNMARSHAL) {

                         context.pushAttrScope();

                         if (!attrUse.isRequired()) {

                                 StatementContainer code = context.getCurrentMethodCode();

                                 String attrAccessExpr = "thisElement."+StringUtils.attrAccess(attrUse.getAttributeDeclaration());

                                 ASIfStatement ifStmt = code.newIf(attrAccessExpr+".length() > 0");

                                 context.setCurrentMethodCode(ifStmt);

                         }

                         XSDAttributeDeclaration attrDecl = attrUse.getAttributeDeclaration();

                         context.generateCode(attrDecl);

                         context.popAttrScope();

                 } else if (role == CodegenRole.MARSHAL) {

                         context.pushAttrScope();

                         if (!attrUse.isRequired()) {

                                 StatementContainer code = context.getCurrentMethodCode();

                                 String sourceExpr = MarshalBuilder.getCurrentSourceExpr(context);

                                 String propertyNameExpr = sourceExpr+"."+context.variableNameFor(attrUse.getAttributeDeclaration());

                                 ASIfStatement ifStmt = code.newIf(propertyNameExpr+" != null");

                                 context.setCurrentMethodCode(ifStmt);

                         }

                         XSDAttributeDeclaration attrDecl = attrUse.getAttributeDeclaration();

                         context.generateCode(attrDecl);

                         context.popAttrScope();

                 } else if (role == CodegenRole.TYPE) {

                         XSDAttributeDeclaration attrDecl = attrUse.getAttributeDeclaration();

                         ASClassType clazz = TypeBuilder.getCurrentType(context);

                         TypeDescriptor typeDesc = context.typeDescriptorFor(attrDecl.getTypeDefinition());

                         ASField field = clazz.newField(context.variableNameFor(attrDecl), Visibility.PUBLIC,

                                                        typeDesc.getTypeName());

                         String doc = DocUtils.findDocumentation(attrDecl.getAnnotation());

                         if (doc == null) {

                                 if (typeDesc.isArray()) {

                                         field.setDescription(typeDesc.getDocumentation());

                                 }

                         } else {

                                 field.setDescription(doc);

                         }

                 } else {

                         super.generateCode(context, component);

                 }

         }

 }