AddTaskNodeAction.java

package org.jboss.tools.smooks.graphical.actions;

import java.util.List;

import org.eclipse.ui.IEditorPart;
import org.jboss.tools.smooks.editor.ISmooksModelProvider;
import org.jboss.tools.smooks.graphical.editors.TaskTypeManager;
import org.jboss.tools.smooks.graphical.editors.process.TaskType;

public class AddTaskNodeAction extends AbstractProcessGraphAction {
	
	protected String taskID = null;
	
	protected TaskTypeRules rules = new TaskTypeRules();

	public AddTaskNodeAction(String taskID, String text, ISmooksModelProvider provider, IEditorPart editorPart) {
		super(text, provider,editorPart);
		this.taskID = taskID;
	}
	
	public String getTaskID() {
		return taskID;
	}

	@Override
	protected void init() {
		// TODO Auto-generated method stub
		super.init();
	}

	@Override
	public void run() {
		// TODO Auto-generated method stub
		super.run();
	}

	@Override
	public void update() {
		this.setEnabled(false);
		List<TaskType> currentSelectedTask = this.getCurrentSelectedTask();
		if (currentSelectedTask != null && currentSelectedTask.size() == 1) {
			TaskType currentTask = currentSelectedTask.get(0);
			
			if(currentTask.getTask().isEmpty()) {
				String taskID = currentTask.getId();
				if (taskID != null) {
					if (taskID.equals(TaskTypeManager.TASK_ID_INPUT) || taskID.equals(TaskTypeManager.TASK_ID_JAVA_MAPPING))
						this.setEnabled(true);
				}
			}
		}
	}

}