com.ecfeed.ui.modelif.RootInterface.java Source code

Introduction

Here is the source code for com.ecfeed.ui.modelif.RootInterface.java

Source

/*******************************************************************************
 *
 * Copyright (c) 2016 ecFeed AS.                                                
 * 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 
 *  
 *******************************************************************************/

package com.ecfeed.ui.modelif;

import java.util.Collection;

import org.eclipse.jdt.core.IType;
import org.eclipse.jface.dialogs.IDialogConstants;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.swt.widgets.Display;

import com.ecfeed.core.adapter.IModelOperation;
import com.ecfeed.core.adapter.ModelOperationException;
import com.ecfeed.core.adapter.operations.RootOperationAddClasses;
import com.ecfeed.core.adapter.operations.RootOperationAddNewClass;
import com.ecfeed.core.model.ClassNode;
import com.ecfeed.core.model.RootNode;
import com.ecfeed.ui.common.Constants;
import com.ecfeed.ui.common.EclipseModelBuilder;
import com.ecfeed.ui.common.Messages;
import com.ecfeed.ui.common.utils.IFileInfoProvider;
import com.ecfeed.ui.dialogs.TestClassImportDialog;

public class RootInterface extends GlobalParametersParentInterface {

    public RootInterface(IModelUpdateContext updateContext, IFileInfoProvider fileInfoProvider) {
        super(updateContext, fileInfoProvider);
    }

    @Override
    protected RootNode getTarget() {
        return (RootNode) super.getTarget();
    }

    public ClassNode addNewClass() {
        return addNewClass(generateClassName());
    }

    public ClassNode addNewClass(String className) {
        ClassNode addedClass = new ClassNode(className);
        if (execute(new RootOperationAddNewClass(getTarget(), addedClass, getTarget().getClasses().size()),
                Messages.DIALOG_ADD_NEW_CLASS_PROBLEM_TITLE)) {
            return addedClass;
        }
        return null;
    }

    public ClassNode addImplementedClass() {
        TestClassImportDialog dialog = new TestClassImportDialog(Display.getCurrent().getActiveShell());

        if (dialog.open() == IDialogConstants.OK_ID) {
            IType selectedClass = (IType) dialog.getFirstResult();
            boolean testOnly = dialog.getTestOnlyFlag();

            if (selectedClass != null) {
                ClassNode classModel;
                try {
                    classModel = new EclipseModelBuilder().buildClassModel(selectedClass, testOnly);
                    if (execute(
                            new RootOperationAddNewClass(getTarget(), classModel, getTarget().getClasses().size()),
                            Messages.DIALOG_ADD_NEW_CLASS_PROBLEM_TITLE)) {
                        return classModel;
                    }
                } catch (ModelOperationException e) {
                    MessageDialog.openError(Display.getCurrent().getActiveShell(),
                            Messages.DIALOG_ADD_NEW_CLASS_PROBLEM_TITLE, e.getMessage());
                }
            }
        }
        return null;
    }

    public boolean removeClasses(Collection<ClassNode> removedClasses) {
        if (MessageDialog.openConfirm(Display.getCurrent().getActiveShell(), Messages.DIALOG_REMOVE_CLASSES_TITLE,
                Messages.DIALOG_REMOVE_CLASSES_MESSAGE)) {
            return removeChildren(removedClasses, Messages.DIALOG_REMOVE_CLASSES_PROBLEM_TITLE);
        }
        return false;
    }

    public boolean addClasses(Collection<ClassNode> classes) {
        IModelOperation operation = new RootOperationAddClasses(getTarget(), classes,
                getTarget().getClasses().size());
        return execute(operation, Messages.DIALOG_ADD_METHODS_PROBLEM_TITLE);
    }

    private String generateClassName() {
        String className = Constants.DEFAULT_NEW_PACKAGE_NAME + "." + Constants.DEFAULT_NEW_CLASS_NAME;
        if (getTarget().getClassModel(className) != null) {
            int i = 0;
            while (getTarget().getClassModel(className + String.valueOf(i)) != null) {
                i++;
            }
            className = className + String.valueOf(i);
        }
        return className;
    }

    @Override
    public boolean goToImplementationEnabled() {
        return false;
    }
}