// QueryManager.h

#ifndef NET_FS_QUERY_MANAGER_H
#define NET_FS_QUERY_MANAGER_H

#include "Locker.h"
#include "QueryIterator.h"

class Volume;
class VolumeManager;

// QueryManager
class QueryManager {
public:
                                                                QueryManager(VolumeManager* volumeManager);
                                                                ~QueryManager();

                        status_t                        Init();

                        status_t                        AddIterator(QueryIterator* iterator);

                        status_t                        AddSubIterator(
                                                                        HierarchicalQueryIterator* iterator,
                                                                        QueryIterator* subIterator);
private:
                        status_t                        RemoveSubIterator(
                                                                        HierarchicalQueryIterator* iterator,
                                                                        QueryIterator* subIterator);
public:
// TODO: Remove.

                        QueryIterator*          GetCurrentSubIterator(
                                                                        HierarchicalQueryIterator* iterator);
                        void                            NextSubIterator(
                                                                        HierarchicalQueryIterator* iterator,
                                                                        QueryIterator* subIterator);
private:
                        void                            RewindSubIterator(
                                                                        HierarchicalQueryIterator* iterator);
public:
// TODO: Remove.

                        void                            PutIterator(QueryIterator* iterator);

                        void                            VolumeUnmounting(Volume* volume);

private:
                        struct IteratorMap;

                        Locker                          fLock;
                        VolumeManager*          fVolumeManager;
                        IteratorMap*            fIterators;
};

// QueryIteratorPutter
class QueryIteratorPutter {
public:
        QueryIteratorPutter(QueryManager* manager, QueryIterator* iterator)
                : fManager(manager),
                  fIterator(iterator)
        {
        }

        ~QueryIteratorPutter()
        {
                if (fManager && fIterator)
                        fManager->PutIterator(fIterator);
        }

        void Detach()
        {
                fManager = NULL;
                fIterator = NULL;
        }

private:
        QueryManager*   fManager;
        QueryIterator*  fIterator;
};

#endif  // NET_FS_QUERY_MANAGER_H