import os
import json
from typing import List, Optional, Dict, Any

import psycopg2
from psycopg2.extras import RealDictCursor
from dotenv import load_dotenv

load_dotenv()

DATABASE_URL = os.getenv(
    "DATABASE_URL",
    "postgresql://user:password@localhost:5432/recipes_db",
)


def get_conn():
    return psycopg2.connect(DATABASE_URL, cursor_factory=RealDictCursor)


def list_recipes_db() -> List[Dict[str, Any]]:
    conn = get_conn()
    try:
        with conn.cursor() as cur:
            cur.execute(
                """
                SELECT id, name, meal_type, time_minutes,
                       tags, ingredients, steps
                FROM recipes
                ORDER BY id
                """
            )
            rows = cur.fetchall()
        return rows
    finally:
        conn.close()


def create_recipe_db(recipe_data: Dict[str, Any]) -> Dict[str, Any]:
    """
    recipe_data keys: name, meal_type, time_minutes, tags, ingredients, steps
    """
    conn = get_conn()
    try:
        with conn.cursor() as cur:
            cur.execute(
                """
                INSERT INTO recipes (name, meal_type, time_minutes, tags, ingredients, steps)
                VALUES (%s, %s, %s, %s, %s, %s)
                RETURNING id, name, meal_type, time_minutes, tags, ingredients, steps
                """,
                (
                    recipe_data["name"],
                    recipe_data["meal_type"],
                    recipe_data["time_minutes"],
                    json.dumps(recipe_data.get("tags", [])),
                    json.dumps(recipe_data.get("ingredients", [])),
                    json.dumps(recipe_data.get("steps", [])),
                ),
            )
            row = cur.fetchone()
        conn.commit()
        return row
    finally:
        conn.close()


def get_recipes_by_filters_db(
    meal_type: Optional[str],
    max_time: Optional[int],
) -> List[Dict[str, Any]]:
    conn = get_conn()
    try:
        query = """
            SELECT id, name, meal_type, time_minutes,
                   tags, ingredients, steps
            FROM recipes
            WHERE 1=1
        """
        params: List = []

        if meal_type:
            query += " AND meal_type = %s"
            params.append(meal_type.lower())

        if max_time:
            query += " AND time_minutes <= %s"
            params.append(max_time)

        with conn.cursor() as cur:
            cur.execute(query, params)
            rows = cur.fetchall()

        return rows
    finally:
        conn.close()