import appdaemon.plugins.hass.hassapi as hass
import debugpy


class Debugger(hass.Hass):
    def initialize(self):
        port = int(self.args.get("port", 5678))
        wait_for_client = bool(self.args.get("wait_for_client", False))
        try:
            debugpy.listen(("0.0.0.0", port))
            self.log(f"debugpy listening on port {port}")
        except RuntimeError:
            self.log("debugpy already listening, skipping")

        if wait_for_client:
            self.log(f"waiting for debug client on port {port}")
            debugpy.wait_for_client()
            self.log("debug client attached")