package com.jetbrains.python.edu.highlighting;

import com.intellij.psi.PsiFile;
import com.jetbrains.python.edu.StudyTaskManager;
import com.jetbrains.python.inspections.PythonVisitorFilter;
import com.jetbrains.python.inspections.unresolvedReference.PyUnresolvedReferencesInspection;
import org.jetbrains.annotations.NotNull;

public class StudyVisitorFilter implements PythonVisitorFilter {
  @Override
  public boolean isSupported(@NotNull final Class visitorClass, @NotNull final PsiFile file) {
    if (StudyTaskManager.getInstance(file.getProject()).getCourse() == null) return true;
    if (visitorClass == PyUnresolvedReferencesInspection.class) {
      return false;
    }
    return true;
  }
}