/*

  * Copyright (c) 2007 David Holroyd

  */

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

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

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

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

 public class DateTimeSimpleTypeComponent extends AbstractSimpleTypeComponent {

         protected TypeDescriptor createTypeDescriptor(CodegenContext context,

                                                       XSDSimpleTypeDefinition simpleType)

         {

                 return new TypeDescriptor(ASConstants.TYPE_DATE, false, null, simpleType);

         }

         protected boolean willAcceptType(XSDSimpleTypeDefinition simpleType) {

                 return simpleType.getVariety() == XSDVariety.ATOMIC_LITERAL

                     && "dateTime".equals(simpleType.getName())

                     && "http://www.w3.org/2001/XMLSchema".equals(simpleType.getTargetNamespace());

         }

         public void generateCode(CodegenContext context,

                                  XSDConcreteComponent component)

         {

                 CodegenRole role = context.getCurrentRole();

                 if (role == CodegenRole.UNMARSHAL) {

                         StatementContainer code = context.getCurrentMethodCode();

                         String expr = "thisValue";

                         code.addStmt("var _result:Date = new Date()");

                         code.addStmt("var _rs:RegExp = /^(\\-?)(\\d+)\\-(\\d+)\\-(\\d+)[Tt](\\d+):(\\d+):(\\d+)(\\.\\d+)?\\s*(.+)?$/");

                         code.addStmt("var _match:Object = _rs.exec("+expr+")");

                         ASIfStatement yearSign = code.newIf("_match[1] == \"\"");

                         yearSign.addStmt("_result.fullYearUTC = Number(_match[2])");

                         yearSign.getElse().addStmt("_result.fullYearUTC = -Number(_match[2])");

                         code.addStmt("_result.monthUTC = Number(_match[3])-1");

                         code.addStmt("_result.dateUTC = Number(_match[4])");

                         code.addStmt("_result.hoursUTC = Number(_match[5])");

                         code.addStmt("_result.minutesUTC = Number(_match[6])");

                         code.addStmt("_result.secondsUTC = Number(_match[7])");

                         ASIfStatement secondsFraction = code.newIf("_match[8] == \"\"");

                         secondsFraction.addStmt("_result.millisecondsUTC = Number(\"0\"+_match[8])*1000");

                         ASIfStatement timezoneOffset = code.newIf("_match[9] == \"\"");

                         timezoneOffset.addStmt("var _tzrs:RegExp = /(\\d\\d):(\\d\\d)/");

                         timezoneOffset.addStmt("_match = _tzrs.exec(_match[9])");

                         ASIfStatement timezoneOffsetPresent = timezoneOffset.newIf("_match != null");

                         // TODO: handle timezone offsets

                         timezoneOffsetPresent.addComment(" Eeek!  handle timezone offset!");

                         code.addStmt("return _result");

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

                         StatementContainer code = context.getCurrentMethodCode();

                         String expr = "thisValue";

                         code.addStmt("var _result:String = \"\"");

                         code.addStmt("_result += "+expr+".fullYearUTC");

                         code.addStmt("_result += \"-\"");

                         code.addStmt("_result += "+expr+".monthUTC+1");

                         code.addStmt("_result += \"-\"");

                         code.addStmt("_result += "+expr+".dateUTC");

                         code.addStmt("_result += \"T\"");

                         code.addStmt("_result += "+expr+".hoursUTC");

                         code.addStmt("_result += \":\"");

                         code.addStmt("_result += "+expr+".minutesUTC");

                         code.addStmt("_result += \":\"");

                         code.addStmt("_result += "+expr+".secondsUTC");

                         ASIfStatement secondsFraction = code.newIf(""+expr+".millisecondsUTC != 0");

                         secondsFraction.addStmt("_result += String("+expr+".millisecondsUTC/1000).slice(1)");

                         code.addStmt("_result += \" Z\"");

                         code.addStmt("return _result");

                 } else {

                         super.generateCode(context, component);

                 }

         }

 }