constrain websockets to frigate topics

LordNex · Feb 20, 2021 · 57ced2c · 57ced2c
1 parent 26a3491
commit 57ced2c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/frigate/http.py b/frigate/http.py
@@ -40,14 +40,14 @@ def register(self, client):
 
  def publish(self, message):
  json_message = json.loads(message)
- self.mqtt_client.publish(json_message['topic'], json_message['payload'], retain=json_message['retain'])
+ self.mqtt_client.publish(f"{self.topic_prefix}/{json_message['topic']}", json_message['payload'], retain=json_message['retain'])
 
  def run(self):
  def send(client, userdata, message):
  """Sends mqtt messages to clients."""
  try:
  ws_message = json.dumps({
- 'topic': message.topic,
+ 'topic': message.topic.replace(f"{self.topic_prefix}/",""),
  'payload': message.payload.decode()
  })
  except: