/*
 * Copyright 2011 by Graz University of Technology, Austria
 * MOCCA has been developed by the E-Government Innovation Center EGIZ, a joint
 * initiative of the Federal Chancellery Austria and Graz University of Technology.
 *
 * Licensed under the EUPL, Version 1.1 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:
 * http://www.osor.eu/eupl/
 *
 * 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 saml;

import java.io.File;
import java.io.FileNotFoundException;
import java.util.List;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBElement;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;
import javax.xml.bind.Unmarshaller;
import oasis.names.tc.saml._1_0.assertion.AnyType;
import oasis.names.tc.saml._1_0.assertion.AssertionType;
import oasis.names.tc.saml._1_0.assertion.AttributeStatementType;
import oasis.names.tc.saml._1_0.assertion.AttributeType;
import oasis.names.tc.saml._1_0.assertion.NameIdentifierType;
import oasis.names.tc.saml._1_0.assertion.ObjectFactory;
import oasis.names.tc.saml._1_0.assertion.StatementAbstractType;
import oasis.names.tc.saml._1_0.assertion.SubjectType;
import org.junit.Ignore;
import org.junit.Test;
import org.w3._2000._09.xmldsig_.RSAKeyValueType;
import org.w3c.dom.Element;

/**
 *
 * @author Clemens Orthacker <clemens.orthacker@iaik.tugraz.at>
 */
@Ignore
public class KeyValueAttributeTypeTest {

  @Test
  public void testAttrStatement() throws FileNotFoundException, JAXBException {
    JAXBContext ctx = JAXBContext.newInstance(ObjectFactory.class, org.w3._2000._09.xmldsig_.ObjectFactory.class);
    Unmarshaller um = ctx.createUnmarshaller();

    JAXBElement<?> assertion = (JAXBElement<?>) um.unmarshal(new File("/home/clemens/workspace/schema-base/src/main/schema/test/saml10.xml"));
    AssertionType value = (AssertionType) assertion.getValue();
    List<StatementAbstractType> statements = ((AssertionType) value).getStatementOrSubjectStatementOrAuthenticationStatement();
    for (StatementAbstractType stmt : statements) {
      if (stmt instanceof AttributeStatementType) {
        System.out.println("AttributeStatement");
        List<AttributeType> attrs = ((AttributeStatementType) stmt).getAttribute();
        for (AttributeType attr : attrs) {
          List<AnyType> attrValue = attr.getAttributeValue();
          System.out.println(attrValue.size() + " AttributeValue(s)");
          for (AnyType attrValueT : attrValue) {
            List<Object> attrValueContent = attrValueT.getContent();
            System.out.println("  AttributeValue: " + attrValueContent.size() + " child nodes");
            for (Object node : attrValueContent) {
              if (node instanceof String) {
                System.out.println("   - CDATA: " + node);
              } else if (node instanceof Element) {
                System.out.println("   - DOM Element: " + ((Element)node).getTagName());
              } else {
                System.out.println("   - " + node.getClass());
              }
            }
          }

        }
      }
    }
  }

  @Test
  public void testAttributeStatement() throws JAXBException {

    org.w3._2000._09.xmldsig_.ObjectFactory dsOF = new org.w3._2000._09.xmldsig_.ObjectFactory();
    RSAKeyValueType rsaKeyValueType = dsOF.createRSAKeyValueType();
    rsaKeyValueType.setExponent("1234".getBytes());
    rsaKeyValueType.setModulus("5678".getBytes());

    JAXBElement<RSAKeyValueType> rsaKeyValue = dsOF.createRSAKeyValue(rsaKeyValueType);


//    KeyValueType kvT = dsOF.createKeyValueType();
//    kvT.getContent().add(rsaKeyValue);
//    JAXBElement<KeyValueType> kv = dsOF.createKeyValue(kvT);

    ObjectFactory saml10OF = new ObjectFactory();
    AssertionType assertionT = saml10OF.createAssertionType();
    
    AttributeStatementType attrStatementT = saml10OF.createAttributeStatementType();
    NameIdentifierType nameIdT = saml10OF.createNameIdentifierType();
    nameIdT.setFormat("format");
    nameIdT.setNameQualifier("qualifier");
    nameIdT.setValue("value");
    JAXBElement<NameIdentifierType> subjNameId = saml10OF.createNameIdentifier(nameIdT);
    SubjectType subjT = saml10OF.createSubjectType();
    subjT.getContent().add(subjNameId);
    attrStatementT.setSubject(subjT);


    AttributeType attrT = saml10OF.createAttributeType();
//    QName keyVal = new QName("testNS", "keyVal");
    attrT.setAttributeName("RSAkeyvalue");
    attrT.setAttributeNamespace("lskdfjlk");
    AnyType attrValueT = saml10OF.createAnyType();
    attrValueT.getContent().add(rsaKeyValue);
    attrT.getAttributeValue().add(attrValueT); //kv); //keyValue); //new JAXBElement(keyVal, declaredType, attrT))
    attrStatementT.getAttribute().add(attrT);
    assertionT.getStatementOrSubjectStatementOrAuthenticationStatement().add(attrStatementT);
    JAXBElement<AssertionType> assertion = saml10OF.createAssertion(assertionT);

    JAXBContext ctx = JAXBContext.newInstance(saml10OF.getClass());
    Marshaller m = ctx.createMarshaller();
    m.marshal(assertion, System.out);
  }
}