@Override
  public Object execute(ExecutionEvent event) throws ExecutionException {
    IEditorPart editor =
        PlatformUI.getWorkbench().getActiveWorkbenchWindow().getActivePage().getActiveEditor();
    final boolean enableRiskyRefactoring =
        Platform.getPreferencesService()
            .getBoolean(
                ProductConstants.PRODUCT_ID_DESIGNER,
                PreferenceConstants.ENABLERISKYREFACTORING,
                false,
                null);
    if (!enableRiskyRefactoring) {
      ErrorReporter.logError("Risky refactoring is not enabled!");
      return null;
    }

    if (editor == null
        || !(editor instanceof TTCN3Editor
            || (editor instanceof TTCNPPEditor && enableRiskyRefactoring))) {
      ErrorReporter.logError("The editor is not found or not a Titan TTCN-3 editor");
      return null;
    }

    IFile file = (IFile) editor.getEditorInput().getAdapter(IFile.class);
    TextFileChange change = null;
    try {
      change = OrganizeImports.organizeImportsChange(file);
    } catch (CoreException e) {
      ErrorReporter.logExceptionStackTrace("Error while creating the needed import changes", e);
      return null;
    }
    try {
      change.perform(new NullProgressMonitor());
    } catch (CoreException e) {
      ErrorReporter.logExceptionStackTrace("Error while performing the needed import changes", e);
    }

    IProject project = file.getProject();
    GlobalParser.getProjectSourceParser(project).reportOutdating(file);
    GlobalParser.getProjectSourceParser(project).analyzeAll();

    return null;
  }
Пример #2
0
  @Override
  public void run(final IAction action) {
    TITANDebugConsole.println("Add import called: ");
    if (targetEditor == null || !(targetEditor instanceof TTCN3Editor)) {
      return;
    }

    targetEditor.getEditorSite().getActionBars().getStatusLineManager().setErrorMessage(null);

    IFile file = (IFile) targetEditor.getEditorInput().getAdapter(IFile.class);
    if (file == null) {
      targetEditor
          .getEditorSite()
          .getActionBars()
          .getStatusLineManager()
          .setErrorMessage(FILENOTIDENTIFIABLE);
      return;
    }

    if (!TITANNature.hasTITANNature(file.getProject())) {
      targetEditor
          .getEditorSite()
          .getActionBars()
          .getStatusLineManager()
          .setErrorMessage(TITANNature.NO_TITAN_FILE_NATURE_FOUND);
      return;
    }

    IPreferencesService prefs = Platform.getPreferencesService();
    boolean reportDebugInformation =
        prefs.getBoolean(
            ProductConstants.PRODUCT_ID_DESIGNER,
            PreferenceConstants.DISPLAYDEBUGINFORMATION,
            true,
            null);

    int offset;
    if (!selection.isEmpty()
        && selection instanceof TextSelection
        && !"".equals(((TextSelection) selection).getText())) {
      if (reportDebugInformation) {
        TITANDebugConsole.println("text selected: " + ((TextSelection) selection).getText());
      }
      TextSelection tSelection = (TextSelection) selection;
      offset = tSelection.getOffset() + tSelection.getLength();
    } else {
      offset = ((TTCN3Editor) targetEditor).getCarretOffset();
    }

    DeclarationCollector declarationCollector =
        OpenDeclarationHelper.findVisibleDeclarations(
            targetEditor,
            new TTCN3ReferenceParser(false),
            ((TTCN3Editor) targetEditor).getDocument(),
            offset,
            false);

    if (declarationCollector == null) {
      return;
    }

    List<DeclarationCollectionHelper> collected = declarationCollector.getCollected();
    if (collected.isEmpty()) {
      // FIXME add semantic check guard on project level.
      ProjectSourceParser projectSourceParser =
          GlobalParser.getProjectSourceParser(file.getProject());
      if (reportDebugInformation) {
        TITANDebugConsole.println("No visible elements found");
      }
      for (String moduleName2 : projectSourceParser.getKnownModuleNames()) {
        Module module2 = projectSourceParser.getModuleByName(moduleName2);
        if (module2 != null) {
          // Visit each file in the project one by
          // one instead of
          // "module2.getAssignments().addDeclaration(declarationCollector)".
          Assignments assignments = module2.getAssignments();
          for (int i = 0; i < assignments.getNofAssignments(); i++) {
            assignments.getAssignmentByIndex(i).addDeclaration(declarationCollector, 0);
          }
        }
      }

      if (declarationCollector.getCollectionSize() == 0) {
        targetEditor
            .getEditorSite()
            .getActionBars()
            .getStatusLineManager()
            .setErrorMessage(NOTTTCN3DECLARATION);
        return;
      }

      if (reportDebugInformation) {
        TITANDebugConsole.println("Elements were only found in not visible modules");
      }

      DeclarationCollectionHelper resultToInsert = null;
      if (collected.size() == 1) {
        resultToInsert = collected.get(0);
      } else {
        OpenDeclarationLabelProvider labelProvider = new OpenDeclarationLabelProvider();
        ElementListSelectionDialog dialog =
            new ElementListSelectionDialog(new Shell(Display.getDefault()), labelProvider);
        dialog.setTitle("Add Import");
        dialog.setMessage("Choose element to generate an import statement for.");
        dialog.setElements(collected.toArray());
        if (dialog.open() == Window.OK) {
          if (reportDebugInformation) {
            TITANDebugConsole.getConsole()
                .newMessageStream()
                .println("Selected: " + dialog.getFirstResult());
          }
          resultToInsert = (DeclarationCollectionHelper) dialog.getFirstResult();
        }
      }

      if (resultToInsert == null) {
        return;
      }

      IFile newfile = (IFile) resultToInsert.location.getFile();
      String moduleName = projectSourceParser.containedModule(newfile);
      Module newModule = projectSourceParser.getModuleByName(moduleName);
      if (newModule == null) {
        targetEditor
            .getEditorSite()
            .getActionBars()
            .getStatusLineManager()
            .setErrorMessage("Could not identify the module in file " + newfile.getName());
        return;
      }

      String ttcnName = newModule.getIdentifier().getTtcnName();
      TITANDebugConsole.println("the new module to insert: " + ttcnName);

      final IFile actualFile = (IFile) targetEditor.getEditorInput().getAdapter(IFile.class);
      String actualModuleName = projectSourceParser.containedModule(actualFile);
      Module actualModule = projectSourceParser.getModuleByName(actualModuleName);

      int insertionOffset =
          ((TTCN3Module) actualModule).getAssignmentsScope().getLocation().getOffset() + 1;

      MultiTextEdit multiEdit = new MultiTextEdit(insertionOffset, 0);
      RewriteSessionEditProcessor processor =
          new RewriteSessionEditProcessor(
              ((TTCN3Editor) targetEditor).getDocument(),
              multiEdit,
              TextEdit.UPDATE_REGIONS | TextEdit.CREATE_UNDO);
      multiEdit.addChild(new InsertEdit(insertionOffset, "\nimport from " + ttcnName + " all;\n"));

      try {
        processor.performEdits();
      } catch (BadLocationException e) {
        ErrorReporter.logExceptionStackTrace(e);
      }
    } else {
      if (reportDebugInformation) {
        for (DeclarationCollectionHelper foundDeclaration : collected) {
          TITANDebugConsole.println(
              "declaration:"
                  + foundDeclaration.location.getFile()
                  + ": "
                  + foundDeclaration.location.getOffset()
                  + " - "
                  + foundDeclaration.location.getEndOffset()
                  + " is available");
        }
      }
    }

    Display.getDefault()
        .asyncExec(
            new Runnable() {
              @Override
              public void run() {
                MessageDialog.openWarning(
                    new Shell(Display.getDefault()),
                    "Study feature",
                    "Adding a missing importation is still under study");
              }
            });
  }