from datetime import datetime

from db.schema import Feedback, Response
from db.setup import db_setup


# Create operation (Ingest data into Firebase with unique user_id check)
def ingest(data: Feedback):
    ref = db_setup()  # Ensure Firebase is initialized
    print(f"Attempting to ingest feedback data from user '{data.user_id}'...")
    # Check if the user_id already exists in the feedback data
    # user_id_str = str(data.user_id)
    # existing_feedback = ref.child('feedback').order_by_child('user_id').equal_to(user_id_str).get()
    #
    #
    # # TODO: This should probably change. If the same user has multiple feedbacks, we should allow it. -> change to update
    # if existing_feedback:
    #     print(f"Feedback from user '{data.user_id}' already exists. Ingestion aborted.")
    # else:
    #     # feedback_data = data.dict()  # Convert Pydantic model to a dictionary
    feedback_data = data.model_dump()
    feedback_data["time_stamp"] = feedback_data['time_stamp'].isoformat()
    ref.child('feedback').push(feedback_data)
    print(f"Feedback data from user '{data.user_id}' pushed to Firebase!")


# Read operation (Fetch feedback data from Firebase)
def read(feedback_id: int):
    ref = db_setup()
    feedback_ref = ref.child('feedback').order_by_child('id').equal_to(feedback_id).get()

    if feedback_ref:
        return feedback_ref
    else:
        print("Feedback not found!")
        return None


# Update operation (Update feedback data in Firebase)
def update(feedback_id: int, updated_data: Feedback):
    ref = db_setup()
    feedback_ref = ref.child('feedback').order_by_child('id').equal_to(feedback_id).get()

    if feedback_ref:
        # Assuming we're updating the first entry found
        for key in feedback_ref:
            ref.child('feedback').child(key).update(updated_data.dict())
        print("Feedback data updated in Firebase!")
    else:
        print("Feedback not found to update!")


# Delete operation (Remove feedback data from Firebase)
def delete(feedback_id: int):
    ref = db_setup()
    feedback_ref = ref.child('feedback').order_by_child('id').equal_to(feedback_id).get()

    if feedback_ref:
        # Assuming we're deleting the first entry found
        for key in feedback_ref:
            ref.child('feedback').child(key).delete()
        print("Feedback data deleted from Firebase!")
    else:
        print("Feedback not found to delete!")


def test():
    # Create a feedback object
    feedback_example = Feedback(
        id=1,
        user_id="user1234",
        time_stamp=datetime.now(),
        responses=[
            {"q_id": "q1", "ans": 5},
            {"q_id": "q2", "ans": 3}
        ]
    )
    feedback = Feedback(
        id=2,
        user_id='tt',
        time_stamp=datetime(2025, 1, 30, 21, 25, 15, 581994),
        responses=[
            Response(config_id='c_p_0_pop_low_easy', rating_v=1, rating_p0=1, rating_p1=1, comment='',
                     timestamp='2025-01-30T21:25:12.642038'),
            Response(config_id='c_p_1_pop_medium_medium', rating_v=1, rating_p0=1, rating_p1=1, comment='',
                     timestamp='2025-01-30T21:25:13.854227'),
            Response(config_id='c_p_2_pop_high_hard', rating_v=1, rating_p0=1, rating_p1=1, comment='',
                     timestamp='2025-01-30T21:25:15.581948'),
        ]
    )
    # Create (Ingest)
    ingest(feedback)

    # Read (Fetch)
    feedback_data = read(1)
    if feedback_data:
        print(feedback_data)

    # Update (Modify)
    updated_feedback = Feedback(
        id=1,
        user_id="user123",
        time_stamp=datetime.now(),
        responses=[
            {"q_id": "q1", "ans": 4},  # Updated answer
            {"q_id": "q2", "ans": 3}
        ]
    )
    update(1, updated_feedback)

    # Delete (Remove)
    delete(1)