All Downloads are FREE. Search and download functionalities are using the official Maven repository.

se.litsec.opensaml.saml2.attribute.AttributeBuilder Maven / Gradle / Ivy

There is a newer version: 1.4.5
Show newest version
/*
 * The opensaml-ext project is an open-source package that extends OpenSAML
 * with useful extensions and utilities.
 *
 * More details on 
 * Copyright (C) 2017 Litsec AB
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see .
 */
package se.litsec.opensaml.saml2.attribute;

import java.io.InputStream;
import java.util.Arrays;
import java.util.List;

import javax.xml.namespace.QName;

import org.opensaml.core.xml.XMLObject;
import org.opensaml.core.xml.XMLObjectBuilder;
import org.opensaml.core.xml.XMLRuntimeException;
import org.opensaml.core.xml.io.MarshallingException;
import org.opensaml.core.xml.io.UnmarshallingException;
import org.opensaml.core.xml.schema.XSString;
import org.opensaml.saml.saml2.core.Attribute;
import org.opensaml.saml.saml2.core.AttributeValue;

import net.shibboleth.utilities.java.support.xml.XMLParserException;
import se.litsec.opensaml.core.AbstractSAMLObjectBuilder;
import se.litsec.opensaml.utils.ObjectUtils;

/**
 * Implements the build pattern to create {@link Attribute} objects.
 * 
 * @author Martin Lindström ([email protected])
 */
public class AttributeBuilder extends AbstractSAMLObjectBuilder {

  /** The default name format for the attribute being built. */
  public static final String DEFAULT_NAME_FORMAT = Attribute.URI_REFERENCE;

  /**
   * Constructor setting the attribute name.
   * 
   * @param name
   *          the attribute name
   */
  public AttributeBuilder(String name) {
    super();
    if (name == null) {
      throw new IllegalArgumentException("name must not be null");
    }
    this.object().setName(name);
  }

  /**
   * Static utility method that creates a default {@code AttributeBuilder}.
   * 
   * @param name
   *          the attribute name
   * @return an {@code AttributeBuilder} instance.
   * @see #AttributeBuilder(String)
   */
  public static AttributeBuilder builder(String name) {
    return new AttributeBuilder(name);
  }

  /**
   * Constructor setting up the builder given an attribute template.
   * 
   * @param template
   *          the attribute template
   * @throws UnmarshallingException
   *           for unmarshalling errors
   * @throws MarshallingException
   *           for marshalling errors
   * @see AbstractSAMLObjectBuilder#AbstractSAMLObjectBuilder(org.opensaml.saml.common.SAMLObject)
   */
  public AttributeBuilder(Attribute template) throws MarshallingException, UnmarshallingException {
    super(template);
  }

  /**
   * Static utility method that creates a {@code AttributeBuilder} given a template attribute.
   * 
   * @param attribute
   *          the attribute template
   * @return an attribute builder
   * @throws UnmarshallingException
   *           for unmarshalling errors
   * @throws MarshallingException
   *           for marshalling errors
   */
  public static AttributeBuilder builder(Attribute attribute) throws MarshallingException, UnmarshallingException {
    return new AttributeBuilder(attribute);
  }

  /**
   * Constructor setting up the builder with a template attribute that is read from an input stream.
   * 
   * @param resource
   *          the attribute template
   * @throws UnmarshallingException
   *           for unmarshalling errors
   * @throws XMLParserException
   *           for XML parsing errors
   * @see AbstractSAMLObjectBuilder#AbstractSAMLObjectBuilder(InputStream)
   */
  public AttributeBuilder(InputStream resource) throws XMLParserException, UnmarshallingException {
    super(resource);
  }

  /**
   * Static utility method that creates a {@code AttributeBuilder} given a template attribute read from an input stream.
   * 
   * @param resource
   *          the attribute template
   * @return an attribute builder
   * @throws UnmarshallingException
   *           for unmarshalling errors
   * @throws XMLParserException
   *           for XML parsing errors
   */
  public static AttributeBuilder builder(InputStream resource) throws XMLParserException, UnmarshallingException {
    return new AttributeBuilder(resource);
  }

  /**
   * Returns the {@link Attribute} object that has been built. If the {@code NameFormat} attribute has not been
   * assigned, the method will inject the default value ({@value #DEFAULT_NAME_FORMAT}).
   */
  @Override
  public Attribute build() {
    if (this.object().getName() == null) {
      throw new RuntimeException("Attribute can not be built without a name");
    }
    if (this.object().getNameFormat() == null) {
      this.object().setNameFormat(DEFAULT_NAME_FORMAT);
    }
    return super.build();
  }

  /**
   * Assigns the attribute name.
   * 
   * @param name
   *          the attribute name
   * @return the builder
   */
  public AttributeBuilder name(String name) {
    this.object().setName(name);
    return this;
  }

  /**
   * Assigns the attribute friendly name.
   * 
   * @param friendlyName
   *          the friendly name
   * @return the builder
   */
  public AttributeBuilder friendlyName(String friendlyName) {
    this.object().setFriendlyName(friendlyName);
    return this;
  }

  /**
   * Assigns the attribute name format.
   * 
   * @param nameFormat
   *          the name format URI
   * @return the builder
   */
  public AttributeBuilder nameFormat(String nameFormat) {
    this.object().setNameFormat(nameFormat);
    return this;
  }

  /**
   * Assigns one (or more) attribute string values.
   * 
   * 

* Note: if {@code null} is passed as a parameter, any previous attribute values are cleared. *

* * @param values * the string value(s) to add * @return the builder */ public AttributeBuilder value(String... values) { return this.value(values != null ? Arrays.asList(values) : null); } /** * @see #value(String...) * * @param values * the string value(s) to add * @return the builder */ public AttributeBuilder value(List values) { if (values == null) { this.object().getAttributeValues().clear(); return this; } for (String s : values) { XSString sv = createValueObject(XSString.TYPE_NAME, XSString.class); sv.setValue(s); this.object().getAttributeValues().add(sv); } return this; } /** * Assigns an attribute value. * * @param value * the value to add * @param * the value type * @return the builder */ public AttributeBuilder value(T value) { this.object().getAttributeValues().add(value); return this; } /** * Creates an {@code AttributeValue} object of the given class. The type of the attribute value will be the field that * is declared as {@code TYPE_NAME} of the given class. *

* After the object has been constructed, its setter methods should be called to setup the value object before adding * it to the attribute itself. *

*

* Note: For attribute having string values, there is no need to explictly create an attribute value. Instead the * {@link #value(String...)} method may be used directly. *

* * @param * the type * @param clazz * the type of attribute value * @return the attribute value * @see #createValueObject(QName, Class) */ public static T createValueObject(Class clazz) { try { QName schemaType = (QName) clazz.getDeclaredField("TYPE_NAME").get(null); return createValueObject(schemaType, clazz); } catch (NoSuchFieldException | IllegalArgumentException | IllegalAccessException | SecurityException e) { throw new XMLRuntimeException(e); } } /** * Creates an {@code AttributeValue} object of the given class and schema type. *

* After the object has been constructed, its setter methods should be called to setup the value object before adding * it to the attribute itself. *

*

* Note: For attribute having string values, there is no need to explictly create an attribute value. Instead the * {@link #value(String...)} method may be used directly. *

* * @param * the type * @param schemaType * the schema type that should be assigned to the attribute value, i.e., {@code xsi:type="ns:ValueType"} * @param clazz * the type of the attribute value * @return the attribute value * @see #createValueObject(Class) */ public static T createValueObject(QName schemaType, Class clazz) { XMLObjectBuilder builder = ObjectUtils.getBuilder(schemaType); return builder.buildObject(AttributeValue.DEFAULT_ELEMENT_NAME, schemaType); } /** {@inheritDoc} */ @Override protected Class getObjectType() { return Attribute.class; } }




© 2015 - 2024 Weber Informatics LLC | Privacy Policy