8defd122fa58ce03eeaaf787f209db3a84fc6c1f
[java-idp.git] / src / edu / internet2 / middleware / shibboleth / xml / SchemasResourceListImpl.java
1 /*
2  * Copyright [2005] [University Corporation for Advanced Internet Development, Inc.]
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package edu.internet2.middleware.shibboleth.xml;
18
19 import java.io.InputStream;
20
21 import org.apache.log4j.Logger;
22 import org.w3c.dom.Document;
23 import org.w3c.dom.Element;
24 import org.xml.sax.InputSource;
25
26 /**
27  *  
28  * Find Schemas as a list of resource files.
29  * 
30  * <p>Java resources are files found in the Classpath of the current
31  * ClassLoader. They may be in directories on disk, in jar files, or
32  * elsewhere. This class must be passed a list of resource names, but
33  * it uses the Java runtime to actually locate the xsd data.
34  * 
35  * @author Howard Gilbert
36  */
37 public class SchemasResourceListImpl extends SchemaStore {
38     
39     private static Logger log = Logger.getLogger(SchemasDirectoryImpl.class);
40     
41     private String resourceprefix = "/schemas/";
42     private String[] resourceNames = null;
43
44     /**
45      * @param resourcedir
46      */
47     public SchemasResourceListImpl(String resourcedir, String[] resources) {
48         this.resourceprefix = resourcedir;
49         this.resourceNames = resources;
50         this.loadBucket();
51     }
52     
53    
54     private void loadBucket() {
55                 // for each .xsd file in the directory
56                 int nextsource=0;
57                 for (int i=0;i<resourceNames.length;i++) {
58             String filename = resourceNames[i];
59             if (!filename.endsWith(".xsd")) {
60                 log.error(filename + " doesn't end in .xsd, ignoring it.");
61                 continue;
62             }
63             String resourceName = resourceprefix+filename;
64             InputStream inputStream =
65                     Parser.class.getResourceAsStream(
66                             resourceName);
67             if (inputStream == null) {
68                 log.error("Resource "+resourceName+" not found, ignoring it.");
69                 continue;
70             }
71             InputSource insrc = new InputSource(inputStream);
72            
73             // Non-validating parse to DOM
74             Document xsddom;
75                         try {
76                                 xsddom = Parser.loadDom(insrc,false);
77                         } catch (Exception e) {
78                                 log.error("Error parsing XML schema (" + resourceName + "): " + e);
79                                 continue;
80                         }
81             
82             // Get the target namespace from the root element
83             Element ele = xsddom.getDocumentElement();
84             if (!ele.getLocalName().equals("schema")) {
85                 log.error("Schema file wrong root element:"+resourceName);
86                 continue;
87             }
88             String targetNamespace = ele.getAttribute("targetNamespace");
89             if (targetNamespace==null) {
90                 log.error("Schema has no targetNamespace: "+resourceName);
91                 continue;
92             }
93             
94             // Put the DOM in the Bucket keyed by namespace
95             if (bucket.containsKey(targetNamespace)) {
96                 log.debug("Replacing XSD for namespace: "+targetNamespace+" "+filename);
97             } else {
98                 log.debug("Defining XSD for namespace:  "+targetNamespace+" "+filename);
99             }
100             bucket.put(targetNamespace,xsddom);
101         }
102         }
103
104 }