feat: add text_search (#215)

* feat: add text_search

* fix: run pre-commit hooks

* test: add tests for text search

* fix: run black

* fix: update poetry deps

---------

Co-authored-by: [email protected] <[email protected]>

Author: J0

poetry.lock

@@ -208,6 +208,29 @@ def maybe_single(self) -> AsyncMaybeSingleRequestBuilder:
             session=self.session,  # type: ignore
         )
 
+    def text_search(
+        self, column: str, query: str, options: Dict[str, any] = {}
+    ) -> AsyncFilterRequestBuilder:
+        type_ = options.get("type")
+        type_part = ""
+        if type_ == "plain":
+            type_part = "pl"
+        elif type_ == "phrase":
+            type_part = "ph"
+        elif type_ == "web_search":
+            type_part = "w"
+        config_part = f"({options.get('config')})" if options.get("config") else ""
+        self.params = self.params.add(column, f"{type_part}fts{config_part}.{query}")
+
+        return AsyncQueryRequestBuilder(
+            headers=self.headers,
+            http_method=self.http_method,
+            json=self.json,
+            params=self.params,
+            path=self.path,
+            session=self.session,  # type: ignore
+        )
+
 
 class AsyncRequestBuilder:
     def __init__(self, session: AsyncClient, path: str) -> None:
@@ -337,3 +360,6 @@ def delete(
         return AsyncFilterRequestBuilder(
             self.session, self.path, method, headers, params, json
         )
+
+    def stub(self):
+        return None

postgrest/_async/request_builder.py

@@ -208,6 +208,29 @@ def maybe_single(self) -> SyncMaybeSingleRequestBuilder:
             session=self.session,  # type: ignore
         )
 
+    def text_search(
+        self, column: str, query: str, options: Dict[str, any] = {}
+    ) -> SyncFilterRequestBuilder:
+        type_ = options.get("type")
+        type_part = ""
+        if type_ == "plain":
+            type_part = "pl"
+        elif type_ == "phrase":
+            type_part = "ph"
+        elif type_ == "web_search":
+            type_part = "w"
+        config_part = f"({options.get('config')})" if options.get("config") else ""
+        self.params = self.params.add(column, f"{type_part}fts{config_part}.{query}")
+
+        return SyncQueryRequestBuilder(
+            headers=self.headers,
+            http_method=self.http_method,
+            json=self.json,
+            params=self.params,
+            path=self.path,
+            session=self.session,  # type: ignore
+        )
+
 
 class SyncRequestBuilder:
     def __init__(self, session: SyncClient, path: str) -> None:
@@ -337,3 +360,6 @@ def delete(
         return SyncFilterRequestBuilder(
             self.session, self.path, method, headers, params, json
         )
+
+    def stub(self):
+        return None