JEntityResolver.java

package org.objectweb.jonas_lib.deployment.digester;


import java.io.IOException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.xml.sax.EntityResolver;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;

import org.objectweb.jonas_lib.deployment.api.DTDs;
import org.objectweb.jonas_lib.deployment.api.Schemas;

public class JEntityResolver implements EntityResolver {

    private HashMap entries = new HashMap();

    private JDigester jDigester = null;


    public JEntityResolver(JDigester jd) {
        super();
        jDigester = jd;
    }

    public JEntityResolver() {
        super();
    }

    private void addSchema(String path) {
        String id = path.substring(path.lastIndexOf('/') + 1);
        entries.put(id, path);
    }



    public InputSource resolveEntity(String publicId, String systemId)
        throws IOException, SAXException {

        // If we have a JDigester instance
        if (jDigester != null) {
            jDigester.setPublicId(publicId);
        }

        // DTD : check if we got this entry
        String localPath = null;
        if (publicId != null) {
            localPath = (String) entries.get(publicId);
        } else if (systemId != null) {
            // Can be a schema
            if (systemId.toLowerCase().endsWith(".xsd")) {
                // Retrieve basename
                String baseName = systemId.substring(systemId.lastIndexOf('/') + 1);

                // Registred ?
                localPath = (String) entries.get(baseName);
            }
        }

        if (localPath == null) {
            // This DTD/Schema was not added to this DTD/Schame local repository
            return null;
        }

        // Return the local path source
        return (new InputSource(localPath));
    }


    public void addDtds(DTDs dtds) {
        if (dtds != null) {
            Map dtdsMapping = dtds.getMapping();
            if (dtdsMapping != null) {
                for (Iterator it = dtdsMapping.entrySet().iterator(); it.hasNext();) {
                    Map.Entry entry = (Map.Entry) it.next();
                    String publicId = (String) entry.getKey();
                    String localURL = (String) entry.getValue();
                    entries.put(publicId, localURL);
                }
            }
        }
    }


    public void addSchemas(Schemas schemas) {
        // no schemas
        if (schemas == null) {
            return;
        }

        List localSchemas = schemas.getlocalSchemas();

        // there are schemas
        if (localSchemas != null) {
            for (Iterator it = localSchemas.iterator(); it.hasNext();) {
               String schema = (String) it.next();
               addSchema(schema);
            }
        }
    }



}

---