import sqlite3
from flask import Flask, render_template, jsonify
from collections import defaultdict
import os

app = Flask(__name__)

def load_pokemon_data():
    pokemon_list = []
    
    conn = sqlite3.connect('pokemon_database.db')
    cursor = conn.cursor()

    cursor.execute('''
        SELECT p.national_dex_number, p.name, pf.form_name, pf.image_path, pf.is_default, p.introduced_in_gen,
               g.name AS earliest_game, m.icon_path AS mark_icon
        FROM pokemon p
        JOIN pokemon_forms pf ON p.national_dex_number = pf.pokemon_id
        LEFT JOIN form_encounters fe ON pf.id = fe.form_id
        LEFT JOIN games g ON fe.game_id = g.id
        LEFT JOIN mark_game_associations mga ON g.id = mga.game_id
        LEFT JOIN marks m ON mga.mark_id = m.id
        GROUP BY p.national_dex_number, pf.id
        ORDER BY p.national_dex_number, pf.is_default DESC, pf.form_name
    ''')
    
    current_group = []
    current_generation = None
    current_dex_number = None
    pokemon_forms = []
    
    for row in cursor.fetchall():
        national_dex_number, name, form_name, image_path, is_default, introduced_in_gen, earliest_game, mark_icon = row
        
        pokemon = {
            'ID': national_dex_number,
            'Name': name,
            'Form': form_name,
            'Image': image_path,
            'IsDefault': is_default,
            'Generation': introduced_in_gen,
            'EarliestGame': earliest_game,
            'MarkIcon': mark_icon
        }
        
        if national_dex_number != current_dex_number:
            if pokemon_forms:
                for form in pokemon_forms:
                    current_group.append(form)
                    if len(current_group) == 30:
                        pokemon_list.append(current_group)
                        current_group = []
            pokemon_forms = []
            current_dex_number = national_dex_number
            
            if is_default:
                if current_generation is None or introduced_in_gen != current_generation:
                    if current_group:
                        while len(current_group) < 30:
                            current_group.append(None)  # Add empty slots
                        pokemon_list.append(current_group)
                    current_group = []
                    current_generation = introduced_in_gen
        
        pokemon_forms.append(pokemon)
    
    # Add the last set of forms
    for form in pokemon_forms:
        current_group.append(form)
        if len(current_group) == 30:
            pokemon_list.append(current_group)
            current_group = []
    
    # Add any remaining Pokémon
    if current_group:
        while len(current_group) < 30:
            current_group.append(None)  # Add empty slots to the last group
        pokemon_list.append(current_group)

    conn.close()
    return pokemon_list

@app.route('/')
def index():
    pokemon_list = load_pokemon_data()
    return render_template('index.html', grouped_pokemon=pokemon_list)

@app.route('/pokemon/<int:dex_number>')
def pokemon_details(dex_number):
    conn = sqlite3.connect('pokemon_database.db')
    cursor = conn.cursor()

    cursor.execute('''
        SELECT pf.form_name, g.name, l.name, em.name
        FROM form_encounters fe
        JOIN games g ON fe.game_id = g.id
        JOIN locations l ON fe.location_id = l.id
        JOIN encounter_methods em ON fe.encounter_method_id = em.id
        JOIN pokemon_forms pf ON fe.form_id = pf.id
        WHERE pf.pokemon_id = ?
        ORDER BY pf.is_default DESC, pf.form_name, g.generation, g.name, l.name
    ''', (dex_number,))

    encounters = [
        {'form': form, 'game': game, 'location': location, 'method': method}
        for form, game, location, method in cursor.fetchall()
    ]

    conn.close()
    return jsonify(encounters)

if __name__ == '__main__':
    extra_files = ['.']
    extra_dirs = ['./templates/', './static/']
    for extra_dir in extra_dirs:
        for dirname, dirs, files in os.walk(extra_dir):
            for filename in files:
                filename = os.path.join(dirname, filename)
                if os.path.isfile(filename):
                    extra_files.append(filename)

    app.run(debug=True, extra_files=extra_files)