Feat/unstract payments support related updates

backend/adapter_processor_v2/adapter_processor.py

@@ -9,6 +9,7 @@
     InValidAdapterId,
     TestAdapterError,
 )
+from cryptography.fernet import Fernet
 from django.conf import settings
 from django.core.exceptions import ObjectDoesNotExist
 from platform_settings_v2.platform_auth_service import PlatformAuthenticationService
@@ -23,6 +24,11 @@
 
 logger = logging.getLogger(__name__)
 
+try:
+    from plugins.subscription.time_trials.subscription_adapter import add_unstract_key
+except ImportError:
+    add_unstract_key = None
+
 
 class AdapterProcessor:
     @staticmethod
@@ -91,6 +97,12 @@ def test_adapter(adapter_id: str, adapter_metadata: dict[str, Any]) -> bool:
             adapter_class = Adapterkit().get_adapter_class_by_adapter_id(adapter_id)
 
             if adapter_metadata.pop(AdapterKeys.ADAPTER_TYPE) == AdapterKeys.X2TEXT:
+
+                if (
+                    adapter_metadata.get("use_platform_provided_unstract_key")
+                    and add_unstract_key
+                ):
+                    adapter_metadata = add_unstract_key(adapter_metadata)
                 adapter_metadata[X2TextConstants.X2TEXT_HOST] = settings.X2TEXT_HOST
                 adapter_metadata[X2TextConstants.X2TEXT_PORT] = settings.X2TEXT_PORT
                 platform_key = PlatformAuthenticationService.get_active_platform_key()
@@ -106,6 +118,21 @@ def test_adapter(adapter_id: str, adapter_metadata: dict[str, Any]) -> bool:
                 e, adapter_name=adapter_metadata[AdapterKeys.ADAPTER_NAME]
             )
 
+    @staticmethod
+    def update_adapter_metadata(adapter_metadata_b: Any) -> Any:
+        if add_unstract_key:
+            encryption_secret: str = settings.ENCRYPTION_KEY
+            f: Fernet = Fernet(encryption_secret.encode("utf-8"))
+
+            adapter_metadata = json.loads(
+                f.decrypt(bytes(adapter_metadata_b).decode("utf-8"))
+            )
+            adapter_metadata = add_unstract_key(adapter_metadata)
+
+            adapter_metadata_b = f.encrypt(json.dumps(adapter_metadata).encode("utf-8"))
+            return adapter_metadata_b
+        return adapter_metadata_b
+
     @staticmethod
     def __fetch_adapters_by_key_value(key: str, value: Any) -> Adapter:
         """Fetches a list of adapters that have an attribute matching key and

backend/adapter_processor_v2/views.py

@@ -169,8 +169,33 @@ def get_serializer_class(
 
     def create(self, request: Any) -> Response:
         serializer = self.get_serializer(data=request.data)
+
+        USE_PLATFORM_PROVIDED_UNSTRACT_KEY = False
+        adapter_metadata = request.data.get(AdapterKeys.ADAPTER_METADATA)
+        if adapter_metadata and adapter_metadata.get(
+            "use_platform_provided_unstract_key", False
+        ):
+            USE_PLATFORM_PROVIDED_UNSTRACT_KEY = True
+
         serializer.is_valid(raise_exception=True)
         try:
+            adapter_type = serializer.validated_data.get(AdapterKeys.ADAPTER_TYPE)
+
+            if (
+                adapter_type == AdapterKeys.X2TEXT
+                and USE_PLATFORM_PROVIDED_UNSTRACT_KEY
+            ):
+                adapter_metadata_b = serializer.validated_data.get(
+                    AdapterKeys.ADAPTER_METADATA_B
+                )
+                adapter_metadata_b = AdapterProcessor.update_adapter_metadata(
+                    adapter_metadata_b
+                )
+                # Update the validated data with the new adapter_metadata
+                serializer.validated_data[AdapterKeys.ADAPTER_METADATA_B] = (
+                    adapter_metadata_b
+                )
+
             instance = serializer.save()
             organization_member = OrganizationMemberService.get_user_by_id(
                 request.user.id
@@ -185,7 +210,6 @@ def create(self, request: Any) -> Response:
                 organization_member=organization_member
             )
 
-            adapter_type = serializer.validated_data.get(AdapterKeys.ADAPTER_TYPE)
             if (adapter_type == AdapterKeys.LLM) and (
                 not user_default_adapter.default_llm_adapter
             ):

backend/middleware/organization_middleware.py

@@ -12,6 +12,11 @@ def process_request(self, request):
         # Check if the URL matches the pattern with organization ID
         match = re.match(pattern, request.path)
         if match:
+            # Check if the request path matches any of the whitelisted paths
+            if any(re.match(path, request.path) for path in settings.WHITELISTED_PATHS):
+                request.path_info = "/" + request.path_info
+                return
+
             org_id = match.group("org_id")
             request.organization_id = org_id
             new_path = re.sub(pattern, "/" + tenant_prefix, request.path_info)