Bug fix #91

Author: SSlinky

server/src/capabilities/capabilities.ts

@@ -17,7 +17,7 @@ import { FoldingRange, FoldingRangeKind } from '../capabilities/folding';
 import { SemanticToken, SemanticTokenModifiers, SemanticTokenTypes } from '../capabilities/semanticTokens';
 import { BaseRuleSyntaxElement, BaseIdentifyableSyntaxElement, BaseSyntaxElement, Context, HasSemanticTokenCapability } from '../project/elements/base';
 import { AmbiguousNameDiagnostic, BaseDiagnostic, DuplicateDeclarationDiagnostic, ShadowDeclarationDiagnostic, SubOrFunctionNotDefinedDiagnostic, UnusedDiagnostic, VariableNotDefinedDiagnostic } from './diagnostics';
-import { isPositionInsideRange, isRangeInsideRange } from '../utils/helpers';
+import { isPositionInsideRange, isRangeInsideRange, rangeEquals } from '../utils/helpers';
 
 
 abstract class BaseCapability {
@@ -795,28 +795,23 @@ export class ScopeItemCapability {
 	 * Recursively removes all scopes with the passed in uri and
 	 * within the range bounds, including where it is linked.
 	 */
-	invalidate(uri: string, range: Range): void {
-		const isInvalidScope = (scope: ScopeItemCapability) =>
-			scope.locationUri === uri
-			&& scope.element?.context.range
-			&& isRangeInsideRange(scope.element.context.range, range);
-
+	invalidate(uri: string, range?: Range): void {
 		const cleanScopes = (scopes?: ScopeItemCapability[]) => {
 			if (scopes === undefined) {
 				return undefined;
 			}
 
 			const result: ScopeItemCapability[] = [];
 			scopes.forEach(scope => {
-				if (isInvalidScope(scope)) {
+				if (scope.isLocatedAt(uri, range)) {
 					Services.logger.debug(`Invalidating ${scope.name}`);
 
 					// Clean the backlinks on the linked item if we have one.
 					if (scope.link) scope.link.backlinks = cleanScopes(
 						scope.link.backlinks);
 
 					// Clean the invaludated scope.
-					scope.invalidate(uri, range);
+					scope.invalidate(uri, scope.range);
 
 					return;
 				}
@@ -857,8 +852,22 @@ export class ScopeItemCapability {
 
 		// Do a basic clean on backlinks that doesn't trigger recursion.
 		if (this.backlinks) {
-			this.backlinks = this.backlinks.filter(scope => !isInvalidScope(scope));
+			this.backlinks = this.backlinks
+				.filter(scope => !scope.isLocatedAt(uri, range));
+		}
+	}
+
+	/** Returns true if the uri matches and, if passed, range is fully inside range. */
+	isLocatedAt(uri: string, range?: Range): boolean {
+		if (uri !== this.locationUri) {
+			return false;
 		}
+
+		if (!range) {
+			return true;
+		}
+
+		return isRangeInsideRange(this.range, range);
 	}
 
 	/** Returns true for public and false for private */

server/src/project/workspace.ts

@@ -153,8 +153,7 @@ export class Workspace implements IWorkspace {
 
 		if (previousDocument) {
 			Services.projectScope.invalidate(
-				previousDocument.uri,
-				previousDocument.range
+				previousDocument.uri
 			);
 		}
 

server/src/utils/helpers.ts

@@ -94,7 +94,12 @@ export function isPositionInsideRange(position: Position, range: Range): boolean
  * @param inner The range to test as enveloped.
  * @param outer The range to test as enveloping.
  */
-export function isRangeInsideRange(inner: Range, outer: Range): boolean {
+export function isRangeInsideRange(inner?: Range, outer?: Range): boolean {
+	// Test we have ranges.
+	if (!inner || !outer) {
+		return false;
+	}
+
 	// Test characters on single-line ranges.
 	const isSingleLine = inner.start.line === inner.end.line
 		&& outer.start.line === outer.end.line
@@ -107,4 +112,12 @@ export function isRangeInsideRange(inner: Range, outer: Range): boolean {
 	// Test lines on multi-line ranges.
 	return inner.start.line >= outer.start.line
 		&& inner.end.line <= outer.end.line;
+}
+
+export function rangeEquals(r1?: Range, r2?: Range): boolean {
+	return !!r1 && !!r2
+		&& r1.start.line === r2.start.line
+		&& r1.start.character === r2.start.character
+		&& r1.end.line === r2.end.line
+		&& r1.end.character === r2.end.character;
 }