db/crud.py

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}'...")
    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!")


def save_feedback(data: Feedback):
    ref = db_setup()  # Ensure Firebase is initialized
    print(f"Processing feedback for user '{data.user_id}'...")

    # Convert Pydantic model to dictionary
    feedback_data = data.model_dump()
    feedback_data["time_stamp"] = feedback_data["time_stamp"].isoformat()

    # Check if the user already has feedback
    existing_feedback = ref.child("feedback").order_by_child("user_id").equal_to(data.user_id).get()

    if existing_feedback:
        # Update existing feedback
        for key in existing_feedback:
            ref.child("feedback").child(key).update(feedback_data)
        print(f"Feedback updated for user '{data.user_id}'!")
    else:
        # Insert new feedback
        ref.child("feedback").push(feedback_data)
        print(f"(Ingestion) New feedback added for user '{data.user_id}'!")


# Read operation (Fetch feedback data from Firebase)
def read(user_id: str):
    """Fetch feedback data for a specific user from Firebase."""
    ref = db_setup()
    feedback_ref = ref.child('feedback').order_by_child('user_id').equal_to(user_id).get()

    if feedback_ref:
        for key, value in feedback_ref.items():
            return value  # Return the first found entry
    return None  # No progress found


# 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 reset_feedback_in_db(user_id: str):
    """Deletes previous feedback for a user, resetting progress in Firebase."""
    ref = db_setup()
    feedback_ref = ref.child('feedback').order_by_child('user_id').equal_to(user_id).get()

    if feedback_ref:
        for key in feedback_ref:
            ref.child('feedback').child(key).delete()
        print(f"Feedback data for user '{user_id}' reset in Firebase.")


def test():
    # Create a feedback object
    # 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("cdSf")
    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)