Add multilanguage support to step definitions

src/language/csharpLanguage.ts

@@ -1,6 +1,6 @@
 import { StringOrRegExp } from '@cucumber/cucumber-expressions'
 
-import { childrenToString, NO_QUOTES } from './helpers.js'
+import { childrenToString, functionNames, NO_QUOTES } from './helpers.js'
 import { Language, TreeSitterSyntaxNode } from './types.js'
 
 export const csharpLanguage: Language = {
@@ -77,7 +77,7 @@ export const csharpLanguage: Language = {
       )
     )
   )
-  (#match? @annotation-name "Given|When|Then|And|But|StepDefinition")
+  (#match? @annotation-name "${functionNames}|And|But|StepDefinition")
 ) @root
 `,
     `
@@ -92,7 +92,7 @@ export const csharpLanguage: Language = {
       )
     )
   )
-  (#match? @annotation-name "Given|When|Then|And|But|StepDefinition")
+  (#match? @annotation-name "${functionNames}|And|But|StepDefinition")
 ) @root
 `,
   ],

src/language/helpers.ts

@@ -1,4 +1,5 @@
 import { ParameterType, RegExps } from '@cucumber/cucumber-expressions'
+import { dialects } from '@cucumber/gherkin'
 import { DocumentUri, LocationLink, Range } from 'vscode-languageserver-types'
 
 import { Link, NodePredicate, TreeSitterQueryMatch, TreeSitterSyntaxNode } from './types'
@@ -70,3 +71,9 @@ export function filter(
 function flatten(node: TreeSitterSyntaxNode): TreeSitterSyntaxNode[] {
   return node.children.reduce((r, o) => [...r, ...flatten(o)], [node])
 }
+
+export const functionNames = Object.values(dialects)
+  .flatMap((dialect) => [...dialect.given, ...dialect.when, ...dialect.then])
+  .map((keyword) => keyword.trim())
+  .filter((keyword) => keyword !== '*')
+  .join('|')

src/language/javaLanguage.ts

@@ -1,3 +1,4 @@
+import { functionNames } from './helpers.js'
 import { Language, TreeSitterSyntaxNode } from './types.js'
 
 export const javaLanguage: Language = {
@@ -90,7 +91,7 @@ export const javaLanguage: Language = {
       )
     )
   )
-  (#match? @annotation-name "Given|When|Then")
+  (#match? @annotation-name "${functionNames}")
 ) @root
 `,
   ],

src/language/phpLanguage.ts

@@ -1,3 +1,4 @@
+import { functionNames } from './helpers.js'
 import { Language } from './types.js'
 
 export const phpLanguage: Language = {
@@ -20,7 +21,7 @@ export const phpLanguage: Language = {
     `
 (
   (comment)+ @expression
-  (#match? @expression "Given|When|Then")
+  (#match? @expression "${functionNames}")
 ) @root
 `,
   ],

src/language/pythonLanguage.ts

@@ -1,5 +1,6 @@
 import { StringOrRegExp } from '@cucumber/cucumber-expressions'
 
+import { functionNames } from './helpers.js'
 import { Language, TreeSitterSyntaxNode } from './types.js'
 
 export const pythonLanguage: Language = {
@@ -72,7 +73,7 @@ export const pythonLanguage: Language = {
           arguments: (argument_list (string) @expression)
         )
       )
-      (#match? @method "(given|when|then)")
+      (#match? @method "(${functionNames.toLowerCase()})")
     ) @root
 `,
   ],

src/language/rubyLanguage.ts

@@ -1,7 +1,7 @@
 import { StringOrRegExp } from '@cucumber/cucumber-expressions'
 import { RegExps } from '@cucumber/cucumber-expressions/dist/cjs/src/ParameterType'
 
-import { childrenToString, filter, NO_QUOTES } from './helpers.js'
+import { childrenToString, filter, functionNames, NO_QUOTES } from './helpers.js'
 import { Language, NodePredicate, TreeSitterSyntaxNode } from './types.js'
 
 export const rubyLanguage: Language = {
@@ -76,7 +76,7 @@ export const rubyLanguage: Language = {
       (regex) @expression
     ]
   )
-  (#match? @method "(Given|When|Then)$")
+  (#match? @method "(${functionNames})$")
 ) @root
 `,
   ],

src/language/rustLanguage.ts

@@ -1,3 +1,4 @@
+import { functionNames } from './helpers.js'
 import { Language, TreeSitterSyntaxNode } from './types.js'
 
 export const rustLanguage: Language = {
@@ -82,7 +83,7 @@ export const rustLanguage: Language = {
       )
     )
   )  
-  (#match? @meta-name "given|when|then")
+  (#match? @meta-name "${functionNames.toLowerCase()}")
 )
 (function_item) ) @root
 `,

src/language/tsxLanguage.ts

@@ -1,7 +1,7 @@
 import { StringOrRegExp } from '@cucumber/cucumber-expressions'
 import { RegExps } from '@cucumber/cucumber-expressions'
 
-import { childrenToString, filter, NO_QUOTES } from './helpers.js'
+import { childrenToString, filter, functionNames, NO_QUOTES } from './helpers.js'
 import { NO_EXPRESSION } from './SourceAnalyzer.js'
 import { Language, TreeSitterSyntaxNode } from './types.js'
 
@@ -80,7 +80,7 @@ export const tsxLanguage: Language = {
       (template_string) @expression
     ]
   )
-  (#match? @function-name "Given|When|Then")
+  (#match? @function-name "${functionNames}")
 ) @root
 `,
   ],

test/language/testdata/javascript/StepDefinitions.js

@@ -2,6 +2,8 @@ import { Given } from '@cucumber/cucumber'
 import assert from 'assert'
 import React from 'react'
 
+const Gegeven = Given;
+
 const dummyJsx = <span>Hello</span>
 
 Given('a {uuid}', async function (uuid) {
@@ -24,6 +26,6 @@ Given('an {undefined-parameter}', async function (date) {
   assert(date)
 })
 
-Given("the bee's knees", async function () {
+Gegeven("the bee's knees", async function () {
   assert(true)
 })