com.testify.ecfeed.ui.wizards.NewEcFileWizard.java Source code

Introduction

Here is the source code for com.testify.ecfeed.ui.wizards.NewEcFileWizard.java

Source

/*******************************************************************************
 * Copyright (c) 2013 Testify 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                                     
 *                                                                               
 * Contributors:                                                                 
 *     Patryk Chamuczynski (p.chamuczynski(at)radytek.com) - initial implementation
 ******************************************************************************/

package com.testify.ecfeed.ui.wizards;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.runtime.IPath;
import org.eclipse.jface.dialogs.IDialogConstants;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.wizard.Wizard;
import org.eclipse.swt.SWT;
import org.eclipse.swt.widgets.Display;
import org.eclipse.ui.INewWizard;
import org.eclipse.ui.IWorkbench;
import org.eclipse.ui.IWorkbenchPage;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.dialogs.WizardNewFileCreationPage;
import org.eclipse.ui.part.FileEditorInput;

import com.testify.ecfeed.model.RootNode;
import com.testify.ecfeed.serialization.ect.EctSerializer;
import com.testify.ecfeed.ui.common.Constants;
import com.testify.ecfeed.ui.common.Messages;

public class NewEcFileWizard extends Wizard implements INewWizard {

    private WizardNewFileCreationPage fPage;
    private IStructuredSelection fSelection;

    public NewEcFileWizard() {
        super();
        setNeedsProgressMonitor(false);
        setHelpAvailable(false);
    }

    public void addPages() {
        fPage = new WizardNewFileCreationPage(Messages.WIZARD_NEW_ECT_FILE_TITLE, fSelection);
        fPage.setFileName(Constants.DEFAULT_NEW_ECT_FILE_NAME);
        fPage.setAllowExistingResources(true);
        fPage.setFileExtension(Constants.EQUIVALENCE_CLASS_FILE_EXTENSION);
        fPage.setTitle(Messages.WIZARD_NEW_ECT_FILE_TITLE);
        fPage.setDescription(Messages.WIZARD_NEW_ECT_FILE_MESSAGE);

        addPage(fPage);
    }

    public boolean performFinish() {
        IFile file = fPage.createNewFile();
        try {

            if (file.getContents().read() != -1) {
                MessageDialog dialog = new MessageDialog(getShell(), Messages.WIZARD_FILE_EXISTS_TITLE,
                        Display.getDefault().getSystemImage(SWT.ICON_QUESTION), Messages.WIZARD_FILE_EXISTS_MESSAGE,
                        MessageDialog.QUESTION_WITH_CANCEL,
                        new String[] { IDialogConstants.OK_LABEL, IDialogConstants.CANCEL_LABEL },
                        IDialogConstants.OK_ID);
                if (dialog.open() != IDialogConstants.OK_ID) {
                    return false;
                }
            }

            final IPath newFileFullPath = fPage.getContainerFullPath().append(fPage.getFileName());
            String modelName = newFileFullPath.removeFileExtension().lastSegment();
            RootNode model = new RootNode(modelName != null ? modelName : Constants.DEFAULT_NEW_ECT_MODEL_NAME);
            ByteArrayOutputStream ostream = new ByteArrayOutputStream();
            new EctSerializer(ostream).serialize(model);
            ByteArrayInputStream istream = new ByteArrayInputStream(ostream.toByteArray());
            file.setContents(istream, true, true, null);

            //open new file in an ect editor
            IWorkbenchPage page = PlatformUI.getWorkbench().getActiveWorkbenchWindow().getActivePage();

            page.openEditor(new FileEditorInput(file), Constants.ECT_EDITOR_ID);
        } catch (Exception e) {
            e.printStackTrace();
        }

        return true;
    }

    public void init(IWorkbench workbench, IStructuredSelection selection) {
        this.fSelection = selection;
    }
}