# micropython REPL
import mip
mip.install("github:surdouski/micropython-sniffs")

import asyncio
from mqtt_as import MQTTClient, config
from usniffs import Sniffs

sniffs = Sniffs()

@sniffs.route('<some_root_path>/<sub_path_name>:{option_1,option_2}/log')
async def sensor_001_message(some_root_path, sub_path_name, topic, message):
    print(f"some root path: {some_root_path}")
    print(f"sub path name: {sub_path_name}")
    print(f"topic: {topic}")
    print(f"message: {message}")
    # Do something else like creating tasks, awaiting functions, etc.

@sniffs.route('<location>:{north,south}/<options>:{option_1,option_2}/etc')
async def sensor_001_message(location, message):
    print(f"location: {location}")
    print(f"message: {message}")
    # You don't need to include all the arguments if you don't need them, just provide the names of what you need in the arguments. 
    
# See for all available config settings https://github.com/peterhinch/micropython-mqtt/blob/master/mqtt_as/README.md#mqtt-parameters
config["client_id"] = "abc-client-id"
config["ssid"] = "my_wifi_name"
config["wifi_pw"] = "my_wifi_password"
config["server"] = "mqtt_broker_host"
config["port"] = 9937
config["user"] = "mqtt_client_username"
config["password"] = "mqtt_client_password"
config["queue_len"] = 1  # queue_len is required to be 1 or more for this library
config["ssl"] = True  # Just an example of another config option you can use, be sure to read mqtt_as docs for details

async def main():
  client = MQTTClient(config)
  await sniffs.bind(client)  # bind usniffs handling to the mqtt_as client
  await client.connect()
  
  while True:
    await asyncio.sleep(100)  # idle wait to ensure program never terminates

asyncio.run(main())

# imports can be found in previous example

@sniffs.route('+/+')  # +/+ just means it matches on any topic/subtopic pair, such as 'any/pair' or 'foo/bar'
async def some_relevant_message(message):
    some_data = mutate_data(message)  # do some conversions/calculations, if needed by your program
    return int(some_data)  # the return will be the value that is sent to the await

async def some_coroutine():
  while True:
    data = await some_relevant_message
    if data != 42:
      print('This number is not the meaning of life.')

async def another_coroutine():
  while True:
    data = await some_relevant_message
    if data < 0:
      print('Negative numbers are scary.')
          
async def main():
  client = MQTTClient(config)  # assumes config is already setup, see previous example
  await sniffs.bind(client)
  await sniffs.connect()
  
  asyncio.create_task(some_coroutine())
  asyncio.create_task(another_coroutine())
  asyncio.sleep(0)  # not technically needed because sleeping below, but we allowing the tasks to begin running here

  while True:
      await asyncio.sleep(100)  # idle wait to ensure program never terminates

asyncio.run(main())

@app.route("<room>:{living_room,kitchen}/temperature")
async def receive_temperature_data(room):
    if room == "living_room":
        # do something
    elif room == "kitchen":
        # do something else

# DOES NOT WORK, DO NOT DO THIS!!!
@app.route("<room>:{living_room,kitchen}/temperature")
async def receive_temperature_data(argument_one):
    ...

@app.route("<room>/temperature")
async def receive_temperature_data(room):
    ...

@app.route("+/temperature")
async def receive_temperature_data(room):
    ...

@app.route("<room>/<sensor>:{sensor_1,sensor_2}/<sensor_type>{temperature,humidity}")
async def receive_temperature_data(room, sensor, sensor_type):
    ...

@app.route("<room>:{living_room,kitchen}/temperature")
async def receive_temperature_data(room, topic, message):
    ...

@app.route("<room>:{living_room,kitchen}/temperature")
async def receive_temperature_data(room):
    ...