Use WEB-INF/security.drl if security.drl not found on classpath

seam · Aug 5, 2011 · 3db9cac · 3db9cac
1 parent 8adb507
commit 3db9cac
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/impl/src/main/java/org/jboss/seam/security/permission/SecurityRuleProducer.java b/impl/src/main/java/org/jboss/seam/security/permission/SecurityRuleProducer.java
@@ -32,6 +32,10 @@ public class SecurityRuleProducer {
     @Resource("security.drl")
     InputStream securityRules;
 
+    @Inject
+    @Resource("WEB-INF/security.drl")
+    InputStream webInfSecurityRules;
+
     @Produces
     @ApplicationScoped
     @Security
@@ -41,7 +45,9 @@ public KnowledgeBase createSecurityKnowledgeBase() {
 
         KnowledgeBuilder kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder(config);
 
-        org.drools.io.Resource resource = ResourceFactory.newInputStreamResource(securityRules);
+        org.drools.io.Resource resource = ResourceFactory.newInputStreamResource(
+            securityRules != null ? securityRules : webInfSecurityRules);
+
         kbuilder.add(resource, ResourceType.DRL);
 
         KnowledgeBuilderErrors kbuildererrors = kbuilder.getErrors();