fairport/pst/folder.h

Go to the documentation of this file.

#ifndef FAIRPORT_PST_FOLDER_H
#define FAIRPORT_PST_FOLDER_H

#include <algorithm>
#include <boost/iterator/filter_iterator.hpp>
#include <boost/iterator/transform_iterator.hpp>

#include "fairport/util/primitives.h"
#include "fairport/util/errors.h"

#include "fairport/ndb/database_iface.h"

#include "fairport/ltp/propbag.h"
#include "fairport/ltp/table.h"

#include "fairport/pst/message.h"
#include "fairport/pst/contact.h"
#include "fairport/pst/task.h"
#include "fairport/pst/pstutil.h"

namespace fairport
{


class search_folder : public detail::itembase
{
    typedef boost::filter_iterator<detail::is_item_type, const_table_row_iter> row_filter_iterator;
public:
    typedef boost::transform_iterator<message_transform_row, const_table_row_iter> message_iterator;
    typedef boost::transform_iterator<contact_transform_row, row_filter_iterator> contact_iterator;
    typedef boost::transform_iterator<task_transform_row, row_filter_iterator> task_iterator;

    search_folder(const node& n)
        : detail::itembase(n) { }

    // subobject discovery/enumeration
    message_iterator message_begin() const
        { return boost::make_transform_iterator(get_contents_table().begin(), message_transform_row(get_db())); }
    message_iterator message_end() const
        { return boost::make_transform_iterator(get_contents_table().end(), message_transform_row(get_db())); }

    contact_iterator contact_begin() const
        { return boost::make_transform_iterator(boost::make_filter_iterator(detail::is_item_type(contact_class), get_contents_table().begin(), get_contents_table().end()), contact_transform_row(get_db())); }
    contact_iterator contact_end() const
        { return boost::make_transform_iterator(boost::make_filter_iterator(detail::is_item_type(contact_class), get_contents_table().end(), get_contents_table().end()), contact_transform_row(get_db())); }

    task_iterator task_begin() const
        { return boost::make_transform_iterator(boost::make_filter_iterator(detail::is_item_type(task_class), get_contents_table().begin(), get_contents_table().end()), task_transform_row(get_db())); }
    task_iterator task_end() const
        { return boost::make_transform_iterator(boost::make_filter_iterator(detail::is_item_type(task_class), get_contents_table().end(), get_contents_table().end()), task_transform_row(get_db())); }

    // property access
    std::wstring get_name() const
        { return get_property_bag().read_prop<std::wstring>(0x3001); }
    size_t get_unread_message_count() const
        { return get_property_bag().read_prop<slong>(0x3603); }
    size_t get_message_count() const
        { return get_property_bag().read_prop<slong>(0x3602); }

    // lower layer access

    table& get_contents_table();
    const table& get_contents_table() const;

private:
    mutable std::tr1::shared_ptr<table> m_contents_table;
};

class folder;
typedef detail::item_transform_row<folder> folder_transform_row;
typedef detail::item_transform_row<search_folder> search_folder_transform_row;

class folder : public detail::itembase
{
    typedef boost::filter_iterator<detail::is_item_type, const_table_row_iter> row_filter_iterator;
    typedef boost::filter_iterator<detail::is_nid_type<nid_type_search_folder>, const_table_row_iter> search_folder_filter_iterator;
    typedef boost::filter_iterator<detail::is_nid_type<nid_type_folder>, const_table_row_iter> folder_filter_iterator;

public:
    typedef boost::transform_iterator<message_transform_row, const_table_row_iter> message_iterator;
    typedef boost::transform_iterator<folder_transform_row, folder_filter_iterator> folder_iterator;
    typedef boost::transform_iterator<search_folder_transform_row, search_folder_filter_iterator> search_folder_iterator;
    typedef boost::transform_iterator<contact_transform_row, row_filter_iterator> contact_iterator;
    typedef boost::transform_iterator<task_transform_row, row_filter_iterator> task_iterator;

    folder(const node& n)
        : detail::itembase(n) { }

    // subobject discovery/enumeration
    folder_iterator sub_folder_begin() const
        { return boost::make_transform_iterator(boost::make_filter_iterator<detail::is_nid_type<nid_type_folder> >(get_hierarchy_table().begin(), get_hierarchy_table().end()), folder_transform_row(get_db())); }
    folder_iterator sub_folder_end() const
        { return boost::make_transform_iterator(boost::make_filter_iterator<detail::is_nid_type<nid_type_folder> >(get_hierarchy_table().end(), get_hierarchy_table().end()), folder_transform_row(get_db())); }

    search_folder_iterator sub_search_folder_begin() const
        { return boost::make_transform_iterator(boost::make_filter_iterator<detail::is_nid_type<nid_type_search_folder> >(get_hierarchy_table().begin(), get_hierarchy_table().end()), search_folder_transform_row(get_db())); }
    search_folder_iterator sub_search_folder_end() const
        { return boost::make_transform_iterator(boost::make_filter_iterator<detail::is_nid_type<nid_type_search_folder> >(get_hierarchy_table().end(), get_hierarchy_table().end()), search_folder_transform_row(get_db())); }

    folder open_sub_folder(const std::wstring& name);

    message_iterator message_begin() const
        { return boost::make_transform_iterator(get_contents_table().begin(), message_transform_row(get_db())); }
    message_iterator message_end() const
        { return boost::make_transform_iterator(get_contents_table().end(), message_transform_row(get_db())); }
    
    contact_iterator contact_begin() const
        { return boost::make_transform_iterator(boost::make_filter_iterator(detail::is_item_type(contact_class), get_contents_table().begin(), get_contents_table().end()), contact_transform_row(get_db())); }
    contact_iterator contact_end() const
        { return boost::make_transform_iterator(boost::make_filter_iterator(detail::is_item_type(contact_class), get_contents_table().end(), get_contents_table().end()), contact_transform_row(get_db())); }

    task_iterator task_begin() const
        { return boost::make_transform_iterator(boost::make_filter_iterator(detail::is_item_type(task_class), get_contents_table().begin(), get_contents_table().end()), task_transform_row(get_db())); }
    task_iterator task_end() const
        { return boost::make_transform_iterator(boost::make_filter_iterator(detail::is_item_type(task_class), get_contents_table().end(), get_contents_table().end()), task_transform_row(get_db())); }

    message_iterator associated_message_begin() const
        { return boost::make_transform_iterator(get_associated_contents_table().begin(), message_transform_row(get_db())); }
    message_iterator associated_message_end() const
        { return boost::make_transform_iterator(get_associated_contents_table().end(), message_transform_row(get_db())); }

    // property access
    std::wstring get_name() const
        { return get_property_bag().read_prop<std::wstring>(0x3001); }
    size_t get_subfolder_count() const
        { return get_hierarchy_table().size(); } 
    size_t get_unread_message_count() const
        { return get_property_bag().read_prop<slong>(0x3603); }
    size_t get_message_count() const
        { return get_property_bag().read_prop<slong>(0x3602); }
    size_t get_associated_message_count() const
        { return get_property_bag().read_prop<slong>(0x3617); }

    // lower layer access

    table& get_hierarchy_table();
    table& get_contents_table();
    table& get_associated_contents_table();
    const table& get_hierarchy_table() const;
    const table& get_contents_table() const;
    const table& get_associated_contents_table() const;

private:
    mutable std::tr1::shared_ptr<table> m_contents_table;
    mutable std::tr1::shared_ptr<table> m_associated_contents_table;
    mutable std::tr1::shared_ptr<table> m_hierarchy_table;
};

typedef detail::item_transform_info<folder> folder_transform_info;

} // end namespace fairport

inline const fairport::table& fairport::search_folder::get_contents_table() const
{
    if(!m_contents_table)
        m_contents_table.reset(new table(get_db()->lookup_node(make_nid(nid_type_search_contents_table, get_nid_index(get_property_bag().get_node().get_id())))));

    return *m_contents_table;
}

inline fairport::table& fairport::search_folder::get_contents_table()
{
    return const_cast<table&>(const_cast<const search_folder*>(this)->get_contents_table());
}


namespace internal
{

struct folder_name_equal : public std::unary_function<bool, const fairport::folder&>
{
    folder_name_equal(const std::wstring& name) : m_name(name) { }
    bool operator()(const fairport::folder& f) const { return f.get_name() == m_name; }
    std::wstring m_name;
};

} // end namespace internal

inline fairport::folder fairport::folder::open_sub_folder(const std::wstring& name)
{
    folder_iterator iter = std::find_if(sub_folder_begin(), sub_folder_end(), internal::folder_name_equal(name));

    if(iter != sub_folder_end())
        return *iter;

    throw key_not_found<std::wstring>(name);
}

inline const fairport::table& fairport::folder::get_contents_table() const
{
    if(!m_contents_table)
        m_contents_table.reset(new table(get_db()->lookup_node(make_nid(nid_type_contents_table, get_nid_index(get_property_bag().get_node().get_id())))));

    return *m_contents_table;
}

inline fairport::table& fairport::folder::get_contents_table()
{
    return const_cast<table&>(const_cast<const folder*>(this)->get_contents_table());
}

inline const fairport::table& fairport::folder::get_hierarchy_table() const
{
    if(!m_hierarchy_table)
        m_hierarchy_table.reset(new table(get_db()->lookup_node(make_nid(nid_type_hierarchy_table, get_nid_index(get_property_bag().get_node().get_id())))));

    return *m_hierarchy_table;
}

inline fairport::table& fairport::folder::get_hierarchy_table()
{
    return const_cast<table&>(const_cast<const folder*>(this)->get_hierarchy_table());
}

inline const fairport::table& fairport::folder::get_associated_contents_table() const
{
    if(!m_associated_contents_table)
        m_associated_contents_table.reset(new table(get_db()->lookup_node(make_nid(nid_type_associated_contents_table, get_nid_index(get_property_bag().get_node().get_id())))));

    return *m_associated_contents_table;
}

inline fairport::table& fairport::folder::get_associated_contents_table()
{
    return const_cast<table&>(const_cast<const folder*>(this)->get_associated_contents_table());
}

#endif