/*
 * Copyright 2017 Graz University of Technology EAAF-Core Components has been developed in a
 * cooperation between EGIZ, A-SIT Plus, 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 at.gv.egiz.eaaf.core.api.idp.IAttributeGenerator;
import at.gv.egiz.eaaf.modules.pvp2.impl.utils.Saml2Utils;

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

public class SamlAttributeGenerator implements IAttributeGenerator<Attribute> {

  private XMLObject buildAttributeStringValue(final String value) {
    final XSStringBuilder stringBuilder =
        (XSStringBuilder) XMLObjectProviderRegistrySupport.getBuilderFactory().getBuilder(XSString.TYPE_NAME);
    final XSString stringValue =
        stringBuilder.buildObject(AttributeValue.DEFAULT_ELEMENT_NAME, XSString.TYPE_NAME);
    stringValue.setValue(value);
    return stringValue;
  }

  private XMLObject buildAttributeIntegerValue(final int value) {
    final XSIntegerBuilder integerBuilder =
        (XSIntegerBuilder) XMLObjectProviderRegistrySupport.getBuilderFactory().getBuilder(
            XSInteger.TYPE_NAME);
    final XSInteger integerValue =
        integerBuilder.buildObject(AttributeValue.DEFAULT_ELEMENT_NAME, XSInteger.TYPE_NAME);
    integerValue.setValue(value);
    return integerValue;
  }

  @Override
  public Attribute buildStringAttribute(final String friendlyName, final String name,
      final String value) {
    final Attribute attribute = Saml2Utils.createSamlObject(Attribute.class);
    attribute.setFriendlyName(friendlyName);
    attribute.setName(name);
    attribute.setNameFormat(Attribute.URI_REFERENCE);
    attribute.getAttributeValues().add(buildAttributeStringValue(value));
    return attribute;
  }

  @Override
  public Attribute buildIntegerAttribute(final String friendlyName, final String name,
      final int value) {
    final Attribute attribute = Saml2Utils.createSamlObject(Attribute.class);
    attribute.setFriendlyName(friendlyName);
    attribute.setName(name);
    attribute.setNameFormat(Attribute.URI_REFERENCE);
    attribute.getAttributeValues().add(buildAttributeIntegerValue(value));
    return attribute;
  }

  @Override
  public Attribute buildEmptyAttribute(final String friendlyName, final String name) {
    final Attribute attribute = Saml2Utils.createSamlObject(Attribute.class);
    attribute.setFriendlyName(friendlyName);
    attribute.setName(name);
    attribute.setNameFormat(Attribute.URI_REFERENCE);
    return attribute;
  }

  @Override
  public Attribute buildLongAttribute(final String friendlyName, final String name,
      final long value) {
    final 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;
  }

}