feat: import rules from LD flags

api/integrations/launch_darkly/client.py

@@ -1,4 +1,4 @@
-from typing import Any, Iterator, Optional
+from typing import Any, Iterator, Optional, TypeVar
 
 from requests import Session
 
@@ -9,6 +9,8 @@
     LAUNCH_DARKLY_API_VERSION,
 )
 
+T = TypeVar("T")
+
 
 class LaunchDarklyClient:
     def __init__(self, token: str) -> None:
@@ -25,7 +27,7 @@ def _get_json_response(
         self,
         endpoint: str,
         params: Optional[dict[str, Any]] = None,
-    ) -> dict[str, Any]:
+    ) -> T:
         full_url = f"{LAUNCH_DARKLY_API_BASE_URL}{endpoint}"
         response = self.client_session.get(full_url, params=params)
         response.raise_for_status()
@@ -35,7 +37,7 @@ def _iter_paginated_items(
         self,
         collection_endpoint: str,
         additional_params: Optional[dict[str, str]] = None,
-    ) -> Iterator[dict[str, Any]]:
+    ) -> Iterator[T]:
         params = {"limit": LAUNCH_DARKLY_API_ITEM_COUNT_LIMIT_PER_PAGE}
         if additional_params:
             params.update(additional_params)
@@ -82,6 +84,7 @@ def get_flags(self, project_key: str) -> list[ld_types.FeatureFlag]:
         return list(
             self._iter_paginated_items(
                 collection_endpoint=endpoint,
+                additional_params={"summary": "0"},
             )
         )
 
@@ -106,3 +109,32 @@ def get_flag_tags(self) -> list[str]:
                 additional_params={"kind": "flag"},
             )
         )
+
+    def get_segment_tags(self) -> list[str]:
+        """operationId: getTags"""
+        endpoint = "/api/v2/tags"
+        return list(
+            self._iter_paginated_items(
+                collection_endpoint=endpoint,
+                additional_params={"kind": "segment"},
+            )
+        )
+
+    def get_segments(
+        self, project_key: str, environment_key: str
+    ) -> list[ld_types.UserSegment]:
+        """operationId: getSegments"""
+        endpoint = f"/api/v2/segments/{project_key}/{environment_key}"
+        return list(
+            self._iter_paginated_items(
+                collection_endpoint=endpoint,
+                additional_params={"limit": "50"},
+            )
+        )
+
+    def get_segment(
+        self, project_key: str, environment_key: str, segment_key: str
+    ) -> ld_types.UserSegment:
+        """operationId: getSegment"""
+        endpoint = f"/api/v2/segments/{project_key}/{environment_key}/{segment_key}"
+        return self._get_json_response(endpoint=endpoint)