PySearchIndexQuery.java example

Explorer
Pydev-master
- plugins
/**
 * Copyright (c) 20015 by Brainwy Software Ltda. All Rights Reserved.
 * Licensed under the terms of the Eclipse Public License (EPL).
 * Please see the license.txt included with this distribution for details.
 * Any modifications to this file must keep this entire header intact.
 */
package com.python.pydev.analysis.search_index;

import java.io.File;
import java.util.List;
import java.util.Set;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.OperationCanceledException;
import org.eclipse.core.runtime.Status;
import org.eclipse.core.runtime.SubProgressMonitor;
import org.eclipse.jface.text.IDocument;
import org.eclipse.search.ui.ISearchResult;
import org.eclipse.search.ui.text.AbstractTextSearchResult;
import org.python.pydev.core.FileUtilsFileBuffer;
import org.python.pydev.core.IPythonNature;
import org.python.pydev.core.MisconfigurationException;
import org.python.pydev.core.ModulesKey;
import org.python.pydev.core.docutils.PySelection;
import org.python.pydev.core.log.Log;
import org.python.pydev.editorinput.PySourceLocatorBase;
import org.python.pydev.shared_core.string.StringUtils;
import org.python.pydev.shared_core.structure.OrderedMap;
import org.python.pydev.shared_ui.search.AbstractSearchIndexQuery;
import org.python.pydev.shared_ui.search.SearchIndexData;
import org.python.pydev.shared_ui.search.SearchIndexResult;
import org.python.pydev.shared_ui.search.SearchResultUpdater;
import org.python.pydev.shared_ui.search.StringMatcherWithIndexSemantics;

import com.python.pydev.analysis.additionalinfo.AbstractAdditionalDependencyInfo;
import com.python.pydev.analysis.additionalinfo.AdditionalProjectInterpreterInfo;
import com.python.pydev.analysis.additionalinfo.IReferenceSearches;

/**
 * Searches the internal indexes from PyDev.
 */
public class PySearchIndexQuery extends AbstractSearchIndexQuery {

    private SearchIndexResult fResult;

    public PySearchIndexQuery(String text) {
        super(text);
    }

    public PySearchIndexQuery(SearchIndexData data) {
        super(data);
    }

    @Override
    public IStatus run(IProgressMonitor monitor) throws OperationCanceledException {
        SearchIndexResult searchResult = (SearchIndexResult) getSearchResult();
        //Remove all so that we don't get duplicates on a search refresh.
        searchResult.removeAll();

        StringMatcherWithIndexSemantics stringMatcher = createStringMatcher();

        Set<String> moduleNamesFilter = scopeAndData.getModuleNamesFilter();
        OrderedMap<String, Set<String>> fieldNameToValues = new OrderedMap<>();
        if (moduleNamesFilter != null && !moduleNamesFilter.isEmpty()) {
            fieldNameToValues.put(IReferenceSearches.FIELD_MODULE_NAME, moduleNamesFilter);
        }
        Set<String> split = makeTextFieldPatternsToSearchFromText();
        fieldNameToValues.put(IReferenceSearches.FIELD_CONTENTS, split);

        final List<IPythonNature> pythonNatures = PyScopeAndData.getPythonNatures(scopeAndData);
        monitor.beginTask("Search indexes", pythonNatures.size());
        try {
            for (IPythonNature nature : pythonNatures) {
                AbstractAdditionalDependencyInfo info;
                try {
                    info = AdditionalProjectInterpreterInfo.getAdditionalInfoForProject(nature);
                } catch (MisconfigurationException e) {
                    Log.log(e);
                    continue;
                }
                IReferenceSearches referenceSearches = info.getReferenceSearches();
                List<ModulesKey> search = referenceSearches.search(nature.getProject(), fieldNameToValues,
                        new SubProgressMonitor(monitor, 1));

                IFile workspaceFile;
                for (ModulesKey modulesKey : search) {
                    File file = modulesKey.file;
                    if (file == null || !file.exists()) {
                        Log.logInfo(StringUtils.format("Ignoring: %s. File no longer exists.", file));
                    }

                    workspaceFile = new PySourceLocatorBase().getWorkspaceFile(file, nature.getProject());
                    if (workspaceFile == null) {
                        Log.logInfo(StringUtils
                                .format("Ignoring: %s. Unable to resolve to a file in the Eclipse workspace.", file));
                        continue;
                    }

                    IDocument doc = FileUtilsFileBuffer.getDocFromResource(workspaceFile);
                    String text = doc.get();
                    createMatches(doc, text, stringMatcher, workspaceFile, searchResult, modulesKey);
                }
            }
        } finally {
            monitor.done();
        }

        return Status.OK_STATUS;
    }

    public void createMatches(IDocument doc, String text, StringMatcherWithIndexSemantics stringMatcher,
            IFile workspaceFile,
            AbstractTextSearchResult searchResult, ModulesKey modulesKey) {

        StringMatcherWithIndexSemantics.Position find = stringMatcher.find(text, 0);
        while (find != null) {
            int offset = find.getStart();
            int end = find.getEnd();
            int length = end - offset;

            PySelection ps = new PySelection(doc, offset);
            int lineNumber = ps.getLineOfOffset();
            String lineContents = ps.getLine(lineNumber);
            int lineStartOffset = ps.getLineOffset(lineNumber);

            PyModuleLineElement element = new PyModuleLineElement(workspaceFile, lineNumber, lineStartOffset,
                    lineContents,
                    modulesKey);
            searchResult.addMatch(new PyModuleMatch(workspaceFile, offset, length, element, modulesKey));
            find = stringMatcher.find(text, end);
        }
    }

    @Override
    public String getLabel() {
        return "PyDev Index Search";
    }

    @Override
    public ISearchResult getSearchResult() {
        if (fResult == null) {
            fResult = new PySearchResult(this);
            new SearchResultUpdater(fResult);
        }
        return fResult;
    }

}