diff --git a/devices.py b/devices.py
index 709b191..9212d35 100644
--- a/devices.py
+++ b/devices.py
@@ -1,9 +1,37 @@
-"""Device management: check status and power off kids' devices."""
-
-from datetime import datetime
+"""Device management: check status, power off, and budget tracking."""
+import json
+import os
import subprocess
+import threading
+from datetime import datetime, timedelta
+from pathlib import Path
+# ── Config ────────────────────────────────────────────────────────────
+
+BUDGET_SECONDS = int(os.environ.get("BUDGET_MINUTES", "60")) * 60 # default 60 min in seconds
+STATE_FILE = Path(__file__).parent / ".device_state.json"
+TICK_INTERVAL = 10 # seconds between budget checks
+
+
+# ── Budget state (persisted to disk) ─────────────────────────────────
+
+def _load_state() -> dict:
+ if STATE_FILE.exists():
+ with open(STATE_FILE) as f:
+ return json.load(f)
+ return {}
+
+
+def _save_state(state: dict):
+ with open(STATE_FILE, "w") as f:
+ json.dump(state, f)
+
+
+
+
+
+# ── Helpers ───────────────────────────────────────────────────────────
def _run(cmd: list[str], *, timeout: int = 10) -> subprocess.CompletedProcess | None:
"""Run a command, return CompletedProcess or None on error."""
@@ -22,24 +50,24 @@ def tv_check() -> dict:
"""
result = _run(["ping", "-c", "1", "-W", "1", "192.168.1.12"])
if not result or result.returncode != 0:
- return {"icon": "❌", "title": "TV", "detail": "Unreachable (ping failed)"}
+ return {"online": False, "detail": "Offline"}
_run(["adb", "connect", "192.168.1.12"])
result = _run(["adb", "shell", "dumpsys", "power"])
if result and "mWakefulness=Awake" in result.stdout:
- return {"icon": "📺", "title": "TV", "detail": "Screen is on"}
+ return {"online": True, "detail": "Online"}
- return {"icon": "📺", "title": "TV", "detail": "Already off"}
+ return {"online": False, "detail": "Offline"}
def tv_turnoff() -> dict | None:
"""Send power key to the TV if it's on. Returns action dict or None."""
state = tv_check()
- if state["detail"] != "Screen is on":
+ if not state["online"]:
return None
_run(["adb", "shell", "input", "keyevent", "26"])
- return {"icon": "📺", "title": "TV", "detail": "Screen turned off (power key sent)"}
+ return {"detail": "Screen turned off (power key sent)"}
# ── Gabi ──────────────────────────────────────────────────────────────
@@ -53,41 +81,44 @@ def gabi_check() -> bool:
def gabi_turnoff() -> dict | None:
"""Log out Gabi if she's logged in. Returns action dict or None."""
if not gabi_check():
- return {"icon": "💻", "title": "Gabi's PC", "detail": "Not logged in — no action needed"}
+ return {"detail": "Not logged in — no action needed"}
result = _run(["doas", "loginctl", "terminate-user", "gabi"])
if result and result.returncode == 0:
- return {"icon": "💻", "title": "Gabi's PC", "detail": "Session terminated"}
+ return {"detail": "Session terminated"}
- return {"icon": "❌", "title": "Gabi's PC", "detail": "Failed to terminate session"}
+ return {"detail": "Failed to terminate session"}
# ── Gaja ──────────────────────────────────────────────────────────────
-def gaja_check() -> bool:
- """Return True if user 'gaja' has an active login session on her PC."""
+def gaja_check() -> dict:
+ """Check if Gaja's PC is online and she has an active desktop session.
+
+ Returns an action dict describing the current state.
+ """
result = _run(["ping", "-c", "1", "-W", "1", "192.168.1.122"])
if not result or result.returncode != 0:
- return False
+ return {"online": False, "detail": "Offline"}
+ # Check for a session with seat assigned (desktop session, not SSH)
check = _run(
["sshpass", "-p", "Nagaja", "ssh", "-o", "ConnectTimeout=5",
"-o", "StrictHostKeyChecking=no", "-F", "/dev/null",
"gaja@192.168.1.122",
- "loginctl list-sessions --no-pager 2>/dev/null | grep -q gaja"]
+ "loginctl list-sessions --no-pager 2>/dev/null | grep gaja | grep seat"]
)
- return check is not None and check.returncode == 0
+ if check is not None and check.stdout.strip():
+ return {"online": True, "detail": "Online"}
+
+ return {"online": False, "detail": "Offline"}
def gaja_turnoff() -> dict | None:
"""Log out Gaja if she's logged in. Returns action dict or None."""
- # Ping check
- result = _run(["ping", "-c", "1", "-W", "1", "192.168.1.122"])
- if not result or result.returncode != 0:
- return {"icon": "❌", "title": "Gaja's PC", "detail": "Unreachable (ping failed)"}
-
- if not gaja_check():
- return {"icon": "💻", "title": "Gaja's PC", "detail": "Not logged in — no action needed"}
+ state = gaja_check()
+ if not state["online"]:
+ return {"detail": "Not online — no action needed"}
_run(
["sshpass", "-p", "Nagaja", "ssh", "-o", "ConnectTimeout=5",
@@ -96,40 +127,154 @@ def gaja_turnoff() -> dict | None:
"loginctl terminate-user gaja"],
timeout=5
)
- return {"icon": "💻", "title": "Gaja's PC", "detail": "Logged out"}
+ return {"detail": "Logged out"}
+
+
+# ── Device registry ───────────────────────────────────────────────────
+
+DEVICES = {
+ "tv": {"icon": "📺", "name": "TV", "check": tv_check, "turnoff": tv_turnoff},
+ "gabi": {"icon": "💻", "name": "Gabi's PC", "check": gabi_check, "turnoff": gabi_turnoff},
+ "gaja": {"icon": "💻", "name": "Gaja's PC", "check": gaja_check, "turnoff": gaja_turnoff},
+}
+
+
+def _get_budget(device: str) -> int:
+ """Get remaining budget in seconds."""
+ state = _load_state()
+ return state.get(device, {}).get("budget", BUDGET_SECONDS)
+
+
+def _set_budget(device: str, budget_seconds: int):
+ state = _load_state()
+ if device not in state:
+ state[device] = {"budget": BUDGET_SECONDS}
+ state[device]["budget"] = budget_seconds
+ state[device]["last_online"] = datetime.now().isoformat()
+ _save_state(state)
+
+
+def _reset_budget(device: str):
+ _set_budget(device, BUDGET_SECONDS)
+
+
+def _budget_to_minutes(budget_seconds: int) -> int:
+ """Convert budget seconds to minutes for display."""
+ return max(0, budget_seconds // 60)
+
+
+# ── Budget timer (runs every 60s in background) ──────────────────────
+
+_timer_running = False
+_timer_lock = threading.Lock()
+
+
+def _budget_tick():
+ """Called every 10 seconds by the background timer."""
+ now = datetime.now()
+ reset_time = now.replace(hour=7, minute=0, second=0)
+
+ # Reset all budgets at 7:00 AM each day
+ if now >= reset_time and now < reset_time + timedelta(minutes=2):
+ for dev_id in DEVICES:
+ _reset_budget(dev_id)
+
+ with _timer_lock:
+ for dev_id, dev_info in DEVICES.items():
+ state = _load_state()
+ current_budget = state.get(dev_id, {}).get("budget", BUDGET_SECONDS)
+
+ is_online = dev_info["check"]()
+ if isinstance(is_online, dict):
+ online = is_online.get("online", False)
+ else:
+ online = bool(is_online)
+
+ # Expired budget + device back online → shut it down
+ if current_budget <= 0 and online:
+ dev_info["turnoff"]()
+ continue
+
+ if online and current_budget > 0:
+ new_budget = current_budget - TICK_INTERVAL
+ _set_budget(dev_id, new_budget)
+
+ if new_budget <= 0:
+ dev_info["turnoff"]()
+
+
+def start_timer():
+ """Start the background budget timer (runs every 10s)."""
+ global _timer_running
+ with _timer_lock:
+ if _timer_running:
+ return
+ _timer_running = True
+
+ def loop():
+ while _timer_running:
+ try:
+ _budget_tick()
+ except Exception:
+ pass # Don't crash the timer on device errors
+ threading.Event().wait(TICK_INTERVAL)
+
+ t = threading.Thread(target=loop, daemon=True)
+ t.start()
+
+
+def stop_timer():
+ """Stop the background timer."""
+ global _timer_running
+ with _timer_lock:
+ _timer_running = False
# ── Orchestrator ──────────────────────────────────────────────────────
def shutdown_all() -> list[dict]:
- """Run all device checks and logouts. Returns ordered list of action dicts."""
+ """Indiscriminately turn off all devices. Returns ordered list of action dicts."""
actions = []
-
- # Time window
- now = datetime.now()
- time_start = now.replace(hour=0, minute=0, second=0)
- time_stop = now.replace(hour=23, minute=59, second=59)
-
- if now < time_start:
- diff = int((time_start - now).total_seconds())
- actions.append({"icon": "⏳", "title": "Time Window", "detail": f"Before active hours — {diff}s until start"})
- elif now > time_stop:
- diff = int((now - time_stop).total_seconds())
- actions.append({"icon": "⏳", "title": "Time Window", "detail": f"After active hours — {diff}s past end"})
- else:
- actions.append({"icon": "🟢", "title": "Active Hours", "detail": f"Executing for the next {time_stop - now}"})
- # Check → act for each device
- for check_fn, action_fn in [
- (tv_check, tv_turnoff),
- (gabi_check, gabi_turnoff),
- (gaja_check, gaja_turnoff),
- ]:
- status = check_fn()
- if isinstance(status, dict):
- actions.append(status) # e.g. unreachable
- elif status is True:
- result = action_fn()
- if result:
- actions.append(result)
-
+ for dev_id, dev_info in DEVICES.items():
+ result = dev_info["turnoff"]()
+ if result:
+ budget_minutes = _budget_to_minutes(_get_budget(dev_id))
+ actions.append({
+ "icon": dev_info["icon"],
+ "title": dev_info["name"],
+ "detail": f"{result['detail']} ({budget_minutes} min left)",
+ })
+ return actions
+
+
+def status_all() -> list[dict]:
+ """Get current status + budget for all devices. Returns ordered list."""
+ actions = []
+ for dev_id, dev_info in DEVICES.items():
+ state = _load_state()
+ budget_seconds = state.get(dev_id, {}).get("budget", BUDGET_SECONDS)
+
+ is_online = dev_info["check"]()
+ if isinstance(is_online, dict):
+ online = is_online.get("online", False)
+ detail = is_online.get("detail", "")
+ else:
+ online = bool(is_online)
+ detail = "Online" if online else "Offline"
+
+ budget_minutes = _budget_to_minutes(budget_seconds)
+
+ # Budget warning
+ if budget_minutes <= 0:
+ icon = "🔴"
+ elif budget_minutes < 5:
+ icon = "🟠"
+ else:
+ icon = dev_info["icon"]
+
+ actions.append({
+ "icon": icon,
+ "title": dev_info["name"],
+ "detail": f"{detail} · {budget_minutes} min left",
+ })
return actions
diff --git a/server.py b/server.py
index 6f1b7d7..1bc83e9 100755
--- a/server.py
+++ b/server.py
@@ -4,7 +4,7 @@
import json
from http.server import HTTPServer, BaseHTTPRequestHandler
-from devices import shutdown_all
+from devices import shutdown_all, status_all, _set_budget, _get_budget, BUDGET_SECONDS, TICK_INTERVAL
class Handler(BaseHTTPRequestHandler):
@@ -14,10 +14,18 @@ class Handler(BaseHTTPRequestHandler):
self.send_header("Content-Type", "text/html; charset=utf-8")
self.end_headers()
self.wfile.write(HTML.encode())
+ elif self.path == "/status":
+ self.send_response(200)
+ self.send_header("Content-Type", "application/json")
+ self.end_headers()
+ self.wfile.write(json.dumps(status_all()).encode())
else:
self.send_error(404)
def do_POST(self):
+ content_length = int(self.headers.get("Content-Length", 0))
+ body = self.rfile.read(content_length).decode() if content_length > 0 else ""
+
if self.path == "/run":
try:
actions = shutdown_all()
@@ -30,6 +38,24 @@ class Handler(BaseHTTPRequestHandler):
self.send_header("Content-Type", "application/json")
self.end_headers()
self.wfile.write(json.dumps({"status": status, "actions": actions}).encode())
+
+ elif self.path == "/budget":
+ try:
+ data = json.loads(body) if body else {}
+ dev_id = data.get("device", "")
+ delta = int(data.get("delta", 0))
+ current = _get_budget(dev_id)
+ new_budget = max(0, min(BUDGET_SECONDS, current + delta))
+ _set_budget(dev_id, new_budget)
+ self.send_response(200)
+ self.send_header("Content-Type", "application/json")
+ self.end_headers()
+ self.wfile.write(json.dumps({"status": "ok"}).encode())
+ except Exception as e:
+ self.send_response(400)
+ self.send_header("Content-Type", "application/json")
+ self.end_headers()
+ self.wfile.write(json.dumps({"status": "error", "detail": str(e)}).encode())
else:
self.send_error(404)
@@ -77,12 +103,39 @@ HTML = """\
.action-item .info { flex: 1; }
.action-item .title { font-weight: 600; font-size: .95rem; }
.action-item .detail { color: #aaa; font-size: .82rem; margin-top: .15rem; }
+ #status {
+ margin-bottom: 2rem;
+ }
+ .status-item {
+ display: flex; align-items: center;
+ padding: .4rem .75rem; margin-bottom: .3rem;
+ border-radius: 6px; background: #0f3460; font-size: .8rem;
+ gap: .5rem;
+ }
+ .status-item .center {
+ display: flex; align-items: center; gap: .5rem;
+ flex: 1; min-width: 0;
+ }
+ .status-item .icon { font-size: 1.1rem; flex-shrink: 0; }
+ .status-item .name { font-weight: 500; white-space: nowrap; overflow: hidden; text-overflow: ellipsis; }
+ .status-item .budget { color: #aaa; font-size: .75rem; margin-left: auto; flex-shrink: 0; }
+ .status-item .actions {
+ display: flex; align-items: center; gap: .25rem;
+ flex-shrink: 0;
+ }
+ .budget-icon {
+ background: none; border: none;
+ color: #666; font-size: 1rem; cursor: pointer;
+ padding: 0; line-height: 1; transition: .15s;
+ }
+ .budget-icon:hover { color: #fff; }
📺 Kids Devices
TV + Gabi's computer + Gaja's computer
+
@@ -120,13 +173,57 @@ HTML = """\
btn.disabled = false;
btn.textContent = 'TURN OFF';
}
+
+ async function refreshStatus() {
+ try {
+ const res = await fetch('https://noom.cc/off/status');
+ const data = await res.json();
+ document.getElementById('status').innerHTML = data.map(s => {
+ const devId = s.title === 'TV' ? 'tv' : s.title.includes("Gabi") ? 'gabi' : 'gaja';
+ return `
+
+ ${s.icon}
+ ${s.title}
+ ${s.detail}
+
+
+
+
+
+
`;
+ }).join('');
+ } catch(e) {
+ // Ignore status errors
+ }
+ }
+
+ async function adjustBudget(device, delta) {
+ try {
+ await fetch('https://noom.cc/off/budget', {
+ method: 'POST',
+ headers: {'Content-Type': 'application/json'},
+ body: JSON.stringify({device, delta})
+ });
+ } catch(e) {}
+ refreshStatus();
+ }
+
+ refreshStatus();
+ setInterval(refreshStatus, 60000);