Add python #123 support

tempestphp · May 23, 2024 · b8b2947 · b8b2947
1 parent 116da0e
commit b8b2947
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 2 deletions.
diff --git a/src/Languages/Python/Patterns/PyBooleanPattern.php b/src/Languages/Python/Patterns/PyBooleanPattern.php
@@ -19,6 +19,6 @@ public function getPattern(): string
 
     public function getTokenType(): TokenTypeEnum
     {
-        return TokenTypeEnum::TYPE;
+        return TokenTypeEnum::BOOLEAN;
     }
 }
diff --git a/src/Languages/Python/Patterns/PyNumberPattern.php b/src/Languages/Python/Patterns/PyNumberPattern.php
@@ -0,0 +1,24 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Tempest\Highlight\Languages\Python\Patterns;
+
+use Tempest\Highlight\IsPattern;
+use Tempest\Highlight\Pattern;
+use Tempest\Highlight\Tokens\TokenTypeEnum;
+
+final readonly class PyNumberPattern implements Pattern
+{
+    use IsPattern;
+
+    public function getPattern(): string
+    {
+        return '(?<match>\b0(?:[bB](?:_?[01])+|[oO](?:_?[0-7])+|[xX](?:_?[a-fA-F0-9])+)\b|(?:\b\d+(?:_\d+)*(?:\.(?:\d+(?:_\d+)*)?)?|\B\.\d+(?:_\d+)*)(?:[eE][+-]?\d+(?:_\d+)*)?j?(?!\w))';
+    }
+
+    public function getTokenType(): TokenTypeEnum
+    {
+        return TokenTypeEnum::NUMBER;
+    }
+}
diff --git a/src/Languages/Python/PythonLanguage.php b/src/Languages/Python/PythonLanguage.php
@@ -15,6 +15,7 @@
 use Tempest\Highlight\Languages\Python\Patterns\PyDecoratorPattern;
 use Tempest\Highlight\Languages\Python\Patterns\PyFunctionPattern;
 use Tempest\Highlight\Languages\Python\Patterns\PyKeywordPattern;
+use Tempest\Highlight\Languages\Python\Patterns\PyNumberPattern;
 use Tempest\Highlight\Languages\Python\Patterns\PyOperatorPattern;
 use Tempest\Highlight\Languages\Python\Patterns\PyTripleDoubleQuoteStringPattern;
 use Tempest\Highlight\Languages\Python\Patterns\PyTripleSingleQuoteStringPattern;
@@ -52,12 +53,17 @@ public function getPatterns(): array
             new PyFunctionPattern(),
 
             // TYPES
-            new PyBooleanPattern(),
             new PyBuiltinPattern(),
 
             // COMMENTS
             new PyCommentPattern(),
 
+            // NUMBERS
+            new PyNumberPattern(),
+
+            // BOOLEANS
+            new PyBooleanPattern(),
+
             // OPERATORS
             new PyOperatorPattern(),