# 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()

# async def update_genshin():
#     async with enka_update:
#         await enka_update.update_assets(lang = ["EN"], path="/tmp")

# 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: 
#             print("prcess" + dt.name)
#             dt.card.save(byte_io, "PNG")
#             byte_io.seek(0)
#             image_url = upload_image(byte_io)
            
#             characters.append({
#                 "name": dt.name,
#                 "url": image_url
#             })
#             print("done" + dt.name)
#         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)
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
import base64

enka_update = EnkaNetworkAPI()

app = Flask(__name__)

async def update_genshin():
    async with enka_update:
        await enka_update.update_assets(lang=["EN"], path="/tmp")

@app.before_serving
def updet():
    """
    Run update_genshin before handling the first request.
    """
    data_dirs = ["/tmp/data", "/tmp/langs"]

    for directory in data_dirs:
        if not os.path.exists(directory):
            os.makedirs(directory)

    asyncio.run(update_genshin())

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:
                print("processing " + dt.name)
                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
    except enkacard.enc_error.ENCardError:
        return jsonify({'error': 'Enable display of the showcase in the game or add characters there.'}), 400
    except Exception as e:
        print(e)
        return jsonify({'error': 'UNKNOWN ERR: ' + str(e)}), 500

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

# @app.route("/update_char")
# def upload():
#     data_dirs = ["/tmp/data", "/tmp/langs"]

#     for directory in data_dirs:
#         if not os.path.exists(directory):
#             os.makedirs(directory)

#     asyncio.run(update_genshin())
    # return 'Update completed!'

def upload_image(data):
    """
    Convert image data to a base64-encoded string.
    
    Parameters:
    - data: A BytesIO object containing the image data.
    Returns:
    - A base64-encoded string of the image.
    """
    # Convert the image data to a base64-encoded string
    image_base64 = base64.b64encode(data.getvalue()).decode('utf-8')

    # Return the base64-encoded string
    return image_base64

if __name__ == "__main__":
    app.run(debug=False, use_reloader=False)