1 files changed, 34 insertions, 5 deletions
diff --git a/src/Store/Query/Parser.hs b/src/Store/Query/Parser.hs
index 5b7bc1b..ce2a4dd 100644
--- a/src/Store/Query/Parser.hs
+++ b/src/Store/Query/Parser.hs
@@ -15,9 +15,11 @@ import Data.Vector qualified as V
 import Data.Void (Void)
 import Store.Exception (ParseError (ParseError))
 import Store.Query.Type
+import System.IO.Unsafe (unsafePerformIO)
 import Text.Megaparsec qualified as P
 import Text.Megaparsec.Char qualified as P
 import Text.Megaparsec.Char.Lexer qualified as P
+import Text.Regex.PCRE.String
 
 instance IsString Query where
   fromString =
@@ -82,6 +84,7 @@ instance IsString Query where
       comma = void $ lexeme (P.string ",")
       embed = void $ lexeme (P.string "EMBED")
       eq = void $ lexeme (P.string "==")
+      match = void $ lexeme (P.string "=~")
       from = void $ lexeme (P.string "FROM")
       on = void $ lexeme (P.string "ON")
       or = void $ lexeme (P.string "OR")
@@ -119,11 +122,22 @@ instance IsString Query where
             Where <$> comparison
           ]
 
-      comparison = do
-        a <- P.choice [Left <$> value, Right <$> field]
-        eq
-        b <- P.choice [Left <$> value, Right <$> field]
-        pure $ Eq a b
+      comparison =
+        P.choice
+          ( map
+              P.try
+              [ do
+                  a <- P.choice [Left <$> value, Right <$> field]
+                  eq
+                  b <- P.choice [Left <$> value, Right <$> field]
+                  pure $ Eq a b,
+                do
+                  a <- P.choice [Left <$> value, Right <$> field]
+                  match
+                  b <- regex
+                  pure $ Regex a b
+              ]
+          )
 
       offsetClause = do
         offset_
@@ -244,6 +258,21 @@ instance IsString Query where
       fieldPart :: P.Parsec Void String String
       fieldPart = P.takeWhile1P Nothing (\c -> not (isSpace c) && c /= ',' && c /= '.')
 
+      regex = lexeme do
+        ( unsafePerformIO . compile compBlank execBlank
+            <$> between (Just (P.string "\\")) (P.string "/") (P.string "/")
+          )
+          >>= either (\e -> error ("regex failed to compile: " <> show e)) pure
+
+      between (Just e) start end = start >> go
+        where
+          go =
+            P.choice
+              [ e >> (:) <$> P.anySingle <*> go,
+                P.try ((const "") <$> end),
+                (:) <$> P.anySingle <*> go
+              ]
+
       joinType :: P.Parsec Void String JoinType
       joinType =
         P.choice