from flask import Blueprint, request, jsonify
from tinydb import Query
from models.reward import Reward
from db.db import reward_db, child_db

reward_api = Blueprint('reward_api', __name__)

# Reward endpoints
@reward_api.route('/reward/add', methods=['PUT'])
def add_reward():
    data = request.get_json()
    name = data.get('name')
    description = data.get('description')
    cost = data.get('cost')
    image = data.get('image_id', '')
    if not name or description is None or cost is None:
        return jsonify({'error': 'Name, description, and cost are required'}), 400
    reward = Reward(name, description, cost, image_id=image)
    reward_db.insert(reward.to_dict())
    return jsonify({'message': f'Reward {name} added.'}), 201

@reward_api.route('/reward/<id>', methods=['GET'])
def get_reward(id):
    RewardQuery = Query()
    result = reward_db.search(RewardQuery.id == id)
    if not result:
        return jsonify({'error': 'Reward not found'}), 404
    return jsonify(result[0]), 200

@reward_api.route('/reward/list', methods=['GET'])
def list_rewards():
    rewards = reward_db.all()
    return jsonify({'rewards': rewards}), 200

@reward_api.route('/reward/<id>', methods=['DELETE'])
def delete_reward(id):
    RewardQuery = Query()
    removed = reward_db.remove(RewardQuery.id == id)
    if removed:
        # remove the reward id from any child's reward list
        ChildQuery = Query()
        for child in child_db.all():
            rewards = child.get('rewards', [])
            if id in rewards:
                rewards.remove(id)
                child_db.update({'rewards': rewards}, ChildQuery.id == child.get('id'))
        return jsonify({'message': f'Reward {id} deleted.'}), 200
    return jsonify({'error': 'Reward not found'}), 404