package com.sun.identity.saml2.protocol.impl;

import com.sun.identity.saml2.common.SAML2Exception;
import com.sun.identity.saml2.common.SAML2SDKUtils;
import com.sun.identity.saml2.protocol.Extensions;
import com.sun.identity.shared.xml.XMLUtils;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;

public class ExtensionsImpl implements Extensions {
   private boolean isMutable;
   private List<String> extensionsList;

   public ExtensionsImpl() {
      this.isMutable = true;
   }

   public ExtensionsImpl(Element element) {
      this.parseElement(element);
   }

   @Override
   public void setAny(List<String> value) throws SAML2Exception {
      if (this.isMutable) {
         this.extensionsList = value;
      } else {
         throw new SAML2Exception(SAML2SDKUtils.BUNDLE.getString("objectImmutable"));
      }
   }

   @Override
   public List<String> getAny() {
      return this.extensionsList;
   }

   @Override
   public String toXMLString() throws SAML2Exception {
      return this.toXMLString(true, false);
   }

   @Override
   public String toXMLString(boolean includeNSPrefix, boolean declareNS) throws SAML2Exception {
      String xmlStr = null;
      if (this.extensionsList != null && !this.extensionsList.isEmpty()) {
         StringBuilder xmlString = new StringBuilder(500);
         xmlString.append("<");
         if (includeNSPrefix) {
            xmlString.append("samlp:");
         }

         xmlString.append("Extensions");
         if (declareNS) {
            xmlString.append(" xmlns:samlp=\"urn:oasis:names:tc:SAML:2.0:protocol\"");
         }

         xmlString.append(">");

         for (String extString : this.extensionsList) {
            xmlString.append("\n").append(extString);
         }

         xmlString.append("\n").append("</samlp:").append("Extensions").append(">");
         xmlStr = xmlString.toString();
      }

      return xmlStr;
   }

   @Override
   public void makeImmutable() {
      if (this.isMutable) {
         this.isMutable = false;
      }
   }

   @Override
   public boolean isMutable() {
      return this.isMutable;
   }

   private void parseElement(Element element) {
      NodeList nList = element.getChildNodes();
      if (this.extensionsList == null || this.extensionsList.isEmpty()) {
         this.extensionsList = new ArrayList<>();
      }

      if (nList != null && nList.getLength() > 0) {
         for (int i = 0; i < nList.getLength(); i++) {
            Node childNode = nList.item(i);
            if (childNode.getLocalName() != null) {
               this.extensionsList.add(XMLUtils.print(childNode));
            }
         }

         if (this.extensionsList != null && !this.extensionsList.isEmpty()) {
            this.extensionsList = Collections.unmodifiableList(this.extensionsList);
         }
      }
   }
}