-
-
Notifications
You must be signed in to change notification settings - Fork 31.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
MQTT: Start embedded server if no config given
- Loading branch information
Showing
4 changed files
with
158 additions
and
14 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,115 @@ | ||
| """MQTT server.""" | ||
| import asyncio | ||
| import logging | ||
| import tempfile | ||
| import threading | ||
|
|
||
| from homeassistant.components.mqtt import PROTOCOL_311 | ||
| from homeassistant.const import EVENT_HOMEASSISTANT_STOP | ||
|
|
||
| REQUIREMENTS = ['hbmqtt==0.6.2'] | ||
| DEPENDENCIES = ['http'] | ||
|
|
||
|
|
||
| @asyncio.coroutine | ||
| def broker_coro(loop, config): | ||
| """Start broker coroutine.""" | ||
| from hbmqtt.broker import Broker | ||
| broker = Broker(config, loop) | ||
| yield from broker.start() | ||
| return broker | ||
|
|
||
|
|
||
| def loop_run(loop, broker, shutdown_complete): | ||
| """Run broker and clean up when done.""" | ||
| # Temp workaround because hbmqtt does not respect passed in loop everywhere | ||
| # https://github.com/beerfactory/hbmqtt/issues/22 | ||
| asyncio.set_event_loop(loop) | ||
| loop.run_forever() | ||
| # run_forever ends when stop is called because we're shutting down | ||
| loop.run_until_complete(broker.shutdown()) | ||
| loop.close() | ||
| shutdown_complete.set() | ||
|
|
||
|
|
||
| def start(hass, server_config=None, http_config=None): | ||
| """Initialize MQTT Server.""" | ||
| from hbmqtt.broker import BrokerException | ||
|
|
||
| loop = asyncio.new_event_loop() | ||
| # Workaround !! | ||
| asyncio.set_event_loop(loop) | ||
|
|
||
| try: | ||
| passwd = tempfile.NamedTemporaryFile() | ||
|
|
||
| if server_config is None: | ||
| server_config, client_config = generate_config(hass, http_config, | ||
| passwd) | ||
| else: | ||
| client_config = None | ||
|
|
||
| start_server = asyncio.gather(broker_coro(loop, server_config), | ||
| loop=loop) | ||
| loop.run_until_complete(start_server) | ||
| # Result raises exception if one was raised during startup | ||
| broker = start_server.result()[0] | ||
| except BrokerException: | ||
| logging.getLogger(__name__).exception('Error initializing MQTT server') | ||
| loop.close() | ||
| return False, None | ||
| finally: | ||
| passwd.close() | ||
|
|
||
| shutdown_complete = threading.Event() | ||
|
|
||
| def shutdown(event): | ||
| """Gracefully shutdown MQTT broker.""" | ||
| loop.call_soon_threadsafe(loop.stop) | ||
| shutdown_complete.wait() | ||
|
|
||
| hass.bus.listen_once(EVENT_HOMEASSISTANT_STOP, shutdown) | ||
|
|
||
| threading.Thread(target=loop_run, args=(loop, broker, shutdown_complete), | ||
| daemon=True, name="MQTT-server").start() | ||
|
|
||
| return True, client_config | ||
|
|
||
|
|
||
| def generate_config(hass, http_config, passwd): | ||
| """Generate a configuration based on current Home Assistant instance.""" | ||
| config = { | ||
| 'listeners': { | ||
| 'default': { | ||
| 'type': 'tcp', | ||
| 'bind': '0.0.0.0:1883', | ||
| }, | ||
| }, | ||
| 'auth': { | ||
| 'allow-anonymous': hass.config.api.api_password is None | ||
| }, | ||
| 'plugins': ['auth_anonymous'], | ||
| } | ||
|
|
||
| if hass.config.api.api_password: | ||
| username = 'homeassistant' | ||
| password = hass.config.api.api_password | ||
|
|
||
| # Encrypt with what hbmqtt uses to verify | ||
| from passlib.apps import custom_app_context | ||
|
|
||
| passwd.write( | ||
| 'homeassistant:{}\n'.format( | ||
| custom_app_context.encrypt( | ||
| hass.config.api.api_password)).encode('utf-8')) | ||
| passwd.flush() | ||
|
|
||
| config['auth']['password-file'] = passwd.name | ||
| config['plugins'].append('auth_file') | ||
| else: | ||
| username = None | ||
| password = None | ||
|
|
||
| client_config = ('localhost', 1883, username, password, None, PROTOCOL_311) | ||
|
|
||
| return config, client_config |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters