com.atlassian.connector.eclipse.ui.actions.AbstractResourceAction.java Source code

Introduction

Here is the source code for com.atlassian.connector.eclipse.ui.actions.AbstractResourceAction.java

Source

/*******************************************************************************
 * Copyright (c) 2009 Atlassian and others.
 * 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:
 *     Atlassian - initial API and implementation
 ******************************************************************************/

package com.atlassian.connector.eclipse.ui.actions;

import com.atlassian.connector.eclipse.ui.commons.AtlassianUiUtil;
import com.atlassian.connector.eclipse.ui.commons.IEditorResource;
import com.atlassian.connector.eclipse.ui.commons.ResourceEditorBean;

import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.jface.action.IAction;
import org.eclipse.jface.text.source.LineRange;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.viewers.StructuredSelection;
import org.eclipse.mylyn.internal.provisional.commons.ui.WorkbenchUtil;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.IActionDelegate;
import org.eclipse.ui.IEditorInput;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.IWorkbenchWindow;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.actions.BaseSelectionListenerAction;

import java.util.ArrayList;
import java.util.List;

public abstract class AbstractResourceAction extends BaseSelectionListenerAction implements IActionDelegate {

    private IWorkbenchWindow workbenchWindow;

    private List<ResourceEditorBean> selectionData;

    protected AbstractResourceAction(String text) {
        super(text);
    }

    protected List<ResourceEditorBean> getSelectionData() {
        return selectionData;
    }

    public void dispose() {
    }

    public void init(IWorkbenchWindow window) {
        this.workbenchWindow = window;
    }

    public void run(IAction action) {
        if (selectionData != null && selectionData.get(0) != null) {
            processResources(selectionData, WorkbenchUtil.getShell());
        }
    }

    private List<ResourceEditorBean> getData(ISelection selection) {
        List<ResourceEditorBean> ret = new ArrayList<ResourceEditorBean>();
        if (selection instanceof IStructuredSelection) {
            final IStructuredSelection structuredSelection = (IStructuredSelection) selection;

            Object[] selectedObjects = structuredSelection.toArray();

            for (Object selectedObject : selectedObjects) {

                if (selectedObject instanceof IEditorResource) {
                    IEditorResource a = (IEditorResource) selectedObject;
                    ret.add(new ResourceEditorBean(a.getResource(), a.getLineRange()));

                } else if (structuredSelection.getFirstElement() instanceof IAdaptable) {
                    IResource resource = null;
                    LineRange lineRange = null;
                    resource = (IResource) ((IAdaptable) structuredSelection.getFirstElement())
                            .getAdapter(IResource.class);
                    lineRange = getJavaEditorSelection(structuredSelection);
                    ret.add(new ResourceEditorBean(resource, lineRange));
                }
            }
        } else {
            IEditorPart activeEditor = getActiveEditor();
            if (activeEditor != null) {
                IEditorInput editorInput = getEditorInputFromSelection(selection);
                if (editorInput != null) {
                    IResource resource = null;
                    LineRange lineRange = null;
                    resource = (IResource) editorInput.getAdapter(IResource.class);
                    // such call:
                    //            lineRange = new LineRange(textSelection.getStartLine(), textSelection.getEndLine()
                    //                  - textSelection.getStartLine());
                    // does not work (i.e. it returns previously selected text region rather than selected now ?!?
                    lineRange = AtlassianUiUtil.getSelectedLineNumberRangeFromEditorInput(activeEditor,
                            activeEditor.getEditorInput());
                    ret.add(new ResourceEditorBean(resource, lineRange));
                }
            }
        }
        return ret;
    }

    @Override
    protected boolean updateSelection(IStructuredSelection selection) {
        selectionData = getData(selection);
        return super.updateSelection(selection);
    }

    public void selectionChanged(IAction action, ISelection selection) {
        if (selection instanceof IStructuredSelection) {
            selectionChanged((IStructuredSelection) selection);
        } else {
            selectionChanged(StructuredSelection.EMPTY);
        }
        action.setEnabled(isEnabled());
    }

    private IEditorPart getActiveEditor() {
        IWorkbenchWindow window = workbenchWindow;
        if (window == null) {
            window = PlatformUI.getWorkbench().getActiveWorkbenchWindow();
        }
        if (window != null && window.getActivePage() != null) {
            return window.getActivePage().getActiveEditor();
        }
        return null;
    }

    private IEditorInput getEditorInputFromSelection(ISelection selection) {
        if (selection instanceof IStructuredSelection) {
            IStructuredSelection structuredSelection = ((IStructuredSelection) selection);
            if (structuredSelection.getFirstElement() instanceof IEditorInput) {
                return (IEditorInput) structuredSelection.getFirstElement();
            }
        }
        return null;
    }

    private LineRange getJavaEditorSelection(ISelection selection) {
        IEditorPart editorPart = getActiveEditor();
        IEditorInput editorInput = getEditorInputFromSelection(selection);
        if (editorInput != null && editorPart != null) {
            return AtlassianUiUtil.getSelectedLineNumberRangeFromEditorInput(editorPart, editorInput);
        }
        return null;
    }

    protected abstract void processResources(List<ResourceEditorBean> selection, final Shell shell);

}