refine logging
[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     // This class is used by both the Idp and SP, so it must use
38     // a conventionally declared logger. The SP has a special logger
39     // setup so this package also logs to the init log.
40     private static Logger log = Logger.getLogger(SchemasDirectoryImpl.class);
41     
42     private String resourcedir = "/schemas/";
43
44         
45     /**
46      * Load the bucket initially from a supplied directory.
47      * 
48      * @param resourcedir
49      */
50     public SchemasDirectoryImpl(String resourcedir) {
51         super();
52         this.resourcedir = resourcedir;
53         this.loadBucket();
54     }
55     
56         private boolean loadBucket() {
57                 // for each .xsd file in the directory
58         URL resource = Parser.class.getResource(resourcedir);
59         String path = resource.getPath();
60         File dir = new File(path);
61         if (!dir.isDirectory()) {
62             log.error("Cannot find the schemas resource directory");
63             return false;
64         }
65         String[] filenames = dir.list();
66                 for (int i=0;i<filenames.length;i++) {
67             String filename = filenames[i];
68             if (!filename.endsWith(".xsd"))
69                 continue;
70             InputStream inputStream =
71                     Parser.class.getResourceAsStream(
72                         "/schemas/" + filename);
73             InputSource insrc = new InputSource(inputStream);
74            
75             // Non-validating parse to DOM
76             Document xsddom;
77                         try {
78                                 xsddom = Parser.loadDom(insrc,false);
79                         } catch (Exception e) {
80                                 log.error("Error parsing XML schema (" + filename + "): " + e);
81                                 continue;
82                         }
83             
84             // Get the target namespace from the root element
85             Element ele = xsddom.getDocumentElement();
86             if (!ele.getLocalName().equals("schema")) {
87                 log.error("Schema file wrong root element:"+filename);
88                 continue;
89             }
90             String targetNamespace = ele.getAttribute("targetNamespace");
91             if (targetNamespace==null) {
92                 log.error("Schema has no targetNamespace: "+filename);
93                 continue;
94             }
95             
96             // Put the DOM in the Bucket keyed by namespace
97             if (bucket.containsKey(targetNamespace)) {
98                 log.info("Replacing XSD for namespace: "+targetNamespace+" "+filename);
99             } else {
100                 log.info("Defining XSD for namespace:  "+targetNamespace+" "+filename);
101             }
102             bucket.put(targetNamespace,xsddom);
103         }
104                 return true;
105         }
106
107 }