From 85259ca00c27ea31067b4445ec62335d7b4a6423 Mon Sep 17 00:00:00 2001
From: blakeblackshear <blakeb@blakeshome.com>
Date: Sat, 9 Mar 2019 13:41:35 -0600
Subject: [PATCH] add support for mqtt user/pass. closes #7

---
 README.md         | 4 ++++
 detect_objects.py | 5 +++++
 2 files changed, 9 insertions(+)

diff --git a/README.md b/README.md
index 05a1404c66..ff06c9fb94 100644
--- a/README.md
+++ b/README.md
@@ -36,6 +36,8 @@ docker run --rm \
 -e RTSP_URL='<rtsp_url>' \
 -e REGIONS='<box_size_1>,<x_offset_1>,<y_offset_1>,<min_person_size_1>,<min_motion_size_1>,<mask_file_1>:<box_size_2>,<x_offset_2>,<y_offset_2>,<min_person_size_2>,<min_motion_size_2>,<mask_file_2>' \
 -e MQTT_HOST='your.mqtthost.com' \
+-e MQTT_USER='username' \
+-e MQTT_PASS='password' \
 -e MQTT_TOPIC_PREFIX='cameras/1' \
 -e DEBUG='0' \
 frigate:latest
@@ -57,6 +59,8 @@ Example docker-compose:
       RTSP_URL: "<rtsp_url>"
       REGIONS: "<box_size_1>,<x_offset_1>,<y_offset_1>,<min_person_size_1>,<min_motion_size_1>,<mask_file_1>:<box_size_2>,<x_offset_2>,<y_offset_2>,<min_person_size_2>,<min_motion_size_2>,<mask_file_2>"
       MQTT_HOST: "your.mqtthost.com"
+      MQTT_USER: "username" #optional
+      MQTT_PASS: "password" #optional
       MQTT_TOPIC_PREFIX: "cameras/1"
       DEBUG: "0"
 ```
diff --git a/detect_objects.py b/detect_objects.py
index 0171e767d8..00468775b7 100644
--- a/detect_objects.py
+++ b/detect_objects.py
@@ -24,6 +24,8 @@
 RTSP_URL = os.getenv('RTSP_URL')
 
 MQTT_HOST = os.getenv('MQTT_HOST')
+MQTT_USER = os.getenv('MQTT_USER')
+MQTT_PASS = os.getenv('MQTT_PASS')
 MQTT_TOPIC_PREFIX = os.getenv('MQTT_TOPIC_PREFIX')
 
 # REGIONS = "350,0,300,50:400,350,250,50:400,750,250,50"
@@ -145,6 +147,9 @@ def on_connect(client, userdata, flags, rc):
     client = mqtt.Client()
     client.on_connect = on_connect
     client.will_set(MQTT_TOPIC_PREFIX+'/available', payload='offline', qos=1, retain=True)
+    if not MQTT_USER is None:
+        client.username_pw_set(MQTT_USER, password=MQTT_PASS)
+
     client.connect(MQTT_HOST, 1883, 60)
     client.loop_start()