okta-expression-parser 0.2.2__tar.gz → 0.4.0__tar.gz

{okta_expression_parser-0.2.2 → okta_expression_parser-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: okta-expression-parser
-Version: 0.2.2
+Version: 0.4.0
 Summary: Returns a boolean by passing Okta profile data to an Okta Expression using Okta Expression Language
 Author: Mathew Moon
 Author-email: me@mathewmoon.net

{okta_expression_parser-0.2.2 → okta_expression_parser-0.4.0}/okta_expression_parser/lexer.py

@@ -10,9 +10,9 @@ class ExpressionLexer(Lexer):
     # reflags = re.IGNORECASE
     # Set of token names.
     tokens = {
+        NAME,
         STRING,
         INT,
-        NAME,
         NE,
         GTE,
         LTE,
@@ -24,8 +24,14 @@ class ExpressionLexer(Lexer):
         NOT,
         NULL,
         BOOL,
+        COLON,
+        QUESTION_MARK,
         MEMBEROF,
         MEMBEROFANY,
+        MEMBEROFNAME,
+        MEMBEROFGROUPSTARTSWITH,
+        MEMBEROFGROUPCONTAINS,
+        MEMBEROFGROUPNAMEREGEX,
         USER,
         CLASS,
     }
@@ -46,14 +52,21 @@ class ExpressionLexer(Lexer):
     OR = r"or|OR|Or"
     EQ = r"(==|eq|EQ|Eq)"
     BOOL = r"(true|false|True|False|TRUE|FALSE)"
+    MEMBEROFGROUPCONTAINS = "isMemberOfGroupNameContains"
+    MEMBEROFGROUPNAMEREGEX = "isMemberOfGroupNameRegex"
+    MEMBEROFGROUPSTARTSWITH = "isMemberOfGroupNameStartsWith"
     MEMBEROFANY = "isMemberOfAnyGroup"
+    MEMBEROFNAME = "isMemberOfGroupName"
     MEMBEROF = "isMemberOfGroup"
     USER = r"user\b"
+    COLON = r"\:"
+    QUESTION_MARK = r"\?"
 
     NULL = r"null|NULL|Null"
     CLASS = r"String|Arrays|Convert|Iso3166Convert|Groups"
     INT = r"\d+"
-    STRING = r""""([^"\\]*(\\.[^"\\]*)*)"|\'([^\'\\]*(\\.[^\'\\]*)*)\'|''|\"\""""
+    #    STRING = r""""([^"\\]*(\\.[^"\\]*)*)"|\'([^\'\\]*(\\.[^\'\\]*)*)\'|''|\"\""""
+    STRING = r'''"(?:\\.|[^"\\])*"'''
     NAME = r"[a-zA-Z_][a-zA-Z0-9\-_]*"
 
     # Line number tracking

{okta_expression_parser-0.2.2 → okta_expression_parser-0.4.0}/okta_expression_parser/parser.py

@@ -1,6 +1,7 @@
 # pyright: reportUndefinedVariable=false
 import logging
 from os import environ
+from re import match
 from sys import stdout
 from typing import Any, Dict, List, Sequence
 from types import ModuleType
@@ -14,9 +15,8 @@ from .lexer import ExpressionLexer
 
 
 class ExpressionParser(sly.Parser):
-    if debug_file := environ.get("OKTA_EXPRESSION_PARSER_LOG_FILE"):
-        #: Writes grammar rules to a file for debugging
-        debugfile: str = environ.get(debug_file)
+    #: Writes grammar rules to a file for debugging
+    debugfile = environ.get("OKTA_EXPRESSION_PARSER_LOG_FILE")
 
     #: Gets the token list from the lexer (required)
     tokens: Sequence = ExpressionLexer.tokens
@@ -50,6 +50,7 @@ class ExpressionParser(sly.Parser):
         ):
             self._expression_classes.Groups.group_data = group_data
 
+        #: A dict, with Group ID's as the keys and group data as the values. Used for operations that require more context than just the group ID.
         self.group_data = group_data
 
         self.__group_ids: str = group_ids
@@ -64,6 +65,12 @@ class ExpressionParser(sly.Parser):
     def parse(self, expression: str):
         return super().parse(ExpressionLexer().tokenize(expression))
 
+    @_("operand")  # noqa: 821
+    @_("condition")  # noqa: 821
+    def result(self, p: YaccProduction) -> bool:  # noqa: 811
+        res = p.condition if hasattr(p, "condition") else p.operand
+        return res
+
     @_("operand EQ operand")  # noqa: 821
     def condition(self, p: YaccProduction) -> bool:
         operand0 = p.operand0
@@ -107,17 +114,39 @@ class ExpressionParser(sly.Parser):
         operand1 = p.operand1
         return type(operand0) == type(operand1) and operand0 > operand1
 
+    @_('"(" condition ")"')  # noqa: 821
+    def condition(self, p: YaccProduction) -> bool:  # noqa: 811
+        return p.condition
+
+    @_('"(" operand ")"')  # noqa: 821
+    def operand(self, p: YaccProduction) -> bool:  # noqa: 811
+        return p.operand
+
     @_("condition AND condition")  # noqa: 821
     def condition(self, p: YaccProduction) -> bool:  # noqa: 811
-        condition0 = p.condition0
-        condition1 = p.condition1
-        return condition0 and condition1
+        return p.condition0 and p.condition1
+
+    @_("operand AND operand")  # noqa: 821
+    def condition(self, p: YaccProduction) -> bool:  # noqa: 811
+        return p.operand0 and p.operand1
+
+    @_("operand AND condition")  # noqa: 821
+    @_("condition AND operand")  # noqa: 821
+    def condition(self, p: YaccProduction) -> bool:  # noqa: 811
+        return p.condition and p.operand
 
     @_("condition OR condition")  # noqa: 821
     def condition(self, p: YaccProduction) -> bool:  # noqa: 811
-        condition0 = p.condition0
-        condition1 = p.condition1
-        return condition0 or condition1
+        return p.condition0 or p.condition1
+
+    @_("operand OR operand")  # noqa: 821
+    def condition(self, p: YaccProduction) -> bool:  # noqa: 811
+        return p.operand0 or p.operand1
+
+    @_("operand OR condition")  # noqa: 821
+    @_("condition OR operand")  # noqa: 821
+    def condition(self, p: YaccProduction) -> bool:  # noqa: 811
+        return p.condition or p.operand
 
     @_("NOT condition")  # noqa: 821
     def condition(self, p: YaccProduction) -> bool:  # noqa: 811
@@ -128,22 +157,11 @@ class ExpressionParser(sly.Parser):
     def condition(self, p: YaccProduction) -> bool:  # noqa: 811
         return not p.operand
 
-    @_('"(" condition ")"')  # noqa: 821
-    def condition(self, p: YaccProduction) -> bool:  # noqa: 811
-        condition = p.condition
-        return condition
-
     @_('operand "," operand')  # noqa: 821
     def operand(self, p: YaccProduction) -> List[Any]:  # noqa: 811
         operand0 = p.operand0
         operand1 = p.operand1
 
-        # if not isinstance(operand0, tuple):
-        #    return (operand0, operand1)
-        #
-        # else:
-        #    return operand0 + (operand1,)
-
         if isinstance(operand0, tuple) and isinstance(operand1, tuple):
             res = operand0 + operand1
         elif isinstance(operand0, tuple) and not isinstance(operand1, tuple):
@@ -309,6 +327,57 @@ class ExpressionParser(sly.Parser):
 
         return p.operand in self.__group_ids
 
+    @_('MEMBEROFNAME "(" operand ")"')
+    def condition(self, p: YaccProduction):
+        """Tests if a user is a member of a group with a specific name"""
+
+        matches = [
+            v for k, v in self.group_data.items() if v["profile"]["name"] == p.operand
+        ]
+
+        return bool(matches)
+
+    @_('MEMBEROFGROUPSTARTSWITH "(" operand ")"')
+    def condition(self, p: YaccProduction):
+        """Tests if a user is a member of a group with a specific name"""
+
+        matches = [
+            v
+            for k, v in self.group_data.items()
+            if v["profile"]["name"].startswith(p.operand)
+        ]
+
+        return bool(matches)
+
+    @_('MEMBEROFGROUPCONTAINS "(" operand ")"')
+    def condition(self, p: YaccProduction):
+        """Tests if a user is a member of a group with a specific name"""
+
+        matches = [
+            v for k, v in self.group_data.items() if p.operand in v["profile"]["name"]
+        ]
+
+        return bool(matches)
+
+    @_('MEMBEROFGROUPNAMEREGEX "(" operand ")"')
+    def condition(self, p: YaccProduction):
+        """Tests if a user is a member of a group with a specific name"""
+
+        for _, group in self.group_data.items():
+            if match(p.operand, group["profile"]["name"]):
+                return True
+
+        return False
+
+    @_("operand QUESTION_MARK operand COLON operand")  # noqa: 821
+    @_("condition QUESTION_MARK operand COLON operand")  # noqa: 821
+    def operand(self, p: YaccProduction) -> Any:  # noqa: 811
+        condition = p.condition if hasattr(p, "condition") else p.operand0
+        if_true = p.operand0 if hasattr(p, "condition") else p.operand1
+        if_false = p.operand1 if hasattr(p, "condition") else p.operand2
+
+        return if_true if condition else if_false
+
     @_("condition error")  # noqa: 821
     def operand(self, x):  # noqa: 811
         raise SyntaxError(

{okta_expression_parser-0.2.2 → okta_expression_parser-0.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "okta-expression-parser"
-version = "0.2.2"
+version = "0.4.0"
 description = "Returns a boolean by passing Okta profile data to an Okta Expression using Okta Expression Language"
 authors = ["Mathew Moon <me@mathewmoon.net>"]
 readme = "README.md"