import requests
import datetime
import telegram

# Configuration
SONARR_API_URL = 'http://localhost:8989/api'
SONARR_API_KEY = 'your_sonarr_api_key'
RADARR_API_URL = 'http://localhost:7878/api'
RADARR_API_KEY = 'your_radarr_api_key'
PLEX_API_URL = 'http://localhost:32400'
PLEX_API_KEY = 'your_plex_api_key'
TELEGRAM_BOT_TOKEN = 'your_telegram_bot_token'
TELEGRAM_CHAT_ID = 'your_telegram_chat_id'
EXCEPTION_SERIES = ['Series1', 'Series2']  # Add your exception series here

# Initialize Telegram bot
bot = telegram.Bot(token=TELEGRAM_BOT_TOKEN)

def get_plex_last_watched():
    # Implement function to get last watched date from Plex
    pass

def disable_series_in_sonarr(series_id):
    url = f"{SONARR_API_URL}/series/{series_id}"
    headers = {'X-Api-Key': SONARR_API_KEY}
    data = {'monitored': False}
    response = requests.put(url, json=data, headers=headers)
    return response.status_code == 200

def disable_movie_in_radarr(movie_id):
    url = f"{RADARR_API_URL}/movie/{movie_id}"
    headers = {'X-Api-Key': RADARR_API_KEY}
    data = {'monitored': False}
    response = requests.put(url, json=data, headers=headers)
    return response.status_code == 200

def list_non_hd_movies():
    # Implement function to list non-HD movies from Radarr
    pass

def send_telegram_message(message):
    bot.send_message(chat_id=TELEGRAM_CHAT_ID, text=message, parse_mode=telegram.ParseMode.MARKDOWN)

def process_series():
    # Get all series from Sonarr
    response = requests.get(f"{SONARR_API_URL}/series", headers={'X-Api-Key': SONARR_API_KEY})
    series_list = response.json()

    processed_series = []
    for series in series_list:
        if series['title'] in EXCEPTION_SERIES:
            continue
        last_watched = get_plex_last_watched(series['title'])
        if last_watched and (datetime.datetime.now() - last_watched).days > 90:
            if disable_series_in_sonarr(series['id']):
                processed_series.append(series['title'])

    return processed_series

def process_movies():
    # Get all movies from Radarr
    response = requests.get(f"{RADARR_API_URL}/movie", headers={'X-Api-Key': RADARR_API_KEY})
    movie_list = response.json()

    processed_movies = []
    for movie in movie_list:
        if get_plex_last_watched(movie['title']):
            if disable_movie_in_radarr(movie['id']):
                processed_movies.append(movie['title'])

    return processed_movies

def main():
    # Process series and movies
    processed_series = process_series()
    processed_movies = process_movies()

    # List non-HD movies
    non_hd_movies = list_non_hd_movies()

    # Send Telegram message
    message = "*Processed TV Series and Movies*\n\n"
    message += "*TV Series:*\n"
    for i, series in enumerate(processed_series, 1):
        message += f"{i}. {series}\n"
    message += "\n*Movies:*\n"
    for i, movie in enumerate(processed_movies, 1):
        message += f"{i}. {movie}\n"
    message += "\n*Non-HD Movies:*\n"
    for movie in non_hd_movies:
        message += f"- {movie}\n"

    send_telegram_message(message)

if __name__ == "__main__":
    main()