diff --git a/src/be/jeffcheasey88/peeratcode/framework/Router.java b/src/be/jeffcheasey88/peeratcode/framework/Router.java index 040d8f2..781d2a0 100644 --- a/src/be/jeffcheasey88/peeratcode/framework/Router.java +++ b/src/be/jeffcheasey88/peeratcode/framework/Router.java @@ -25,7 +25,7 @@ import org.jose4j.jwt.consumer.JwtConsumer; import org.jose4j.jwt.consumer.JwtConsumerBuilder; import org.jose4j.lang.JoseException; -public class Router{ +public class Router{l private Map> responses; private Map patterns; @@ -89,7 +89,7 @@ public class Router{ } } - public void register(Response response){ + public Router register(Response response){ try{ Method method = response.getClass().getDeclaredMethod("exec", Response.class.getDeclaredMethods()[0].getParameterTypes()); @@ -100,13 +100,15 @@ public class Router{ }catch(Exception e){ throw new IllegalArgumentException(e); } + return this; } - public void setDefault(Response response){ + public Router setDefault(Response response){ this.noFileFound = response; + return this; } - public void exec(RequestType type, String path, User user, HttpReader reader, HttpWriter writer) throws Exception{ + void exec(RequestType type, String path, User user, HttpReader reader, HttpWriter writer) throws Exception{ if(type == null) return; for(Entry routes : this.responses.get(type).entrySet()){ Matcher matcher = this.patterns.get(routes.getKey()).matcher(path); @@ -127,9 +129,10 @@ public class Router{ if(noFileFound != null) noFileFound.exec(null, user, reader, writer); } - public void configureSSL(String keyStore, String keyStorePassword){ + public Router configureSSL(String keyStore, String keyStorePassword){ System.setProperty("javax.net.ssl.keyStore", keyStore); System.setProperty("javax.net.ssl.keyStorePassword", keyStorePassword); + return this; } public U getUser(String token) throws Exception{