1
0

state.py 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672
  1. # state.py
  2. import asyncio
  3. import threading
  4. import json
  5. import os
  6. import logging
  7. import uuid
  8. from typing import Optional, Literal
  9. logger = logging.getLogger(__name__)
  10. # Debounce timer for state saves (reduces SD card wear on Pi)
  11. _save_timer = None
  12. _save_lock = threading.Lock()
  13. class AppState:
  14. def __init__(self):
  15. # Private variables for properties
  16. self._current_playing_file = None
  17. self._current_coordinates = None # Cache parsed coordinates for current file (avoids re-parsing large files)
  18. self._current_preview = None # Cache (file_name, base64_data) for current pattern preview
  19. self._next_preview = None # Cache (file_name, base64_data) for next pattern preview
  20. self._pause_requested = False
  21. self._speed = 100
  22. self._current_playlist = None
  23. self._current_playlist_name = None # New variable for playlist name
  24. # Execution control flags (with event support for async waiting)
  25. self._stop_requested = False
  26. self._skip_requested = False
  27. self._stop_event: Optional[asyncio.Event] = None
  28. self._skip_event: Optional[asyncio.Event] = None
  29. self._event_loop: Optional[asyncio.AbstractEventLoop] = None
  30. # Regular state variables
  31. self.pause_condition = threading.Condition()
  32. self.execution_progress = None
  33. self.is_clearing = False
  34. self.current_theta = 0
  35. self.current_rho = 0
  36. self.current_playlist_index = 0
  37. self.playlist_mode = "loop"
  38. self.pause_time_remaining = 0
  39. self.active_clear_pattern = None # Runtime: clear pattern mode for current playlist (not persisted)
  40. # Machine position variables
  41. self.machine_x = 0.0
  42. self.machine_y = 0.0
  43. self.x_steps_per_mm = 0.0
  44. self.y_steps_per_mm = 0.0
  45. self.gear_ratio = 10
  46. # Homing mode: 0 = crash homing, 1 = sensor homing ($H)
  47. self.homing = 0
  48. # Track if user has explicitly set homing preference (vs auto-detected)
  49. # When False/None, homing mode can be auto-detected from firmware ($22 setting)
  50. self.homing_user_override = False
  51. # Homing state tracking (for sensor mode)
  52. self.homed_x = False # Set to True when [MSG:Homed:X] is received
  53. self.homed_y = False # Set to True when [MSG:Homed:Y] is received
  54. # Homing in progress flag - blocks other movement operations
  55. self.is_homing = False
  56. # Sensor homing failure flag - set when sensor homing fails
  57. # This indicates to the UI that sensor homing failed and user action is needed
  58. self.sensor_homing_failed = False
  59. # Angular homing compass reference point
  60. # This is the angular offset in degrees where the sensor is placed
  61. # After homing, theta will be set to this value
  62. self.angular_homing_offset_degrees = 0.0
  63. # Auto-homing settings for playlists
  64. # When enabled, performs homing after X patterns during playlist execution
  65. self.auto_home_enabled = False
  66. self.auto_home_after_patterns = 5 # Number of patterns after which to auto-home
  67. self.patterns_since_last_home = 0 # Counter for patterns played since last home
  68. # Hard reset on theta reset (sends $Bye to FluidNC to reset machine position)
  69. # When False (default), only normalizes theta to [0, 2π) without machine reset
  70. # When True, also performs soft reset which clears all position counters
  71. self.hard_reset_theta = False
  72. self.STATE_FILE = "state.json"
  73. self.mqtt_handler = None # Will be set by the MQTT handler
  74. self.conn = None
  75. self.port = None
  76. self.preferred_port = None # User's preferred port for auto-connect
  77. self.wled_ip = None
  78. self.led_provider = "none" # "wled", "dw_leds", or "none"
  79. self.led_controller = None
  80. # DW LED settings
  81. self.dw_led_num_leds = 60 # Number of LEDs in strip
  82. self.dw_led_gpio_pin = 18 # GPIO pin (12, 13, 18, or 19)
  83. self.dw_led_pixel_order = "RGB" # Pixel color order for WS281x (RGB for WS2815, GRB for WS2812)
  84. self.dw_led_brightness = 35 # Brightness 0-100
  85. self.dw_led_speed = 50 # Effect speed 0-255
  86. self.dw_led_intensity = 128 # Effect intensity 0-255
  87. # Idle effect settings (all parameters)
  88. self.dw_led_idle_effect = None # Full effect configuration dict or None
  89. # Playing effect settings (all parameters)
  90. self.dw_led_playing_effect = None # Full effect configuration dict or None
  91. # Idle timeout settings
  92. self.dw_led_idle_timeout_enabled = False # Enable automatic LED turn off after idle period
  93. self.dw_led_idle_timeout_minutes = 30 # Idle timeout duration in minutes
  94. self.dw_led_last_activity_time = None # Last activity timestamp (runtime only, not persisted)
  95. self.table_type = None
  96. self.table_type_override = None # User override for table type detection
  97. self._playlist_mode = "loop"
  98. self._pause_time = 0
  99. self._clear_pattern = "none"
  100. self._clear_pattern_speed = None # None means use state.speed as default
  101. self._shuffle = False # Shuffle playlist order
  102. self.custom_clear_from_in = None # Custom clear from center pattern
  103. self.custom_clear_from_out = None # Custom clear from perimeter pattern
  104. # Application name setting
  105. self.app_name = "Dune Weaver" # Default app name
  106. # Multi-table identity (for network discovery)
  107. self.table_id = str(uuid.uuid4()) # UUID generated on first run, persistent across restarts
  108. self.table_name = "Dune Weaver" # User-customizable table name
  109. # Known remote tables (for multi-table management)
  110. # List of dicts: [{id, name, url, host?, port?, version?}, ...]
  111. self.known_tables = []
  112. # Custom branding settings (filenames only, files stored in static/custom/)
  113. # Favicon is auto-generated from logo as logo-favicon.ico
  114. self.custom_logo = None # Custom logo filename (e.g., "logo-abc123.png")
  115. # auto_play mode settings
  116. self.auto_play_enabled = False
  117. self.auto_play_playlist = None # Playlist to auto-play in auto_play mode
  118. self.auto_play_run_mode = "loop" # "single" or "loop"
  119. self.auto_play_pause_time = 5.0 # Pause between patterns in seconds
  120. self.auto_play_clear_pattern = "adaptive" # Clear pattern option
  121. self.auto_play_shuffle = False # Shuffle playlist
  122. # Still Sands settings
  123. self.scheduled_pause_enabled = False
  124. self.scheduled_pause_time_slots = [] # List of time slot dictionaries
  125. self.scheduled_pause_control_wled = False # Turn off WLED during pause periods
  126. self.scheduled_pause_finish_pattern = False # Finish current pattern before pausing
  127. self.scheduled_pause_timezone = None # User-selected timezone (None = use system timezone)
  128. # Server port setting (requires restart to take effect)
  129. self.server_port = 8080 # Default server port
  130. # Machine timezone setting (IANA timezone, e.g., "America/New_York", "UTC")
  131. # Used for logging timestamps and scheduling features
  132. self.timezone = "UTC" # Default to UTC
  133. # MQTT settings (UI-configurable, overrides .env if set)
  134. self.mqtt_enabled = False # Master enable/disable for MQTT
  135. self.mqtt_broker = "" # MQTT broker IP/hostname
  136. self.mqtt_port = 1883 # MQTT broker port
  137. self.mqtt_username = "" # MQTT authentication username
  138. self.mqtt_password = "" # MQTT authentication password
  139. self.mqtt_client_id = "dune_weaver" # MQTT client ID
  140. self.mqtt_discovery_prefix = "homeassistant" # Home Assistant discovery prefix
  141. self.mqtt_device_id = "dune_weaver" # Device ID for Home Assistant
  142. self.mqtt_device_name = "Dune Weaver" # Device display name
  143. self.load()
  144. @property
  145. def current_playing_file(self):
  146. return self._current_playing_file
  147. @current_playing_file.setter
  148. def current_playing_file(self, value):
  149. # Clear cached data when file changes or is unset
  150. if value != self._current_playing_file or value is None:
  151. self._current_coordinates = None
  152. self._current_preview = None
  153. self._next_preview = None
  154. self._current_playing_file = value
  155. # force an empty string (and not None) if we need to unset
  156. if value is None:
  157. value = ""
  158. if self.mqtt_handler:
  159. is_running = bool(value and not self._pause_requested)
  160. self.mqtt_handler.update_state(current_file=value, is_running=is_running)
  161. @property
  162. def pause_requested(self):
  163. return self._pause_requested
  164. @pause_requested.setter
  165. def pause_requested(self, value):
  166. self._pause_requested = value
  167. if self.mqtt_handler:
  168. is_running = bool(self._current_playing_file and not value)
  169. self.mqtt_handler.update_state(is_running=is_running)
  170. @property
  171. def speed(self):
  172. return self._speed
  173. @speed.setter
  174. def speed(self, value):
  175. self._speed = value
  176. if self.mqtt_handler and self.mqtt_handler.is_enabled:
  177. self.mqtt_handler.client.publish(f"{self.mqtt_handler.speed_topic}/state", value, retain=True)
  178. @property
  179. def current_playlist(self):
  180. return self._current_playlist
  181. @current_playlist.setter
  182. def current_playlist(self, value):
  183. self._current_playlist = value
  184. # force an empty string (and not None) if we need to unset
  185. if value is None:
  186. value = ""
  187. # Also clear the playlist name when playlist is cleared
  188. self._current_playlist_name = None
  189. if self.mqtt_handler:
  190. self.mqtt_handler.update_state(playlist=value, playlist_name=None)
  191. @property
  192. def current_playlist_name(self):
  193. return self._current_playlist_name
  194. @current_playlist_name.setter
  195. def current_playlist_name(self, value):
  196. self._current_playlist_name = value
  197. if self.mqtt_handler:
  198. self.mqtt_handler.update_state(playlist_name=value)
  199. @property
  200. def playlist_mode(self):
  201. return self._playlist_mode
  202. @playlist_mode.setter
  203. def playlist_mode(self, value):
  204. self._playlist_mode = value
  205. @property
  206. def pause_time(self):
  207. return self._pause_time
  208. @pause_time.setter
  209. def pause_time(self, value):
  210. self._pause_time = value
  211. @property
  212. def clear_pattern(self):
  213. return self._clear_pattern
  214. @clear_pattern.setter
  215. def clear_pattern(self, value):
  216. self._clear_pattern = value
  217. @property
  218. def clear_pattern_speed(self):
  219. return self._clear_pattern_speed
  220. @clear_pattern_speed.setter
  221. def clear_pattern_speed(self, value):
  222. self._clear_pattern_speed = value
  223. # --- Execution Control Properties (stop/skip with event support) ---
  224. def _ensure_events(self):
  225. """Lazily create asyncio.Event objects in the current event loop."""
  226. try:
  227. loop = asyncio.get_running_loop()
  228. except RuntimeError:
  229. # No running loop - skip event creation (sync code path)
  230. return
  231. # Recreate events if the event loop changed
  232. if self._event_loop != loop:
  233. self._event_loop = loop
  234. self._stop_event = asyncio.Event()
  235. self._skip_event = asyncio.Event()
  236. # Sync event state with current flags
  237. if self._stop_requested:
  238. self._stop_event.set()
  239. if self._skip_requested:
  240. self._skip_event.set()
  241. @property
  242. def stop_requested(self) -> bool:
  243. return self._stop_requested
  244. @stop_requested.setter
  245. def stop_requested(self, value: bool):
  246. self._stop_requested = value
  247. self._ensure_events()
  248. if self._stop_event and self._event_loop:
  249. # asyncio.Event.set()/clear() are NOT thread-safe
  250. # Use call_soon_threadsafe when called from non-async threads (e.g., motion thread)
  251. try:
  252. if asyncio.get_running_loop() == self._event_loop:
  253. # Same loop - safe to call directly
  254. if value:
  255. self._stop_event.set()
  256. else:
  257. self._stop_event.clear()
  258. else:
  259. # Different loop - use thread-safe call
  260. if value:
  261. self._event_loop.call_soon_threadsafe(self._stop_event.set)
  262. else:
  263. self._event_loop.call_soon_threadsafe(self._stop_event.clear)
  264. except RuntimeError:
  265. # No running loop (sync context) - use thread-safe call
  266. if self._event_loop.is_running():
  267. if value:
  268. self._event_loop.call_soon_threadsafe(self._stop_event.set)
  269. else:
  270. self._event_loop.call_soon_threadsafe(self._stop_event.clear)
  271. @property
  272. def skip_requested(self) -> bool:
  273. return self._skip_requested
  274. @skip_requested.setter
  275. def skip_requested(self, value: bool):
  276. self._skip_requested = value
  277. self._ensure_events()
  278. if self._skip_event and self._event_loop:
  279. # asyncio.Event.set()/clear() are NOT thread-safe
  280. # Use call_soon_threadsafe when called from non-async threads (e.g., motion thread)
  281. try:
  282. if asyncio.get_running_loop() == self._event_loop:
  283. # Same loop - safe to call directly
  284. if value:
  285. self._skip_event.set()
  286. else:
  287. self._skip_event.clear()
  288. else:
  289. # Different loop - use thread-safe call
  290. if value:
  291. self._event_loop.call_soon_threadsafe(self._skip_event.set)
  292. else:
  293. self._event_loop.call_soon_threadsafe(self._skip_event.clear)
  294. except RuntimeError:
  295. # No running loop (sync context) - use thread-safe call
  296. if self._event_loop.is_running():
  297. if value:
  298. self._event_loop.call_soon_threadsafe(self._skip_event.set)
  299. else:
  300. self._event_loop.call_soon_threadsafe(self._skip_event.clear)
  301. def get_stop_event(self) -> Optional[asyncio.Event]:
  302. """Get the stop event for async waiting. Returns None if no event loop."""
  303. self._ensure_events()
  304. return self._stop_event
  305. def get_skip_event(self) -> Optional[asyncio.Event]:
  306. """Get the skip event for async waiting. Returns None if no event loop."""
  307. self._ensure_events()
  308. return self._skip_event
  309. async def wait_for_interrupt(
  310. self,
  311. timeout: float = 1.0,
  312. check_stop: bool = True,
  313. check_skip: bool = True,
  314. ) -> Literal['timeout', 'stopped', 'skipped']:
  315. """
  316. Wait for a stop/skip interrupt or timeout.
  317. This provides instant response to stop/skip requests by waiting on
  318. asyncio.Event objects rather than polling flags.
  319. Args:
  320. timeout: Maximum time to wait in seconds
  321. check_stop: Whether to check for stop requests
  322. check_skip: Whether to check for skip requests
  323. Returns:
  324. 'stopped' if stop was requested
  325. 'skipped' if skip was requested
  326. 'timeout' if timeout elapsed without interrupt
  327. """
  328. # Quick flag check first (handles edge cases and sync code)
  329. if check_stop and self._stop_requested:
  330. return 'stopped'
  331. if check_skip and self._skip_requested:
  332. return 'skipped'
  333. self._ensure_events()
  334. # Build list of event wait tasks
  335. tasks = []
  336. if check_stop and self._stop_event:
  337. tasks.append(asyncio.create_task(self._stop_event.wait(), name='stop'))
  338. if check_skip and self._skip_event:
  339. tasks.append(asyncio.create_task(self._skip_event.wait(), name='skip'))
  340. if not tasks:
  341. # No events available, fall back to simple sleep
  342. await asyncio.sleep(timeout)
  343. return 'timeout'
  344. # Add timeout task
  345. timeout_task = asyncio.create_task(asyncio.sleep(timeout), name='timeout')
  346. tasks.append(timeout_task)
  347. pending = set() # Initialize to empty set to avoid UnboundLocalError
  348. try:
  349. done, pending = await asyncio.wait(tasks, return_when=asyncio.FIRST_COMPLETED)
  350. finally:
  351. # Cancel all pending tasks
  352. for task in pending:
  353. task.cancel()
  354. # Await cancelled tasks to suppress warnings
  355. for task in pending:
  356. try:
  357. await task
  358. except asyncio.CancelledError:
  359. pass
  360. # Check which event fired (flags are authoritative)
  361. if check_stop and self._stop_requested:
  362. return 'stopped'
  363. if check_skip and self._skip_requested:
  364. return 'skipped'
  365. return 'timeout'
  366. @property
  367. def shuffle(self):
  368. return self._shuffle
  369. @shuffle.setter
  370. def shuffle(self, value):
  371. self._shuffle = value
  372. def to_dict(self):
  373. """Return a dictionary representation of the state."""
  374. return {
  375. "stop_requested": self.stop_requested,
  376. "pause_requested": self._pause_requested,
  377. "current_playing_file": self._current_playing_file,
  378. "execution_progress": self.execution_progress,
  379. "is_clearing": self.is_clearing,
  380. "current_theta": self.current_theta,
  381. "current_rho": self.current_rho,
  382. "speed": self._speed,
  383. "machine_x": self.machine_x,
  384. "machine_y": self.machine_y,
  385. "x_steps_per_mm": self.x_steps_per_mm,
  386. "y_steps_per_mm": self.y_steps_per_mm,
  387. "gear_ratio": self.gear_ratio,
  388. "homing": self.homing,
  389. "homing_user_override": self.homing_user_override,
  390. "angular_homing_offset_degrees": self.angular_homing_offset_degrees,
  391. "auto_home_enabled": self.auto_home_enabled,
  392. "auto_home_after_patterns": self.auto_home_after_patterns,
  393. "hard_reset_theta": self.hard_reset_theta,
  394. "current_playlist": self._current_playlist,
  395. "current_playlist_name": self._current_playlist_name,
  396. "current_playlist_index": self.current_playlist_index,
  397. "playlist_mode": self._playlist_mode,
  398. "pause_time": self._pause_time,
  399. "clear_pattern": self._clear_pattern,
  400. "clear_pattern_speed": self._clear_pattern_speed,
  401. "shuffle": self._shuffle,
  402. "custom_clear_from_in": self.custom_clear_from_in,
  403. "custom_clear_from_out": self.custom_clear_from_out,
  404. "port": self.port,
  405. "preferred_port": self.preferred_port,
  406. "wled_ip": self.wled_ip,
  407. "led_provider": self.led_provider,
  408. "dw_led_num_leds": self.dw_led_num_leds,
  409. "dw_led_gpio_pin": self.dw_led_gpio_pin,
  410. "dw_led_pixel_order": self.dw_led_pixel_order,
  411. "dw_led_brightness": self.dw_led_brightness,
  412. "dw_led_speed": self.dw_led_speed,
  413. "dw_led_intensity": self.dw_led_intensity,
  414. "dw_led_idle_effect": self.dw_led_idle_effect,
  415. "dw_led_playing_effect": self.dw_led_playing_effect,
  416. "dw_led_idle_timeout_enabled": self.dw_led_idle_timeout_enabled,
  417. "dw_led_idle_timeout_minutes": self.dw_led_idle_timeout_minutes,
  418. "app_name": self.app_name,
  419. "table_id": self.table_id,
  420. "table_name": self.table_name,
  421. "known_tables": self.known_tables,
  422. "custom_logo": self.custom_logo,
  423. "auto_play_enabled": self.auto_play_enabled,
  424. "auto_play_playlist": self.auto_play_playlist,
  425. "auto_play_run_mode": self.auto_play_run_mode,
  426. "auto_play_pause_time": self.auto_play_pause_time,
  427. "auto_play_clear_pattern": self.auto_play_clear_pattern,
  428. "auto_play_shuffle": self.auto_play_shuffle,
  429. "scheduled_pause_enabled": self.scheduled_pause_enabled,
  430. "scheduled_pause_time_slots": self.scheduled_pause_time_slots,
  431. "scheduled_pause_control_wled": self.scheduled_pause_control_wled,
  432. "scheduled_pause_finish_pattern": self.scheduled_pause_finish_pattern,
  433. "scheduled_pause_timezone": self.scheduled_pause_timezone,
  434. "timezone": self.timezone,
  435. "mqtt_enabled": self.mqtt_enabled,
  436. "mqtt_broker": self.mqtt_broker,
  437. "mqtt_port": self.mqtt_port,
  438. "mqtt_username": self.mqtt_username,
  439. "mqtt_password": self.mqtt_password,
  440. "mqtt_client_id": self.mqtt_client_id,
  441. "mqtt_discovery_prefix": self.mqtt_discovery_prefix,
  442. "mqtt_device_id": self.mqtt_device_id,
  443. "mqtt_device_name": self.mqtt_device_name,
  444. "table_type_override": self.table_type_override,
  445. }
  446. def from_dict(self, data):
  447. """Update state from a dictionary."""
  448. self.stop_requested = data.get("stop_requested", False)
  449. self._pause_requested = data.get("pause_requested", False)
  450. self._current_playing_file = data.get("current_playing_file", None)
  451. self.execution_progress = data.get("execution_progress")
  452. self.is_clearing = data.get("is_clearing", False)
  453. self.current_theta = data.get("current_theta", 0)
  454. self.current_rho = data.get("current_rho", 0)
  455. self._speed = data.get("speed", 150)
  456. self.machine_x = data.get("machine_x", 0.0)
  457. self.machine_y = data.get("machine_y", 0.0)
  458. self.x_steps_per_mm = data.get("x_steps_per_mm", 0.0)
  459. self.y_steps_per_mm = data.get("y_steps_per_mm", 0.0)
  460. self.gear_ratio = data.get('gear_ratio', 10)
  461. self.homing = data.get('homing', 0)
  462. self.homing_user_override = data.get('homing_user_override', False)
  463. self.angular_homing_offset_degrees = data.get('angular_homing_offset_degrees', 0.0)
  464. self.auto_home_enabled = data.get('auto_home_enabled', False)
  465. self.auto_home_after_patterns = data.get('auto_home_after_patterns', 5)
  466. self.hard_reset_theta = data.get('hard_reset_theta', False)
  467. self._current_playlist = data.get("current_playlist", None)
  468. self._current_playlist_name = data.get("current_playlist_name", None)
  469. self.current_playlist_index = data.get("current_playlist_index", None)
  470. self._playlist_mode = data.get("playlist_mode", "loop")
  471. self._pause_time = data.get("pause_time", 0)
  472. self._clear_pattern = data.get("clear_pattern", "none")
  473. self._clear_pattern_speed = data.get("clear_pattern_speed", None)
  474. self._shuffle = data.get("shuffle", False)
  475. self.custom_clear_from_in = data.get("custom_clear_from_in", None)
  476. self.custom_clear_from_out = data.get("custom_clear_from_out", None)
  477. self.port = data.get("port", None)
  478. self.preferred_port = data.get("preferred_port", None)
  479. self.wled_ip = data.get('wled_ip', None)
  480. self.led_provider = data.get('led_provider', "none")
  481. self.dw_led_num_leds = data.get('dw_led_num_leds', 60)
  482. self.dw_led_gpio_pin = data.get('dw_led_gpio_pin', 18)
  483. self.dw_led_pixel_order = data.get('dw_led_pixel_order', "RGB")
  484. self.dw_led_brightness = data.get('dw_led_brightness', 35)
  485. self.dw_led_speed = data.get('dw_led_speed', 50)
  486. self.dw_led_intensity = data.get('dw_led_intensity', 128)
  487. # Load effect settings (handle both old string format and new dict format)
  488. idle_effect_data = data.get('dw_led_idle_effect', None)
  489. if isinstance(idle_effect_data, str):
  490. # Old format: just effect name
  491. self.dw_led_idle_effect = None if idle_effect_data == "off" else {"effect_id": 0}
  492. else:
  493. # New format: full dict or None
  494. self.dw_led_idle_effect = idle_effect_data
  495. playing_effect_data = data.get('dw_led_playing_effect', None)
  496. if isinstance(playing_effect_data, str):
  497. # Old format: just effect name
  498. self.dw_led_playing_effect = None if playing_effect_data == "off" else {"effect_id": 0}
  499. else:
  500. # New format: full dict or None
  501. self.dw_led_playing_effect = playing_effect_data
  502. # Load idle timeout settings
  503. self.dw_led_idle_timeout_enabled = data.get('dw_led_idle_timeout_enabled', False)
  504. self.dw_led_idle_timeout_minutes = data.get('dw_led_idle_timeout_minutes', 30)
  505. self.app_name = data.get("app_name", "Dune Weaver")
  506. # Load or generate table_id (UUID persisted once generated)
  507. self.table_id = data.get("table_id", None)
  508. if self.table_id is None:
  509. self.table_id = str(uuid.uuid4())
  510. self.table_name = data.get("table_name", "Dune Weaver")
  511. self.known_tables = data.get("known_tables", [])
  512. self.custom_logo = data.get("custom_logo", None)
  513. self.auto_play_enabled = data.get("auto_play_enabled", False)
  514. self.auto_play_playlist = data.get("auto_play_playlist", None)
  515. self.auto_play_run_mode = data.get("auto_play_run_mode", "loop")
  516. self.auto_play_pause_time = data.get("auto_play_pause_time", 5.0)
  517. self.auto_play_clear_pattern = data.get("auto_play_clear_pattern", "adaptive")
  518. self.auto_play_shuffle = data.get("auto_play_shuffle", False)
  519. self.scheduled_pause_enabled = data.get("scheduled_pause_enabled", False)
  520. self.scheduled_pause_time_slots = data.get("scheduled_pause_time_slots", [])
  521. self.scheduled_pause_control_wled = data.get("scheduled_pause_control_wled", False)
  522. self.scheduled_pause_finish_pattern = data.get("scheduled_pause_finish_pattern", False)
  523. self.scheduled_pause_timezone = data.get("scheduled_pause_timezone", None)
  524. self.timezone = data.get("timezone", "UTC")
  525. self.mqtt_enabled = data.get("mqtt_enabled", False)
  526. self.mqtt_broker = data.get("mqtt_broker", "")
  527. self.mqtt_port = data.get("mqtt_port", 1883)
  528. self.mqtt_username = data.get("mqtt_username", "")
  529. self.mqtt_password = data.get("mqtt_password", "")
  530. self.mqtt_client_id = data.get("mqtt_client_id", "dune_weaver")
  531. self.mqtt_discovery_prefix = data.get("mqtt_discovery_prefix", "homeassistant")
  532. self.mqtt_device_id = data.get("mqtt_device_id", "dune_weaver")
  533. self.mqtt_device_name = data.get("mqtt_device_name", "Dune Weaver")
  534. self.table_type_override = data.get("table_type_override", None)
  535. def save(self):
  536. """Save the current state to a JSON file."""
  537. try:
  538. with open(self.STATE_FILE, "w") as f:
  539. json.dump(self.to_dict(), f)
  540. except Exception as e:
  541. print(f"Error saving state to {self.STATE_FILE}: {e}")
  542. def save_debounced(self, delay: float = 2.0):
  543. """
  544. Schedule a state save after a delay, coalescing multiple rapid saves.
  545. This reduces SD card writes on Raspberry Pi.
  546. Args:
  547. delay: Seconds to wait before saving (default 2.0)
  548. """
  549. global _save_timer
  550. with _save_lock:
  551. # Cancel any pending save
  552. if _save_timer is not None:
  553. _save_timer.cancel()
  554. # Schedule new save
  555. _save_timer = threading.Timer(delay, self._do_debounced_save)
  556. _save_timer.daemon = True # Don't block shutdown
  557. _save_timer.start()
  558. def _do_debounced_save(self):
  559. """Internal method called by debounce timer."""
  560. global _save_timer
  561. with _save_lock:
  562. _save_timer = None
  563. self.save()
  564. logger.debug("Debounced state save completed")
  565. def load(self):
  566. """Load state from a JSON file. If the file doesn't exist, create it with default values."""
  567. if not os.path.exists(self.STATE_FILE):
  568. # File doesn't exist: create one with the current (default) state.
  569. self.save()
  570. return
  571. try:
  572. with open(self.STATE_FILE, "r") as f:
  573. data = json.load(f)
  574. self.from_dict(data)
  575. except Exception as e:
  576. print(f"Error loading state from {self.STATE_FILE}: {e}")
  577. def update_steps_per_mm(self, x_steps, y_steps):
  578. """Update and save steps per mm values."""
  579. self.x_steps_per_mm = x_steps
  580. self.y_steps_per_mm = y_steps
  581. self.save()
  582. def reset_state(self):
  583. """Reset all state variables to their default values."""
  584. self.__init__() # Reinitialize the state
  585. self.save()
  586. # Create a singleton instance that you can import elsewhere:
  587. state = AppState()