/*******************************************************************************
 * Copyright 2017 Graz University of Technology
 * EAAF-Core Components has been developed in a cooperation between EGIZ,  
 * A-SIT+, A-SIT, and Graz University of Technology.
 *
 * Licensed under the EUPL, Version 1.2 or - as soon they will be approved by
 * the European Commission - subsequent versions of the EUPL (the "Licence");
 * You may not use this work except in compliance with the Licence.
 * You may obtain a copy of the Licence at:
 * https://joinup.ec.europa.eu/news/understanding-eupl-v12
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the Licence is distributed on an "AS IS" basis,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the Licence for the specific language governing permissions and
 * limitations under the Licence.
 *
 * This product combines work with different licenses. See the "NOTICE" text
 * file for details on the various modules and licenses.
 * The "NOTICE" text file is part of the distribution. Any derivative works
 * that you distribute must include a readable copy of the "NOTICE" text file.
 *******************************************************************************/
/*******************************************************************************
 *******************************************************************************/
/*******************************************************************************
 *******************************************************************************/
package at.gv.egiz.eaaf.modules.pvp2.impl.builder;

import org.opensaml.saml2.core.Attribute;
import org.opensaml.saml2.core.AttributeValue;
import org.opensaml.xml.Configuration;
import org.opensaml.xml.XMLObject;
import org.opensaml.xml.schema.XSInteger;
import org.opensaml.xml.schema.XSString;
import org.opensaml.xml.schema.impl.XSIntegerBuilder;
import org.opensaml.xml.schema.impl.XSStringBuilder;

import at.gv.egiz.eaaf.core.api.idp.IAttributeGenerator;
import at.gv.egiz.eaaf.modules.pvp2.impl.utils.SAML2Utils;

public class SamlAttributeGenerator implements IAttributeGenerator<Attribute> {
	
	private XMLObject buildAttributeStringValue(String value) {
		XSStringBuilder stringBuilder = (XSStringBuilder) Configuration.getBuilderFactory().getBuilder(XSString.TYPE_NAME);
		XSString stringValue = stringBuilder.buildObject(AttributeValue.DEFAULT_ELEMENT_NAME, XSString.TYPE_NAME);
		stringValue.setValue(value);
		return stringValue;
	}
	
	private XMLObject buildAttributeIntegerValue(int value) {
		XSIntegerBuilder integerBuilder = (XSIntegerBuilder) Configuration.getBuilderFactory().getBuilder(XSInteger.TYPE_NAME);
		XSInteger integerValue = integerBuilder.buildObject(AttributeValue.DEFAULT_ELEMENT_NAME, XSInteger.TYPE_NAME);
		integerValue.setValue(value);
		return integerValue;
	}
	
	public Attribute buildStringAttribute(final String friendlyName, final String name, final String value) {
		Attribute attribute = SAML2Utils.createSAMLObject(Attribute.class);
		attribute.setFriendlyName(friendlyName);
		attribute.setName(name);
		attribute.setNameFormat(Attribute.URI_REFERENCE);
		attribute.getAttributeValues().add(buildAttributeStringValue(value));
		return attribute;
	}
	
	public Attribute buildIntegerAttribute(final String friendlyName, final String name, final int value) {
		Attribute attribute = SAML2Utils.createSAMLObject(Attribute.class);
		attribute.setFriendlyName(friendlyName);
		attribute.setName(name);
		attribute.setNameFormat(Attribute.URI_REFERENCE);
		attribute.getAttributeValues().add(buildAttributeIntegerValue(value));
		return attribute;
	}
	
	public Attribute buildEmptyAttribute(final String friendlyName, final String name) {
		Attribute attribute = SAML2Utils.createSAMLObject(Attribute.class);
		attribute.setFriendlyName(friendlyName);
		attribute.setName(name);
		attribute.setNameFormat(Attribute.URI_REFERENCE);
		return attribute;
	}

	public Attribute buildLongAttribute(String friendlyName, String name, long value) {
		Attribute attribute = SAML2Utils.createSAMLObject(Attribute.class);
		attribute.setFriendlyName(friendlyName);
		attribute.setName(name);
		attribute.setNameFormat(Attribute.URI_REFERENCE);
		attribute.getAttributeValues().add(buildAttributeIntegerValue((int) value));
		return attribute;
	}
	
}