floppy-ytube-player/main.py

import os
import sys
import json
from dotenv import load_dotenv
from lib.mqtt import create_client, update_disc, control_player, create_config


# Path to your .env file
dotenv_path = '/home/dietpi/floppy-ytube-player/.env'

# Load the .env file
load_dotenv(dotenv_path)

# Define the MQTT server details
broker = os.environ.get("broker")
port = int(os.environ.get("port"))
# MQTT username and password
username = os.environ.get('username')
password = os.environ.get('password')

print(f"Broker: {broker}, Port: {port}, Username: {username}")

client = create_client(broker, port, username, password)

create_config(client)

# Check if any arguments are passed
if len(sys.argv) > 1:
    # Get the JSON argument from the command line
    json_str = sys.argv[1]
    
    try:
        # Convert the JSON string into a dictionary
        disc_data = json.loads(json_str)
        # Pass the parsed data to the update_disc function
        update_disc(client, disc_data)
    except json.JSONDecodeError as e:
      try:
          if json_str == "EJECT":
            control_player(client, "EJECT")
            update_disc(client, {"type":"album", "id":"EJECT"})
          elif json_str == "PLAY":
            control_player(client, "PLAY")
      except Exception:
        print(f"Invalid JSON format: {e}")
else:
    print("No JSON argument passed, running default control_player action.")
    
# You can still have the default behavior if no JSON is passed
# control_player(client, "EJECT")