package uk.co.badgersinfoil.asxsd.components;

 import org.eclipse.xsd.XSDAttributeDeclaration;

 import org.eclipse.xsd.XSDConcreteComponent;

 import uk.co.badgersinfoil.asxsd.CodegenContext;

 import uk.co.badgersinfoil.asxsd.CodegenRole;

 import uk.co.badgersinfoil.asxsd.MappingFunction;

 import uk.co.badgersinfoil.asxsd.MarshalBuilder;

 import uk.co.badgersinfoil.asxsd.StringUtils;

 import uk.co.badgersinfoil.asxsd.UnmarshalBuilder;

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

 public class AttributeDeclarationComponent extends AbstractMappingComponent {

         public boolean willAccept(XSDConcreteComponent component) {

                 return component instanceof XSDAttributeDeclaration;

         }

         public void generateCode(CodegenContext context,

                                  XSDConcreteComponent component)

         {

                 XSDAttributeDeclaration attrDecl = (XSDAttributeDeclaration)component;

                 CodegenRole role = context.getCurrentRole();

                 if (role == CodegenRole.UNMARSHAL) {

                         StatementContainer code = context.getCurrentMethodCode();

                         String sourceExpr = UnmarshalBuilder.getCurrentSourceExpr(context);

                         String attrAccessExpr = sourceExpr+"."+StringUtils.attrAccess(attrDecl);

                         MappingFunction function = context.functionFor(attrDecl.getType());

                         String converted = function.appliedTo(attrAccessExpr);

                         code.addStmt("_result."+context.variableNameFor(attrDecl)+" = "+converted);

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

                         StatementContainer code = context.getCurrentMethodCode();

                         String sourceExpr = MarshalBuilder.getCurrentSourceExpr(context);

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

                         MappingFunction function = context.functionFor(attrDecl.getType());

                         String converted = function.appliedTo(propertyNameExpr);

                         code.addStmt("_result."+StringUtils.attrAccess(attrDecl)+" = "+converted);

                 } else {

                         super.generateCode(context, component);

                 }

         }

         public String variableNameFor(CodegenContext context,

                                       XSDConcreteComponent component)

         {

                 // TODO: just handle XSDNamedComponent in superclass?

                 XSDAttributeDeclaration attrDecl = (XSDAttributeDeclaration)component;

                 return context.toVariableName(attrDecl.getName());

         }

 }