lsp/org.eclipse.cdt.lsp.core/src/org/eclipse/cdt/lsp/core/Server2ClientProtocolExtension.java - cdt/org.eclipse.cdt - Git at Google

 /*******************************************************************************
  * Copyright (c) 2018 Manish Khurana , Nathan Ridge and others.
  *
  * This program and the accompanying materials
  * are made available under the terms of the Eclipse Public License 2.0
  * which accompanies this distribution, and is available at
  * https://www.eclipse.org/legal/epl-2.0/
  *
  * SPDX-License-Identifier: EPL-2.0
  *******************************************************************************/

 package org.eclipse.cdt.lsp.core;

 import java.net.URI;
 import java.util.ArrayList;
 import java.util.List;

 import org.eclipse.cdt.internal.ui.editor.SemanticHighlightingManager.HighlightedPosition;
 import org.eclipse.cdt.internal.ui.editor.SemanticHighlightingManager.HighlightingStyle;
 import org.eclipse.cdt.lsp.core.cquery.CqueryInactiveRegions;
 import org.eclipse.cdt.lsp.core.cquery.CquerySemanticHighlights;
 import org.eclipse.cdt.lsp.core.cquery.HighlightSymbol;
 import org.eclipse.cdt.lsp.core.cquery.IndexingProgressStats;
 import org.eclipse.cdt.ui.CUIPlugin;
 import org.eclipse.cdt.ui.PreferenceConstants;
 import org.eclipse.core.resources.IFile;
 import org.eclipse.jface.action.StatusLineContributionItem;
 import org.eclipse.jface.action.StatusLineManager;
 import org.eclipse.jface.preference.IPreferenceStore;
 import org.eclipse.jface.preference.PreferenceConverter;
 import org.eclipse.jface.text.BadLocationException;
 import org.eclipse.jface.text.BadPositionCategoryException;
 import org.eclipse.jface.text.IDocument;
 import org.eclipse.jface.text.Position;
 import org.eclipse.jface.text.Region;
 import org.eclipse.jface.text.TextAttribute;
 import org.eclipse.jface.text.TextPresentation;
 import org.eclipse.lsp4e.LSPEclipseUtils;
 import org.eclipse.lsp4e.LanguageClientImpl;
 import org.eclipse.lsp4j.Range;
 import org.eclipse.lsp4j.jsonrpc.services.JsonNotification;
 import org.eclipse.osgi.util.NLS;
 import org.eclipse.swt.SWT;
 import org.eclipse.swt.graphics.Color;
 import org.eclipse.swt.widgets.Display;
 import org.eclipse.ui.IWorkbenchWindow;
 import org.eclipse.ui.PlatformUI;
 import org.eclipse.ui.internal.WorkbenchWindow;

 @SuppressWarnings("restriction")
 public class Server2ClientProtocolExtension extends LanguageClientImpl {

 	@JsonNotification("$cquery/progress")
 	public final void indexingProgress(IndexingProgressStats stats) {

 		Display.getDefault().asyncExec(new Runnable() {
 			@Override
 			public void run() {
 				final String cqueryStatusFieldId = "org.eclipse.lsp4e.cpp.status"; //$NON-NLS-1$
 				final int width = 28;
 				IWorkbenchWindow[] workbenchWindows = PlatformUI.getWorkbench().getWorkbenchWindows();
 				for (IWorkbenchWindow window : workbenchWindows) {
 					StatusLineManager statusLine = ((WorkbenchWindow) window).getStatusLineManager();
 					StatusLineContributionItem cqueryStatusField = (StatusLineContributionItem) statusLine
 							.find(cqueryStatusFieldId);
 					if (cqueryStatusField == null) {
 						cqueryStatusField = new StatusLineContributionItem(cqueryStatusFieldId, width);
 						statusLine.add(cqueryStatusField);
 					}
 					String msg = stats.getTotalJobs() > 0 ? NLS.bind(Messages.CqueryStateBusy, stats.getTotalJobs())
 							: Messages.CqueryStateIdle;
 					cqueryStatusField.setText(msg);
 				}
 			}
 		});
 	}

 	@JsonNotification("$cquery/setInactiveRegions")
 	public final void setInactiveRegions(CqueryInactiveRegions regions) {
 		URI uriReceived = regions.getUri();
 		List<Range> inactiveRegions = regions.getInactiveRegions();
 		IDocument doc = null;

 		// To get the document for the received URI.
 		for (PresentationReconcilerCPP eachReconciler : PresentationReconcilerCPP.presentationReconcilers) {
 			IDocument currentReconcilerDoc = eachReconciler.getTextViewer().getDocument();
 			URI currentReconcilerUri = getUri(currentReconcilerDoc);

 			if (currentReconcilerUri == null) {
 				continue;
 			}

 			if (uriReceived.equals(currentReconcilerUri)) {
 				doc = currentReconcilerDoc;
 				break;
 			}
 		}

 		if (doc == null) {
 			return;
 		}

 		// Removing inactive code highlighting position category and old positions from document.
 		try {
 			doc.removePositionCategory(PresentationReconcilerCPP.INACTIVE_CODE_HIGHLIGHTING_POSITION_CATEGORY);
 		} catch (BadPositionCategoryException e) {
 			Activator.log(e);
 		}
 		// Again add Inactive Code Position Category to the document.
 		doc.addPositionCategory(PresentationReconcilerCPP.INACTIVE_CODE_HIGHLIGHTING_POSITION_CATEGORY);

 		for (Range region : inactiveRegions) {
 			int offset = 0, length = 0;
 			try {
 				offset = doc.getLineOffset(region.getStart().getLine());
 				length = doc.getLineOffset(region.getEnd().getLine()) - offset;
 			} catch (BadLocationException e) {
 				Activator.log(e);
 			}

 			Position inactivePosition = new Position(offset, length);
 			try {
 				doc.addPosition(PresentationReconcilerCPP.INACTIVE_CODE_HIGHLIGHTING_POSITION_CATEGORY,
 						inactivePosition);
 			} catch (BadLocationException | BadPositionCategoryException e) {
 				Activator.log(e);
 			}
 		}
 	}

 	@JsonNotification("$cquery/publishSemanticHighlighting")
 	public final void semanticHighlights(CquerySemanticHighlights highlights) {
 		URI uriReceived = highlights.getUri();

 		// List of PresentationReconcilerCPP objects attached with same C++ source file.
 		List<PresentationReconcilerCPP> matchingReconcilers = new ArrayList<>();

 		for (PresentationReconcilerCPP eachReconciler : PresentationReconcilerCPP.presentationReconcilers) {
 			IDocument currentReconcilerDoc = eachReconciler.getTextViewer().getDocument();
 			URI currentReconcilerUri = getUri(currentReconcilerDoc);

 			if (currentReconcilerUri == null) {
 				continue;
 			}

 			if (uriReceived.equals(currentReconcilerUri)) {
 				matchingReconcilers.add(eachReconciler);
 			}
 		}

 		if (matchingReconcilers.size() == 0) {
 			return;
 		}

 		// Using only first object of matchingReconcilers because all reconciler objects share same document object.
 		IDocument doc = matchingReconcilers.get(0).getTextViewer().getDocument();
 		IPreferenceStore store = CUIPlugin.getDefault().getPreferenceStore();

 		// Removing semantic highlighting position category and old positions from document.
 		try {
 			doc.removePositionCategory(PresentationReconcilerCPP.SEMANTIC_HIGHLIGHTING_POSITION_CATEGORY);
 		} catch (BadPositionCategoryException e) {
 			Activator.log(e);
 		}
 		// Again add Semantic Highlighting Position Category to the document.
 		doc.addPositionCategory(PresentationReconcilerCPP.SEMANTIC_HIGHLIGHTING_POSITION_CATEGORY);

 		for (HighlightSymbol highlight : highlights.getSymbols()) {

 			String highlightingName = HighlightSymbol.getHighlightingName(highlight.getKind(),
 					highlight.getParentKind(), highlight.getStorage(), highlight.getRole());
 			String colorKey = PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_PREFIX + highlightingName
 					+ PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_COLOR_SUFFIX;

 			boolean isEnabled = store.getBoolean(PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_PREFIX
 					+ highlightingName + PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_ENABLED_SUFFIX);
 			boolean isBold = store.getBoolean(PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_PREFIX + highlightingName
 					+ PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_BOLD_SUFFIX);
 			boolean isItalic = store.getBoolean(PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_PREFIX
 					+ highlightingName + PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_ITALIC_SUFFIX);
 			boolean isUnderline = store.getBoolean(PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_PREFIX
 					+ highlightingName + PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_UNDERLINE_SUFFIX);
 			boolean isStrikethrough = store.getBoolean(PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_PREFIX
 					+ highlightingName + PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_STRIKETHROUGH_SUFFIX);

 			// TODO: Use IColorManager to cache Color objects so that only one object per color is created.
 			Color color = new Color(Display.getCurrent(),
 					PreferenceConverter.getColor(CUIPlugin.getDefault().getPreferenceStore(), colorKey));

 			List<Range> ranges = highlight.getRanges();
 			for (Range range : ranges) {

 				int offset = 0, length = 0;
 				try {
 					offset = doc.getLineOffset(range.getStart().getLine()) + range.getStart().getCharacter();
 					length = doc.getLineOffset(range.getEnd().getLine()) + range.getEnd().getCharacter() - offset;
 				} catch (BadLocationException e) {
 					Activator.log(e);
 				}

 				int textStyle = SWT.NORMAL;

 				if (isBold) {
 					textStyle = SWT.BOLD;
 				}
 				if (isItalic) {
 					textStyle |= SWT.ITALIC;
 				}
 				if (isUnderline) {
 					textStyle |= TextAttribute.UNDERLINE;
 				}
 				if (isStrikethrough) {
 					textStyle |= TextAttribute.STRIKETHROUGH;
 				}

 				TextAttribute textAttribute = new TextAttribute(color, null, textStyle);
 				HighlightingStyle highlightingStyle = new HighlightingStyle(textAttribute, isEnabled);
 				HighlightedPosition highlightedPosition = new HighlightedPosition(offset, length, highlightingStyle,
 						matchingReconcilers.get(0).getSemanticHighlightingPositionUpdater());
 				try {
 					doc.addPosition(PresentationReconcilerCPP.SEMANTIC_HIGHLIGHTING_POSITION_CATEGORY,
 							highlightedPosition);
 				} catch (BadLocationException | BadPositionCategoryException e) {
 					Activator.log(e);
 				}
 			}
 		}

 		Display.getDefault().asyncExec(new Runnable() {
 			@Override
 			public void run() {
 				for (PresentationReconcilerCPP eachReconciler : matchingReconcilers) {
 					TextPresentation presentation = eachReconciler.createPresentation(new Region(0, doc.getLength()),
 							doc);
 					eachReconciler.getTextViewer().changeTextPresentation(presentation, false);
 				}
 			}
 		});
 	}

 	/*
 	 * Returns the URI of the Document provided.
 	 * Can return null value.
 	 */
 	public static URI getUri(IDocument document) {
 		URI uri = null;
 		IFile file = LSPEclipseUtils.getFile(document);
 		if (file != null) {
 			uri = LSPEclipseUtils.toUri(file);
 		}
 		return uri;
 	}
 }
	/*******************************************************************************
	* Copyright (c) 2018 Manish Khurana , Nathan Ridge and others.
	*
	* This program and the accompanying materials
	* are made available under the terms of the Eclipse Public License 2.0
	* which accompanies this distribution, and is available at
	* https://www.eclipse.org/legal/epl-2.0/
	*
	* SPDX-License-Identifier: EPL-2.0
	*******************************************************************************/

	package org.eclipse.cdt.lsp.core;

	import java.net.URI;
	import java.util.ArrayList;
	import java.util.List;

	import org.eclipse.cdt.internal.ui.editor.SemanticHighlightingManager.HighlightedPosition;
	import org.eclipse.cdt.internal.ui.editor.SemanticHighlightingManager.HighlightingStyle;
	import org.eclipse.cdt.lsp.core.cquery.CqueryInactiveRegions;
	import org.eclipse.cdt.lsp.core.cquery.CquerySemanticHighlights;
	import org.eclipse.cdt.lsp.core.cquery.HighlightSymbol;
	import org.eclipse.cdt.lsp.core.cquery.IndexingProgressStats;
	import org.eclipse.cdt.ui.CUIPlugin;
	import org.eclipse.cdt.ui.PreferenceConstants;
	import org.eclipse.core.resources.IFile;
	import org.eclipse.jface.action.StatusLineContributionItem;
	import org.eclipse.jface.action.StatusLineManager;
	import org.eclipse.jface.preference.IPreferenceStore;
	import org.eclipse.jface.preference.PreferenceConverter;
	import org.eclipse.jface.text.BadLocationException;
	import org.eclipse.jface.text.BadPositionCategoryException;
	import org.eclipse.jface.text.IDocument;
	import org.eclipse.jface.text.Position;
	import org.eclipse.jface.text.Region;
	import org.eclipse.jface.text.TextAttribute;
	import org.eclipse.jface.text.TextPresentation;
	import org.eclipse.lsp4e.LSPEclipseUtils;
	import org.eclipse.lsp4e.LanguageClientImpl;
	import org.eclipse.lsp4j.Range;
	import org.eclipse.lsp4j.jsonrpc.services.JsonNotification;
	import org.eclipse.osgi.util.NLS;
	import org.eclipse.swt.SWT;
	import org.eclipse.swt.graphics.Color;
	import org.eclipse.swt.widgets.Display;
	import org.eclipse.ui.IWorkbenchWindow;
	import org.eclipse.ui.PlatformUI;
	import org.eclipse.ui.internal.WorkbenchWindow;

	@SuppressWarnings("restriction")
	public class Server2ClientProtocolExtension extends LanguageClientImpl {

	@JsonNotification("$cquery/progress")
	public final void indexingProgress(IndexingProgressStats stats) {

	Display.getDefault().asyncExec(new Runnable() {
	@Override
	public void run() {
	final String cqueryStatusFieldId = "org.eclipse.lsp4e.cpp.status"; //$NON-NLS-1$
	final int width = 28;
	IWorkbenchWindow[] workbenchWindows = PlatformUI.getWorkbench().getWorkbenchWindows();
	for (IWorkbenchWindow window : workbenchWindows) {
	StatusLineManager statusLine = ((WorkbenchWindow) window).getStatusLineManager();
	StatusLineContributionItem cqueryStatusField = (StatusLineContributionItem) statusLine
	.find(cqueryStatusFieldId);
	if (cqueryStatusField == null) {
	cqueryStatusField = new StatusLineContributionItem(cqueryStatusFieldId, width);
	statusLine.add(cqueryStatusField);
	}
	String msg = stats.getTotalJobs() > 0 ? NLS.bind(Messages.CqueryStateBusy, stats.getTotalJobs())
	: Messages.CqueryStateIdle;
	cqueryStatusField.setText(msg);
	}
	}
	});
	}

	@JsonNotification("$cquery/setInactiveRegions")
	public final void setInactiveRegions(CqueryInactiveRegions regions) {
	URI uriReceived = regions.getUri();
	List<Range> inactiveRegions = regions.getInactiveRegions();
	IDocument doc = null;

	// To get the document for the received URI.
	for (PresentationReconcilerCPP eachReconciler : PresentationReconcilerCPP.presentationReconcilers) {
	IDocument currentReconcilerDoc = eachReconciler.getTextViewer().getDocument();
	URI currentReconcilerUri = getUri(currentReconcilerDoc);

	if (currentReconcilerUri == null) {
	continue;
	}

	if (uriReceived.equals(currentReconcilerUri)) {
	doc = currentReconcilerDoc;
	break;
	}
	}

	if (doc == null) {
	return;
	}

	// Removing inactive code highlighting position category and old positions from document.
	try {
	doc.removePositionCategory(PresentationReconcilerCPP.INACTIVE_CODE_HIGHLIGHTING_POSITION_CATEGORY);
	} catch (BadPositionCategoryException e) {
	Activator.log(e);
	}
	// Again add Inactive Code Position Category to the document.
	doc.addPositionCategory(PresentationReconcilerCPP.INACTIVE_CODE_HIGHLIGHTING_POSITION_CATEGORY);

	for (Range region : inactiveRegions) {
	int offset = 0, length = 0;
	try {
	offset = doc.getLineOffset(region.getStart().getLine());
	length = doc.getLineOffset(region.getEnd().getLine()) - offset;
	} catch (BadLocationException e) {
	Activator.log(e);
	}

	Position inactivePosition = new Position(offset, length);
	try {
	doc.addPosition(PresentationReconcilerCPP.INACTIVE_CODE_HIGHLIGHTING_POSITION_CATEGORY,
	inactivePosition);
	} catch (BadLocationException \| BadPositionCategoryException e) {
	Activator.log(e);
	}
	}
	}

	@JsonNotification("$cquery/publishSemanticHighlighting")
	public final void semanticHighlights(CquerySemanticHighlights highlights) {
	URI uriReceived = highlights.getUri();

	// List of PresentationReconcilerCPP objects attached with same C++ source file.
	List<PresentationReconcilerCPP> matchingReconcilers = new ArrayList<>();

	for (PresentationReconcilerCPP eachReconciler : PresentationReconcilerCPP.presentationReconcilers) {
	IDocument currentReconcilerDoc = eachReconciler.getTextViewer().getDocument();
	URI currentReconcilerUri = getUri(currentReconcilerDoc);

	if (currentReconcilerUri == null) {
	continue;
	}

	if (uriReceived.equals(currentReconcilerUri)) {
	matchingReconcilers.add(eachReconciler);
	}
	}

	if (matchingReconcilers.size() == 0) {
	return;
	}

	// Using only first object of matchingReconcilers because all reconciler objects share same document object.
	IDocument doc = matchingReconcilers.get(0).getTextViewer().getDocument();
	IPreferenceStore store = CUIPlugin.getDefault().getPreferenceStore();

	// Removing semantic highlighting position category and old positions from document.
	try {
	doc.removePositionCategory(PresentationReconcilerCPP.SEMANTIC_HIGHLIGHTING_POSITION_CATEGORY);
	} catch (BadPositionCategoryException e) {
	Activator.log(e);
	}
	// Again add Semantic Highlighting Position Category to the document.
	doc.addPositionCategory(PresentationReconcilerCPP.SEMANTIC_HIGHLIGHTING_POSITION_CATEGORY);

	for (HighlightSymbol highlight : highlights.getSymbols()) {

	String highlightingName = HighlightSymbol.getHighlightingName(highlight.getKind(),
	highlight.getParentKind(), highlight.getStorage(), highlight.getRole());
	String colorKey = PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_PREFIX + highlightingName
	+ PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_COLOR_SUFFIX;

	boolean isEnabled = store.getBoolean(PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_PREFIX
	+ highlightingName + PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_ENABLED_SUFFIX);
	boolean isBold = store.getBoolean(PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_PREFIX + highlightingName
	+ PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_BOLD_SUFFIX);
	boolean isItalic = store.getBoolean(PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_PREFIX
	+ highlightingName + PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_ITALIC_SUFFIX);
	boolean isUnderline = store.getBoolean(PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_PREFIX
	+ highlightingName + PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_UNDERLINE_SUFFIX);
	boolean isStrikethrough = store.getBoolean(PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_PREFIX
	+ highlightingName + PreferenceConstants.EDITOR_SEMANTIC_HIGHLIGHTING_STRIKETHROUGH_SUFFIX);

	// TODO: Use IColorManager to cache Color objects so that only one object per color is created.
	Color color = new Color(Display.getCurrent(),
	PreferenceConverter.getColor(CUIPlugin.getDefault().getPreferenceStore(), colorKey));

	List<Range> ranges = highlight.getRanges();
	for (Range range : ranges) {

	int offset = 0, length = 0;
	try {
	offset = doc.getLineOffset(range.getStart().getLine()) + range.getStart().getCharacter();
	length = doc.getLineOffset(range.getEnd().getLine()) + range.getEnd().getCharacter() - offset;
	} catch (BadLocationException e) {
	Activator.log(e);
	}

	int textStyle = SWT.NORMAL;

	if (isBold) {
	textStyle = SWT.BOLD;
	}
	if (isItalic) {
	textStyle \|= SWT.ITALIC;
	}
	if (isUnderline) {
	textStyle \|= TextAttribute.UNDERLINE;
	}
	if (isStrikethrough) {
	textStyle \|= TextAttribute.STRIKETHROUGH;
	}

	TextAttribute textAttribute = new TextAttribute(color, null, textStyle);
	HighlightingStyle highlightingStyle = new HighlightingStyle(textAttribute, isEnabled);
	HighlightedPosition highlightedPosition = new HighlightedPosition(offset, length, highlightingStyle,
	matchingReconcilers.get(0).getSemanticHighlightingPositionUpdater());
	try {
	doc.addPosition(PresentationReconcilerCPP.SEMANTIC_HIGHLIGHTING_POSITION_CATEGORY,
	highlightedPosition);
	} catch (BadLocationException \| BadPositionCategoryException e) {
	Activator.log(e);
	}
	}
	}

	Display.getDefault().asyncExec(new Runnable() {
	@Override
	public void run() {
	for (PresentationReconcilerCPP eachReconciler : matchingReconcilers) {
	TextPresentation presentation = eachReconciler.createPresentation(new Region(0, doc.getLength()),
	doc);
	eachReconciler.getTextViewer().changeTextPresentation(presentation, false);
	}
	}
	});
	}

	/*
	* Returns the URI of the Document provided.
	* Can return null value.
	*/
	public static URI getUri(IDocument document) {
	URI uri = null;
	IFile file = LSPEclipseUtils.getFile(document);
	if (file != null) {
	uri = LSPEclipseUtils.toUri(file);
	}
	return uri;
	}
	}