test2

Author: Napalys

javascript/extractor/src/com/semmle/js/ast/regexp/StringDisjunction.java

@@ -0,0 +1,25 @@
+package com.semmle.js.ast.regexp;
+
+import com.semmle.js.ast.SourceLocation;
+
+/**
+ * A '\q{}' escape sequence in a regular expression, which is a special extension
+ * to standard regular expressions.
+ */
+public class StringDisjunction extends RegExpTerm {
+  private final RegExpTerm term;
+
+  public StringDisjunction(SourceLocation loc, RegExpTerm term) {
+    super(loc, "StringDisjunction");
+    this.term = term;
+  }
+
+  public RegExpTerm getTerm() {
+    return term;
+  }
+
+  @Override
+  public void accept(Visitor v) {
+    v.visit(this);
+  }
+}

javascript/extractor/src/com/semmle/js/ast/regexp/Subtraction.java

@@ -0,0 +1,22 @@
+package com.semmle.js.ast.regexp;
+
+import com.semmle.js.ast.SourceLocation;
+import java.util.List;
+
+public class Subtraction extends RegExpTerm {
+    private final List<RegExpTerm> subtraction;
+
+    public Subtraction(SourceLocation loc, List<RegExpTerm> subtraction) {
+        super(loc, "Intersection");
+        this.subtraction = subtraction;
+    }
+
+    @Override
+    public void accept(Visitor v) {
+        v.visit(this);
+    }
+
+    public List<RegExpTerm> getSubtraction() {
+        return subtraction;
+    }
+}

javascript/extractor/src/com/semmle/js/ast/regexp/Union.java

@@ -0,0 +1,22 @@
+package com.semmle.js.ast.regexp;
+
+import com.semmle.js.ast.SourceLocation;
+import java.util.List;
+
+public class Union extends RegExpTerm {
+    private final List<RegExpTerm> union;
+
+    public Union(SourceLocation loc, List<RegExpTerm> union) {
+        super(loc, "Union");
+        this.union = union;
+    }
+
+    @Override
+    public void accept(Visitor v) {
+        v.visit(this);
+    }
+
+    public List<RegExpTerm> getUnion() {
+        return union;
+    }
+}

javascript/extractor/src/com/semmle/js/ast/regexp/Visitor.java

@@ -63,4 +63,10 @@ public interface Visitor {
   public void visit(UnicodePropertyEscape nd);
 
   public void visit(Intersection nd);
+
+  public void visit(Subtraction nd);
+
+  public void visit(Union nd);
+
+  public void visit(StringDisjunction nd);
 }

javascript/extractor/src/com/semmle/js/extractor/RegExpExtractor.java

@@ -23,6 +23,8 @@
 import com.semmle.js.ast.regexp.HexEscapeSequence;
 import com.semmle.js.ast.regexp.IdentityEscape;
 import com.semmle.js.ast.regexp.Intersection;
+import com.semmle.js.ast.regexp.Subtraction;
+import com.semmle.js.ast.regexp.Union;
 import com.semmle.js.ast.regexp.Literal;
 import com.semmle.js.ast.regexp.NamedBackReference;
 import com.semmle.js.ast.regexp.NonWordBoundary;
@@ -34,6 +36,7 @@
 import com.semmle.js.ast.regexp.RegExpTerm;
 import com.semmle.js.ast.regexp.Sequence;
 import com.semmle.js.ast.regexp.Star;
+import com.semmle.js.ast.regexp.StringDisjunction;
 import com.semmle.js.ast.regexp.UnicodeEscapeSequence;
 import com.semmle.js.ast.regexp.UnicodePropertyEscape;
 import com.semmle.js.ast.regexp.Visitor;
@@ -94,6 +97,9 @@ public RegExpExtractor(TrapWriter trapwriter, LocationManager locationManager) {
     termkinds.put("ZeroWidthNegativeLookbehind", 26);
     termkinds.put("UnicodePropertyEscape", 27);
     termkinds.put("Intersection", 28);
+    termkinds.put("Subtraction", 29);
+    termkinds.put("Union", 30);
+    termkinds.put("StringDisjunction", 31); // Add StringDisjunction to the kind map
   }
 
   private static final String[] errmsgs =
@@ -292,6 +298,12 @@ public void visit(UnicodePropertyEscape nd) {
       if (nd.hasValue()) trapwriter.addTuple("unicode_property_escapevalue", lbl, nd.getValue());
     }
 
+    @Override
+    public void visit(StringDisjunction nd) {
+      Label lbl = extractTerm(nd, parent, idx);
+      visit(nd.getTerm(), lbl, 0);
+    }
+
     @Override
     public void visit(DecimalEscape nd) {
       visit((Literal) nd);
@@ -354,6 +366,22 @@ public void visit(Intersection nd) {
       for (RegExpTerm element : nd.getIntersections())
         visit(element, lbl, i++);
     }
+
+    @Override
+    public void visit(Subtraction nd) {
+      Label lbl = extractTerm(nd, parent, idx);
+      int i = 0;
+      for (RegExpTerm element : nd.getSubtraction())
+        visit(element, lbl, i++);
+    }
+
+    @Override
+    public void visit(Union nd) {
+      Label lbl = extractTerm(nd, parent, idx);
+      int i = 0;
+      for (RegExpTerm element : nd.getUnion())
+        visit(element, lbl, i++);
+    }
   }
 
   public void extract(String src, SourceMap sourceMap, Node parent, boolean isSpeculativeParsing) {