config.py 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  1. from __future__ import annotations
  2. import json
  3. from dataclasses import dataclass
  4. from pathlib import Path
  5. from typing import Literal
  6. Strategy = Literal["broadcast", "top2", "top3", "top4", "backup"]
  7. KernelMode = Literal["auto", "20", "24"]
  8. @dataclass
  9. class RelayNode:
  10. name: str
  11. host: str
  12. port: int
  13. token: str
  14. weight: int = 100
  15. @dataclass
  16. class Config:
  17. relays: list[RelayNode]
  18. strategy: Strategy = "top3"
  19. kernel_mode: KernelMode = "auto"
  20. redundancy: int = 3
  21. tcp_warmup_bytes: int = 1048576
  22. probe_interval: float = 15.0
  23. tcp_loser_grace_ms: int = 1500
  24. direct_open_timeout: float = 10.0
  25. relay_open_timeout: float = 10.0
  26. tcp_connect_happy_eyeballs_delay: float | None = None
  27. relay_reconnect_delay: float = 3.0
  28. relay_reconnect_attempts: int = 5
  29. relay_reconnect_max_delay: float = 30.0
  30. relay_ping_interval: float = 10.0
  31. relay_ping_timeout: float = 25.0
  32. relay_tcp_nodelay: bool = True
  33. direct_redundancy: int = 2
  34. direct_redundancy_v4: int | None = None
  35. direct_redundancy_v6: int | None = None
  36. direct_max_redundancy: int = 3
  37. udp_redundancy: int = 1
  38. udp_direct_redundancy: int = 2
  39. udp_always_broadcast: bool = True
  40. udp_copy_interval_ms: int = 8
  41. socks_host: str = "127.0.0.1"
  42. socks_port: int = 0
  43. @classmethod
  44. def load(cls, path: str) -> "Config":
  45. raw = json.loads(Path(path).read_text())
  46. relays = [RelayNode(**item) for item in raw.get("relays", [])]
  47. return cls(
  48. relays=relays,
  49. strategy=raw.get("strategy", "top3"),
  50. kernel_mode=raw.get("kernel_mode", "auto"),
  51. redundancy=raw.get("redundancy", 3),
  52. tcp_warmup_bytes=raw.get("tcp_warmup_bytes", 1048576),
  53. probe_interval=raw.get("probe_interval", 15.0),
  54. tcp_loser_grace_ms=raw.get("tcp_loser_grace_ms", 1500),
  55. direct_open_timeout=raw.get("direct_open_timeout", 10.0),
  56. relay_open_timeout=raw.get("relay_open_timeout", 10.0),
  57. tcp_connect_happy_eyeballs_delay=raw.get("tcp_connect_happy_eyeballs_delay"),
  58. relay_reconnect_delay=raw.get("relay_reconnect_delay", 3.0),
  59. relay_reconnect_attempts=max(1, raw.get("relay_reconnect_attempts", 5)),
  60. relay_reconnect_max_delay=max(raw.get("relay_reconnect_delay", 3.0), raw.get("relay_reconnect_max_delay", 30.0)),
  61. relay_ping_interval=max(1.0, raw.get("relay_ping_interval", 10.0)),
  62. relay_ping_timeout=max(1.0, raw.get("relay_ping_timeout", 25.0)),
  63. relay_tcp_nodelay=raw.get("relay_tcp_nodelay", True),
  64. direct_redundancy=max(1, raw.get("direct_redundancy", 2)),
  65. direct_redundancy_v4=raw.get("direct_redundancy_v4"),
  66. direct_redundancy_v6=raw.get("direct_redundancy_v6"),
  67. direct_max_redundancy=max(1, raw.get("direct_max_redundancy", 3)),
  68. udp_redundancy=max(0, raw.get("udp_redundancy", 1)),
  69. udp_direct_redundancy=max(1, raw.get("udp_direct_redundancy", 2)),
  70. udp_always_broadcast=raw.get("udp_always_broadcast", True),
  71. udp_copy_interval_ms=max(0, raw.get("udp_copy_interval_ms", 8)),
  72. socks_host=raw.get("socks_host", "127.0.0.1"),
  73. socks_port=max(0, raw.get("socks_port", 0)),
  74. )