RemoveVariableCommand.java

/*
 * Copyright 2015 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.model.cmd.variables;

import static com.google.common.collect.Lists.newArrayList;

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

import org.rf.ide.core.testdata.model.table.VariableTable;
import org.robotframework.ide.eclipse.main.plugin.model.RobotModelEvents;
import org.robotframework.ide.eclipse.main.plugin.model.RobotSuiteFileSection;
import org.robotframework.ide.eclipse.main.plugin.model.RobotVariable;
import org.robotframework.ide.eclipse.main.plugin.tableeditor.EditorCommand;

public class RemoveVariableCommand extends EditorCommand {

    private final List<RobotVariable> variablesToDelete;

    private List<Integer> deletedVariablesIndexes = newArrayList();

    public RemoveVariableCommand(final List<RobotVariable> variablesToDelete) {
        this.variablesToDelete = variablesToDelete;
    }

    @Override
    public void execute() throws CommandExecutionException {
        if (variablesToDelete.isEmpty()) {
            return;
        }

        for (final RobotVariable var : variablesToDelete) {
            deletedVariablesIndexes.add(var.getIndex());
        }

        final RobotSuiteFileSection variablesSection = variablesToDelete.get(0).getParent();
        variablesSection.getChildren().removeAll(variablesToDelete);

        final VariableTable table = (VariableTable) variablesSection.getLinkedElement();
        for (final RobotVariable var : variablesToDelete) {
            table.removeVariable(var.getLinkedElement());

        }
        eventBroker.send(RobotModelEvents.ROBOT_VARIABLE_REMOVED, variablesSection);
    }

    @Override
    public List<EditorCommand> getUndoCommands() {
        return newUndoCommands(setupUndoCommandsForDeletedVariables());
    }

    private List<EditorCommand> setupUndoCommandsForDeletedVariables() {
        final List<EditorCommand> commands = new ArrayList<>();
        if (variablesToDelete.size() == deletedVariablesIndexes.size()) {
            for (int i = 0; i < variablesToDelete.size(); i++) {
                final RobotVariable var = variablesToDelete.get(i);
                commands.add(new InsertVariablesCommand(var.getParent(), deletedVariablesIndexes.get(i),
                        new RobotVariable[] { var }));
            }
        }
        return commands;
    }
}