from flask import Flask, jsonify
import requests

app = Flask(__name__)

BASE_URL = "https://gameyfin-instance.local/v1"

def fetch_games():
    response = requests.get(f"{BASE_URL}/games")
    if response.status_code != 200:
        return []
    return response.json()

def human_readable_size(bytes_size):
    for unit in ['B', 'KB', 'MB', 'GB', 'TB']:
        if bytes_size < 1024:
            return f"{bytes_size:.1f} {unit}"
        bytes_size /= 1024
    return f"{bytes_size:.1f} PB"
    
def build_download_entry(game):
    return {
        "title": game.get("title", "Unknown Title"),
        "fileSize": human_readable_size(game.get("diskSize", 0)),
        "uris": [f"{BASE_URL}/games/game/{game.get('slug')}/download"],
        "uploadDate": game.get("addedToLibrary", "Unknown Date")
    }

def is_pc_game(game):
    platforms = game.get("platforms", [])
    return any(platform.get("slug") == "win" for platform in platforms)

@app.route("/gameyfin.json")
def get_downloads():
    games = fetch_games()
    confirmed_pc_games = [
        game for game in games
        if game.get("confirmedMatch") and is_pc_game(game)
    ]
    output = {
        "name": "Gameyfin",
        "downloads": [build_download_entry(game) for game in confirmed_pc_games]
    }
    return jsonify(output)

if __name__ == "__main__":
    app.run(host="127.0.0.1", port=5000, debug=True)