Allow identifiers to have a LABEL reference again

MarkusAmshove · May 1, 2023 · 52f2231 · 52f2231
1 parent 7b81394
commit 52f2231
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/libs/natparse/src/main/java/org/amshove/natparse/parsing/AbstractParser.java b/libs/natparse/src/main/java/org/amshove/natparse/parsing/AbstractParser.java
@@ -781,6 +781,15 @@ protected IVariableReferenceNode consumeVariableReferenceNode(BaseSyntaxNode nod
 			consumeMandatory(reference, SyntaxKind.RPAREN);
 		}
 
+		if (peekKind(SyntaxKind.LPAREN) && peekKind(1, SyntaxKind.LABEL_IDENTIFIER))
+		{
+			while (!isAtEnd() && !peekKind(SyntaxKind.RPAREN))
+			{
+				consume(reference);
+			}
+			consumeMandatory(reference, SyntaxKind.RPAREN);
+		}
+
 		unresolvedReferences.add(reference);
 		return reference;
 	}

diff --git a/libs/natparse/src/test/java/org/amshove/natparse/parsing/StatementListParserShould.java b/libs/natparse/src/test/java/org/amshove/natparse/parsing/StatementListParserShould.java
@@ -1983,4 +1983,11 @@ void parseDownloadPcCommandAsync()
 	{
 		assertParsesSingleStatement("DOWNLOAD PC 5 COMMAND 'Hi' ASYNC", IWritePcNode.class);
 	}
+
+	@Test
+	void allowLabelIdentifierAsVariableOperand()
+	{
+		var assignment = assertParsesSingleStatement("#VAR(R1.) := 5", IAssignmentStatementNode.class);
+		assertIsVariableReference(assignment.target(), "#VAR");
+	}
 }