add authorisation lsitener

miltonio · Jan 24, 2018 · 9bacade · 9bacade
1 parent 661bedf
commit 9bacade
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/milton-server-ce/src/main/java/io/milton/config/HttpManagerBuilder.java b/milton-server-ce/src/main/java/io/milton/config/HttpManagerBuilder.java
@@ -1800,9 +1800,18 @@ private Object findOrCreateObject(Class c) throws CreationException {
 			o = createObject(c);
 		}
 		return o;
+	}
+
+	public AuthorisationListener getAuthorisationListener() {
+		return authorisationListener;
+	}
 
+	public void setAuthorisationListener(AuthorisationListener authorisationListener) {
+		this.authorisationListener = authorisationListener;
 	}
 
+
+
 	public class CreationException extends Exception {
 
 		private final Class attemptedToCreate;

diff --git a/milton-server-ce/src/main/java/io/milton/http/AuthorisationListener.java b/milton-server-ce/src/main/java/io/milton/http/AuthorisationListener.java
@@ -22,6 +22,6 @@
 public interface AuthorisationListener {
 
 
-	void onAuthorisationResult(Request request, Request.Method method, Auth auth, boolean authorised);
-	
+	boolean onAuthorisationResult(Request request, Request.Method method, Auth auth, boolean authorised);
+
 }
diff --git a/milton-server-ce/src/main/java/io/milton/http/HandlerHelper.java b/milton-server-ce/src/main/java/io/milton/http/HandlerHelper.java
@@ -148,7 +148,7 @@ public boolean checkAuthorisation(HttpManager manager, Resource resource, Reques
 			boolean authorised = resource.authorise(request, method, auth);
 
 			if( authorisationListener != null ) {
-				authorisationListener.onAuthorisationResult(request, method, auth, authorised);
+				authorised = authorisationListener.onAuthorisationResult(request, method, auth, authorised);
 			}
 
 			if (!authorised) {