import enkacard
import enkacard.encbanner
import enkanetwork
import asyncio
from flask import Flask, jsonify, request
import requests
from io import BytesIO
from enkanetwork import EnkaNetworkAPI
import os

enka_update = EnkaNetworkAPI()

app = Flask(__name__)

async def card(id, designtype):
    async with enkacard.encbanner.ENC(uid = str(id)) as encard:
        return await encard.creat(template = (2 if str(designtype) == "2" else 1))
    
@app.route("/<int:id>")
def characters(id):
    try:
        characters = []
        result = asyncio.run(card(id, request.args.get('design')))
        for dt in result.card:
         with BytesIO() as byte_io: 
            dt.card.save(byte_io, "PNG")
            byte_io.seek(0)
            image_url = upload_image(byte_io)
            
            characters.append({
                "name": dt.name,
                "url": image_url
            })
        return jsonify({'response': characters}) 
    except enkanetwork.exception.VaildateUIDError:
        # return jsonify({'error': 'Invalid UID. Please check your UID.'}), 400 

        return jsonify({'error': 'Invalid UID. Please check your UID.'})

    except enkacard.enc_error.ENCardError:
        # return jsonify({'error': 'Enable display of the showcase in the game or add characters there.'}), 400 
        return jsonify({'error': 'Enable display of the showcase in the game or add characters there.'})
    except Exception as e:

        return jsonify({'error': 'UNKNOWN ERR' + e.__str__()})


@app.route("/")
def hello_world():
    return 'AMERICA YA HALLO!!'

@app.route("/update_char")
def upload():
    data_dir = "/tmp/data"

    if not os.path.exists(data_dir):
        os.makedirs(data_dir)

    data_dir = "/tmp/langs"

    if not os.path.exists(data_dir):
        os.makedirs(data_dir)
    asyncio.run(update_genshin())
    return 'Update smth ig!!'

def upload_image(data):
    url = "https://telegra.ph/upload"
    files = {'file': ('file', data, "image/png")}  
    response = requests.post(url, files=files)
    
    if response.status_code != 200:
        raise Exception(f"HTTP Error: {response.status_code}")
    
    try:
        body = response.json()
        if isinstance(body, list) and 'src' in body[0]:
            return "https://telegra.ph" + body[0]['src']
        else:
            raise Exception(f"Telegraph error: {body.get('error', 'Unknown error')}")
    except (ValueError, KeyError, IndexError) as e:
        raise Exception(f"Failed to parse response: {str(e)}")
    
if __name__ == "__main__":
    app.run(debug=False, use_reloader=False)