powershell: Parse enum labels

https://learn.microsoft.com/en-us/powershell/module/microsoft.powershell.core/about/about_enum?view=powershell-7.4
universal-ctags · May 13, 2024 · bad24b7 · bad24b7
1 parent 8da7ff0
commit bad24b7
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 0 deletions.
diff --git a/Units/parser-powershell.r/enum-powershell.d/expected.tags b/Units/parser-powershell.r/enum-powershell.d/expected.tags
@@ -1,4 +1,12 @@
 EnumName1	input.ps1	/^enum EnumName1 {$/;"	g
+Label11	input.ps1	/^    Label11$/;"	e	enum:EnumName1
+Label12	input.ps1	/^    Label12 = 10$/;"	e	enum:EnumName1
 EnumName2	input.ps1	/^enum EnumName2 {$/;"	g
+Label21	input.ps1	/^    Label21$/;"	e	enum:EnumName2
+Label22	input.ps1	/^    Label22 = 20$/;"	e	enum:EnumName2
 EnumName3	input.ps1	/^Enum EnumName3 {$/;"	g
+Label31	input.ps1	/^    Label31$/;"	e	enum:EnumName3
+Label32	input.ps1	/^    Label32 = 30$/;"	e	enum:EnumName3
 EnumName4	input.ps1	/^[Flags()] enum EnumName4 {$/;"	g
+Label41	input.ps1	/^    Label41$/;"	e	enum:EnumName4
+Label42	input.ps1	/^    Label42 = 40$/;"	e	enum:EnumName4
diff --git a/parsers/powershell.c b/parsers/powershell.c
@@ -42,6 +42,7 @@ typedef enum {
 	K_CLASS,
 	K_FILTER,
 	K_ENUM,
+	K_ENUMLABEL,
 	COUNT_KIND
 } powerShellKind;
 
@@ -51,6 +52,7 @@ static kindDefinition PowerShellKinds[COUNT_KIND] = {
 	{ true, 'c', "class",		"classes" },
 	{ true, 'i', "filter",		"filter" },
 	{ true, 'g', "enum",		"enum names" },
+	{ true, 'e', "enumerator",	"enum labels" },
 };
 
 
@@ -648,6 +650,30 @@ static bool parseEnum (tokenInfo *const token)
 	return readNext;
 }
 
+/* parses declarations of the form
+ * 	<label> [= <int-value>]
+ * that is, contents of an enum
+ */
+static bool parseEnumLabel (tokenInfo *const token)
+{
+	bool readNext = true;
+
+	if (token->parentKind != K_ENUM)
+		return false;
+
+	if (token->type != TOKEN_IDENTIFIER)
+		return false;
+
+	makeSimplePowerShellTag (token, K_ENUMLABEL, ACCESS_UNDEFINED);
+	readToken (token);
+	if (token->type != TOKEN_EQUAL_SIGN)
+		readNext = false;
+	else /* skip int-value */
+		readToken (token);
+
+	return readNext;
+}
+
 /* parses declarations of the form
  * 	$var = VALUE
  */
@@ -735,6 +761,11 @@ static void enterScope (tokenInfo *const parentToken,
 				readNext = parseVariable (token);
 				break;
 
+			case TOKEN_IDENTIFIER:
+				if (parentKind == K_ENUM)
+					readNext = parseEnumLabel (token);
+				break;
+
 			default: break;
 		}