Java Code Examples for com.intellij.openapi.editor.Document#getUserData()

public long set(@Nonnull Document document, @Nonnull ID<?, ?> indexId, long value) {
  List<IdVersionInfo> list = document.getUserData(KEY);
  if (list == null) {
    list = ((UserDataHolderEx)document).putUserDataIfAbsent(KEY, new ArrayList<>());
  }

  synchronized (list) {
    for (IdVersionInfo info : list) {
      if (info.id == indexId) {
        long old = info.docVersion;
        if (info.mapVersion != mapVersion) {
          old = INVALID_STAMP;
          info.mapVersion = mapVersion;
        }
        info.docVersion = value;
        return old;
      }
    }
    list.add(new IdVersionInfo(indexId, value, mapVersion));
    return INVALID_STAMP;
  }
}
 

@Nullable
public static EditorHighlighter getEditorHighlighterForCachesBuilding(Document document) {
  if (document == null) {
    return null;
  }
  final WeakReference<EditorHighlighter> editorHighlighterWeakReference = document.getUserData(ourSomeEditorSyntaxHighlighter);
  final EditorHighlighter someEditorHighlighter = SoftReference.dereference(editorHighlighterWeakReference);

  if (someEditorHighlighter instanceof LexerEditorHighlighter &&
      ((LexerEditorHighlighter)someEditorHighlighter).isValid()
          ) {
    return someEditorHighlighter;
  }
  document.putUserData(ourSomeEditorSyntaxHighlighter, null);
  return null;
}
 

@Nonnull
private LinkedList<UndoableGroup> doGetStackForDocument(@Nonnull DocumentReference r) {
  // If document is not associated with file, we have to store its stack in document
  // itself to avoid memory leaks caused by holding stacks of all documents, ever created, here.
  // And to know, what documents do exist now, we have to maintain weak reference list of them.

  Document d = r.getDocument();
  LinkedList<UndoableGroup> result = d.getUserData(STACK_IN_DOCUMENT_KEY);
  if (result == null) {
    result = new LinkedList<UndoableGroup>();
    d.putUserData(STACK_IN_DOCUMENT_KEY, result);
    myDocumentsWithStacks.add(d);
  }
  return result;
}
 

private static List<Runnable> getAndClearActionsAfterCommit(@Nonnull Document document) {
  List<Runnable> list;
  synchronized (ACTION_AFTER_COMMIT) {
    list = document.getUserData(ACTION_AFTER_COMMIT);
    if (list != null) {
      list = new ArrayList<>(list);
      document.putUserData(ACTION_AFTER_COMMIT, null);
    }
  }
  return list;
}
 

public static void addRunOnCommit(@Nonnull Document document, @Nonnull Runnable action) {
  synchronized (ACTION_AFTER_COMMIT) {
    List<Runnable> list = document.getUserData(ACTION_AFTER_COMMIT);
    if (list == null) {
      document.putUserData(ACTION_AFTER_COMMIT, list = new SmartList<>());
    }
    list.add(action);
  }
}
 

@Override
public boolean maySaveDocument(@Nonnull Document document, boolean isSaveExplicit) {
  final Object beingCommitted = document.getUserData(CommitHelper.DOCUMENT_BEING_COMMITTED_KEY);
  if (beingCommitted == VetoSavingCommittingDocumentsAdapter.SAVE_DENIED) {
    return false;
  }
  if (beingCommitted instanceof Project) {
    boolean allowSave = myAdapter.showAllowSaveDialog(Collections.singletonMap(document, (Project)beingCommitted));
    if (!allowSave) {
      return false;
    }
  }
  return true;
}
 

public static <T extends PsiElement> T getOriginalElement(@Nonnull T psi, PsiFile containingFile) {
  if (containingFile != null && containingFile != containingFile.getOriginalFile() && psi.getTextRange() != null) {
    TextRange range = psi.getTextRange();
    Integer start = range.getStartOffset();
    Integer end = range.getEndOffset();
    final Document document = containingFile.getViewProvider().getDocument();
    if (document != null) {
      Document hostDocument = document instanceof DocumentWindow ? ((DocumentWindow)document).getDelegate() : document;
      OffsetTranslator translator = hostDocument.getUserData(OffsetTranslator.RANGE_TRANSLATION);
      if (translator != null) {
        if (document instanceof DocumentWindow) {
          TextRange translated = ((DocumentWindow)document).injectedToHost(new TextRange(start, end));
          start = translated.getStartOffset();
          end = translated.getEndOffset();
        }

        start = translator.translateOffset(start);
        end = translator.translateOffset(end);
        if (start == null || end == null) {
          return null;
        }

        if (document instanceof DocumentWindow) {
          start = ((DocumentWindow)document).hostToInjected(start);
          end = ((DocumentWindow)document).hostToInjected(end);
        }
      }
    }
    //noinspection unchecked
    return (T)PsiTreeUtil.findElementOfClassAtRange(containingFile.getOriginalFile(), start, end, psi.getClass());
  }

  return psi;
}
 

private static boolean shouldRecordActions(@Nonnull Document document) {
  if (document.getUserData(UndoConstants.DONT_RECORD_UNDO) == Boolean.TRUE) return false;

  VirtualFile vFile = FileDocumentManager.getInstance().getFile(document);
  if (vFile == null) return true;
  return vFile.getUserData(AbstractFileViewProvider.FREE_THREADED) != Boolean.TRUE && vFile.getUserData(UndoConstants.DONT_RECORD_UNDO) != Boolean.TRUE;
}
 

private static int addNumberOfRequestedRedetects(@Nonnull Document document, int delta) {
  AtomicInteger oldData = document.getUserData(RUNNING_REDETECTS_KEY);
  if (oldData == null) {
    oldData = ((UserDataHolderEx)document).putUserDataIfAbsent(RUNNING_REDETECTS_KEY, new AtomicInteger());
  }
  return oldData.addAndGet(delta);
}
 

private static ConcurrentMap<Project, MarkupModelImpl> getMarkupModelMap(@Nonnull Document document) {
  ConcurrentMap<Project, MarkupModelImpl> markupModelMap = document.getUserData(MARKUP_MODEL_MAP_KEY);
  if (markupModelMap == null) {
    ConcurrentMap<Project, MarkupModelImpl> newMap = ContainerUtil.newConcurrentMap();
    markupModelMap = ((UserDataHolderEx)document).putUserDataIfAbsent(MARKUP_MODEL_MAP_KEY, newMap);
  }
  return markupModelMap;
}
 

@Nonnull
private static Document getOriginal(@Nonnull Document document) {
  Document result = document.getUserData(ORIGINAL_DOCUMENT);
  return result == null ? document : result;
}
 

static boolean isCopy(@Nonnull Document d) {
  return d.getUserData(ORIGINAL_DOCUMENT) != null;
}
 

private void removeDocumentListener(Document document) {
  if (isAcceptableDocument(document) && document.getUserData(ourListenerKey) != null) {
    document.removeDocumentListener(myDocumentListener);
    document.putUserData(ourListenerKey, null);
  }
}
 

private static void handleBeforeDocumentChange(@Nonnull UndoManagerImpl undoManager, @Nonnull Document document) {
  if (undoManager.isActive() && isUndoable(undoManager, document) && undoManager.isUndoOrRedoInProgress() && document.getUserData(UNDOING_EDITOR_CHANGE) != Boolean.TRUE) {
    throw new IllegalStateException("Do not change documents during undo as it will break undo sequence.");
  }
}
 

@Override
@Nullable
public Charset getCachedCharsetFromContent(@Nonnull Document document) {
  return document.getUserData(CACHED_CHARSET_FROM_CONTENT);
}
 

@Override
public boolean isPartialPreviewOfALargeFile(@Nonnull Document document) {
  return document.getUserData(BIG_FILE_PREVIEW) == Boolean.TRUE;
}
 

@Override
@Nullable
public VirtualFile getFile(@Nonnull Document document) {
  if (document instanceof FrozenDocument) return null;
  return document.getUserData(FILE_KEY);
}
 

static boolean isWhitespaceOptimizationAllowed(@Nonnull Document document) {
  return document.getUserData(TYPING_INSIDE_HIGHLIGHTER_OCCURRED) == null;
}
 

/**
 * @return an internal lock object to prevent read & write phases of commit from running simultaneously for free-threaded PSI
 */
private static Lock getDocumentLock(Document document) {
  Lock lock = document.getUserData(DOCUMENT_LOCK);
  return lock != null ? lock : ((UserDataHolderEx)document).putUserDataIfAbsent(DOCUMENT_LOCK, new ReentrantLock());
}
 

@Override
public VirtualFile getFile(@Nonnull Document document) {
  return document.getUserData(MOCK_VIRTUAL_FILE_KEY);
}