/*
    * Copyright 2001-2004 The Apache Software Foundation.
    * 
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    * 
    *      http://www.apache.org/licenses/LICENSE-2.0
    * 
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.apache.juddi.handler;
  
  import java.util.Vector;
  
  import org.apache.juddi.datatype.CategoryBag;
  import org.apache.juddi.datatype.Description;
  import org.apache.juddi.datatype.IdentifierBag;
  import org.apache.juddi.datatype.KeyedReference;
  import org.apache.juddi.datatype.OverviewDoc;
  import org.apache.juddi.datatype.RegistryObject;
  import org.apache.juddi.datatype.UploadRegister;
  import org.apache.juddi.datatype.request.AuthInfo;
  import org.apache.juddi.datatype.request.SaveTModel;
  import org.apache.juddi.datatype.tmodel.TModel;
  import org.apache.juddi.util.xml.XMLUtils;
  import org.w3c.dom.Element;
  
  /***
   * @author Steve Viens (sviens@apache.org)
   */
  public class SaveTModelHandler extends AbstractHandler
  {
    public static final String TAG_NAME = "save_tModel";
  
    private HandlerMaker maker = null;
  
    protected SaveTModelHandler(HandlerMaker maker)
    {
      this.maker = maker;
    }
  
    public RegistryObject unmarshal(Element element)
    {
      SaveTModel obj = new SaveTModel();
      Vector nodeList = null;
      AbstractHandler handler = null;
  
      // Attributes
      String generic = element.getAttribute("generic");
      if ((generic != null && (generic.trim().length() > 0)))
        obj.setGeneric(generic);
  
      // Text Node Value
      // {none}
  
      // Child Elements
      nodeList = XMLUtils.getChildElementsByTagName(element,AuthInfoHandler.TAG_NAME);
      if (nodeList.size() > 0)
      {
        handler = maker.lookup(AuthInfoHandler.TAG_NAME);
        obj.setAuthInfo((AuthInfo)handler.unmarshal((Element)nodeList.elementAt(0)));
      }
  
      nodeList = XMLUtils.getChildElementsByTagName(element,TModelHandler.TAG_NAME);
      for (int i=0; i<nodeList.size(); i++)
      {
        handler = maker.lookup(TModelHandler.TAG_NAME);
        obj.addTModel((TModel)handler.unmarshal((Element)nodeList.elementAt(i)));
      }
  
      nodeList = XMLUtils.getChildElementsByTagName(element,UploadRegisterHandler.TAG_NAME);
      for (int i=0; i<nodeList.size(); i++)
      {
        handler = maker.lookup(UploadRegisterHandler.TAG_NAME);
        obj.addUploadRegister((UploadRegister)handler.unmarshal((Element)nodeList.elementAt(i)));
      }
  
      return obj;
    }
  
    public void marshal(RegistryObject object,Element parent)
    {
      SaveTModel request = (SaveTModel)object;
      String generic = request.getGeneric();
      generic = getGeneric(generic);
      String namespace = getUDDINamespace(generic);
      Element element = parent.getOwnerDocument().createElementNS(namespace,TAG_NAME);
      AbstractHandler handler = null;
  
      element.setAttribute("generic",generic);
  
      AuthInfo authInfo = request.getAuthInfo();
      if (authInfo != null)
      {
       handler = maker.lookup(AuthInfoHandler.TAG_NAME);
       handler.marshal(authInfo,element);
     }
 
     Vector tModelVector = request.getTModelVector();
     if ((tModelVector!=null) && (tModelVector.size() > 0))
     {
       handler = maker.lookup(TModelHandler.TAG_NAME);
       for (int i=0; i<tModelVector.size(); i++)
         handler.marshal((TModel)tModelVector.elementAt(i),element);
     }
 
     Vector urVector = request.getUploadRegisterVector();
     if ((urVector!=null) && (urVector.size() > 0))
     {
       handler = maker.lookup(UploadRegisterHandler.TAG_NAME);
       for (int i=0; i<urVector.size(); i++)
         handler.marshal((UploadRegister)urVector.elementAt(i),element);
     }
 
     parent.appendChild(element);
   }
 
 
   /****************************************************************************/
   /****************************** TEST DRIVER *********************************/
   /****************************************************************************/
 
 
   public static void main(String args[])
     throws Exception
   {
     HandlerMaker maker = HandlerMaker.getInstance();
     AbstractHandler handler = maker.lookup(SaveTModelHandler.TAG_NAME);
 
     Element parent = XMLUtils.newRootElement();
     Element child = null;
 
     AuthInfo authInfo = new AuthInfo();
     authInfo.setValue("6f157513-844e-4a95-a856-d257e6ba9726");
 
     OverviewDoc overDoc = new OverviewDoc();
     overDoc.setOverviewURL("http://www.sviens.com/service.html");
     overDoc.addDescription(new Description("over-doc Descr"));
     overDoc.addDescription(new Description("over-doc Descr Two","en"));
 
     CategoryBag catBag = new CategoryBag();
     catBag.addKeyedReference(new KeyedReference("catBagKeyName","catBagKeyValue"));
     catBag.addKeyedReference(new KeyedReference("uuid:dfddb58c-4853-4a71-865c-f84509017cc7","catBagKeyName2","catBagKeyValue2"));
 
     IdentifierBag idBag = new IdentifierBag();
     idBag.addKeyedReference(new KeyedReference("idBagKeyName","idBagkeyValue"));
     idBag.addKeyedReference(new KeyedReference("uuid:f78a135a-4769-4e79-8604-54d440314bc0","idBagKeyName2","idBagkeyValue2"));
 
     TModel tModel = new TModel();
     tModel.setTModelKey("uuid:269855db-62eb-4862-8e5a-1b06f2753038");
     tModel.setOperator("jUDDI");
     tModel.setAuthorizedName("Steve Viens");
     tModel.setName("jUDDI TModel");
     tModel.addDescription(new Description("tModel whatever"));
     tModel.addDescription(new Description("tModel whatever too","fr"));
     tModel.setCategoryBag(catBag);
     tModel.setIdentifierBag(idBag);
     tModel.setOverviewDoc(overDoc);
 
     SaveTModel request = new SaveTModel();
     request.setAuthInfo(authInfo);
     request.addTModel(tModel);
     request.addTModel(tModel);
     request.addUploadRegister(new UploadRegister("http://www.juddi.org/businessEntity.xml"));
     request.addUploadRegister(new UploadRegister("http://www.sourceforge.net/businessService.xml"));
     request.addUploadRegister(new UploadRegister("http://www.uddi.org/bindingTemplate.xml"));
 
     System.out.println();
 
     RegistryObject regObject = request;
     handler.marshal(regObject,parent);
     child = (Element)parent.getFirstChild();
     parent.removeChild(child);
     XMLUtils.writeXML(child,System.out);
 
     System.out.println();
 
     regObject = handler.unmarshal(child);
     handler.marshal(regObject,parent);
     child = (Element)parent.getFirstChild();
     parent.removeChild(child);
     XMLUtils.writeXML(child,System.out);
   }
 }