Files
mudfish-decky/main.py
T
ysj7705andClaude Opus 5.5 8c595389a4 fix: turn VPN off on sleep/resume and unload; harden teardown
- Detect resume (boottime-monotonic jump) and tear down the tunnel: on wake
  NetworkManager drops Mudfish's relay route and all traffic black-holes.
- Turn the tunnel off on plugin unload (Deck shutdown/restart).
- One verified teardown path for disconnect, connect failure/timeout and
  node-change failure; keep the route snapshot if tap0 stays up.
- Toggle can always turn off a stuck tunnel; status failure with tap0 up
  is reported as an error, not off.
- Re-sign in once when the cloud rejects an expired jwt.
- Expire cached pings after 10 minutes.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
2026-10-24 21:21:21 +09:00

64 lines
2.3 KiB
Python

import asyncio
import logging
import os
import sys
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "py_modules"))
import decky # noqa: E402
from mudfish_decky import system # noqa: E402
from mudfish_decky.client import MudfishClient # noqa: E402
from mudfish_decky.controller import Controller # noqa: E402
from mudfish_decky.store import Store # noqa: E402
RESUME_POLL_SECONDS = 2
class Plugin:
controller = None
def _get_controller(self) -> Controller:
if self.controller is None:
installed = system.find_mudfish()
base = system.api_base(installed[1]) if installed else system.DEFAULT_API
store = Store(os.path.join(decky.DECKY_PLUGIN_SETTINGS_DIR, "settings.json"))
self.controller = Controller(MudfishClient(base), system, store)
return self.controller
async def get_state(self) -> dict:
return await asyncio.to_thread(self._get_controller().state)
async def connect(self) -> dict:
return await asyncio.to_thread(self._get_controller().connect)
async def disconnect(self) -> dict:
return await asyncio.to_thread(self._get_controller().disconnect)
async def set_node(self, choice: str) -> dict:
return await asyncio.to_thread(self._get_controller().set_node, str(choice))
async def list_nodes(self) -> list:
return await asyncio.to_thread(self._get_controller().nodes)
async def _main(self):
log = logging.getLogger("mudfish_decky")
log.setLevel(logging.INFO)
for handler in decky.logger.handlers:
log.addHandler(handler)
controller = self._get_controller()
# Bring up the idle launcher and sign in so nodes are listed. Never starts the tunnel.
await asyncio.to_thread(controller.prepare)
decky.logger.info("Mudfish VPN plugin loaded")
# Watch for sleep/resume even while the panel is closed; a resumed tunnel is turned off.
while True:
try:
await asyncio.to_thread(controller.check_resume, system.sleep_offset())
except Exception as e:
decky.logger.warning("resume check failed: %s", e)
await asyncio.sleep(RESUME_POLL_SECONDS)
async def _unload(self):
await asyncio.to_thread(self._get_controller().shutdown)
decky.logger.info("Mudfish VPN plugin unloaded")