package lombok;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import lombok.javac.CapturingDiagnosticListener.CompilerMessage;
public class CompilerMessageMatcher {
private final List<Integer> lineNumbers = new ArrayList<Integer>();
private final List<List<String>> messages = new ArrayList<List<String>>();
private boolean optional;
private CompilerMessageMatcher() {}
public boolean isOptional() {
return optional;
}
public static CompilerMessageMatcher asCompilerMessageMatcher(CompilerMessage message) {
CompilerMessageMatcher cmm = new CompilerMessageMatcher();
cmm.lineNumbers.add((int) message.getLine());
cmm.messages.add(Arrays.asList(message.getMessage().split("\\s+")));
return cmm;
}
@Override public String toString() {
StringBuilder out = new StringBuilder();
for (int i = 0; i < lineNumbers.size(); i++) {
out.append(lineNumbers.get(i)).append(" ");
for (String part : messages.get(i)) out.append(part).append(" ");
if (out.length() > 0) out.setLength(out.length() - 1);
out.append(" |||| ");
}
if (out.length() > 0) out.setLength(out.length() - 6);
return out.toString();
}
public boolean matches(CompilerMessage message) {
outer:
for (int i = 0; i < lineNumbers.size(); i++) {
if (message.getLine() != lineNumbers.get(i) && message.getLine() -1 != lineNumbers.get(i)) continue;
for (String token : messages.get(i)) {
if (!message.getMessage().contains(token)) continue outer;
}
return true;
}
return false;
}
public static List<CompilerMessageMatcher> readAll(InputStream rawIn) throws IOException {
BufferedReader in = new BufferedReader(new InputStreamReader(rawIn, "UTF-8"));
List<CompilerMessageMatcher> out = new ArrayList<CompilerMessageMatcher>();
for (String line = in.readLine(); line != null; line = in.readLine()) {
CompilerMessageMatcher cmm = read(line);
if (cmm != null) out.add(cmm);
}
return out;
}
private static final Pattern PATTERN = Pattern.compile("^(\\d+) (.*)$");
private static CompilerMessageMatcher read(String line) {
line = line.trim();
if (line.isEmpty()) return null;
boolean optional = false;
if (line.startsWith("OPTIONAL ")) {
line = line.substring(9);
optional = true;
}
String[] parts = line.split("\\s*\\|\\|\\|\\|\\s*");
CompilerMessageMatcher cmm = new CompilerMessageMatcher();
cmm.optional = optional;
for (String part : parts) {
Matcher m = PATTERN.matcher(part);
if (!m.matches()) throw new IllegalArgumentException("Typo in test file: " + line);
cmm.lineNumbers.add(Integer.parseInt(m.group(1)));
cmm.messages.add(Arrays.asList(m.group(2).split("\\s+")));
}
return cmm;
}
}