/*
    * Copyright 2001-2013 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.v3.client.cli;
  
  import java.io.File;
  import java.net.URL;
  import java.util.List;
  import java.util.Properties;
  import java.util.Set;
  
  import javax.xml.namespace.QName;
  
  import org.apache.juddi.jaxb.PrintUDDI;
  import org.apache.juddi.v3.client.config.UDDIClerk;
  import org.apache.juddi.v3.client.config.UDDIClient;
  import org.apache.juddi.v3.client.mapping.URLLocalizerDefaultImpl;
  import org.apache.juddi.v3.client.mapping.wadl.Application;
  import org.apache.juddi.v3.client.mapping.wadl.WADL2UDDI;
  import org.apache.juddi.v3.client.transport.Transport;
  import org.apache.juddi.v3_service.JUDDIApiPortType;
  import org.uddi.api_v3.AuthToken;
  import org.uddi.api_v3.BusinessDetail;
  import org.uddi.api_v3.BusinessEntity;
  import org.uddi.api_v3.BusinessService;
  import org.uddi.api_v3.BusinessServices;
  import org.uddi.api_v3.GetAuthToken;
  import org.uddi.api_v3.Name;
  import org.uddi.api_v3.SaveBusiness;
  import org.uddi.api_v3.SaveService;
  import org.uddi.api_v3.SaveTModel;
  import org.uddi.api_v3.TModel;
  import org.uddi.v3_service.UDDIPublicationPortType;
  import org.uddi.v3_service.UDDISecurityPortType;
  
  /**
   * This class shows how to perform a WSDL2UDDI import manually. More
   * specifically, this is WSDL2UDDI without using annotations.
   *
   * @author <a href="mailto:alexoree@apache.org">Alex O'Ree</a>
   */
  public class WadlImport {
  
          private Properties properties = new Properties();
          private UDDISecurityPortType security = null;
          private UDDIPublicationPortType publish = null;
  
          public void fire(String pathOrURL, String businessKey, String token, Transport transport) throws Exception {
  
                  if (transport == null) {
                  // create a manager and read the config in the archive; 
                          // you can use your config file name
                          UDDIClient clerkManager = new UDDIClient("META-INF/simple-publish-uddi.xml");
                          transport = clerkManager.getTransport();
                  }
                  // Now you create a reference to the UDDI API
                  security = transport.getUDDISecurityService();
                  publish = transport.getUDDIPublishService();
  
                  if (token == null) {
                          //step one, get a token
                          GetAuthToken getAuthTokenRoot = new GetAuthToken();
                          getAuthTokenRoot.setUserID("uddi");
                          getAuthTokenRoot.setCred("uddi");
  
                          // Making API call that retrieves the authentication token for the 'root' user.
                          AuthToken rootAuthToken = security.getAuthToken(getAuthTokenRoot);
                          token = rootAuthToken.getAuthInfo();
                  }
  
                  //step two, identify the key used for all your stuff
                  //you must have a key generator created already
                  //here, we are assuming that you don't have one
                  //NOTE: these are some of the publicly available WSDLs that were used to test WSDL2UDDI
                  //publish.saveTModel(stm);
                  //step three, we have two options
                  //1) import the wsdl's services into a brand new business
                  //2) import the wsdl's services into an existing business
                  //in either case, we're going to have to parse the WSDL
                  //Application app = WADL2UDDI.parseWadl(new URL("http://server/wsdl.wsdl"), "username", "password", clerkManager.getClientConfig().isX_To_Wsdl_Ignore_SSL_Errors() );
                  Application app = null;
                  if (!pathOrURL.startsWith("http")) {
                          File f = new File("test.wadl");
                          if (!f.exists()) {
                                  System.out.println(pathOrURL + " doesn't exist!");
                                  return;
                         } else {
                                 System.out.println("Attempting to parse " + f.getAbsolutePath());
                                 app = WADL2UDDI.parseWadl(f);
                         }
                 } else {
                         app = WADL2UDDI.parseWadl(new URL(pathOrURL));
                 }
 
                 List<URL> urls = WADL2UDDI.getBaseAddresses(app);
                 URL url = urls.get(0);
                 String domain = url.getHost();
                 PrintUDDI<TModel> tmodelPrinter = new PrintUDDI<TModel>();
                 TModel keygen = UDDIClerk.createKeyGenator("uddi:" + domain + ":keygenerator", domain, "en");
 
                 //save the keygen
                 SaveTModel stm = new SaveTModel();
                 stm.setAuthInfo(token);
                 stm.getTModel().add(keygen);
                 System.out.println("Saving the following tModel keygen");
                 System.out.println(tmodelPrinter.print(keygen));
                 publish.saveTModel(stm);
 
                 properties.put("keyDomain", domain);
                 properties.put("businessName", domain);
                 properties.put("serverName", url.getHost());
                 properties.put("serverPort", url.getPort());
                 //wsdlURL = wsdlDefinition.getDocumentBaseURI();
                 WADL2UDDI wadl2UDDI = new WADL2UDDI(null, new URLLocalizerDefaultImpl(), properties);
 
                 BusinessService businessServices = wadl2UDDI.createBusinessService(new QName(domain, domain), app);
 
                 Set<TModel> portTypeTModels = wadl2UDDI.createWADLPortTypeTModels(pathOrURL, app);
 
                 // Set<TModel> createWSDLBindingTModels = wadl2UDDI.createWSDLBindingTModels(wsdlURL, allBindings);
                 //When parsing a WSDL, there's really two things going on
                 //1) convert a bunch of stuff (the portTypes) to tModels
                 //2) convert the service definition to a BusinessService
                 //Since the service depends on the tModel, we have to save the tModels first
                 stm = new SaveTModel();
                 stm.setAuthInfo(token);
 
                 TModel[] tmodels = portTypeTModels.toArray(new TModel[0]);
                 for (int i = 0; i < tmodels.length; i++) {
                         System.out.println(tmodelPrinter.print(tmodels[i]));
                         stm.getTModel().add(tmodels[i]);
                 }
 
                 tmodels = wadl2UDDI.createWADLTModels(pathOrURL, app).toArray(new TModel[0]);
                 for (int i = 0; i < tmodels.length; i++) {
                         System.out.println(tmodelPrinter.print(tmodels[i]));
                         stm.getTModel().add(tmodels[i]);
                 }
                 //important, you'll need to save your new tModels, or else saving the business/service may fail
                 System.out.println("Saving the following tModels " + stm.getTModel().size());
                 publish.saveTModel(stm);
 
                 //finaly, we're ready to save all of the services defined in the WSDL
                 //again, we're creating a new business, if you have one already, look it up using the Inquiry getBusinessDetails
                 PrintUDDI<BusinessService> servicePrinter = new PrintUDDI<BusinessService>();
 
                 System.out.println("here's our new service: " + servicePrinter.print(businessServices));
 
                 if (businessKey == null || businessKey.length() == 0) {
                         BusinessEntity be = new BusinessEntity();
                         be.setBusinessKey(businessServices.getBusinessKey());
                         be.getName().add(new Name());
                         be.getName().get(0).setValue(domain);
                         be.getName().get(0).setLang("en");
                         be.setBusinessServices(new BusinessServices());
                         be.getBusinessServices().getBusinessService().add(businessServices);
                         SaveBusiness sb = new SaveBusiness();
                         sb.setAuthInfo(token);
                         sb.getBusinessEntity().add(be);
                         BusinessDetail saveBusiness = publish.saveBusiness(sb);
                         System.out.println("new business created, key = " + saveBusiness.getBusinessEntity().get(0).getBusinessKey());
                 }
                 SaveService ss = new SaveService();
                 ss.setAuthInfo(token);
                 businessServices.setBusinessKey(businessKey);
                 ss.getBusinessService().add(businessServices);
                 publish.saveService(ss);
                 System.out.println("Saved! " + businessServices.getServiceKey());
 
                 //and we're done
                 //Be sure to report any problems to the jUDDI JIRA bug tracker at 
                 //https://issues.apache.org/jira/browse/JUDDI
         }
 
         public static void main(String[] args) throws Exception {
 
                 new WadlImport().fire("http://svn.apache.org/repos/asf/cxf/trunk/systests/jaxrs/src/test/resources/wadl/bookstoreImportResource.wadl", null, null, null);
 
         }
 }