Exception : Add target constructor + add exception when named dependency not found

.classpath

@@ -2,9 +2,9 @@
 <classpath>
 	<classpathentry kind="con" path="org.eclipse.jdt.launching.JRE_CONTAINER/org.eclipse.jdt.internal.debug.ui.launcher.StandardVMType/JavaSE-1.8"/>
 	<classpathentry kind="src" path="src"/>
-	<classpathentry exported="true" kind="lib" path="jose4j-0.9.3.jar"/>
-	<classpathentry exported="true" kind="lib" path="password4j-1.6.3.jar"/>
-	<classpathentry exported="true" kind="lib" path="slf4j-api-2.0.6.jar"/>
-	<classpathentry exported="true" kind="lib" path="C:/Users/jeffc/eclipse-workspace/peer-at-code-framework/PeerAtCodeParser.jar"/>
+	<classpathentry exported="true" kind="lib" path="jose4j-0.9.6.jar"/>
+	<classpathentry exported="true" kind="lib" path="password4j-1.8.1.jar"/>
+	<classpathentry exported="true" kind="lib" path="slf4j-api-2.0.12.jar"/>
+	<classpathentry exported="true" kind="lib" path="PeerAtCodeParser.jar"/>
 	<classpathentry kind="output" path="bin"/>
 </classpath>

src/dev/peerat/framework/Client.java

@@ -9,13 +9,11 @@ public class Client<U extends User> extends Thread{
 	private HttpReader reader;
 	private HttpWriter writer;
 	private Router<U> router;
-	private Locker<Context> logger;
 
 	public Client(Socket socket, Router<U> router) throws Exception{
 		this.reader = new HttpReader(socket);
 		this.writer = new HttpWriter(socket);
 		this.router = router;
-		this.logger = router.getLogger();
 	}
 
 	@Override
@@ -26,25 +24,28 @@ public class Client<U extends User> extends Thread{
 			reader.readHeaders();
 
 			RequestType type = RequestType.valueOf(headers[0]);
-			context = new Context(type, headers[1], isLogin(router, type, reader), writer, router.getDefaultHeaders(type));
+			context = new Context(type, headers[1], isLogin(type, reader), writer, router.getDefaultHeaders(type));
 			router.exec(context, reader, writer);
 			writer.flush();
 			writer.close();
 		}catch(InvalidJwtException e){
-		}catch(Exception e){
-			e.printStackTrace();
+			this.router.getExceptionLogger().setValue(e);
+		}catch(Throwable e){
+			this.router.getExceptionLogger().setValue(e);
 			if(context != null && context.getResponseCode() == 0){
 				try{
 					context.response(500);
 					writer.flush();
 					writer.close();
-				}catch(Exception ex){}
+				}catch(Exception ex){
+					this.router.getExceptionLogger().setValue(ex);
+				}
 			}
 		}
-		if(context != null) logger.setValue(context);
+		if(context != null) router.getLogger().setValue(context);
 	}
 
-	private User isLogin(Router<U> router, RequestType type, HttpReader reader) throws InvalidJwtException{
+	private User isLogin(RequestType type, HttpReader reader) throws InvalidJwtException{
 		String auth = reader.getHeader("Authorization");
 		if(auth == null) return null;
 		auth = auth.substring(7);
@@ -52,10 +53,12 @@ public class Client<U extends User> extends Thread{
 			return this.router.getUser(auth);
 		}catch(InvalidJwtException e){
 			try{
-				writer.response(401, router.getDefaultHeaders(type));
+				writer.response(401, this.router.getDefaultHeaders(type));
 				writer.flush();
 				writer.close();
-			}catch(Exception ex){}
+			}catch(Exception ex){
+				this.router.getExceptionLogger().setValue(ex);
+			}
 			throw e;
 		}
 	}

src/dev/peerat/framework/DependencyInjector.java

@@ -0,0 +1,70 @@
+package dev.peerat.framework;
+
+import java.lang.reflect.Constructor;
+import java.lang.reflect.Parameter;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.function.BiFunction;
+
+public class DependencyInjector{
+	
+	private Map<String, Object> map;
+	private List<BiFunction<Constructor<?>, Parameter, Object>> builders;
+	private Object[] injections;
+	
+	public DependencyInjector(){
+		this.map = new HashMap<>();
+		this.builders = new ArrayList<>();
+	}
+	
+	public DependencyInjector of(String name, Object value){
+		this.map.put(name, value);
+		return this;
+	}
+	
+	public DependencyInjector of(BiFunction<Constructor<?>, Parameter, Object> builder){
+		this.builders.add(builder);
+		return this;
+	}
+	
+	public DependencyInjector of(Object... injections){
+		if(this.injections == null){
+			this.injections = injections;
+			return this;
+		}
+		Object[] copy = new Object[this.injections.length+injections.length];
+		System.arraycopy(this.injections, 0, copy, 0, this.injections.length);
+		System.arraycopy(injections, 0, copy, this.injections.length, injections.length);
+		this.injections = copy;
+		return this;
+	}
+
+	Object applyDependency(Constructor<?> constructor, Parameter parameter, Map<Class<?>, Object> cache){
+		Injection annotation = parameter.getAnnotation(Injection.class);
+		if(annotation != null){
+			Object result = this.map.get(annotation.value());
+			if(result == null) throw new IllegalArgumentException("No depdency named "+annotation.value()+" ("+this.map.keySet()+") in constructor "+constructor);
+			return result;
+		}
+		for(BiFunction<Constructor<?>, Parameter, Object> function : builders){
+			Object result = function.apply(constructor, parameter);
+			if(result != null) return result;
+		}
+		if(this.injections == null) return null;
+		Class<?> type = parameter.getType();
+		Object result = null;
+		for(Object injection : injections){
+			if(type.isAssignableFrom(injection.getClass())){
+				if(cache.containsKey(type)){
+					throw new IllegalArgumentException("Double dependency for type "+type+", can be in constructor "+constructor);
+				}else{
+					cache.put(type, injection);
+					result = injection;
+				}
+			}
+		}
+		return result;
+	}
+}

src/dev/peerat/framework/HttpReader.java

@@ -9,6 +9,7 @@ import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.Set;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 
@@ -52,6 +53,10 @@ public class HttpReader{
 		}
 	}
 	
+	public Set<String> getHeaders(){
+		return this.headers.keySet();
+	}
+	
 	public String getHeader(String key){
 		return this.headers.get(key.toLowerCase());
 	}

src/dev/peerat/framework/HttpWriter.java

@@ -46,7 +46,7 @@ public class HttpWriter{
 		flush();
 	}
 	
-	private static String[] HEIGHTBITS = new String[9];
+	private static String[] HEIGHTBITS = new String[27];
 	private static String[] NINEBITS = new String[212];
 	
 	static {
@@ -59,6 +59,7 @@ public class HttpWriter{
 		HEIGHTBITS[6] = " Partial Content";
 		HEIGHTBITS[7] = " Mutli-Status";
 		HEIGHTBITS[8] = " Already Reported";
+		HEIGHTBITS[26] = " IM Used";
 		
 		NINEBITS[0] = " Multiple Choices";
 		NINEBITS[1] = " Moved Permanently";
@@ -88,6 +89,8 @@ public class HttpWriter{
 		NINEBITS[116] = " Range Not Satisfiable";
 		NINEBITS[117] = " Expectation Failed";
 		NINEBITS[118] = " I'm a Teapot";
+		NINEBITS[119] = " Insufficient Space On Resource";
+		NINEBITS[120] = " Method Failure";
 		NINEBITS[121] = " Misdirected Request";
 		NINEBITS[122] = " Unprocessable Content";
 		NINEBITS[123] = " Locked";
@@ -108,6 +111,7 @@ public class HttpWriter{
 		NINEBITS[206] = " Variant Also Negotiates";
 		NINEBITS[207] = " Insufficient Storage";
 		NINEBITS[208] = " Loop Detected";
+		NINEBITS[209] = " Bandwith Limit Exceeded";
 		NINEBITS[210] = " Not Extended";
 		NINEBITS[211] = " Network Authentication Required";
 	}

src/dev/peerat/framework/Injection.java

@@ -0,0 +1,11 @@
+package dev.peerat.framework;
+
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+
+@Retention(RetentionPolicy.RUNTIME)
+public @interface Injection{
+
+	String value();
+	
+}

src/dev/peerat/framework/Locker.java

@@ -8,8 +8,6 @@ import java.util.concurrent.LinkedBlockingQueue;
 import java.util.function.Consumer;
 import java.util.function.Supplier;
 
-import dev.peerat.framework.Locker.Key;
-
 public class Locker<V>{
 	
 	private Map<Key, BlockingQueue<V>> map;

src/dev/peerat/framework/RouteInterceptor.java

@@ -0,0 +1,10 @@
+package dev.peerat.framework;
+
+import java.lang.reflect.Method;
+import java.util.regex.Matcher;
+
+public interface RouteInterceptor{
+	
+	boolean intercept(Matcher matcher, Context context, HttpReader reader, HttpWriter writer, Method method);
+
+}

src/dev/peerat/framework/RouteMapper.java

@@ -1,6 +1,8 @@
 package dev.peerat.framework;
 
+import java.lang.reflect.Method;
 import java.security.MessageDigest;
+import java.util.List;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 
@@ -11,6 +13,7 @@ public class RouteMapper<U extends User>{
 	private Router<U> router;
 	
 	private Response[] responses;
+	private Method[] methods;
 	private Route[] routes;
 	private Pattern[] patterns;
 	private int[] calls;
@@ -19,13 +22,15 @@ public class RouteMapper<U extends User>{
 	public RouteMapper(Router<U> router){
 		this.router = router;
 		this.responses = new Response[0];
+		this.methods = new Method[0];
 		this.routes = new Route[0];
 		this.patterns = new Pattern[0];
 	}
 
-	public void register(Response response, Route route, Pattern pattern){
+	public void register(Response response, Method method, Route route, Pattern pattern){
 		int length = responses.length+1;
 		this.responses = addElement(responses, new Response[length], response);
+		this.methods = addElement(methods, new Method[length], method);
 		this.routes = addElement(routes, new Route[length], route);
 		this.patterns = addElement(patterns, new Pattern[length], pattern);
 		if(this.calls != null) this.calls = new int[length];
@@ -36,15 +41,15 @@ public class RouteMapper<U extends User>{
 		this.dif = 2;
 	}
 	
-	public boolean exec(Context context, HttpReader reader, HttpWriter writer) throws Exception{
-		Response result = null;
+	public boolean exec(Context context, HttpReader reader, HttpWriter writer, List<RouteInterceptor> interceptors) throws Exception{
+		Response response = null;
+		Method method = null;
 		Matcher matcher = null;
 		
 		String path = context.getPath();
 		
 		synchronized(responses){
 			for(int i = 0; i < responses.length; i++){
-				Response response = responses[i];
 				Route route = routes[i];
 				Pattern pattern = patterns[i];
 				matcher = pattern.matcher(path);
@@ -62,14 +67,18 @@ public class RouteMapper<U extends User>{
 						writer.response(401, router.getDefaultHeaders(context.getType()));
 						return true;
 					}
+					response = responses[i];
+					method = methods[i];
 					order(i);
-					result = response;
 					break;
 				}
 			}
 		}
-		if(result != null){
-			result.exec(matcher, context, reader, writer);
+		if(response != null){
+			for(RouteInterceptor interceptor : interceptors){
+				if(!interceptor.intercept(matcher, context, reader, writer, method)) return true;
+			}
+			response.exec(matcher, context, reader, writer);
 			return true;
 		}
 		return false;
@@ -86,6 +95,7 @@ public class RouteMapper<U extends User>{
 		if(index < 1) return;
 		if(call > this.calls[index-1]){
 			switchElement(responses, index, index-1);
+			switchElement(methods, index, index-1);
 			switchElement(routes, index, index-1);
 			switchElement(patterns, index, index-1);
 			

src/dev/peerat/framework/Router.java

@@ -1,8 +1,18 @@
 package dev.peerat.framework;
 
+import java.io.BufferedReader;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.lang.reflect.Constructor;
 import java.lang.reflect.Method;
+import java.lang.reflect.Parameter;
 import java.net.ServerSocket;
 import java.net.Socket;
+import java.security.Key;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
 import java.util.function.Consumer;
 import java.util.function.Function;
 import java.util.regex.Pattern;
@@ -11,6 +21,7 @@ import javax.net.ssl.SSLServerSocket;
 import javax.net.ssl.SSLServerSocketFactory;
 
 import org.jose4j.jwa.AlgorithmConstraints.ConstraintType;
+import org.jose4j.jwk.JsonWebKey.OutputControlLevel;
 import org.jose4j.jwk.RsaJsonWebKey;
 import org.jose4j.jwk.RsaJwkGenerator;
 import org.jose4j.jws.AlgorithmIdentifiers;
@@ -28,7 +39,9 @@ public class Router<U extends User>{
 	}
 	
 	private Locker<Context> logger;
+	private Locker<Throwable> exceptions;
 	private RouteMapper<U>[] mappers;
+	private List<RouteInterceptor> interceptors;
 	private Response noFileFound;
 	private RsaJsonWebKey rsaJsonWebKey;
 	private JwtConsumer jwtConsumer;
@@ -37,30 +50,46 @@ public class Router<U extends User>{
 	private String[][] headers;
 	private ServerSocket serverSocket;
 
-	public Router() throws Exception{
+	public Router(){
 		this.logger = new Locker<>();
+		this.exceptions = new Locker<>();
 		int types = RequestType.values().length;
 		this.mappers = new RouteMapper[types];
+		this.interceptors = new ArrayList<>();
 		for(RequestType type : RequestType.values()) this.mappers[type.ordinal()] = new RouteMapper<>(this);
-		this.rsaJsonWebKey = RsaJwkGenerator.generateJwk(2048);
 		this.headers = new String[types][0];
 	}
 	
-	public Router<U> configureJwt(Consumer<JwtConsumerBuilder> consumer, Consumer<JwtClaims> claims, Function<JwtClaims, U> userCreator){
-		JwtConsumerBuilder builder = new JwtConsumerBuilder()
-		.setRequireExpirationTime()
-		.setAllowedClockSkewInSeconds(30)
-		.setVerificationKey(rsaJsonWebKey.getKey())
-		.setJwsAlgorithmConstraints(ConstraintType.PERMIT, AlgorithmIdentifiers.RSA_USING_SHA256);
-		
-		consumer.accept(builder);
-		
-		this.jwtConsumer = builder.build();
-		this.claims = claims;
-		this.userCreator = userCreator;
+	public Router<U> configureJwt(Consumer<JwtConsumerBuilder> consumer, Consumer<JwtClaims> claims, Function<JwtClaims, U> userCreator) throws Exception{
+		this.rsaJsonWebKey = RsaJwkGenerator.generateJwk(2048);
+		configureJwtWithKey(consumer, claims, userCreator, this.rsaJsonWebKey.getKey());
 		return this;
 	}
 	
+	public Router<U> configureJwt(Consumer<JwtConsumerBuilder> consumer, Consumer<JwtClaims> claims, Function<JwtClaims, U> userCreator, Map<String, Object> keyParams) throws Exception{
+		this.rsaJsonWebKey = new RsaJsonWebKey(keyParams);
+		configureJwtWithKey(consumer, claims, userCreator, this.rsaJsonWebKey.getKey());
+		return this;
+	}
+	
+	private void configureJwtWithKey(Consumer<JwtConsumerBuilder> consumer, Consumer<JwtClaims> claims, Function<JwtClaims, U> userCreator, Key key) throws Exception{
+		JwtConsumerBuilder builder = new JwtConsumerBuilder()
+			.setRequireExpirationTime()
+			.setAllowedClockSkewInSeconds(30)
+			.setVerificationKey(key)
+			.setJwsAlgorithmConstraints(ConstraintType.PERMIT, AlgorithmIdentifiers.RSA_USING_SHA256);
+				
+			consumer.accept(builder);
+				
+			this.jwtConsumer = builder.build();
+			this.claims = claims;
+			this.userCreator = userCreator;
+	}
+	
+	public Map<String, Object> exportJwtKey(){
+		return this.rsaJsonWebKey.toParams(OutputControlLevel.INCLUDE_PRIVATE);
+	}
+	
 	public Router<U> addDefaultHeaders(RequestType type, String... headers){
 		String[] origin = this.headers[type.ordinal()]; 
 		String[] copy = new String[origin.length+headers.length];
@@ -76,18 +105,84 @@ public class Router<U extends User>{
 					Response.class.getDeclaredMethods()[0].getParameterTypes());
 			Route route = method.getAnnotation(Route.class);
 
-			this.mappers[route.type().ordinal()].register(response, route, Pattern.compile(route.path()));
+			this.mappers[route.type().ordinal()].register(response, method, route, Pattern.compile(route.path()));
 		}catch(Exception e){
 			throw new IllegalArgumentException(e);
 		}
 		return this;
 	}
 	
+	public Router<U> registerPackages(Object... injections){
+		return registerPackages(new DependencyInjector().of(injections));
+	}
+	
+	public Router<U> registerPackages(DependencyInjector injector){
+		String clazz = Thread.currentThread().getStackTrace()[2].getClassName();
+		String pack = clazz.substring(0, clazz.lastIndexOf('.'));
+		return registerPackages(pack, injector);
+	}
+	
+	public Router<U> registerPackages(String name, Object... injections){
+		return registerPackages(name, new DependencyInjector().of(injections));
+	}
+	
+	public Router<U> registerPackages(String name, DependencyInjector injector){
+		InputStream stream = ClassLoader.getSystemClassLoader().getResourceAsStream(name.replace(".", "/"));
+		try{
+			BufferedReader reader = new BufferedReader(new InputStreamReader(stream));
+			String line;
+			Map<Class<?>, Object> cache = new HashMap<>();
+			while((line = reader.readLine()) != null){
+				if(line.endsWith(".class")){
+					Class<?> clazz = Class.forName(name+"."+line.substring(0, line.lastIndexOf('.')));
+					if(Response.class.isAssignableFrom(clazz)){
+						if(clazz.equals(Response.class)) continue;
+						Constructor<?>[] constructors = clazz.getDeclaredConstructors();
+						if(constructors.length != 1) continue;
+						injectDependencies(constructors[0], cache, injector);
+					}
+				}else{
+					registerPackages(name+"."+line, injector);
+				}
+			}
+			reader.close();
+		}catch(Exception e){
+			System.err.println("Failed to read "+name);
+			e.printStackTrace();
+		}
+		return this;
+	}
+	
+	private void injectDependencies(Constructor<?> constructor, Map<Class<?>, Object> cache, DependencyInjector injector){
+		Parameter[] parameters = constructor.getParameters();
+		Object[] injections = new Object[parameters.length];
+		for(int i = 0; i < parameters.length; i++){
+			Parameter parameter = parameters[i];
+			Class<?> type = parameter.getType();
+			Object dependency = cache.get(type);
+			if(dependency != null) injections[i] = dependency;
+			else if((injections[i] = injector.applyDependency(constructor, parameter, cache)) == null) throw new IllegalArgumentException("No dependency for type "+type+" in constructor "+constructor);
+		}
+		try {
+			Response response = (Response) constructor.newInstance(injections);
+			register(response);
+			System.out.println("registerd "+response.getClass());
+		} catch (Exception e){
+			e.printStackTrace();
+		}
+		
+	}
+
 	public Router<U> setDefault(Response response){
 		this.noFileFound = response;
 		return this;
 	}
 	
+	public Router<U> addInterceptor(RouteInterceptor interceptor){
+		this.interceptors.add(interceptor);
+		return this;
+	}
+	
 	public Router<U> activeReOrdering(){
 		for(RouteMapper<?> mapper : this.mappers) mapper.activeReOrdering();
 		return this;
@@ -98,7 +193,7 @@ public class Router<U extends User>{
 	}
 
 	void exec(Context context, HttpReader reader, HttpWriter writer) throws Exception{
-		if(this.mappers[context.getType().ordinal()].exec(context, reader, writer)) return;
+		if(this.mappers[context.getType().ordinal()].exec(context, reader, writer, this.interceptors)) return;
 		if(noFileFound != null) noFileFound.exec(null, context, reader, writer);
 	}
 
@@ -133,6 +228,10 @@ public class Router<U extends User>{
 		return this.logger;
 	}
 	
+	public Locker<Throwable> getExceptionLogger(){
+		return this.exceptions;
+	}
+	
 	public void listen(int port, boolean ssl) throws Exception{
 		if (ssl) { // Not needed with the use of a proxy
 			try {

src/dev/peerat/framework/utils/json/JsonParser.java

@@ -45,7 +45,11 @@ public class JsonParser extends Parser<Json>{
 		content.then(new RedirectStateTree<>(base, (global, local) -> global.set(local.get()))).end();
 		content.then((validator) -> validator.validate((token) -> token.getType().equals(TokenType.STRING), (bag, token) -> bag.set(token))).end();
 		content.then((validator) -> validator.validate((token) -> token.getType().equals(TokenType.CHAR), (bag, token) -> bag.set(token))).end();
-		content.then((validator) -> validator.validate((token) -> token.getType().equals(TokenType.NAME), (bag, token) -> bag.set(token))).end();
+		StateTree<Json> number = content.then((validator) -> validator.validate((token) -> token.getType().equals(TokenType.NAME), (bag, token) -> bag.set(token)));
+		number.end();
+		number.then((validator) -> validator.validate((token) -> token.getValue().equals("."), (bag, token) -> bag.set(bag.<Token>get().concat(token))))
+			.then((validator) -> validator.validate((token) -> token.getType().equals(TokenType.NAME), (bag, token) -> bag.set(bag.<Token>get().concat(token))))
+			.end();
 		
 		StateTree<Json> mapper = new StateTree<>();
 		StateTree<Json> mapper_key = mapper.then((validator) -> validator.validate((token) -> token.getType().equals(TokenType.STRING), (bag, token) -> {
@@ -110,10 +114,10 @@ public class JsonParser extends Parser<Json>{
 					return Long.parseLong(content);
 				}catch(Exception _){
 					try {
-						return Boolean.parseBoolean(content);
+						return  Double.parseDouble(content);
 					}catch(Exception __){
 						try{
-							return Double.parseDouble(content);
+							return Boolean.parseBoolean(content);
 						}catch(Exception ___){}
 					}
 				}

src/dev/peerat/framework/utils/json/JsonTokenizer.java

@@ -29,7 +29,22 @@ public class JsonTokenizer extends Tokenizer{
 					for(; j < line.length(); j++){
 						c = line.charAt(j);
 						if(c == '\\'){
-							buffer+=c+line.charAt(++j);
+							c = line.charAt(++j);
+							switch(c){
+								case 'n':
+									buffer+='\n';
+									break;
+								case 'r':
+									buffer+='\r';
+									break;
+								case 't':
+									buffer+='\t';
+									break;
+								default:
+									buffer+=c;
+									break;
+							}
+							continue;
 						}
 						if(c == '\"') break;
 						buffer+=c;
@@ -38,21 +53,6 @@ public class JsonTokenizer extends Tokenizer{
 					i=j;
 					continue;
 				}
-				if(c == '\''){
-					String buffer = "";
-					int j = i+1;
-					for(; j < line.length(); j++){
-						c = line.charAt(j);
-						if(c == '\\'){
-							buffer+=c+line.charAt(++j);
-						}
-						if(c == '\'') break;
-						buffer+=c;
-					}
-					getTokens().add(new Token(1, i+1, buffer, TokenType.CHAR));
-					i=j;
-					continue;
-				}
 				getTokens().add(new Token(1, i+1, ""+c, TokenType.DELIMITER));
 			}
 		}