CollapseAllHandler.java

/*
 * Copyright 2016 Nokia Solutions and Networks
 * Licensed under the Apache License, Version 2.0,
 * see license.txt file for details.
 */
package org.robotframework.ide.eclipse.main.plugin.tableeditor.source.handler;

import javax.inject.Named;

import org.eclipse.e4.core.di.annotations.Execute;
import org.eclipse.jface.text.source.projection.ProjectionViewer;
import org.eclipse.ui.ISources;
import org.robotframework.ide.eclipse.main.plugin.tableeditor.RobotFormEditor;
import org.robotframework.ide.eclipse.main.plugin.tableeditor.source.handler.CollapseAllHandler.E4CollapseAllHandler;
import org.robotframework.red.commands.DIParameterizedHandler;
import org.robotframework.red.jface.text.ProjectionViewerWrapper;

import com.google.common.annotations.VisibleForTesting;

public class CollapseAllHandler extends DIParameterizedHandler<E4CollapseAllHandler> {

    public CollapseAllHandler() {
        super(E4CollapseAllHandler.class);
    }

    public static class E4CollapseAllHandler {

        @Execute
        public void collapseAll(final @Named(ISources.ACTIVE_EDITOR_NAME) RobotFormEditor editor) {
            collapseAll(ProjectionViewerWrapper.from(editor));
        }

        @VisibleForTesting
        void collapseAll(final ProjectionViewerWrapper projectionViewer) {
            if (projectionViewer.canDoOperation(ProjectionViewer.COLLAPSE_ALL)) {
                projectionViewer.doOperation(ProjectionViewer.COLLAPSE_ALL);
            }
        }
    }
}