update

Author: erkanrzgc

core/engine.py

@@ -246,9 +246,16 @@ async def _check_platform(
             if platform.url_probe:
                 probe_url = platform.url_probe.replace("{username}", username)
                 try:
-                    _probe_status, _probe_data, _probe_elapsed = await client.get_json(
-                        probe_url, platform.headers
-                    )
+                    if platform.probe_method == "POST" and platform.probe_body:
+                        import json as _json
+                        _body_str = _json.dumps(platform.probe_body).replace("{username}", username)
+                        _probe_status, _probe_data, _probe_elapsed = await client.post_json(
+                            probe_url, _json.loads(_body_str), platform.headers
+                        )
+                    else:
+                        _probe_status, _probe_data, _probe_elapsed = await client.get_json(
+                            probe_url, platform.headers
+                        )
                 except Exception:
                     _probe_status, _probe_data, _probe_elapsed = -1, None, 0.0
                 if _probe_status != 200:

core/platform_loader.py

@@ -40,6 +40,8 @@ class Platform:
     presence_strings: tuple[str, ...] = ()
     url_probe: str | None = None
     check_method: str = "status"
+    probe_method: str = "GET"
+    probe_body: dict | None = None
 
 
 BUILTIN_YAML = Path(__file__).resolve().parent.parent / "modules" / "platforms.yaml"
@@ -84,6 +86,12 @@ def _coerce(entry: dict[str, Any]) -> Platform:
     check_method = entry.get("check_method", "status")
     if check_method not in ("status", "message", "response_url"):
         raise ValueError(f"platform {name!r} invalid check_method {check_method!r}")
+    probe_method = entry.get("probe_method", "GET")
+    if probe_method not in ("GET", "POST"):
+        raise ValueError(f"platform {name!r} invalid probe_method {probe_method!r}")
+    probe_body = entry.get("probe_body")
+    if probe_body is not None and not isinstance(probe_body, dict):
+        raise ValueError(f"platform {name!r} probe_body must be a mapping")
     return Platform(
         name=name,
         url=url,
@@ -100,6 +108,8 @@ def _coerce(entry: dict[str, Any]) -> Platform:
         presence_strings=tuple(str(s) for s in presence if str(s).strip()),
         url_probe=url_probe,
         check_method=check_method,
+        probe_method=probe_method,
+        probe_body=probe_body,
     )
 
 

modules/platforms.yaml

@@ -171,6 +171,11 @@ platforms:
   name: LeetCode
   presence_strings:
   - '"username":'
+  probe_body:
+    query: 'query getUserProfile($username: String!) { matchedUser(username: $username) { username } }'
+    variables:
+      username: '{username}'
+  probe_method: POST
   url: https://leetcode.com/u/{username}/
   url_probe: https://leetcode.com/graphql/
 - absence_strings: