mirror of
https://github.com/kuhyx/steam-backlog-enforcer.git
synced 2026-11-11 15:07:58 +01:00
Krzysztof kuhy Rudnicki
b5b4b35c07
Each of these was one file doing several jobs, so they are split along what they actually do rather than at arbitrary line offsets: - _total_block: the lock file and the status derived from it stay in _total_block; making Steam *not installed* moves to _total_block_purge, making it *unreachable* to _total_block_hosts / _total_block_iptables. The generic pacman verbs (owner, installed, remove) become _pacman, since nothing about them is total-block-specific. Domain and process name tables become data modules. - _playtime: budget state and persistence (_playtime_state), pure transitions (_playtime_budget), cmdline process detection (_playtime_procs), the warn/cutoff/kill ladder (_playtime_cutoff), desktop notifications (_playtime_notify), process-tree walking and killing (_playtime_kill), mountinfo parsing (_playtime_mountinfo) and the shared privileged-command helper (_playtime_run). - _mcp: read-only tools (_mcp_query) split from mutating ones (_mcp_actions, _mcp_gaming_time), so "what can this server change?" is answerable from one file. _mcp keeps the server object and entry point. No re-export shims: callers import from the module that defines the name. That keeps a stale patch target loud (AttributeError) instead of letting it pass while intercepting nothing. `ruff check --select F821,F401` is run after every extraction -- an earlier mechanical attempt moved functions without the module-level names they referenced, which imported fine and only failed when called (349 errors), so it was reverted and redone by hand. The autouse fixtures in conftest are now imported by name and held by __all__; ruff --unsafe-fixes had deleted them as unused, since pytest resolves fixtures by name. 1324 tests pass at 100.00% coverage in the clean venv; `main status` is byte-identical to a worktree at HEAD. 49 -> 39 files over the cap. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01PPpmMpTSkp8UziRc5u8B7P
87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
"""/etc/hosts null-routing for the total gaming block.
|
|
|
|
Deliberately distinct from :mod:`store_blocker`'s own BLOCKED_DOMAINS /
|
|
STEAM_ENFORCER hosts section, so ending a total block never disturbs the
|
|
user's normal ``config.block_store`` entries. The iptables half of the same
|
|
block lives in :mod:`steam_backlog_enforcer._total_block_iptables`.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
import subprocess
|
|
|
|
from steam_backlog_enforcer._total_block_domains import (
|
|
_ALL_TOTAL_BLOCK_DOMAINS,
|
|
NULL_ROUTE_IP,
|
|
)
|
|
from steam_backlog_enforcer.config import HOSTS_FILE
|
|
from steam_backlog_enforcer.store_blocker import (
|
|
_disable_hosts_protection,
|
|
_enable_hosts_protection,
|
|
_sudo_write_hosts,
|
|
)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
_HOSTS_BLOCK_BEGIN = "# BEGIN steam-backlog-enforcer total-block\n"
|
|
_HOSTS_BLOCK_END = "# END steam-backlog-enforcer total-block\n"
|
|
|
|
|
|
def apply_total_block_hosts() -> bool:
|
|
"""Append the total-block domain block to /etc/hosts, if not present."""
|
|
try:
|
|
content = HOSTS_FILE.read_text(encoding="utf-8")
|
|
except OSError:
|
|
logger.exception("Failed to read /etc/hosts")
|
|
return False
|
|
|
|
if _HOSTS_BLOCK_BEGIN in content:
|
|
return True
|
|
|
|
block_lines = [_HOSTS_BLOCK_BEGIN]
|
|
block_lines += [
|
|
f"{NULL_ROUTE_IP} {domain}\n" for domain in _ALL_TOTAL_BLOCK_DOMAINS
|
|
]
|
|
block_lines.append(_HOSTS_BLOCK_END)
|
|
|
|
new_content = content if content.endswith("\n") else content + "\n"
|
|
new_content += "".join(block_lines)
|
|
|
|
try:
|
|
_disable_hosts_protection()
|
|
_sudo_write_hosts(new_content)
|
|
except (OSError, subprocess.SubprocessError):
|
|
logger.exception("Failed to write total-block hosts entries")
|
|
return False
|
|
finally:
|
|
_enable_hosts_protection()
|
|
return True
|
|
|
|
|
|
def remove_total_block_hosts() -> bool:
|
|
"""Remove the total-block domain block from /etc/hosts, if present."""
|
|
try:
|
|
content = HOSTS_FILE.read_text(encoding="utf-8")
|
|
except OSError:
|
|
logger.exception("Failed to read /etc/hosts")
|
|
return False
|
|
|
|
if _HOSTS_BLOCK_BEGIN not in content:
|
|
return True
|
|
|
|
start = content.index(_HOSTS_BLOCK_BEGIN)
|
|
end_marker_at = content.index(_HOSTS_BLOCK_END, start)
|
|
end = end_marker_at + len(_HOSTS_BLOCK_END)
|
|
new_content = content[:start] + content[end:]
|
|
|
|
try:
|
|
_disable_hosts_protection()
|
|
_sudo_write_hosts(new_content)
|
|
except (OSError, subprocess.SubprocessError):
|
|
logger.exception("Failed to remove total-block hosts entries")
|
|
return False
|
|
finally:
|
|
_enable_hosts_protection()
|
|
return True
|