CH-226 fix cookie auth issue

Author: filippomc

applications/samples/backend/samples/controllers/auth_controller.py

@@ -26,7 +26,5 @@ def valid_cookie():  # noqa: E501
     :rtype: List[Valid]
     """
     from cloudharness.middleware import get_authentication_token
-    from cloudharness.auth import decode_token
     token = get_authentication_token()
-    assert decode_token(token)
     return 'OK'

applications/samples/backend/samples/controllers/security_controller_.py

@@ -1,4 +1,6 @@
 from typing import List
+from cloudharness.auth import decode_token
+from cloudharness.middleware import set_authentication_token
 
 
 def info_from_bearerAuth(token):
@@ -13,3 +15,31 @@ def info_from_bearerAuth(token):
     :rtype: dict | None
     """
     return {'uid': 'user_id'}
+
+
+def info_from_cookieAuth(api_key):
+    """
+    Check and retrieve authentication information from cookie-based API key.
+    This function is called by Connexion when cookieAuth security is used.
+    
+    :param api_key Token provided by the kc-access cookie
+    :type api_key: str
+    :return: Decoded token information or None if token is invalid
+    :rtype: dict | None
+    """
+    if not api_key:
+        return None
+    
+    # Set the authentication token in the middleware context
+    # so that get_authentication_token() can access it
+    set_authentication_token(api_key)
+    
+    # Decode and validate the token
+    try:
+        decoded = decode_token(api_key)
+        if decoded:
+            return decoded
+    except Exception:
+        pass
+    
+    return None

applications/samples/backend/samples/openapi/openapi.yaml

@@ -336,4 +336,4 @@ components:
       in: cookie
       name: kc-access
       type: apiKey
-      x-apikeyInfoFunc: cloudharness.auth.decode_token
+      x-apikeyInfoFunc: samples.controllers.security_controller_.info_from_cookieAuth

libraries/cloudharness-common/cloudharness/utils/server.py

@@ -170,7 +170,16 @@ def handle_exception(request, exc: Exception):
 
         # Register error handler with Flask app directly for better compatibility
         @app.errorhandler(Exception)
-        def flask_handle_exception(exc: Exception):
+        def flask_handle_exception(*args):
+            # Flask error handlers can be called with different signatures
+            # Handle both single argument (exc) and multiple arguments flexibly
+            if len(args) == 1:
+                exc = args[0]
+            elif len(args) >= 2:
+                exc = args[0] if isinstance(args[0], Exception) else args[1]
+            else:
+                exc = Exception("Unknown error")
+            
             # For Flask error handlers, we don't get the request object, 
             # but we can access it via flask.request if needed
             try: