integration/intellij/src/main/java/ai/vespa/intellij/schema/findUsages/SdFindUsagesHandlerFactory.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package ai.vespa.intellij.schema.findUsages;

import com.intellij.find.findUsages.FindUsagesHandler;
import com.intellij.find.findUsages.FindUsagesHandlerFactory;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiNamedElement;

/**
 * This class is used for the extension (in plugin.xml) to the class SdFindUsagesHandler.
 *
 * @author Shahar Ariel
 */
public class SdFindUsagesHandlerFactory extends FindUsagesHandlerFactory {
    
    @Override
    public boolean canFindUsages(PsiElement element) {
        if (element == null) return false;
        if (element.getContainingFile() == null) return false;
        if (element.getContainingFile().getVirtualFile() == null) return false;
        String extension = element.getContainingFile().getVirtualFile().getExtension();
        return ("sd".equals(extension) || "profile".equals(extension));
    }
    
    @Override
    public FindUsagesHandler createFindUsagesHandler(PsiElement element, boolean forHighlightUsages) {
        return new SdFindUsagesHandler(element);
    }

}