# app/__init__.py

import os
import json
import glob
import importlib
import importlib.util

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate
from flask_login import LoginManager
from flask_wtf.csrf import CSRFProtect
from dotenv import load_dotenv

# Load environment variables from .env or system
load_dotenv()

# ─── Initialize core extensions ─────────────────────────────────────────────────
db            = SQLAlchemy()
migrate       = Migrate()
login_manager = LoginManager()
csrf          = CSRFProtect()

from plugins.media.routes import generate_image_url  # Import it here


def create_app():
    app = Flask(__name__)
    app.config.from_object('app.config.Config')

    # ─── Initialize extensions with the app ───────────────────────────────────────
    csrf.init_app(app)
    db.init_app(app)
    migrate.init_app(app, db)
    login_manager.init_app(app)
    login_manager.login_view = 'auth.login'

    # ─── Register user_loader for Flask-Login ───────────────────────────────────
    from plugins.auth.models import User

    @login_manager.user_loader
    def load_user(user_id):
        try:
            return User.query.get(int(user_id))
        except Exception:
            return None

    # ─── Register error handlers ─────────────────────────────────────────────────
    from .errors import bp as errors_bp
    app.register_blueprint(errors_bp)

    # ─── 1) Auto‐import plugin models by their package names ─────────────────────
    # This ensures that every plugins/<plugin>/models.py is imported exactly once
    plugin_model_paths = glob.glob(
        os.path.join(os.path.dirname(__file__), '..', 'plugins', '*', 'models.py')
    )
    for path in plugin_model_paths:
        # path looks like ".../plugins/plant/models.py"
        rel = path.split(os.sep)[-2]           # e.g. "plant"
        pkg = f"plugins.{rel}.models"          # e.g. "plugins.plant.models"
        try:
            importlib.import_module(pkg)
            print(f"✅ (Startup) Loaded: {pkg}")
        except Exception as e:
            print(f"❌ (Startup) Failed to load {pkg}: {e}")

    # ─── 2) Auto‐discover & register plugin routes, CLI, entry‐points ────────────
    plugin_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', 'plugins'))
    for plugin in os.listdir(plugin_path):
        if plugin.endswith('.noload'):
            print(f"[⏭] Skipping plugin '{plugin}' (marked as .noload)")
            continue

        plugin_dir = os.path.join(plugin_path, plugin)
        if not os.path.isdir(plugin_dir):
            continue

        # (a) Register routes.py
        route_file = os.path.join(plugin_dir, 'routes.py')
        if os.path.isfile(route_file):
            try:
                spec = importlib.util.spec_from_file_location(f"plugins.{plugin}.routes", route_file)
                mod  = importlib.util.module_from_spec(spec)
                spec.loader.exec_module(mod)
                if hasattr(mod, 'bp'):
                    app.register_blueprint(mod.bp, strict_slashes=False)
                    print(f"✔️ Registered routes for plugin '{plugin}'")
            except Exception as e:
                print(f"❌ Failed to load routes from plugin '{plugin}': {e}")

        # (b) Register CLI and entry‐point
        init_file   = os.path.join(plugin_dir, '__init__.py')
        plugin_json = os.path.join(plugin_dir, 'plugin.json')
        if os.path.isfile(init_file):
            try:
                cli_module = importlib.import_module(f"plugins.{plugin}")
                if hasattr(cli_module, 'register_cli'):
                    cli_module.register_cli(app)
                    print(f"✔️ Registered CLI for plugin '{plugin}'")
                if os.path.isfile(plugin_json):
                    with open(plugin_json, 'r') as f:
                        meta = json.load(f)
                    entry = meta.get('entry_point')
                    if entry and hasattr(cli_module, entry):
                        getattr(cli_module, entry)(app)
                        print(f"✔️ Ran entry point '{entry}' for plugin '{plugin}'")
            except Exception as e:
                print(f"❌ Failed to load CLI for plugin '{plugin}': {e}")

    # ─── Inject current year into templates ────────────────────────────────────────
    @app.context_processor
    def inject_current_year():
        from datetime import datetime
        return {'current_year': datetime.now().year}
    
    app.jinja_env.globals['generate_image_url'] = generate_image_url

    return app
