/*
 * Copyright 2011, Haiku, Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *              Clemens Zeidler <haiku@clemens-zeidler.de>
 */


#include "QueryMonitor.h"


QueryHandler::QueryHandler(EntryViewInterface* listener)
        :
        FileMonitor(listener)
{
        
}


void
QueryHandler::MessageReceived(BMessage* message)
{
        int32 opcode;
        if (message->what == B_QUERY_UPDATE
                && message->FindInt32("opcode", &opcode) == B_OK) {
                switch (opcode) {
                        case B_ENTRY_CREATED:
                        case B_ENTRY_REMOVED:
                                message->what = B_NODE_MONITOR;
                                break;
                }
        }

        FileMonitor::MessageReceived(message);
}


QueryReader::QueryReader(QueryHandler* handler)
        :
        ReadThread(handler)
{

}


QueryReader::~QueryReader()
{
        Reset();
}


bool
QueryReader::AddQuery(BQuery* query)
{
        query->SetTarget(fTarget);
        query->Fetch();
        return fQueries.AddItem(query);
}


void
QueryReader::Reset()
{
        Stop();
        Wait();

        for (int32 i = 0; i < fLiveQueries.CountItems(); i++)
                delete fLiveQueries.ItemAt(i);
        fLiveQueries.MakeEmpty();

        for (int32 i = 0; i < fQueries.CountItems(); i++)
                delete fQueries.ItemAt(i);
        fQueries.MakeEmpty();   
}


bool
QueryReader::ReadNextEntry(entry_ref& entry)
{
        BQuery* query = fQueries.ItemAt(0);
        if (query == NULL)
                return false;
        if (query->GetNextRef(&entry) != B_OK) {
                fQueries.RemoveItemAt(0);
                fLiveQueries.AddItem(query);
                return ReadNextEntry(entry);
        }
        return true;
}