/*
    * 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.GregorianCalendar;
  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.uddi.api_v3.*;
  import org.uddi.custody_v3.KeyBag;
  import org.uddi.custody_v3.TransferEntities;
  import org.uddi.custody_v3.TransferToken;
  import org.uddi.v3_service.UDDICustodyTransferPortType;
  import org.uddi.v3_service.UDDIInquiryPortType;
  import org.uddi.v3_service.UDDIPublicationPortType;
  import org.uddi.v3_service.UDDISecurityPortType;
  
  /**
   * This provides an example of how to transfer custody of a business from one
   * user to another on the same UDDI node. All child objects are also transfer
   *
   * @author <a href="mailto:alexoree@apache.org">Alex O'Ree</a>
   */
  public class UddiCustodyTransfer {
  
          private UDDISecurityPortType security = null;
          private UDDIPublicationPortType publish = null;
          private UDDIInquiryPortType uddiInquiryService = null;
          private UDDICustodyTransferPortType custodyTransferPortType = null;
  
          public UddiCustodyTransfer() {
                  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();
                          uddiInquiryService = transport.getUDDIInquiryService();
                          custodyTransferPortType = transport.getUDDICustodyTransferService();
                  } catch (Exception e) {
                          e.printStackTrace();
                  }
          }
  
          public void main(String args[]) throws Exception {
                  UddiCustodyTransfer sp = new UddiCustodyTransfer();
  
                  GetAuthToken getAuthTokenRoot = new GetAuthToken();
                  getAuthTokenRoot.setUserID("root");
                  getAuthTokenRoot.setCred("root");
  
                  // Making API call that retrieves the authentication token for the 'root' user.
                  AuthToken rootAuthToken = security.getAuthToken(getAuthTokenRoot);
                  System.out.println("root AUTHTOKEN = " + "don't log auth tokens!");
  
                  getAuthTokenRoot = new GetAuthToken();
                  getAuthTokenRoot.setUserID("uddi");
                  getAuthTokenRoot.setCred("uddi");
  
                  // Making API call that retrieves the authentication token for the 'root' user.
                  AuthToken uddiAuthToken = security.getAuthToken(getAuthTokenRoot);
                  System.out.println("uddi AUTHTOKEN = " + "don't log auth tokens!");
                  BusinessEntity biz = sp.createBusiness("uddi");
  
                  //save user uddi's business
                  SaveBusiness sb = new SaveBusiness();
                  sb.setAuthInfo(uddiAuthToken.getAuthInfo());
                  sb.getBusinessEntity().add(biz);
                  BusinessDetail saveBusiness = publish.saveBusiness(sb);
  
                  sp.transferBusiness(uddiAuthToken.getAuthInfo(), "uddi", rootAuthToken.getAuthInfo(), "root", saveBusiness.getBusinessEntity().get(0).getBusinessKey());
          }
  
          public void transferBusiness(String fromUser, String fromUserAuthToken, String toUser, String toUserAuthToken,
                  String BusinessKey) throws Exception {
  
                  System.out.println("Transfering business key " + BusinessKey);
                  GregorianCalendar gcal = new GregorianCalendar();
                  gcal.setTimeInMillis(System.currentTimeMillis());
         
                 //Create a transfer token from fromUser to toUser
                 KeyBag kb = new KeyBag();
                 kb.getKey().add(BusinessKey);
                 Holder<String> nodeidOUT = new Holder<String>();
                 Holder<XMLGregorianCalendar> expiresOUT = new Holder<XMLGregorianCalendar>();
                 Holder<byte[]> tokenOUT = new Holder<byte[]>();
                 custodyTransferPortType.getTransferToken(fromUserAuthToken, kb, nodeidOUT, expiresOUT, tokenOUT);
 
                 System.out.println("Transfer token obtained. Give this to user " + toUser);
                 System.out.println("Expires " + expiresOUT.value.toXMLFormat());
                 System.out.println("Node " + nodeidOUT.value);
                 System.out.println("Token " + org.apache.commons.codec.binary.Base64.encodeBase64String(tokenOUT.value));
 
                 if (toUser == null || toUser.length() == 0 || toUserAuthToken == null || toUserAuthToken.length() == 0) {
                         System.out.println("The toUser parameters are either null or empty, I can't complete the transfer here");
                         return;
                 }
 
                 //The magic part happens here, the user ROOT needs to give the user UDDI the token information out of band
                 //in practice, all values must match exactly
                 //UDDI now accepts the transfer
                 TransferEntities te = new TransferEntities();
                 te.setAuthInfo(toUserAuthToken);
                 te.setKeyBag(kb);
                 TransferToken tt = new TransferToken();
                 tt.setExpirationTime(expiresOUT.value);
                 tt.setNodeID(nodeidOUT.value);
                 tt.setOpaqueToken(tokenOUT.value);
                 te.setTransferToken(tt);
                 System.out.println("Excuting transfer...");
                 custodyTransferPortType.transferEntities(te);
                 System.out.println("Complete! verifing ownership change...");
 
                 //confirm the transfer
                 GetOperationalInfo go = new GetOperationalInfo();
                 go.setAuthInfo(fromUserAuthToken);
                 go.getEntityKey().add(BusinessKey);
                 OperationalInfos operationalInfo = uddiInquiryService.getOperationalInfo(go);
                 boolean ok = false;
                 boolean found = false;
                 for (int i = 0; i < operationalInfo.getOperationalInfo().size(); i++) {
                         if (operationalInfo.getOperationalInfo().get(i).getEntityKey().equalsIgnoreCase(BusinessKey)) {
                                 found = true;
                                 if (operationalInfo.getOperationalInfo().get(i).getAuthorizedName().equalsIgnoreCase(fromUser)) {
                                         System.out.println("Transfer unexpected failed");
                                 }
                                 if (operationalInfo.getOperationalInfo().get(i).getAuthorizedName().equalsIgnoreCase(toUser)) {
                                         ok = true;
                                 }
 
                         }
                 }
                 if (!found) {
                         System.out.println("Could get the operational info the transfed business");
                 }
                 System.out.println("Transfer " + (ok ? "success" : " failed"));
         }
 
         private BusinessEntity createBusiness(String user) {
                 BusinessEntity be = new BusinessEntity();
                 be.getName().add(new Name(user + "'s business", null));
                 return be;
         }
 
 }