package com.intellij.codeInsight.daemon.impl;

import com.intellij.lang.ASTNode;
import com.intellij.lang.annotation.Annotation;
import com.intellij.lang.annotation.AnnotationHolder;
import com.intellij.lang.annotation.HighlightSeverity;
import com.intellij.openapi.util.TextRange;
import com.intellij.psi.PsiElement;
import com.intellij.util.SmartList;
import com.intellij.xml.util.XmlUtil;

/**
 * Created by IntelliJ IDEA.
 * User: max
 * Date: Feb 6, 2005
 * Time: 5:37:26 PM
 * To change this template use File | Settings | File Templates.
 */
public class AnnotationHolderImpl extends SmartList<Annotation> implements AnnotationHolder {
  public Annotation createErrorAnnotation(PsiElement elt, String message) {
    return createAnnotation(elt.getTextRange(), HighlightSeverity.ERROR, message);
  }

  public Annotation createErrorAnnotation(ASTNode node, String message) {
    return createAnnotation(node.getTextRange(), HighlightSeverity.ERROR, message);
  }

  public Annotation createErrorAnnotation(TextRange range, String message) {
    return createAnnotation(range, HighlightSeverity.ERROR, message);
  }

  public Annotation createWarningAnnotation(PsiElement elt, String message) {
    return createAnnotation(elt.getTextRange(), HighlightSeverity.WARNING, message);
  }

  public Annotation createWarningAnnotation(ASTNode node, String message) {
    return createAnnotation(node.getTextRange(), HighlightSeverity.WARNING, message);
  }

  public Annotation createWarningAnnotation(TextRange range, String message) {
    return createAnnotation(range, HighlightSeverity.WARNING, message);
  }

  public Annotation createInfoAnnotation(PsiElement elt, String message) {
    return createAnnotation(elt.getTextRange(), HighlightSeverity.INFORMATION, message);
  }

  public Annotation createInfoAnnotation(ASTNode node, String message) {
    return createAnnotation(node.getTextRange(), HighlightSeverity.INFORMATION, message);
  }

  public Annotation createInfoAnnotation(TextRange range, String message) {
    return createAnnotation(range, HighlightSeverity.INFORMATION, message);
  }

  private final Annotation createAnnotation(TextRange range, HighlightSeverity severity, String message) {
    final String tooltip = message == null ? null : "<html><body>" + XmlUtil.escapeString(message) + "</body></html>";
    Annotation annotation = new Annotation(range.getStartOffset(), range.getEndOffset(), severity, message, tooltip);
    add(annotation);
    return annotation;
  }

  public boolean hasAnnotations() {
    return size() > 0;
  }

  public Annotation[] getResult() {
    return toArray(new Annotation[size()]);
  }
}