FaultException.java

// Copyright (C) 2012 jOVAL.org.  All rights reserved.
// This software is licensed under the AGPL 3.0 license available at http://www.joval.org/agpl_v3.txt

package jwsmv.wsman;

import javax.xml.bind.JAXBElement;

import com.microsoft.wsman.fault.WSManFaultType;
import com.microsoft.wsman.fault.ProviderFaultType;
import org.w3c.soap.envelope.Detail;
import org.w3c.soap.envelope.Fault;

/**
 * A Web-Service (SOAP) Fault-derived exception class
 *
 * @author David A. Solin
 * @version %I% %G%
 */
public class FaultException extends Exception {
    Fault fault;

    public FaultException(Fault fault) {
	super();
	this.fault = fault;
    }

    public Fault getFault() {
	return fault;
    }

    /**
     * Extract the (required) detail message from the fault.
     */
    @Override
    public String getMessage() {
	if (fault.isSetDetail() && fault.getDetail().isSetAny()) {
	    StringBuffer sb = new StringBuffer();
	    for (Object obj : fault.getDetail().getAny()) {
		if (obj instanceof JAXBElement) {
		    obj = ((JAXBElement)obj).getValue();
		}
		if (obj instanceof WSManFaultType) {
		    return toString((WSManFaultType)obj);
		} else if (obj != null) {
		    sb.append(obj.toString());
		}
	    }
	    return sb.toString();
	} else {
	    return null;
	}
    }

    // Private

    private String toString(WSManFaultType wsmft) {
	StringBuffer sb = new StringBuffer();
	if (wsmft.isSetMessage() && wsmft.getMessage().isSetContent()) {
	    for (Object obj : wsmft.getMessage().getContent()) {
		if (obj instanceof JAXBElement) {
		    obj = ((JAXBElement)obj).getValue();
		}
		if (obj instanceof ProviderFaultType) {
		    ProviderFaultType pft = (ProviderFaultType)obj;
		    if (pft.isSetProviderId()) {
			sb.append("Provider: ").append(pft.getProviderId());
		    }
		    if (pft.isSetContent()) {
			for (Object content : pft.getContent()) {
			    if (content instanceof JAXBElement) {
				content = ((JAXBElement)content).getValue();
			    }
			    if (content instanceof WSManFaultType) {
				sb.append(toString((WSManFaultType)content));
			    }
			}
		    }
		} else if (obj != null) {
		    sb.append(obj.toString().trim());
		}
	    }
	}
	if (wsmft.isSetCode()) {
	    String code = new StringBuffer("Code: ").append(Long.toString(wsmft.getCode())).toString();
	    if (sb.length() == 0) {
		sb.append(code);
	    } else if (sb.indexOf(code) == -1) {
		sb.append(" [").append(code).append("]");
	    }
	}
	return sb.toString();
    }
}