/*
    * 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.concurrent.atomic.AtomicReference;
  
  import org.apache.juddi.v3.client.config.UDDIClient;
  import org.apache.juddi.v3.client.cryptor.DigSigUtil;
  import org.apache.juddi.v3.client.transport.Transport;
  import org.uddi.api_v3.*;
  import org.uddi.v3_service.UDDIInquiryPortType;
  import org.uddi.v3_service.UDDIPublicationPortType;
  import org.uddi.v3_service.UDDISecurityPortType;
  
  /**
   * This class shows you how to digital sign a business
   *
   * @author <a href="mailto:alexoree@apache.org">Alex O'Ree</a>
   */
  public class UddiDigitalSignatureBusiness {
  
          private UDDISecurityPortType security = null;
          private UDDIInquiryPortType inquiry = null;
          private UDDIPublicationPortType publish = null;
          private UDDIClient clerkManager = null;
  
          /**
           * This sets up the ws proxies using uddi.xml in META-INF
           */
          public UddiDigitalSignatureBusiness() {
                  try {
                          // create a manager and read the config in the archive; 
                          // you can use your config file name
                          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();
                          inquiry = transport.getUDDIInquiryService();
                          publish = transport.getUDDIPublishService();
                  } catch (Exception e) {
                          e.printStackTrace();
                  }
          }
          
          public UddiDigitalSignatureBusiness(Transport transport) {
                  try {
                         
                          // Now you create a reference to the UDDI API
                          security = transport.getUDDISecurityService();
                          inquiry = transport.getUDDIInquiryService();
                          publish = transport.getUDDIPublishService();
                  } catch (Exception e) {
                          e.printStackTrace();
                  }
          }
  
          /**
           * Main entry point
           *
           * @param args
           */
          public static void main(String args[]) {
  
                  UddiDigitalSignatureBusiness sp = new UddiDigitalSignatureBusiness();
                  sp.fire(null, null);
          }
  
          public void fire(String token, String key) {
                  try {
  
                          DigSigUtil ds = null;
  
                          //option 1), set everything manually
                          ds = new DigSigUtil();
                          ds.put(DigSigUtil.SIGNATURE_KEYSTORE_FILE, "keystore.jks");
                          ds.put(DigSigUtil.SIGNATURE_KEYSTORE_FILETYPE, "JKS");
                          ds.put(DigSigUtil.SIGNATURE_KEYSTORE_FILE_PASSWORD, "Test");
                          ds.put(DigSigUtil.SIGNATURE_KEYSTORE_KEY_ALIAS, "Test");
                          ds.put(DigSigUtil.SIGNATURE_OPTION_CERT_INCLUSION_BASE64, "true");
  
                          ds.put(DigSigUtil.SIGNATURE_OPTION_CERT_INCLUSION_SERIAL, "true");
                          ds.put(DigSigUtil.SIGNATURE_OPTION_CERT_INCLUSION_SUBJECTDN, "true");
                          ds.put(DigSigUtil.TRUSTSTORE_FILE, "truststore.jks");
                          ds.put(DigSigUtil.TRUSTSTORE_FILETYPE, "JKS");
                          ds.put(DigSigUtil.TRUSTSTORE_FILE_PASSWORD, "Test");
 
                         //option 2), load it from the juddi config file
                         //ds = new DigSigUtil(clerkManager.getClientConfig().getDigitalSignatureConfiguration());
                         //login
                         if (token == null) //option, load from juddi config
                         {
                                 token = getAuthKey(clerkManager.getClerk("default").getPublisher(),
                                         clerkManager.getClerk("default").getPassword());
                         }
 
                         if (key == null) {
                                 //make a new business
                                 SaveBusiness sb = new SaveBusiness();
                                 sb.setAuthInfo(token);
                                 BusinessEntity ob = new BusinessEntity();
                                 Name name = new Name();
                                 name.setValue("My Signed Business");
                                 ob.getName().add(name);
                                 sb.getBusinessEntity().add(ob);
                                 //save it
                                 BusinessDetail saveBusiness = publish.saveBusiness(sb);
 
                                 System.out.println("business created with key " + saveBusiness.getBusinessEntity().get(0).getBusinessKey());
 
                                 BusinessEntity be = saveBusiness.getBusinessEntity().get(0);
                                 key = be.getBusinessKey();
                         }
                         BusinessEntity be = clerkManager.getClerk("default").getBusinessDetail(key);
                         //sign the copy returned from the UDDI node (it may have made changes)
                         DigSigUtil.JAXB_ToStdOut(be);
                         if (!be.getSignature().isEmpty())
                         {
                                 System.out.println("WARN, the entity with the key " + key + " is already signed! aborting");
                                 return;
                         }
 
                         //if it's already signed, remove all existing signatures
                         
                         System.out.println("signing");
                         BusinessEntity signUDDI_JAXBObject = ds.signUddiEntity(be);
                         DigSigUtil.JAXB_ToStdOut(signUDDI_JAXBObject);
                         System.out.println("signed, saving");
 
                         SaveBusiness sb = new SaveBusiness();
                         sb.setAuthInfo(token);
                         sb.getBusinessEntity().add(signUDDI_JAXBObject);
                         publish.saveBusiness(sb);
                         System.out.println("saved, fetching");
 
                         //validate it again from the server, confirming that it was transformed correctly
                         GetBusinessDetail gb = new GetBusinessDetail();
                         gb.setAuthInfo(token);
                         gb.getBusinessKey().add(be.getBusinessKey());
                         be = inquiry.getBusinessDetail(gb).getBusinessEntity().get(0);
                         DigSigUtil.JAXB_ToStdOut(be);
                         System.out.println("verifing");
                         AtomicReference<String> msg = new AtomicReference<String>();
                         boolean verifySigned_UDDI_JAXB_Object = ds.verifySignedUddiEntity(be, msg);
                         if (verifySigned_UDDI_JAXB_Object) {
                                 System.out.println("signature validation passed (expected)");
                         } else {
                                 System.out.println("signature validation failed (not expected)");
                         }
                         System.out.println(msg.get());
 
                 } catch (Exception e) {
                         e.printStackTrace();
                 }
         }
 
         /**
          * Gets a UDDI style auth token, otherwise, appends credentials to the
          * ws proxies (not yet implemented)
          *
          * @param username
          * @param password
          * @param style
          * @return
          */
         private String getAuthKey(String username, String password) {
                 try {
 
                         GetAuthToken getAuthTokenRoot = new GetAuthToken();
                         getAuthTokenRoot.setUserID(username);
                         getAuthTokenRoot.setCred(password);
 
                         // 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!");
                         return rootAuthToken.getAuthInfo();
                 } catch (Exception ex) {
                         System.out.println("Could not authenticate with the provided credentials " + ex.getMessage());
                 }
                 return null;
         }
 }