package com.bala.custom.wih;

import java.util.ArrayList;
import java.util.List;

import javax.xml.bind.JAXBException;
import javax.xml.namespace.QName;

import org.apache.cxf.endpoint.Client;
import org.apache.cxf.headers.Header;
import org.apache.cxf.jaxb.JAXBDataBinding;
import org.jbpm.process.workitem.webservice.WebServiceWorkItemHandler;
import org.kie.api.runtime.KieSession;
import org.kie.api.runtime.process.WorkItem;

/**
 *
 * @author bbalasub
 *
 *
 *         Custom Work Item Handler to set the custom SOAP headers to the
 *         request context
 */
public class CustomWSWIH extends WebServiceWorkItemHandler {

	public CustomWSWIH(KieSession ksession, ClassLoader classloader) {
		super(ksession, classloader);
		System.out.println("Invoked the constructor in custome ws wih....");
	}

	@Override
	protected Client getWSClient(WorkItem workItem, String interfaceRef) {
		Client client = super.getWSClient(workItem, interfaceRef);

		Object headerToken = workItem.getParameter("Header");
		System.out.println("The value passed in for header is " + headerToken);

		List<Header> headers = new ArrayList<Header>();
		Header dummyHeader = null;
		try {
			dummyHeader = new Header(new QName("uri:org.apache.cxf", "dummy"), headerToken,
					new JAXBDataBinding(String.class));
		} catch (JAXBException e) {
			e.printStackTrace();
		}
		headers.add(dummyHeader);

		client.getRequestContext().put(Header.HEADER_LIST, headers);

		return client;
	}
}