WsEndpointDDModifier.java

package org.objectweb.jonas_ws.wsgen.ddmodifier;

import org.objectweb.util.monolog.api.BasicLevel;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.w3c.dom.Text;

public class WsEndpointDDModifier extends DeploymentDescModifier {

    private static final String SERVLET = "servlet";

    private static final String SERVLET_NAME = "servlet-name";

    private static final String SERVLET_CLASS = "servlet-class";

    private static final String SERVLET_MAPPING = "servlet-mapping";

    private static final String URL_MAPPING = "url-pattern";

    private static final String INIT_PARAM = "init-param";

    private static final String PARAM_NAME = "param-name";

    private static final String PARAM_VALUE = "param-value";

    private static final String SECURITY_ROLE_REF = "security-role-ref";

    private static final String URL_PATTERN = "url-pattern";

    private static final String SECURITY_CONSTRAINT = "security-constraint";

    public WsEndpointDDModifier(Document web) {
        super(web.getDocumentElement(), web);
    }

    public void addServlet(String name, String classname) {
        Element servlet = newJ2EEElement(SERVLET);
        Element servletName = newJ2EEElement(SERVLET_NAME, name);
        Element servletClass = newJ2EEElement(SERVLET_CLASS, classname);

        servlet.appendChild(servletName);
        servlet.appendChild(servletClass);

        // add servlet in the webapp Element
        getElement().appendChild(servlet);

    }

    public void removeServlet(String name) {
        Element servlet = findServlet(name);

        if (servlet != null) {
            getElement().removeChild(servlet);
        }
    }

    public NodeList removeServletWithSecurity(String name) {
        Element servlet = findServlet(name);
        NodeList elements = null;

        if (servlet != null) {

            elements = servlet.getElementsByTagNameNS(J2EE_NS, SECURITY_ROLE_REF);
            // if the list is empty, return null
            if (elements.getLength() == 0) {
                elements = null;
            }
            getElement().removeChild(servlet);
        }
        return elements;
    }

    public void addServletMapping(String name, String mapping) {
        Element servletMapping = newJ2EEElement(SERVLET_MAPPING);
        Element servletName = newJ2EEElement(SERVLET_NAME, name);
        Element urlMapping = newJ2EEElement(URL_MAPPING, mapping);

        servletMapping.appendChild(servletName);
        servletMapping.appendChild(urlMapping);

        // add servletMapping in the webapp Element
        getElement().appendChild(servletMapping);
    }

    public void addServletParam(String servletName, String pName, String pValue) {
        Element ip = newJ2EEElement(INIT_PARAM);
        Element pn = newJ2EEElement(PARAM_NAME, pName);
        Element pv = newJ2EEElement(PARAM_VALUE, pValue);

        ip.appendChild(pn);
        ip.appendChild(pv);

        Element servlet = findServlet(servletName);
        servlet.appendChild(ip);

    }

    public void addServletSecurityRoleRefs(String servletName, NodeList securityRoleRefs) {
        // find the servlet
        Element servlet = findServlet(servletName);
        // add all security-role-ref
        for (int i = 0; i < securityRoleRefs.getLength(); i++) {
            Node securityRoleRefItem = securityRoleRefs.item(i);
            servlet.appendChild(securityRoleRefItem);
        }
    }

    private Element findServlet(String name) {
        NodeList nl = getElement().getElementsByTagNameNS(J2EE_NS, SERVLET);
        Element servlet = null;

        for (int i = 0; (i < nl.getLength()) && (servlet == null); i++) {
            Element e = (Element) nl.item(i);

            NodeList names = e.getElementsByTagNameNS(J2EE_NS, SERVLET_NAME);

            // test servlet/servlet-name/#text-node.value
            if (names.item(0).getFirstChild().getNodeValue().equals(name)) {
                servlet = e;
            }
        }

        return servlet;
    }

    public String removeServletMapping(String sName) {
        NodeList nl = getElement().getElementsByTagNameNS(J2EE_NS, SERVLET_MAPPING);
        Element mapping = null;
        String urlPatternValue = null;

        for (int i = 0; (i < nl.getLength()) && (mapping == null); i++) {
            Element e = (Element) nl.item(i);

            NodeList names = e.getElementsByTagNameNS(J2EE_NS, SERVLET_NAME);

            // test servlet-mapping/servlet-name/#text-node.value
            if (names.item(0).getFirstChild().getNodeValue().equals(sName)) {
                mapping = e;
            }
        }

        if (mapping != null) {
            getLogger().log(BasicLevel.DEBUG, "mapping element found : " + mapping);
            // pick up the url-pattern before removing
            NodeList urlPatterns = mapping.getElementsByTagNameNS(J2EE_NS, URL_PATTERN);
            urlPatternValue = urlPatterns.item(0).getFirstChild().getNodeValue();

            getElement().removeChild(mapping);
        }

        return urlPatternValue;
    }

    public void updateSecurityConstraint(String oldUrlPatter, String newUrlPatterValue) {

        NodeList nl = getElement().getElementsByTagNameNS(J2EE_NS, SECURITY_CONSTRAINT);

        // loop over security constraints
        for (int i = 0; i < nl.getLength(); i++) {
            Element e = (Element) nl.item(i);

            // look at nested url-pattern Element(s)
            NodeList urlPatternCollection = e.getElementsByTagNameNS(J2EE_NS, URL_PATTERN);
            for (int j = 0; j < urlPatternCollection.getLength(); j++) {

                Element urlPatternElement = (Element) urlPatternCollection.item(j);
                Text urlPatternText = (Text) urlPatternElement.getFirstChild();

                // if found urlPattern Element, replace its value
                if (urlPatternText.getNodeValue().equals(oldUrlPatter)) {
                    urlPatternText.setNodeValue(newUrlPatterValue);
                }
            }
        }
    }

    public void addEjbRef(String name, String home, String remote, String link) {

        Element ejbRef = newJ2EEElement("ejb-ref");
        Element ejbRefName = newJ2EEElement("ejb-ref-name", name);
        Element ejbRefType = newJ2EEElement("ejb-ref-type", "Session");
        Element ejbHome = newJ2EEElement("home", home);
        Element ejbRemote = newJ2EEElement("remote", remote);
        Element ejbLink = newJ2EEElement("ejb-link", link);

        ejbRef.appendChild(ejbRefName);
        ejbRef.appendChild(ejbRefType);
        ejbRef.appendChild(ejbHome);
        ejbRef.appendChild(ejbRemote);
        ejbRef.appendChild(ejbLink);

        getElement().appendChild(ejbRef);

    }

    public void addEjbLocalRef(String name, String home, String remote, String link) {

        Element ejbRef = newJ2EEElement("ejb-local-ref");
        Element ejbRefName = newJ2EEElement("ejb-ref-name", name);
        Element ejbRefType = newJ2EEElement("ejb-ref-type", "Session");
        Element ejbHome = newJ2EEElement("local-home", home);
        Element ejbRemote = newJ2EEElement("local", remote);
        Element ejbLink = newJ2EEElement("ejb-link", link);

        ejbRef.appendChild(ejbRefName);
        ejbRef.appendChild(ejbRefType);
        ejbRef.appendChild(ejbHome);
        ejbRef.appendChild(ejbRemote);
        ejbRef.appendChild(ejbLink);

        getElement().appendChild(ejbRef);

    }
}

---