config.py

import os
from dotenv import load_dotenv

load_dotenv()

def get_env_or_default(key, default=None, cast_func=str):
    value = os.getenv(key)
    if value is not None and value.strip() != "":
        try:
            return cast_func(value)
        except (ValueError, TypeError):
            return default
    return default

# Load main credentials
API_ID = get_env_or_default("API_ID", cast_func=int)
API_HASH = get_env_or_default("API_HASH")
BOT_TOKEN = get_env_or_default("BOT_TOKEN")
SESSION_STRING = get_env_or_default("SESSION_STRING")
OWNER_ID = get_env_or_default("OWNER_ID", cast_func=int)
CHANNEL_ID = get_env_or_default("CHANNEL_ID", cast_func=int)

# Load command prefixes
raw_prefixes = get_env_or_default("COMMAND_PREFIX", "!|.|#|,|/")
COMMAND_PREFIX = [prefix.strip() for prefix in raw_prefixes.split("|") if prefix.strip()]

# Update channel
UPDATE_CHANNEL_URL = get_env_or_default("UPDATE_CHANNEL_URL", "t.me/TheSmartDev")

# Required variable validation
required_vars = {
    "API_ID": API_ID,
    "API_HASH": API_HASH,
    "BOT_TOKEN": BOT_TOKEN,
    "SESSION_STRING": SESSION_STRING,
    "OWNER_ID": OWNER_ID,
    "CHANNEL_ID": CHANNEL_ID,
}

for var_name, var_value in required_vars.items():
    if var_value is None or (isinstance(var_value, str) and var_value.strip() == ""):
        raise ValueError(f"Required variable {var_name} is missing or invalid.")

if not COMMAND_PREFIX:
    raise ValueError("No command prefixes found. Set COMMAND_PREFIX in .env.")

print("Loaded COMMAND_PREFIX:", COMMAND_PREFIX)