/*
    * 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.util.ArrayList;
  import java.util.GregorianCalendar;
  import java.util.List;
  import javax.xml.datatype.DatatypeFactory;
  import javax.xml.datatype.XMLGregorianCalendar;
  import javax.xml.ws.Holder;
  import org.apache.juddi.v3.client.config.UDDIClient;
  import org.apache.juddi.v3.client.transport.Transport;
  import org.apache.juddi.v3_service.JUDDIApiPortType;
  import org.uddi.api_v3.*;
  import org.uddi.v3_service.UDDIPublicationPortType;
  import org.uddi.v3_service.UDDISecurityPortType;
  
  /**
   * This will create two businesses under different users, then setup a
   * relationship between the two
   *
   * @author <a href="mailto:alexoree@apache.org">Alex O'Ree</a>
   */
  public class UddiRelatedBusinesses {
  
          private UDDISecurityPortType security = null;
          private UDDIPublicationPortType publish = null;
  
          public UddiRelatedBusinesses() {
                  try {
                          // 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 transport = clerkManager.getTransport();
                          // Now you create a reference to the UDDI API
                          security = transport.getUDDISecurityService();
                          publish = transport.getUDDIPublishService();
                  } catch (Exception e) {
                          e.printStackTrace();
                  }
          }
  
          public void fire(String businessKey, String authInfo, String businessKey1, String authInfo1, String relationship) throws Exception {
                  try {
  
                          GregorianCalendar gcal = new GregorianCalendar();
                          gcal.setTimeInMillis(System.currentTimeMillis());
                          //ROOT creates half of the relationship
                          //create a business relationship (publisher assertion)
                          Holder<List<PublisherAssertion>> x = new Holder<List<PublisherAssertion>>();
                          PublisherAssertion pa = new PublisherAssertion();
                          pa.setFromKey(businessKey);
                          pa.setToKey(businessKey1);
                          pa.setKeyedReference(new KeyedReference());
                          pa.getKeyedReference().setKeyName("Subsidiary");
                          pa.getKeyedReference().setKeyValue(relationship);
  
                          pa.getKeyedReference().setTModelKey("uddi:uddi.org:relationships");
                          x.value = new ArrayList<PublisherAssertion>();
                          x.value.add(pa);
                          publish.setPublisherAssertions(authInfo, x);
  
                          //now "UDDI" the user, creates the other half. It has to be mirrored exactly
                          x = new Holder<List<PublisherAssertion>>();
                          pa = new PublisherAssertion();
                          pa.setFromKey(businessKey);
                          pa.setToKey(businessKey1);
                          pa.setKeyedReference(new KeyedReference());
                          pa.getKeyedReference().setKeyName("Subsidiary");
                          pa.getKeyedReference().setKeyValue(relationship);
  
                          pa.getKeyedReference().setTModelKey("uddi:uddi.org:relationships");
                          x.value = new ArrayList<PublisherAssertion>();
                          x.value.add(pa);
                          publish.setPublisherAssertions(authInfo1, x);
  
                          System.out.println("Success!");
                          /*
                           * Here's some notes:
                           * You can use
                           * List<AssertionStatusItem> assertionStatusReport = publish.getAssertionStatusReport(UDDIAuthToken.getAuthInfo(), CompletionStatus.STATUS_FROM_KEY_INCOMPLETE);
                           * to determine if there's any assertions/relationships requests that are pending
                           * this should have one item it in, the relationship that's incomplete
                           * 
                           * There's also publish.deletePublisherAssertions();
                          */
                 } catch (Exception e) {
                         e.printStackTrace();
                 }
         }
 
         public static void main(String args[]) throws Exception {
                 UddiRelatedBusinesses sp = new UddiRelatedBusinesses();
 
                 GetAuthToken getAuthTokenRoot = new GetAuthToken();
                 getAuthTokenRoot.setUserID("root");
                 getAuthTokenRoot.setCred("root");
 
                 // Making API call that retrieves the authentication token for the 'root' user.
                 AuthToken rootAuthToken = sp.security.getAuthToken(getAuthTokenRoot);
                 System.out.println("root AUTHTOKEN = " + "don't log auth tokens!");
                 BusinessEntity rootbiz = sp.createBusiness("root");
 
                 getAuthTokenRoot = new GetAuthToken();
                 getAuthTokenRoot.setUserID("uddi");
                 getAuthTokenRoot.setCred("uddi");
 
                 // Making API call that retrieves the authentication token for the 'root' user.
                 AuthToken uddiAuthToken = sp.security.getAuthToken(getAuthTokenRoot);
                 System.out.println("uddi AUTHTOKEN = " + "don't log auth tokens!");
                 BusinessEntity uddibiz = sp.createBusiness("uddi");
 
                 //save user uddi's business
                 SaveBusiness sb = new SaveBusiness();
                 sb.setAuthInfo(uddiAuthToken.getAuthInfo());
                 sb.getBusinessEntity().add(uddibiz);
                 BusinessDetail uddibize = sp.publish.saveBusiness(sb);
 
                 sb = new SaveBusiness();
                 sb.setAuthInfo(rootAuthToken.getAuthInfo());
                 sb.getBusinessEntity().add(rootbiz);
                 BusinessDetail rootbize = sp.publish.saveBusiness(sb);
 
                 sp.fire(rootbize.getBusinessEntity().get(0).getBusinessKey(), rootAuthToken.getAuthInfo(),
                         uddibize.getBusinessEntity().get(0).getBusinessKey(), uddiAuthToken.getAuthInfo(),
                         "parent-child");
         }
 
         private BusinessEntity createBusiness(String user) {
                 BusinessEntity be = new BusinessEntity();
                 be.getName().add(new Name(user + "'s business", null));
                 return be;
         }
 
 }