/*
 * Copyright 2005-2014 the original author or authors.
 *
 * 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.springframework.ws.soap.axiom;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Locale;
import javax.xml.namespace.QName;

import org.apache.axiom.om.OMNamespace;
import org.apache.axiom.soap.SOAPFactory;
import org.apache.axiom.soap.SOAPFault;
import org.apache.axiom.soap.SOAPFaultCode;
import org.apache.axiom.soap.SOAPFaultNode;
import org.apache.axiom.soap.SOAPFaultReason;
import org.apache.axiom.soap.SOAPFaultSubCode;
import org.apache.axiom.soap.SOAPFaultText;
import org.apache.axiom.soap.SOAPFaultValue;
import org.apache.axiom.soap.SOAPProcessingException;

import org.springframework.util.StringUtils;
import org.springframework.ws.soap.axiom.support.AxiomUtils;
import org.springframework.ws.soap.soap12.Soap12Fault;

/** Axiom-specific version of {@code org.springframework.ws.soap.Soap12Fault}. */
class AxiomSoap12Fault extends AxiomSoapFault implements Soap12Fault {

	AxiomSoap12Fault(SOAPFault axiomFault, SOAPFactory axiomFactory) {
		super(axiomFault, axiomFactory);
	}

	@Override
	public QName getFaultCode() {
		return getAxiomFault().getCode().getValue().getTextAsQName();
	}

	@Override
	public Iterator<QName> getFaultSubcodes() {
		List<QName> subcodes = new ArrayList<QName>();
		SOAPFaultSubCode subcode = getAxiomFault().getCode().getSubCode();
		while (subcode != null) {
			subcodes.add(subcode.getValue().getTextAsQName());
			subcode = subcode.getSubCode();
		}
		return subcodes.iterator();
	}

	@Override
	public void addFaultSubcode(QName subcode) {
		SOAPFaultCode faultCode = getAxiomFault().getCode();
		SOAPFaultSubCode faultSubCode = null;
		if (faultCode.getSubCode() == null) {
			faultSubCode = getAxiomFactory().createSOAPFaultSubCode(faultCode);
		}
		else {
			faultSubCode = faultCode.getSubCode();
			while (true) {
				if (faultSubCode.getSubCode() != null) {
					faultSubCode = faultSubCode.getSubCode();
				}
				else {
					faultSubCode = getAxiomFactory().createSOAPFaultSubCode(faultSubCode);
					break;
				}
			}
		}
		SOAPFaultValue faultValue = getAxiomFactory().createSOAPFaultValue(faultSubCode);
		setValueText(subcode, faultValue);
	}

	private void setValueText(QName code, SOAPFaultValue faultValue) {
		String prefix = code.getPrefix();
		if (StringUtils.hasLength(code.getNamespaceURI()) && StringUtils.hasLength(prefix)) {
			OMNamespace namespace = getAxiomFault().findNamespaceURI(prefix);
			if (namespace == null) {
				getAxiomFault().declareNamespace(code.getNamespaceURI(), prefix);
			}
		}
		else if (StringUtils.hasLength(code.getNamespaceURI())) {
			OMNamespace namespace = getAxiomFault().findNamespace(code.getNamespaceURI(), null);
			if (namespace == null) {
				throw new IllegalArgumentException("Could not resolve namespace of code [" + code + "]");
			}
			code = new QName(code.getNamespaceURI(), code.getLocalPart(),
					namespace.getPrefix());
		}
		faultValue.setText(prefix + ":" + code.getLocalPart());
	}

	@Override
	public String getFaultNode() {
		SOAPFaultNode faultNode = getAxiomFault().getNode();
		if (faultNode == null) {
			return null;
		}
		else {
			return faultNode.getFaultNodeValue();
		}
	}

	@Override
	public void setFaultNode(String uri) {
		try {
			SOAPFaultNode faultNode = getAxiomFactory().createSOAPFaultNode(getAxiomFault());
			faultNode.setFaultNodeValue(uri);
			getAxiomFault().setNode(faultNode);
		}
		catch (SOAPProcessingException ex) {
			throw new AxiomSoapFaultException(ex);
		}
	}

	@Override
	public String getFaultStringOrReason() {
		return getFaultReasonText(Locale.getDefault());
	}

	@Override
	public String getFaultReasonText(Locale locale) {
		SOAPFaultReason faultReason = getAxiomFault().getReason();
		String language = AxiomUtils.toLanguage(locale);
		SOAPFaultText faultText = faultReason.getSOAPFaultText(language);
		return faultText != null ? faultText.getText() : null;
	}

	@Override
	public void setFaultReasonText(Locale locale, String text) {
		SOAPFaultReason faultReason = getAxiomFault().getReason();
		String language = AxiomUtils.toLanguage(locale);
		try {
			SOAPFaultText faultText = getAxiomFactory().createSOAPFaultText(faultReason);
			faultText.setLang(language);
			faultText.setText(text);
		}
		catch (SOAPProcessingException ex) {
			throw new AxiomSoapFaultException(ex);
		}
	}

}