eu.modelwriter.marker.ui.internal.wizards.markallinwswizard.MarkAllInWsWizard.java Source code

Introduction

Here is the source code for eu.modelwriter.marker.ui.internal.wizards.markallinwswizard.MarkAllInWsWizard.java

Source

/*******************************************************************************
 * Copyright (c) 2015 UNIT Information Technologies R&D Ltd 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: Ferhat Erata - initial API and implementation H. Emre Kirmizi - initial API and
 * implementation Serhat Celik - initial API and implementation U. Anil Ozturk - initial API and
 * implementation
 *******************************************************************************/
package eu.modelwriter.marker.ui.internal.wizards.markallinwswizard;

import java.util.Scanner;
import java.util.UUID;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IMarker;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.dialogs.ProgressMonitorDialog;
import org.eclipse.jface.text.Document;
import org.eclipse.jface.text.IDocument;
import org.eclipse.jface.text.ITextSelection;
import org.eclipse.jface.text.TextSelection;
import org.eclipse.jface.wizard.Wizard;
import org.eclipse.swt.graphics.Image;

import eu.modelwriter.configuration.internal.AlloyUtilities;
import eu.modelwriter.marker.MarkerActivator;
import eu.modelwriter.marker.internal.AnnotationFactory;
import eu.modelwriter.marker.internal.MarkUtilities;
import eu.modelwriter.marker.internal.MarkerFactory;
import eu.modelwriter.marker.ui.Activator;

public class MarkAllInWsWizard extends Wizard {

    private final IFile file;
    private MarkAllInWsPage page;
    private final ITextSelection textSelection;

    public MarkAllInWsWizard(final ITextSelection textSelection, final IFile file) {
        super();
        this.textSelection = textSelection;
        this.file = file;
    }

    @Override
    public void addPages() {
        this.page = new MarkAllInWsPage();
        super.addPages();
        this.addPage(this.page);
    }

    private void addToAlloyXML(final IMarker mymarker) {
        AlloyUtilities.addMarkerToRepository(mymarker);
    }

    @Override
    public String getWindowTitle() {
        return "Mark All In Workspace";
    }

    @Override
    public boolean performFinish() {
        final Object[] checkedElements = MarkAllInWsPage.checkboxTreeViewer.getCheckedElements();
        final String text = this.textSelection.getText();
        final String leader_id = UUID.randomUUID().toString();
        boolean success = false;

        final Image image = MarkerActivator.getDefault().getImageRegistry()
                .get(MarkerActivator.Images.IMAGE_MODELWRITER_ID);
        final ProgressMonitorDialog pmd = new ProgressMonitorDialog(
                Activator.getDefault().getWorkbench().getActiveWorkbenchWindow().getShell());
        ProgressMonitorDialog.setDefaultImage(image);
        pmd.open();
        final IProgressMonitor pmdmoni = pmd.getProgressMonitor();
        try {
            pmdmoni.beginTask("Marking", checkedElements.length);
            for (int i = 0; i < checkedElements.length && !pmdmoni.isCanceled(); i++) {
                try {
                    if (checkedElements[i] instanceof IFile) {
                        final IFile iFile = (IFile) checkedElements[i];
                        pmdmoni.subTask(iFile.getName());
                        pmdmoni.worked(1);

                        if (!iFile.getFileExtension().equals("xmi") && !iFile.getFileExtension().equals("ecore")
                                && !iFile.getFileExtension().equals("reqif")) {
                            final String charSet = iFile.getCharset();
                            final Scanner scanner = new Scanner(iFile.getContents(), charSet);
                            final IDocument document = new Document(scanner.useDelimiter("\\A").next());
                            scanner.close();

                            final String fullText = document.get();
                            boolean hasLeader = false;
                            int index = 0;
                            int offset = 0;
                            final int lenght = this.textSelection.getLength();
                            final String id = UUID.randomUUID().toString();

                            if (lenght != 0) {
                                while ((offset = fullText.indexOf(text, index)) != -1) {
                                    final TextSelection nextSelection = new TextSelection(document, offset, lenght);
                                    if (MarkerFactory.findMarkerByOffset(iFile, offset) == null) {
                                        final IMarker mymarker = MarkerFactory.createMarker(iFile, nextSelection);
                                        MarkUtilities.setGroupId(mymarker, id);
                                        if (!iFile.equals(this.file)) {
                                            if (hasLeader == false) {
                                                MarkUtilities.setLeaderId(mymarker, leader_id);
                                                hasLeader = true;
                                            }
                                        } else {
                                            if (this.textSelection.getOffset() == offset) {
                                                MarkUtilities.setLeaderId(mymarker, leader_id);
                                            }
                                        }

                                        this.addToAlloyXML(mymarker);
                                        AnnotationFactory.addAnnotation(mymarker,
                                                AnnotationFactory.ANNOTATION_MARKING);
                                    }
                                    index = offset + lenght;
                                }
                            }
                            success = true;
                        } else {
                            final MessageDialog dialog = new MessageDialog(MarkerActivator.getShell(),
                                    "Mark All In Workspace", null,
                                    iFile.getName() + " doesn't supported for this command.",
                                    MessageDialog.INFORMATION, new String[] { "OK" }, 0);
                            dialog.open();
                            MarkAllInWsPage.checkboxTreeViewer.setChecked(iFile, false);
                        }
                    }
                } catch (final CoreException e) {
                    e.printStackTrace();
                }
            }
        } finally {
            pmdmoni.done();
        }
        pmd.close();

        if (success == true) {
            final MessageDialog dialog = new MessageDialog(MarkerActivator.getShell(), "Mark All In Workspace",
                    null, "Selected project(s) have been marked by selected text", MessageDialog.INFORMATION,
                    new String[] { "OK" }, 0);
            dialog.open();
            return true;
        } else {
            return false;
        }
    }
}