appmesh 1.6.9__py3-none-any.whl → 1.6.10__py3-none-any.whl

appmesh/__init__.py

@@ -11,6 +11,7 @@ Example:
 
 import sys
 from types import ModuleType
+from typing import TYPE_CHECKING
 
 __all__ = ["App", "AppMeshClient", "AppMeshClientTCP", "AppMeshClientOAuth", "AppMeshServer", "AppMeshServerTCP"]
 
@@ -24,6 +25,17 @@ _LAZY_IMPORTS = {
 }
 
 
+if TYPE_CHECKING:
+    # Provide explicit imports for static analyzers and type checkers
+    # These imports are only executed during type checking and won't affect runtime.
+    from .app import App  # noqa: F401
+    from .client_http import AppMeshClient  # noqa: F401
+    from .client_tcp import AppMeshClientTCP  # noqa: F401
+    from .client_http_oauth import AppMeshClientOAuth  # noqa: F401
+    from .server_http import AppMeshServer  # noqa: F401
+    from .server_tcp import AppMeshServerTCP  # noqa: F401
+
+
 def _lazy_import(name):
     """Helper function for lazy importing."""
     if name in _LAZY_IMPORTS:

appmesh/client_http.py

@@ -129,6 +129,10 @@ class AppMeshClient(metaclass=abc.ABCMeta):
     HTTP_HEADER_KEY_USER_AGENT = "User-Agent"
     HTTP_HEADER_KEY_X_TARGET_HOST = "X-Target-Host"
     HTTP_HEADER_KEY_X_FILE_PATH = "X-File-Path"
+    HTTP_HEADER_JWT_set_cookie = "X-Set-Cookie"
+    HTTP_HEADER_NAME_CSRF_TOKEN = "X-CSRF-Token"
+    COOKIE_TOKEN = "appmesh_auth_token"
+    COOKIE_CSRF_TOKEN = "appmesh_csrf_token"
 
     @unique
     class Method(Enum):
@@ -234,16 +238,7 @@ class AppMeshClient(metaclass=abc.ABCMeta):
         # Session and cookie management
         self._lock = threading.Lock()
         self.session = requests.Session()
-        self.cookie_file = rest_cookie_file
-        if self.cookie_file:
-            self.session.cookies = cookiejar.MozillaCookieJar(self.cookie_file)
-            if os.path.exists(self.cookie_file):
-                self.session.cookies.load(ignore_discard=True, ignore_expires=True)
-            else:
-                os.makedirs(os.path.dirname(self.cookie_file), exist_ok=True)
-                self.session.cookies.save(ignore_discard=True, ignore_expires=True)
-                if os.name == "posix":
-                    os.chmod(self.cookie_file, 0o600)  # User read/write only
+        self.cookie_file = self._load_cookies(rest_cookie_file)
 
         # Token auto-refresh
         self._token_refresh_timer = None
@@ -259,6 +254,44 @@ class AppMeshClient(metaclass=abc.ABCMeta):
     def _get_access_token(self) -> str:
         return self.jwt_token
 
+    def _load_cookies(self, cookie_file: Optional[str]) -> str:
+        """Load cookies from the cookie file and return the file path ."""
+        if not cookie_file:
+            return ""
+
+        self.session.cookies = cookiejar.MozillaCookieJar(cookie_file)
+        if os.path.exists(cookie_file):
+            self.session.cookies.load(ignore_discard=True, ignore_expires=True)
+        else:
+            os.makedirs(os.path.dirname(cookie_file), exist_ok=True)
+            self.session.cookies.save(ignore_discard=True, ignore_expires=True)
+            if os.name == "posix":
+                os.chmod(cookie_file, 0o600)  # User read/write only
+        return cookie_file
+
+    @staticmethod
+    def _get_cookie_value(cookies, name, check_expiry=True) -> Optional[str]:
+        """Get cookie value by name, checking expiry if requested."""
+        # If it's a RequestsCookieJar, use .get() but check expiry manually if requested
+        if hasattr(cookies, "get") and not isinstance(cookies, list):
+            cookie = cookies.get(name)
+            if cookie is None:
+                return None
+            if check_expiry and getattr(cookie, "expires", None):
+                if cookie.expires < time.time():
+                    return None  # expired
+            return cookie.value if hasattr(cookie, "value") else cookie
+
+        # Otherwise, assume it's a MozillaCookieJar — iterate manually
+        for c in cookies:
+            if c.name == name:
+                if check_expiry and getattr(c, "expires", None):
+                    if c.expires < time.time():
+                        return None  # expired
+                return c.value
+
+        return None
+
     def _check_and_refresh_token(self):
         """Check and refresh token if needed, then schedule next check.
 
@@ -407,6 +440,8 @@ class AppMeshClient(metaclass=abc.ABCMeta):
             - Refresh tokens before expiration
             - Validate token format before setting
         """
+        if self._jwt_token == token:
+            return  # No change
         self._jwt_token = token
 
         # handle refresh
@@ -499,6 +534,7 @@ class AppMeshClient(metaclass=abc.ABCMeta):
                 self.HTTP_HEADER_KEY_AUTH: "Basic " + base64.b64encode(f"{user_name}:{user_pwd}".encode()).decode(),
                 "X-Expire-Seconds": str(self._parse_duration(timeout_seconds)),
                 **({"X-Audience": audience} if audience else {}),
+                **({self.HTTP_HEADER_JWT_set_cookie: "true"} if self.cookie_file else {}),
                 # **({"X-Totp-Code": totp_code} if totp_code else {}),
             },
         )
@@ -1422,9 +1458,13 @@ class AppMeshClient(metaclass=abc.ABCMeta):
 
         # Prepare headers
         header = {} if header is None else header
-        token = self._get_access_token()
-        if token:
-            header[self.HTTP_HEADER_KEY_AUTH] = f"Bearer {token}"
+
+        # JWT or Cookie token
+        if self.cookie_file and self._get_cookie_value(self.session.cookies, self.COOKIE_CSRF_TOKEN):
+            header[self.HTTP_HEADER_NAME_CSRF_TOKEN] = self._get_cookie_value(self.session.cookies, self.COOKIE_CSRF_TOKEN)
+        elif self._get_access_token():
+            header[self.HTTP_HEADER_KEY_AUTH] = f"Bearer {self._get_access_token()}"
+
         if self.forward_to and len(self.forward_to) > 0:
             if ":" in self.forward_to:
                 header[self.HTTP_HEADER_KEY_X_TARGET_HOST] = self.forward_to

{appmesh-1.6.9.dist-info → appmesh-1.6.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: appmesh
-Version: 1.6.9
+Version: 1.6.10
 Summary: Client SDK for App Mesh
 Home-page: https://github.com/laoshanxi/app-mesh
 Author: laoshanxi

{appmesh-1.6.9.dist-info → appmesh-1.6.10.dist-info}/RECORD

@@ -1,16 +1,16 @@
-appmesh/__init__.py,sha256=TY1y5B5cE57uhraEzCFOZRWuo9SY1R-fYNRan8hCZOM,1670
+appmesh/__init__.py,sha256=uJ5LOadwZW2nOXkSuPOy2S3WH9Bx0BUn5wUvPBeFzoc,2217
 appmesh/app.py,sha256=crD4DRFZJuHtZMfSsz7C-EwvjPmGZbFXYXvA_wCdvdI,10734
 appmesh/app_output.py,sha256=vfn322AyixblI8DbXds08h6L_ybObiaRSifsA1-Xcoo,1035
 appmesh/app_run.py,sha256=aYq852a29OThIi32Xtx5s0sTXZ97T0lHD5WXH8yfPoc,2018
 appmesh/appmesh_client.py,sha256=ywB2222PtJUffdfdxZcBfdhZs1KYyc7JvzMxwuK2qyI,378
-appmesh/client_http.py,sha256=OiYM4CXn21S4e_jx71X5_31krLes6UuHHueEYk6UIu4,58690
+appmesh/client_http.py,sha256=KD4AMcMbHqaLJWSrra0J03kMqWAiwYHiyusUc5kpr6o,60443
 appmesh/client_http_oauth.py,sha256=1d51o0JX_xtB8d2bEuM7_XJHcwMnhcjkbIq7GE1Zxm8,6120
 appmesh/client_tcp.py,sha256=aq6UUzytZA4ibE9WQMMWdo1uW8sHETEhJjsbM6IYSno,11457
 appmesh/server_http.py,sha256=rBIYO9rbR-r3x1Jcry440Sp--IM-OWKRaOhNpGdkxh8,4299
 appmesh/server_tcp.py,sha256=-CU5tw97WJmDcUNsNPWqpdZ0wxRzRD6kUP3XyNZUTHc,1444
 appmesh/tcp_messages.py,sha256=H9S_iCy0IuufY2v50_SUgRvcyQmJsySG65tBe_xb3Ko,1878
 appmesh/tcp_transport.py,sha256=0hRSp5fpL9wKB05JIyIRIuyBC8w1IdokryhMDHqtN4M,8946
-appmesh-1.6.9.dist-info/METADATA,sha256=aT2BPT5BM_5FkkFKhR-h_NBWMQMeVBP3-Wo2QwA-j6Y,11763
-appmesh-1.6.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-appmesh-1.6.9.dist-info/top_level.txt,sha256=-y0MNQOGJxUzLdHZ6E_Rfv5_LNCkV-GTmOBME_b6pg8,8
-appmesh-1.6.9.dist-info/RECORD,,
+appmesh-1.6.10.dist-info/METADATA,sha256=do80dYMa4qK9tg1qKDM4M0klzIgzfcEGlWo6w1OIBfg,11764
+appmesh-1.6.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+appmesh-1.6.10.dist-info/top_level.txt,sha256=-y0MNQOGJxUzLdHZ6E_Rfv5_LNCkV-GTmOBME_b6pg8,8
+appmesh-1.6.10.dist-info/RECORD,,