FindReferencesAction.java

/*******************************************************************************
 * Copyright (c) 2012-2017 Codenvy, S.A.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   Codenvy, S.A. - initial API and implementation
 *******************************************************************************/
package org.eclipse.che.plugin.languageserver.ide.navigation.references;

import com.google.inject.Inject;

import org.eclipse.che.api.promises.client.Promise;
import org.eclipse.che.ide.api.action.AbstractPerspectiveAction;
import org.eclipse.che.ide.api.action.ActionEvent;
import org.eclipse.che.ide.api.editor.EditorAgent;
import org.eclipse.che.ide.api.editor.EditorPartPresenter;
import org.eclipse.che.ide.api.editor.editorconfig.TextEditorConfiguration;
import org.eclipse.che.ide.api.editor.texteditor.TextEditor;
import org.eclipse.che.ide.dto.DtoFactory;
import org.eclipse.che.plugin.languageserver.ide.editor.LanguageServerEditorConfiguration;
import org.eclipse.che.plugin.languageserver.ide.location.OpenLocationPresenter;
import org.eclipse.che.plugin.languageserver.ide.location.OpenLocationPresenterFactory;
import org.eclipse.che.plugin.languageserver.ide.service.TextDocumentServiceClient;
import org.eclipse.lsp4j.Location;
import org.eclipse.lsp4j.Position;
import org.eclipse.lsp4j.ReferenceContext;
import org.eclipse.lsp4j.ReferenceParams;
import org.eclipse.lsp4j.ServerCapabilities;
import org.eclipse.lsp4j.TextDocumentIdentifier;

import javax.validation.constraints.NotNull;
import java.util.List;

import static java.util.Collections.singletonList;
import static org.eclipse.che.ide.workspace.perspectives.project.ProjectPerspective.PROJECT_PERSPECTIVE_ID;

/**
 * @author Evgen Vidolob
 */
public class FindReferencesAction extends AbstractPerspectiveAction {

    private final EditorAgent               editorAgent;
    private final TextDocumentServiceClient client;
    private final DtoFactory                dtoFactory;
    private final OpenLocationPresenter     presenter;

    @Inject
    public FindReferencesAction(EditorAgent editorAgent, OpenLocationPresenterFactory presenterFactory,
                                TextDocumentServiceClient client, DtoFactory dtoFactory) {
        super(singletonList(PROJECT_PERSPECTIVE_ID), "Find References", "Find References", null, null);
        this.editorAgent = editorAgent;
        this.client = client;
        this.dtoFactory = dtoFactory;
        presenter = presenterFactory.create("Find References");
    }

    @Override
    public void updateInPerspective(@NotNull ActionEvent event) {
        EditorPartPresenter activeEditor = editorAgent.getActiveEditor();
        if (activeEditor instanceof TextEditor) {
            TextEditorConfiguration configuration = ((TextEditor)activeEditor).getConfiguration();
            if (configuration instanceof LanguageServerEditorConfiguration) {
                ServerCapabilities capabilities = ((LanguageServerEditorConfiguration)configuration).getServerCapabilities();
                event.getPresentation()
                     .setEnabledAndVisible(capabilities.getReferencesProvider() != null && capabilities.getReferencesProvider());
                return;
            }
        }
        event.getPresentation().setEnabledAndVisible(false);
    }

    @Override
    public void actionPerformed(ActionEvent e) {
        EditorPartPresenter activeEditor = editorAgent.getActiveEditor();

        //TODO replace this
        if (!(activeEditor instanceof TextEditor)) {
            return;
        }
        TextEditor textEditor = ((TextEditor)activeEditor);
        String path = activeEditor.getEditorInput().getFile().getLocation().toString();
        ReferenceParams paramsDTO = dtoFactory.createDto(ReferenceParams.class);

        Position Position = dtoFactory.createDto(Position.class);
        Position.setLine(textEditor.getCursorPosition().getLine());
        Position.setCharacter(textEditor.getCursorPosition().getCharacter());

        TextDocumentIdentifier identifierDTO = dtoFactory.createDto(TextDocumentIdentifier.class);
        identifierDTO.setUri(path);

        ReferenceContext contextDTO = dtoFactory.createDto(ReferenceContext.class);
        contextDTO.setIncludeDeclaration(true);

        paramsDTO.setUri(path);
        paramsDTO.setPosition(Position);
        paramsDTO.setTextDocument(identifierDTO);
        paramsDTO.setContext(contextDTO);
        Promise<List<Location>> promise = client.references(paramsDTO);
        presenter.openLocation(promise);
    }
}