StateTree -> import (only firstone)

src/be/jeffcheasey88/peeratcode/parser/java/JavaParser.java

@@ -37,20 +37,44 @@ public class JavaParser  extends Parser<JavaFile> {
 		Tokenizer tokenizer = new Tokenizer();
 		
 		
+		StateTree<JavaFile> importState = new StateTree<>();
+		importState.then((validator) -> {
+			if(validator.validate(
+					(token) -> token.getValue().equals("import"),
+					(bag, token) -> bag.set(new LinkedList<>()))){
+				
+				validator.validate(
+						(token) -> token.getValue().equals("static"),
+						(bag, token) -> bag.set("static", null));
+				
+				while(validator.validate(
+						(token) -> !token.getValue().equals(";"),
+						(bag, token) -> bag.<List<Token>>get().add(token))
+					);
+				
+				return validator.validate((token) -> token.getValue().equals(";"));
+			}
+			return false;
+		}).<JavaElement>end((javafile, bag) -> javafile.addImport(bag));
+
+		
 		StateTree<JavaFile> main = new StateTree<>();
 		
 		main.then((validator) -> {			
 			if(validator.validate(
 					(token) -> token.getValue().equals("package"),
 					(bag, token) -> bag.set(new LinkedList<>()))){
+				
 				while(validator.validate(
 						(token) -> !token.getValue().equals(";"),
 						(bag, token) -> bag.<List<Token>>get().add(token))
 					);
+				
 				return validator.validate((token) -> token.getValue().equals(";"));
 			}
 			return false;
-		}).<JavaElement>end((javafile, bag) -> javafile.setPackage(bag));
+		}).<JavaElement>end((javafile, bag) -> javafile.setPackage(bag))
+			.thenNoChild(importState);
 		
 		
 		System.out.println((System.currentTimeMillis()-time)+"ms");