Load gherkin parser considering language comment

cucumber · Jan 28, 2013 · 4b1cf13 · 4b1cf13
1 parent 1d16d85
commit 4b1cf13
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/lib/cucumber/parser.js b/lib/cucumber/parser.js
@@ -1,6 +1,5 @@
 var Parser = function(featureSources, astFilter) {
   var Gherkin      = require('gherkin');
-  var GherkinLexer = require('gherkin/lib/gherkin/lexer/en');
   var Cucumber     = require('../cucumber');
 
   var features     = Cucumber.Ast.Features();
@@ -9,13 +8,25 @@ var Parser = function(featureSources, astFilter) {
 
   var self = {
     parse: function parse() {
-      var eventHandler = self.getEventHandlers();
-      var lexer = new GherkinLexer(self.getEventHandlers());
+      var lexers = {};
+      var lexer = function (lang) {
+        if (!(lang in lexers)) {
+          lexers[lang] = new (Gherkin.Lexer(lang))(self.getEventHandlers());
+        }
+
+        return lexers[lang];
+      };
+
       for (i in featureSources) {
         var currentSourceUri = featureSources[i][Parser.FEATURE_NAME_SOURCE_PAIR_URI_INDEX];
         var featureSource    = featureSources[i][Parser.FEATURE_NAME_SOURCE_PAIR_SOURCE_INDEX];
         self.setCurrentSourceUri(currentSourceUri);
-        lexer.scan(featureSource);
+        featureSource = featureSource.toString();
+
+        var languageMatch = /^# language: (.*)\n/.exec(featureSource);
+        var language = languageMatch == null ? 'en' : languageMatch[1];
+
+        lexer(language).scan(featureSource);
       }
       return features;
     },