package jdk.javadoc.internal.tool;
import java.io.PrintWriter;
import java.util.Locale;
import java.util.ResourceBundle;
import javax.lang.model.element.Element;
import javax.tools.Diagnostic.Kind;
import com.sun.tools.javac.util.Context.Factory;
import jdk.javadoc.doclet.Reporter;
import com.sun.source.tree.CompilationUnitTree;
import com.sun.source.util.DocSourcePositions;
import com.sun.source.util.DocTreePath;
import com.sun.source.util.TreePath;
import com.sun.tools.javac.api.JavacTrees;
import com.sun.tools.javac.tree.JCTree;
import com.sun.tools.javac.util.Context;
import com.sun.tools.javac.util.JCDiagnostic;
import com.sun.tools.javac.util.JCDiagnostic.DiagnosticType;
import com.sun.tools.javac.util.JavacMessages;
import com.sun.tools.javac.util.Log;
public class Messager extends Log implements Reporter {
final Context context;
public static Messager instance0(Context context) {
Log instance = context.get(logKey);
if (instance == null || !(instance instanceof Messager))
throw new InternalError("no messager instance!");
return (Messager)instance;
}
public static void preRegister(Context context,
final String programName) {
context.put(logKey, (Factory<Log>)c -> new Messager(c, programName));
}
public static void preRegister(Context context, final String programName,
final PrintWriter outWriter, final PrintWriter errWriter) {
context.put(logKey, (Factory<Log>)c -> new Messager(c, programName, outWriter, errWriter));
}
@Override
public void print(Kind kind, String msg) {
switch (kind) {
case ERROR:
printError(msg);
return;
case WARNING:
case MANDATORY_WARNING:
printWarning(msg);
return;
default:
printNotice(msg);
return;
}
}
@Override
public void print(Kind kind, DocTreePath path, String msg) {
switch (kind) {
case ERROR:
printError(path, msg);
return;
case WARNING:
case MANDATORY_WARNING:
printWarning(path, msg);
return;
default:
printWarning(path, msg);
return;
}
}
@Override
public void print(Kind kind, Element e, String msg) {
switch (kind) {
case ERROR:
printError(e, msg);
return;
case WARNING:
case MANDATORY_WARNING:
printWarning(e, msg);
return;
default:
printWarning(e, msg);
return;
}
}
final String programName;
private Locale locale;
private final JavacMessages messages;
private final JCDiagnostic.Factory javadocDiags;
static final PrintWriter defaultOutWriter = new PrintWriter(System.out);
static final PrintWriter defaultErrWriter = new PrintWriter(System.err);
public Messager(Context context, String programName) {
this(context, programName, defaultOutWriter, defaultErrWriter);
}
@SuppressWarnings("deprecation")
public Messager(Context context, String programName, PrintWriter outWriter, PrintWriter errWriter) {
super(context, errWriter, errWriter, outWriter);
messages = JavacMessages.instance(context);
messages.add(locale -> ResourceBundle.getBundle("jdk.javadoc.internal.tool.resources.javadoc",
locale));
javadocDiags = new JCDiagnostic.Factory(messages, "javadoc");
this.programName = programName;
this.context = context;
locale = Locale.getDefault();
}
public void setLocale(Locale locale) {
this.locale = locale;
}
String getText(String key, Object... args) {
return messages.getLocalizedString(locale, key, args);
}
private String getDiagSource(DocTreePath path) {
if (path == null || path.getTreePath() == null) {
return programName;
}
JavacTrees trees = JavacTrees.instance(context);
DocSourcePositions sourcePositions = trees.getSourcePositions();
CompilationUnitTree cu = path.getTreePath().getCompilationUnit();
long spos = sourcePositions.getStartPosition(cu, path.getDocComment(), path.getLeaf());
long lineNumber = cu.getLineMap().getLineNumber(spos);
String fname = cu.getSourceFile().getName();
String posString = fname + ":" + lineNumber;
return posString;
}
private String getDiagSource(Element e) {
if (e == null) {
return programName;
}
JavacTrees trees = JavacTrees.instance(context);
TreePath path = trees.getPath(e);
DocSourcePositions sourcePositions = trees.getSourcePositions();
JCTree tree = trees.getTree(e);
CompilationUnitTree cu = path.getCompilationUnit();
long spos = sourcePositions.getStartPosition(cu, tree);
long lineNumber = cu.getLineMap().getLineNumber(spos);
String fname = cu.getSourceFile().getName();
String posString = fname + ":" + lineNumber;
return posString;
}
public void printError(String msg) {
printError((DocTreePath)null, msg);
}
public void printError(DocTreePath path, String msg) {
String prefix = getDiagSource(path);
if (diagListener != null) {
report(DiagnosticType.ERROR, prefix, msg);
return;
}
printError(prefix, msg);
}
public void printError(Element e, String msg) {
String prefix = getDiagSource(e);
if (diagListener != null) {
report(DiagnosticType.ERROR, prefix, msg);
return;
}
printError(prefix, msg);
}
public void printErrorUsingKey(String key, Object... args) {
printError((Element)null, getText(key, args));
}
private void printError(String prefix, String msg) {
if (nerrors < MaxErrors) {
PrintWriter errWriter = getWriter(WriterKind.ERROR);
printRawLines(errWriter, prefix + ": " + getText("javadoc.error") + " - " + msg);
errWriter.flush();
prompt();
nerrors++;
}
}
public void printWarning(String msg) {
printWarning((DocTreePath)null, msg);
}
public void printWarningUsingKey(String key, Object... args) {
printWarning((Element)null, getText(key, args));
}
public void printWarning(Element e, String key, Object... args) {
printWarning(getText(key, args));
}
public void printWarning(DocTreePath path, String msg) {
String prefix = getDiagSource(path);
if (diagListener != null) {
report(DiagnosticType.WARNING, prefix, msg);
return;
}
printWarning(prefix, msg);
}
public void printWarning(Element e, String msg) {
String prefix = getDiagSource(e);
if (diagListener != null) {
report(DiagnosticType.WARNING, prefix, msg);
return;
}
printWarning(prefix, msg);
}
private void printWarning(String prefix, String msg) {
if (nwarnings < MaxWarnings) {
PrintWriter warnWriter = getWriter(WriterKind.WARNING);
printRawLines(warnWriter, prefix + ": " + getText("javadoc.warning") + " - " + msg);
warnWriter.flush();
nwarnings++;
}
}
public void printNotice(String msg) {
printNotice((DocTreePath)null, msg);
}
public void printNotice(DocTreePath path, String msg) {
String prefix = getDiagSource(path);
if (diagListener != null) {
report(DiagnosticType.NOTE, null, prefix + ": " + msg);
return;
}
PrintWriter noticeWriter = getWriter(WriterKind.NOTICE);
if (path == null) {
printRawLines(noticeWriter, msg);
} else {
printRawLines(noticeWriter, prefix + ": " + msg);
}
noticeWriter.flush();
}
public void printNotice(Element e, String msg) {
String pos = getDiagSource(e);
if (diagListener != null) {
report(DiagnosticType.NOTE, pos, msg);
return;
}
PrintWriter noticeWriter = getWriter(WriterKind.NOTICE);
if (e == null) {
printRawLines(noticeWriter, msg);
} else {
printRawLines(noticeWriter, pos + ": " + msg);
}
noticeWriter.flush();
}
public void notice(String key, Object... args) {
printNotice(getText(key, args));
}
public boolean hasErrors() {
return nerrors != 0;
}
public boolean hasWarnings() {
return nwarnings != 0;
}
public void printErrorWarningCounts() {
if (nerrors > 0) {
notice((nerrors > 1) ? "main.errors" : "main.error",
"" + nerrors);
}
if (nwarnings > 0) {
notice((nwarnings > 1) ? "main.warnings" : "main.warning",
"" + nwarnings);
}
}
private void report(DiagnosticType type, String pos, String msg) {
switch (type) {
case ERROR:
case WARNING:
Object prefix = (pos == null) ? programName : pos;
report(javadocDiags.create(type, null, null, "msg", prefix, msg));
break;
case NOTE:
String key = (pos == null) ? "msg" : "pos.msg";
report(javadocDiags.create(type, null, null, key, pos, msg));
break;
default:
throw new IllegalArgumentException(type.toString());
}
}
}