Fix validUntil processing.
[java-idp.git] / src / edu / internet2 / middleware / shibboleth / xml / SchemasDirectoryImpl.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.File;
20 import java.io.InputStream;
21 import java.net.URL;
22
23 import org.apache.log4j.Logger;
24 import org.w3c.dom.Document;
25 import org.w3c.dom.Element;
26 import org.xml.sax.InputSource;
27
28 /**
29  * 
30  * Find Schemas in a Resource directory
31  * 
32  * @author Howard Gilbert
33  * 
34  */
35 public class SchemasDirectoryImpl extends SchemaStore {
36     
37     private static Logger log = Logger.getLogger(SchemasDirectoryImpl.class);
38     
39     private String resourcedir = "/schemas/";
40
41         
42     /**
43      * Load the bucket initially from a supplied directory.
44      * 
45      * @param resourcedir
46      */
47     public SchemasDirectoryImpl(String resourcedir) {
48         super();
49         this.resourcedir = resourcedir;
50         this.loadBucket();
51     }
52     
53         private boolean loadBucket() {
54                 // for each .xsd file in the directory
55         URL resource = Parser.class.getResource(resourcedir);
56         String path = resource.getPath();
57         File dir = new File(path);
58         if (!dir.isDirectory()) {
59             log.error("Cannot find the schemas resource directory");
60             return false;
61         }
62         String[] filenames = dir.list();
63                 int nextsource=0;
64                 for (int i=0;i<filenames.length;i++) {
65             String filename = filenames[i];
66             if (!filename.endsWith(".xsd"))
67                 continue;
68             InputStream inputStream =
69                     Parser.class.getResourceAsStream(
70                         "/schemas/" + filename);
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 (" + filename + "): " + 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:"+filename);
86                 continue;
87             }
88             String targetNamespace = ele.getAttribute("targetNamespace");
89             if (targetNamespace==null) {
90                 log.error("Schema has no targetNamespace: "+filename);
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                 return true;
103         }
104
105 }