code2vec/JavaExtractor/JPredict/src/main/java/JavaExtractor/Common/Common.java
package JavaExtractor.Common;

import java.util.ArrayList;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import com.github.javaparser.ast.Node;
import com.github.javaparser.ast.UserDataKey;

import JavaExtractor.FeaturesEntities.ProgramNode;
import JavaExtractor.FeaturesEntities.Property;

public final class Common {
	public static final UserDataKey<Property> PropertyKey = new UserDataKey<Property>() {
	};
	public static final UserDataKey<ProgramNode> ProgramNodeKey = new UserDataKey<ProgramNode>() {
	};
	public static final UserDataKey<Integer> ChildId = new UserDataKey<Integer>() {
	};
	public static final String EmptyString = "";
	public static final String UTF8 = "UTF-8";
	public static final String EvaluateTempDir = "EvalTemp";

	public static final String FieldAccessExpr = "FieldAccessExpr";
	public static final String ClassOrInterfaceType = "ClassOrInterfaceType";
	public static final String MethodDeclaration = "MethodDeclaration";
	public static final String NameExpr = "NameExpr";
	public static final String MethodCallExpr = "MethodCallExpr";
	public static final String DummyNode = "DummyNode";
	public static final String BlankWord = "BLANK";

	public static final int c_MaxLabelLength = 50;
	public static final String methodName = "METHOD_NAME";
	public static final String internalSeparator = "|";

	public static String normalizeName(String original, String defaultString) {
		original = original.toLowerCase().replaceAll("\\\\n", "") // escaped new
																	// lines
				.replaceAll("//s+", "") // whitespaces
				.replaceAll("[\"',]", "") // quotes, apostrophies, commas
				.replaceAll("\\P{Print}", ""); // unicode weird characters
		String stripped = original.replaceAll("[^A-Za-z]", "");
		if (stripped.length() == 0) {
			String carefulStripped = original.replaceAll(" ", "_");
			if (carefulStripped.length() == 0) {
				return defaultString;
			} else {
				return carefulStripped;
			}
		} else {
			return stripped;
		}
	}

	public static boolean isMethod(Node node) {
		String type = node.getUserData(Common.PropertyKey).getType();

		return isMethod(node, type);
	}

	public static boolean isMethod(Node node, String type) {
		Property parentProperty = node.getParentNode().getUserData(Common.PropertyKey);
		if (parentProperty == null) {
			return false;
		}

		String parentType = parentProperty.getType();
		return Common.NameExpr.equals(type) && Common.MethodDeclaration.equals(parentType);
	}

	public static ArrayList<String> splitToSubtokens(String str1) {
		String str2 = str1.trim();
		return Stream.of(str2.split("(?<=[a-z])(?=[A-Z])|_|[0-9]|(?<=[A-Z])(?=[A-Z][a-z])|\\s+"))
				.filter(s -> s.length() > 0).map(s -> Common.normalizeName(s, Common.EmptyString))
				.filter(s -> s.length() > 0).collect(Collectors.toCollection(ArrayList::new));
	}
}
