Added dependency support to DataConnectors
[java-idp.git] / src / edu / internet2 / middleware / shibboleth / aa / attrresolv / provider / BaseResolutionPlugIn.java
1 /* 
2  * The Shibboleth License, Version 1. 
3  * Copyright (c) 2002 
4  * University Corporation for Advanced Internet Development, Inc. 
5  * All rights reserved
6  * 
7  * 
8  * Redistribution and use in source and binary forms, with or without 
9  * modification, are permitted provided that the following conditions are met:
10  * 
11  * Redistributions of source code must retain the above copyright notice, this 
12  * list of conditions and the following disclaimer.
13  * 
14  * Redistributions in binary form must reproduce the above copyright notice, 
15  * this list of conditions and the following disclaimer in the documentation 
16  * and/or other materials provided with the distribution, if any, must include 
17  * the following acknowledgment: "This product includes software developed by 
18  * the University Corporation for Advanced Internet Development 
19  * <http://www.ucaid.edu>Internet2 Project. Alternately, this acknowledegement 
20  * may appear in the software itself, if and wherever such third-party 
21  * acknowledgments normally appear.
22  * 
23  * Neither the name of Shibboleth nor the names of its contributors, nor 
24  * Internet2, nor the University Corporation for Advanced Internet Development, 
25  * Inc., nor UCAID may be used to endorse or promote products derived from this 
26  * software without specific prior written permission. For written permission, 
27  * please contact shibboleth@shibboleth.org
28  * 
29  * Products derived from this software may not be called Shibboleth, Internet2, 
30  * UCAID, or the University Corporation for Advanced Internet Development, nor 
31  * may Shibboleth appear in their name, without prior written permission of the 
32  * University Corporation for Advanced Internet Development.
33  * 
34  * 
35  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 
36  * AND WITH ALL FAULTS. ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 
37  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A 
38  * PARTICULAR PURPOSE, AND NON-INFRINGEMENT ARE DISCLAIMED AND THE ENTIRE RISK 
39  * OF SATISFACTORY QUALITY, PERFORMANCE, ACCURACY, AND EFFORT IS WITH LICENSEE. 
40  * IN NO EVENT SHALL THE COPYRIGHT OWNER, CONTRIBUTORS OR THE UNIVERSITY 
41  * CORPORATION FOR ADVANCED INTERNET DEVELOPMENT, INC. BE LIABLE FOR ANY DIRECT, 
42  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 
43  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 
44  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 
45  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 
46  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 
47  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
48  */
49
50 package edu.internet2.middleware.shibboleth.aa.attrresolv.provider;
51
52 import java.util.HashSet;
53 import java.util.Set;
54
55 import org.apache.log4j.Logger;
56 import org.w3c.dom.Element;
57 import org.w3c.dom.NodeList;
58
59 import edu.internet2.middleware.shibboleth.aa.attrresolv.AttributeResolver;
60 import edu.internet2.middleware.shibboleth.aa.attrresolv.ResolutionPlugInException;
61
62 /**
63  * Base class for Resolution PlugIns, both <code>AttributeDefinitionPlugIn</code> & <code>DataConnectorPlugIn</code>
64  * 
65  * @author Walter Hoehn (wassa@columbia.edu)
66  */
67
68 public abstract class BaseResolutionPlugIn {
69
70         private static Logger log = Logger.getLogger(BaseResolutionPlugIn.class.getName());
71         
72         /** The identifier for this PlugIn. */
73         protected String id;
74
75         /** Time, in seconds, for which the Attribute Resolver should cache resolutions of this PlugIn. */
76         protected long ttl = 0;
77
78     /** Dependencies. */
79     protected Set connectorDependencyIds = new HashSet();
80     protected Set attributeDependencyIds = new HashSet();
81
82         protected BaseResolutionPlugIn(Element e) throws ResolutionPlugInException {
83                 String id = e.getAttribute("id");
84                 if (id == null || id.equals("")) {
85                         log.error("Attribute \"id\" required to configure plugin.");
86                         throw new ResolutionPlugInException("Failed to initialize Resolution PlugIn.");
87                 }
88                 this.id = id;
89
90                 String cacheTime = e.getAttribute("cacheTime");
91                 if (cacheTime != null && !cacheTime.equals("")) {
92                         try {
93                                 this.ttl = Long.parseLong(cacheTime);
94                         } catch (NumberFormatException nfe) {
95                                 log.error("Attribute \"cacheTime\" must be an integer between 0 and " + Long.MAX_VALUE + ".");
96                                 throw new ResolutionPlugInException("Failed to initialize Resolution PlugIn.");
97                         }
98                 }
99
100         NodeList connectorNodes = e.getElementsByTagNameNS(AttributeResolver.resolverNamespace, "DataConnectorDependency");
101
102         for (int i = 0; connectorNodes.getLength() > i; i++) {
103             Element connector = (Element) connectorNodes.item(i);
104             String connectorName = connector.getAttribute("requires");
105             if (connectorName != null && !connectorName.equals("")) {
106                 addDataConnectorDependencyId(connectorName);
107             } else {
108                 log.error("Data Connector dependency must be accompanied by a \"requires\" attribute.");
109                 throw new ResolutionPlugInException("Failed to initialize Resolution PlugIn.");
110             }
111         }
112
113         NodeList attributeNodes = e.getElementsByTagNameNS(AttributeResolver.resolverNamespace, "AttributeDependency");
114         for (int i = 0; attributeNodes.getLength() > i; i++) {
115             Element attribute = (Element) attributeNodes.item(i);
116             String attributeName = attribute.getAttribute("requires");
117             if (attributeName != null && !attributeName.equals("")) {
118                 addAttributeDefinitionDependencyId(attributeName);
119             } else {
120                 log.error("Attribute Definition dependency must be accompanied by a \"requires\" attribute.");
121                 throw new ResolutionPlugInException("Failed to initialize Resolution PlugIn.");
122             }
123         }
124         }
125         
126         /** Returns the identifier for this PlugIn. */
127         public String getId() {
128                 return id;
129         }
130
131         /** Returns the time, in seconds, for which the Attribute Resolver should cache resolutions of this PlugIn. */
132         public long getTTL() {
133                 return ttl;
134         }
135
136     protected void addDataConnectorDependencyId(String id) {
137         connectorDependencyIds.add(id);
138     }
139
140     protected void addAttributeDefinitionDependencyId(String id) {
141         attributeDependencyIds.add(id);
142     }
143
144     /**
145      * @see edu.internet2.middleware.shibboleth.aa.attrresolv.ResolutionPlugIn#getAttributeDependencyIds()
146      */
147     public String[] getAttributeDefinitionDependencyIds() {
148         return (String[]) attributeDependencyIds.toArray(new String[0]);
149     }
150
151     /**
152      * @see edu.internet2.middleware.shibboleth.aa.attrresolv.ResolutionPlugIn#getConnectorDependencyIds()
153      */
154     public String[] getDataConnectorDependencyIds() {
155         return (String[]) connectorDependencyIds.toArray(new String[0]);
156     }
157 }