designer/src/com/android/tools/idea/uibuilder/lint/LintAnnotationsModel.java

/*
 * Copyright (C) 2015 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.android.tools.idea.uibuilder.lint;

import com.android.annotations.NonNull;
import com.android.annotations.Nullable;
import com.android.tools.idea.uibuilder.model.NlComponent;
import com.google.common.collect.ArrayListMultimap;
import com.google.common.collect.ListMultimap;
import com.intellij.codeHighlighting.HighlightDisplayLevel;
import icons.AndroidIcons;
import org.jetbrains.android.inspections.lint.AndroidLintInspectionBase;

import javax.swing.*;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;

public class LintAnnotationsModel {
  /** A map from a component to a list of issues for that component. */
  private ListMultimap<NlComponent, IssueData> myIssues;

  @NonNull
  public Collection<NlComponent> getComponentsWithIssues() {
    return myIssues == null ? Collections.<NlComponent>emptyList() : myIssues.keySet();
  }

  @Nullable
  public Icon getIssueIcon(@NonNull NlComponent component) {
    List<IssueData> issueData = myIssues.get(component);
    if (issueData == null || issueData.isEmpty()) {
      return null;
    }

    IssueData max = findHighestSeverityIssue(issueData);
    return HighlightDisplayLevel.ERROR.equals(max.level) ? AndroidIcons.ErrorBadge : AndroidIcons.WarningBadge;
  }

  public String getIssueMessage(@NonNull NlComponent component) {
    List<IssueData> issueData = myIssues.get(component);
    if (issueData == null || issueData.isEmpty()) {
      return null;
    }

    IssueData max = findHighestSeverityIssue(issueData);
    return max.inspection.getStaticDescription();
  }

  private static IssueData findHighestSeverityIssue(List<IssueData> issueData) {
    return Collections.max(issueData, new Comparator<IssueData>() {
      @Override
      public int compare(IssueData o1, IssueData o2) {
        return o1.level.getSeverity().compareTo(o2.level.getSeverity());
      }
    });
  }

  public void addIssue(@NonNull NlComponent component,
                       @NonNull AndroidLintInspectionBase inspection,
                       @NonNull HighlightDisplayLevel level) {
    if (myIssues == null) {
      myIssues = ArrayListMultimap.create();
    }

    myIssues.put(component, new IssueData(inspection, level));
  }

  private static class IssueData {
    @NonNull public final AndroidLintInspectionBase inspection;
    @NonNull public final HighlightDisplayLevel level;

    private IssueData(@NonNull AndroidLintInspectionBase inspection, @NonNull HighlightDisplayLevel level) {
      this.inspection = inspection;
      this.level = level;
    }
  }
}