Java TemplateState类代码示例

import com.intellij.codeInsight.template.impl.TemplateState; //导入依赖的package包/类
public void testSingleExpressionTemplate() {
  TemplateManagerImpl.setTemplateTesting(getProject(), getTestRootDisposable());

  myFixture.configureByFile(getTestName(true) + ".java");
  myFixture.type('\t');

  TemplateState templateState = TemplateManagerImpl.getTemplateState(myFixture.getEditor());
  assertNotNull(templateState);
  assertFalse(templateState.isFinished());

  myFixture.type("Integer");
  templateState.nextTab();
  assertTrue(templateState.isFinished());

  myFixture.checkResultByFile(getTestName(true) + "_after.java", true);
}
 

import com.intellij.codeInsight.template.impl.TemplateState; //导入依赖的package包/类
private void doTestStopEditing(Pass<AbstractInplaceIntroducer> pass) {
  String name = getTestName(true);
  configureByFile(getBasePath() + name + getExtension());
  final boolean enabled = getEditor().getSettings().isVariableInplaceRenameEnabled();
  try {
    TemplateManagerImpl.setTemplateTesting(getProject(), getTestRootDisposable());
    getEditor().getSettings().setVariableInplaceRenameEnabled(true);

    final AbstractInplaceIntroducer introducer = invokeRefactoring();
    pass.pass(introducer);
    checkResultByFile(getBasePath() + name + "_after" + getExtension());
  }
  finally {
    TemplateState state = TemplateManagerImpl.getTemplateState(getEditor());
    if (state != null) {
      state.gotoEnd(true);
    }
    getEditor().getSettings().setVariableInplaceRenameEnabled(enabled);
  }
}
 

import com.intellij.codeInsight.template.impl.TemplateState; //导入依赖的package包/类
@Override
public void doApplyInformationToEditor() {
  ApplicationManager.getApplication().assertIsDispatchThread();

  if (!ApplicationManager.getApplication().isUnitTestMode() && !myEditor.getContentComponent().hasFocus()) return;

  // do not show intentions if caret is outside visible area
  LogicalPosition caretPos = myEditor.getCaretModel().getLogicalPosition();
  Rectangle visibleArea = myEditor.getScrollingModel().getVisibleArea();
  Point xy = myEditor.logicalPositionToXY(caretPos);
  if (!visibleArea.contains(xy)) return;

  TemplateState state = TemplateManagerImpl.getTemplateState(myEditor);
  if (myShowBulb && (state == null || state.isFinished()) && !HintManager.getInstance().hasShownHintsThatWillHideByOtherHint(false)) {
    DaemonCodeAnalyzerImpl codeAnalyzer = (DaemonCodeAnalyzerImpl)DaemonCodeAnalyzer.getInstance(myProject);
    codeAnalyzer.setLastIntentionHint(myProject, myFile, myEditor, myIntentionsInfo, myHasToRecreate);
  }
}
 

import com.intellij.codeInsight.template.impl.TemplateState; //导入依赖的package包/类
@Override
public void execute(Editor editor, DataContext dataContext) {
  final TemplateState templateState = TemplateManagerImpl.getTemplateState(editor);
  if (templateState != null && !templateState.isFinished()) {
    final TextRange range = templateState.getCurrentVariableRange();
    final int caretOffset = editor.getCaretModel().getOffset();
    if (range != null && range.getStartOffset() <= caretOffset && caretOffset <= range.getEndOffset()) {
      int offsetToMove = myIsHomeHandler ? range.getStartOffset() : range.getEndOffset();
      if (offsetToMove != caretOffset) {
        editor.getCaretModel().moveToOffset(offsetToMove);
      }
      editor.getSelectionModel().removeSelection();
    } else {
      myOriginalHandler.execute(editor, dataContext);
    }
  } else {
    myOriginalHandler.execute(editor, dataContext);
  }
}
 

import com.intellij.codeInsight.template.impl.TemplateState; //导入依赖的package包/类
private void highlightTemplateVariables(Template template, Editor topLevelEditor) {
  //add highlights
  if (myHighlighters != null) { // can be null if finish is called during testing
    Map<TextRange, TextAttributes> rangesToHighlight = new HashMap<TextRange, TextAttributes>();
    final TemplateState templateState = TemplateManagerImpl.getTemplateState(topLevelEditor);
    if (templateState != null) {
      EditorColorsManager colorsManager = EditorColorsManager.getInstance();
      for (int i = 0; i < templateState.getSegmentsCount(); i++) {
        final TextRange segmentOffset = templateState.getSegmentRange(i);
        final String name = template.getSegmentName(i);
        TextAttributes attributes = null;
        if (name.equals(PRIMARY_VARIABLE_NAME)) {
          attributes = colorsManager.getGlobalScheme().getAttributes(EditorColors.WRITE_SEARCH_RESULT_ATTRIBUTES);
        }
        else if (name.equals(OTHER_VARIABLE_NAME)) {
          attributes = colorsManager.getGlobalScheme().getAttributes(EditorColors.SEARCH_RESULT_ATTRIBUTES);
        }
        if (attributes == null) continue;
        rangesToHighlight.put(segmentOffset, attributes);
      }
    }
    addHighlights(rangesToHighlight, topLevelEditor, myHighlighters, HighlightManager.getInstance(myProject));
  }
}
 

import com.intellij.codeInsight.template.impl.TemplateState; //导入依赖的package包/类
@Override
public void beforeTemplateFinished(final TemplateState templateState, Template template) {
  try {
    final TextResult value = templateState.getVariableValue(PRIMARY_VARIABLE_NAME);
    myInsertedName = value != null ? value.toString() : null;

    TextRange range = templateState.getCurrentVariableRange();
    final int currentOffset = myEditor.getCaretModel().getOffset();
    if (range == null && myRenameOffset != null) {
      range = new TextRange(myRenameOffset.getStartOffset(), myRenameOffset.getEndOffset());
    }
    myBeforeRevert =
      range != null && range.getEndOffset() >= currentOffset && range.getStartOffset() <= currentOffset
      ? myEditor.getDocument().createRangeMarker(range.getStartOffset(), currentOffset)
      : null;
    if (myBeforeRevert != null) {
      myBeforeRevert.setGreedyToRight(true);
    }
    finish(true);
  }
  finally {
    restoreDaemonUpdateState();
  }
}
 

import com.intellij.codeInsight.template.impl.TemplateState; //导入依赖的package包/类
@Override
public void execute(Editor editor, DataContext dataContext) {
  final SelectionModel selectionModel = editor.getSelectionModel();
  if (selectionModel.hasSelection()) {
    final TemplateState state = TemplateManagerImpl.getTemplateState(editor);
    if (state != null && editor.getUserData(InplaceRefactoring.INPLACE_RENAMER) != null) {
      final LookupImpl lookup = (LookupImpl)LookupManager.getActiveLookup(editor);
      if (lookup != null) {
        selectionModel.removeSelection();
        lookup.hideLookup(true);
        return;
      }
    }
  }

  myOriginalHandler.execute(editor, dataContext);
}
 

import com.intellij.codeInsight.template.impl.TemplateState; //导入依赖的package包/类
@Override
public void finish(boolean success) {
  myFinished = true;
  final TemplateState templateState = TemplateManagerImpl.getTemplateState(myEditor);
  if (templateState != null) {
    myEditor.putUserData(ACTIVE_INTRODUCE, null);
  }
  if (myDocumentAdapter != null) {
    myEditor.getDocument().removeDocumentListener(myDocumentAdapter);
  }
  if (myBalloon == null) {
    releaseIfNotRestart();
  }
  super.finish(success);
  if (success) {
    PsiDocumentManager.getInstance(myProject).commitAllDocuments();
    final V variable = getVariable();
    if (variable == null) {
      return;
    }
    restoreState(variable);
  }
}
 

import com.intellij.codeInsight.template.impl.TemplateState; //导入依赖的package包/类
@Override
public void execute(Editor editor, DataContext dataContext) {
  final SelectionModel selectionModel = editor.getSelectionModel();
  if (selectionModel.hasSelection()) {
    final TemplateState state = TemplateManagerImpl.getTemplateState(editor);
    if (state != null && editor.getUserData(InplaceRefactoring.INPLACE_RENAMER) != null) {
      final LookupImpl lookup = (LookupImpl)LookupManager.getActiveLookup(editor);
      if (lookup != null) {
        selectionModel.removeSelection();
        lookup.hide();
        return;
      }
    }
  }

  myOriginalHandler.execute(editor, dataContext);
}
 

import com.intellij.codeInsight.template.impl.TemplateState; //导入依赖的package包/类
public void format(PsiFile psiFile, int startOffset, int endOffset) throws FileDoesNotExistsException {
	LOG.debug("#format " + startOffset + "-" + endOffset);
	boolean wholeFile = FileUtils.isWholeFile(startOffset, endOffset, psiFile.getText());
	Range range = new Range(startOffset, endOffset, wholeFile);

	final Editor editor = PsiUtilBase.findEditor(psiFile);
	if (editor != null) {
		TemplateState templateState = TemplateManagerImpl.getTemplateState(editor);
		if (templateState != null && !settings.isUseForLiveTemplates()) {
			throw new ReformatItInIntelliJ();
		}
		formatWhenEditorIsOpen(editor, range, psiFile);
	} else {
		formatWhenEditorIsClosed(range, psiFile);
	}

}
 

import com.intellij.codeInsight.template.impl.TemplateState; //导入依赖的package包/类
@Override
protected void doExecute(Editor editor, @Nullable Caret caret, DataContext dataContext) {
  final TemplateState templateState = TemplateManagerImpl.getTemplateState(editor);
  if (templateState != null && !templateState.isFinished()) {
    final TextRange range = templateState.getCurrentVariableRange();
    final int caretOffset = editor.getCaretModel().getOffset();
    if (range != null && shouldStayInsideVariable(range, caretOffset)) {
      int selectionOffset = editor.getSelectionModel().getLeadSelectionOffset();
      int offsetToMove = myIsHomeHandler ? range.getStartOffset() : range.getEndOffset();
      LogicalPosition logicalPosition = editor.offsetToLogicalPosition(offsetToMove).leanForward(myIsHomeHandler);
      editor.getCaretModel().moveToLogicalPosition(logicalPosition);
      EditorModificationUtil.scrollToCaret(editor);
      if (myWithSelection) {
        editor.getSelectionModel().setSelection(selectionOffset, offsetToMove);
      }
      else {
        editor.getSelectionModel().removeSelection();
      }
      return;
    }
  }
  myOriginalHandler.execute(editor, caret, dataContext);
}
 

import com.intellij.codeInsight.template.impl.TemplateState; //导入依赖的package包/类
@Override
public void execute(Editor editor, DataContext dataContext) {
  TemplateState templateState = TemplateManagerImpl.getTemplateState(editor);
  if (templateState != null && !templateState.isFinished()) {
    SelectionModel selectionModel = editor.getSelectionModel();
    LookupImpl lookup = (LookupImpl)LookupManager.getActiveLookup(editor);

    // the idea behind lookup checking is that if there is a preselected value in lookup
    // then user might want just to close lookup but not finish a template.
    // E.g. user wants to move to the next template segment by Tab without completion invocation.
    // If there is no selected value in completion that user definitely wants to finish template
    boolean lookupIsEmpty = lookup == null || lookup.getCurrentItem() == null;
    if (!selectionModel.hasSelection() && lookupIsEmpty) {
      CommandProcessor.getInstance().setCurrentCommandName(CodeInsightBundle.message("finish.template.command"));
      templateState.gotoEnd(true);
      return;
    }
  }

  if (myOriginalHandler.isEnabled(editor, dataContext)) {
    myOriginalHandler.execute(editor, dataContext);
  }
}
 

import com.intellij.codeInsight.template.impl.TemplateState; //导入依赖的package包/类
private static void finishTemplate(Editor editor) {
  final TemplateState templateState = TemplateManagerImpl.getTemplateState(editor);
  final InplaceRefactoring renamer = editor.getUserData(InplaceRefactoring.INPLACE_RENAMER);
  if (templateState != null && renamer != null) {
    templateState.gotoEnd(true);
    editor.putUserData(InplaceRefactoring.INPLACE_RENAMER, null);
  }
}
 

import com.intellij.codeInsight.template.impl.TemplateState; //导入依赖的package包/类
private void doTestWithTemplateFinish(@NotNull String fileName, Surrounder surrounder, @Nullable String textToType) {
  TemplateManagerImpl.setTemplateTesting(getProject(), getTestRootDisposable());
  configureByFile(BASE_PATH + fileName + ".java");
  SurroundWithHandler.invoke(getProject(), getEditor(), getFile(), surrounder);
  if (textToType != null) {
    type(textToType);
  }
  TemplateState templateState = TemplateManagerImpl.getTemplateState(getEditor());
  assertNotNull(templateState);
  templateState.nextTab();
  checkResultByFile(BASE_PATH + fileName + "_after.java");
}
 

import com.intellij.codeInsight.template.impl.TemplateState; //导入依赖的package包/类
public void testValueTyping() {
  configureByFile(getBasePath() + "/beforeValueTyping.java");
  TemplateManagerImpl.setTemplateTesting(getProject(), getTestRootDisposable());
  doAction("Add missing annotation parameters - value3, value2, value1");
  final TemplateState state = TemplateManagerImpl.getTemplateState(getEditor());
  assertNotNull(state);
  type("\"value33\"");
  state.nextTab();
  type("\"value22\"");
  state.nextTab();
  type("\"value11\"");
  state.nextTab();
  assertTrue(state.isFinished());
  checkResultByFile(getBasePath() + "/afterValueTyping.java");
}