bivashy/Blitz-Proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(status): improve rule checking logic for warp configuration

Browse Source
This commit is contained in:
ReturnFI
2025-12-19 08:36:11 +00:00
parent 5747a893a7
commit 75db42d166
1 changed files with 28 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
core/scripts/warp/status.py
View File

@ -3,6 +3,7 @@
import json import json
from pathlib import Path from pathlib import Path
import sys import sys
import re
core_scripts_dir = Path(__file__).resolve().parents[1] core_scripts_dir = Path(__file__).resolve().parents[1]
if str(core_scripts_dir) not in sys.path: if str(core_scripts_dir) not in sys.path:
@ -26,25 +27,36 @@ def check_warp_configuration():
print(json.dumps(status_data, indent=4)) print(json.dumps(status_data, indent=4))
return return
acl_inline = config.get("acl", {}).get("inline", []) acl_inline = config.get("acl", {}).get("inline", [])
def contains_warp(rule_prefixes): def contains_any_rule(rule_patterns):
return any(rule.startswith(prefix) for rule in acl_inline for prefix in rule_prefixes) """Checks if any of the provided rule patterns exist in the acl_inline list."""
for rule in acl_inline:
for pattern in rule_patterns:
if re.search(pattern, rule):
return True
return False
status_data["all_traffic_via_warp"] = contains_warp(["warps(all)"]) status_data["all_traffic_via_warp"] = "warps(all)" in acl_inline
status_data["popular_sites_via_warp"] = contains_warp([
"warps(geosite:google)", popular_site_patterns = [
"warps(geoip:google)", r"warps\(geosite:google\)",
"warps(geosite:netflix)", r"warps\(geoip:google\)",
"warps(geosite:spotify)", r"warps\(geosite:netflix\)",
"warps(geosite:openai)", r"warps\(geosite:spotify\)"
"warps(geoip:openai)" ]
]) status_data["popular_sites_via_warp"] = contains_any_rule(popular_site_patterns)
status_data["domestic_sites_via_warp"] = contains_warp([
"warps(geosite:ir)", domestic_site_patterns = [
"warps(geoip:ir)" r"warps\(geosite:ir\)",
]) r"warps\(geoip:ir\)",
r"warps\(geosite:cn\)",
r"warps\(geoip:cn\)",
r"warps\(geosite:ru-available-only-inside\)",
r"warps\(geoip:ru\)"
]
status_data["domestic_sites_via_warp"] = contains_any_rule(domestic_site_patterns)
status_data["block_adult_content"] = "reject(geosite:nsfw)" in acl_inline status_data["block_adult_content"] = "reject(geosite:nsfw)" in acl_inline
print(json.dumps(status_data, indent=4)) print(json.dumps(status_data, indent=4))