Merge pull request #605 from mesonbuild/ternary

Added ternary operator support

1 files changed, 20 insertions, 0 deletions

diff --git a/mesonbuild/mparser.py b/mesonbuild/mparser.py
index fd720fb..2f0eb2d 100644
--- a/mesonbuild/mparser.py
+++ b/mesonbuild/mparser.py
@@ -67,6 +67,7 @@ class Lexer:
             ('lt', re.compile(r'<')),
             ('ge', re.compile(r'>=')),
             ('gt', re.compile(r'>')),
+            ('questionmark', re.compile(r'\?')),
         ]
 
     def lex(self, code):
@@ -282,6 +283,14 @@ class IfNode():
         self.condition = condition
         self.block = block
 
+class TernaryNode():
+    def __init__(self, lineno, colno, condition, trueblock, falseblock):
+        self.lineno = lineno
+        self.colno = colno
+        self.condition = condition
+        self.trueblock = trueblock
+        self.falseblock = falseblock
+
 class ArgumentNode():
     def __init__(self, token):
         self.lineno = token.lineno
@@ -344,6 +353,7 @@ class Parser:
     def __init__(self, code):
         self.stream = Lexer().lex(code)
         self.getsym()
+        self.in_ternary = False
 
     def getsym(self):
         try:
@@ -383,6 +393,16 @@ class Parser:
                 raise ParseException('Assignment target must be an id.',
                                      left.lineno, left.colno)
             return AssignmentNode(left.lineno, left.colno, left.value, value)
+        elif self.accept('questionmark'):
+            if self.in_ternary:
+                raise ParseException('Nested ternary operators are not allowed.',
+                                     left.lineno, left.colno)
+            self.in_ternary = True
+            trueblock = self.e1()
+            self.expect('colon')
+            falseblock = self.e1()
+            self.in_ternary = False
+            return TernaryNode(left.lineno, left.colno, left, trueblock, falseblock)
         return left
 
     def e2(self):