package uk.co.badgersinfoil.asxsd.components;

 import org.eclipse.xsd.XSDConcreteComponent;

 import org.eclipse.xsd.XSDSimpleTypeDefinition;

 import org.eclipse.xsd.XSDVariety;

 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.TypeDescriptor;

 import uk.co.badgersinfoil.asxsd.UnmarshalBuilder;

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

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

 public class ListComponent extends AbstractSimpleTypeComponent {

         public TypeDescriptor typeDescriptorFor(CodegenContext context,

                                                 XSDSimpleTypeDefinition listType)

         {

                 return context.getOrCreateType(listType);

         }

         public TypeDescriptor createTypeDescriptor(CodegenContext context,

                                                    XSDSimpleTypeDefinition listType)

         {

                 XSDSimpleTypeDefinition itemType = listType.getItemTypeDefinition();

                 TypeDescriptor itemTypeDesc = context.typeDescriptorFor(itemType);

                 String docs = "Items of type "+itemTypeDesc.getTypeName()+" ("+itemType.getURI()+")";

                 return new TypeDescriptor("Array", true, docs, listType);

         }

         public void generateCode(CodegenContext context,

                                  XSDConcreteComponent component)

         {

                 XSDSimpleTypeDefinition listType = (XSDSimpleTypeDefinition)component;

                 XSDSimpleTypeDefinition itemType = listType.getItemTypeDefinition();

                 CodegenRole role = context.getCurrentRole();

                 if (role == CodegenRole.UNMARSHAL) {

                         StatementContainer block = context.getCurrentMethodCode();

                         String srcExpr = UnmarshalBuilder.getCurrentSourceExpr(context);

                         block.addStmt("var _tmp = new Array()");

                         StatementContainer loop = block.newForEachIn("var listItem", srcExpr+".split(\" \")");

                         MappingFunction fn = context.functionFor(itemType);

                         String converted = fn.appliedTo("listItem");

                         loop.addStmt("_tmp.push("+converted+")");

                         block.newReturn("_tmp");

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

                         StatementContainer block = context.getCurrentMethodCode();

                         String srcExpr = MarshalBuilder.getCurrentSourceExpr(context);

                         block.addStmt("var _tmp = null");

                         StatementContainer loop = block.newForEachIn("var listItem", srcExpr);

                         MappingFunction fn = context.functionFor(itemType);

                         String converted = fn.appliedTo("listItem");

                         ASIfStatement ifFirst = loop.newIf("_tmp == null");

                         ifFirst.addStmt("_tmp = " + converted);

                         ifFirst.getElse().addStmt("_tmp += \" \" + "+converted);

                         block.newReturn("_tmp");

                 } else {

                         super.generateCode(context, component);

                 }

         }

         public boolean willAcceptType(XSDSimpleTypeDefinition simpleType) {

                 return simpleType.getVariety() == XSDVariety.LIST_LITERAL;

         }

 }