from sqlalchemy import select
from db.models import User_Meta, Metadata, Category
from db.query.base_query import BaseQuery


class UserMetaQuery(BaseQuery):
    def __init__(self, user):
        super().__init__(user)

    # def get_user_meta_entries(self, db):
    #     """Fetch all user meta entries joined with metadata and category."""
    #     join_models = [Metadata, Category]
    #     print(join_models)
    #     join_conditions = [
    #         User_Meta.metadata_id == Metadata.id,
    #         Metadata.category_id == Category.id,
    #     ]
    #     print(join_conditions)

    #     result = self.get_all_with_joins(
    #         db,
    #         model=User_Meta,
    #         join_models=join_models,
    #         join_conditions=join_conditions,
    #     )
    #     return result
    
    def get_user_meta_entries(self, db):
        """Fetch all user meta entries joined with metadata and category."""
        join_models = [Metadata, Category]
        join_conditions = [
            User_Meta.metadata_id == Metadata.id,
            Metadata.category_id == Category.id,
        ]
        
        filter_conditions = [User_Meta.user_id == self.user_id]

        result = self.get_with_joins(
            db,
            model=User_Meta,
            join_models=join_models,
            join_conditions=join_conditions,
            filter_conditions=filter_conditions,
            multiple=True
        )
        return result


    def insert_user_meta_entries(self, db, metadata_ids):
        """Insert new user meta entries."""
        user_meta_entries = [
            User_Meta(user_id=self.user.get("id"), metadata_id=mid)
            for mid in metadata_ids
        ]

        # Use the method from BaseQuery to insert entries
        self.insert_entries(db, user_meta_entries)
        return {
            "message": "User meta entries added successfully.",
            "metadata_ids": metadata_ids,  # Include the metadata IDs in the result
        }
    
    def update_user_meta_entries(self, db, metadata_ids):
        """Update user meta entries: keep, delete, or add new entries based on metadata_ids."""
        filter_conditions = [User_Meta.user_id == self.user_id]
        
        # Fetch existing user meta entries
        existing_user_meta = self.get(db, model=User_Meta, filter_conditions=filter_conditions, multiple=True)
        existing_user_meta = [user_meta[0] for user_meta in existing_user_meta]
        existing_meta_ids = [entry.metadata_id for entry in existing_user_meta]

        # Convert both lists to sets once for efficiency
        metadata_ids_set = set(metadata_ids)
        existing_meta_ids_set = set(existing_meta_ids)

        # Find metadata to add, keep, or delete
        metadata_to_add = list(metadata_ids_set - existing_meta_ids_set)
        metadata_to_keep = list(metadata_ids_set & existing_meta_ids_set)
        metadata_to_delete = list(existing_meta_ids_set - metadata_ids_set)

        # Delete entries that are no longer in the updated metadata_ids list
        if metadata_to_delete:
            db.query(User_Meta).filter(User_Meta.user_id == self.user_id, User_Meta.metadata_id.in_(metadata_to_delete)).delete(synchronize_session=False)

        # Add new entries for metadata that are not in the existing user meta
        for meta_id in metadata_to_add:
            new_entry = User_Meta(user_id=self.user_id, metadata_id=meta_id)
            self.add(db, new_entry)

        db.commit()
        
        return {
            "status": "success",
            "added_meta": list(metadata_to_add),
            "deleted_meta": list(metadata_to_delete),
            "kept_meta": list(metadata_to_keep),
        }

    def delete_user_meta(self, db, metadata_id):
        """Delete user meta entries by metadata_id."""
        self.delete(db, model=User_Meta, id=metadata_id)
        return {"message": f"Book user with id {id} deleted successfully."}

    def delete_all_user_meta(self, db):
        """Delete all user meta entries for a user."""
        self.delete_all(db, model=User_Meta)