/*
 * Created on Aug 29, 2004
 *
 * TODO To change the template for this generated file go to
 * Window - Preferences - Java - Code Style - Code Templates
 */
package org.apache.ws.sandbox.axis.security.trust.service;

import java.net.URL;
import java.util.Enumeration;
import java.util.Hashtable;
import java.util.Properties;

import javax.xml.soap.SOAPHeader;

import org.apache.axis.AxisFault;
import org.apache.axis.Message;
import org.apache.axis.MessageContext;
import org.apache.axis.SOAPPart;
import org.apache.axis.message.SOAPEnvelope;
import org.apache.axis.utils.ByteArrayOutputStream;
import org.apache.ws.sandbox.security.trust.STSManager;
import org.apache.ws.security.util.Loader;
import org.apache.xml.security.utils.XMLUtils;
import org.w3c.dom.Document;

/**
 * @author Ruchith
 *
 * TODO To change the template for this generated type comment go to
 * Window - Preferences - Java - Code Style - Code Templates
 */
public class SecurityTokenService {

	protected Hashtable options;
	
	public SecurityTokenService() {
		this.loadProperties("STS.properties");
	}
	
	public void requestSecurityToken(org.apache.ws.sandbox.axis.security.trust.service.RequestSecurityTokenType request) throws java.rmi.RemoteException{
    	try{
    		MessageContext msgCntxt = MessageContext.getCurrentContext();
	    	msgCntxt.getRequestMessage();
	    	
	        SOAPHeader sHeader = null;
	        //get the request msg    
	        Message smReq = msgCntxt.getRequestMessage();
	        //get the response msg
	       // Message smCurr = msgCntxt.getCurrentMessage();
		   Message smCurr = msgCntxt.getResponseMessage();//.getRequestMessage();
	        //get the request msg as a SOAP part
	        SOAPPart sPartReq = (org.apache.axis.SOAPPart) smReq.getSOAPPart();
	        //get the response msg as a SOAP part
	        SOAPPart sPartRes = (org.apache.axis.SOAPPart) smCurr.getSOAPPart();

	        Document docReq, docRes = null;

	        try {
	            //initialize xml security
	            org.apache.xml.security.Init.init();
	            docReq = ((SOAPEnvelope) sPartReq.getEnvelope()).getAsDocument();
	            docRes = ((SOAPEnvelope) sPartRes.getEnvelope()).getAsDocument();

	            STSManager stsMgr =
	                    new STSManager(this.options);
	            docRes = stsMgr.handleRequest(docReq, docRes);
//	            log.debug("STSServerHandler: STSManager has done the job");
	            ByteArrayOutputStream os = new ByteArrayOutputStream();
	            XMLUtils.outputDOM(docRes, os, true); 
	            //modify the current message
	            sPartRes.setCurrentMessage(os.toByteArray(), SOAPPart.FORM_BYTES);   
	            //set current message to the context
				//msgCntxt.setCurrentMessage(sPartRes.getMessage());
				//msgCntxt.setPastPivot(true);
				//msgCntxt.setPastPivot(true);
				msgCntxt.setCurrentMessage(sPartRes.getMessage());
	          // msgCntxt.setResponseMessage(sPartRes.getMessage());
			 

	        } catch (Exception ex) {
	            throw new AxisFault("STSServerHandler-dorequest:Response failed due to a problem in issuence process",
	                    ex);
	        }

    	} catch (Exception e) {
    		System.out.println("Exception is "+e.getMessage());
    		throw new AxisFault();
    	}
    }

    protected void loadProperties(String propFilename) {
        Properties properties = new Properties();
        try {
            URL url = Loader.getResource(propFilename);
            properties.load(url.openStream());
        } catch (Exception e) {
            throw new RuntimeException("SecurityTokenSErvice: Cannot load properties: " + propFilename);
        }
        this.options = new Hashtable();
        Enumeration enumKeys = properties.keys();
        while(enumKeys.hasMoreElements()) {
        	String key = (String)enumKeys.nextElement();
        	this.options.put(key,properties.getProperty(key));
        }
    }

}