#include "./filteredfilesystemitemmodel.h"

namespace RenamingUtility {

/*
    TRANSLATOR RenamingUtility::FilteredFileSystemItemModel
    Necessary for lupdate.
*/

FilteredFileSystemItemModel::FilteredFileSystemItemModel(ItemStatus statusFilter, QObject *parent) :
    QSortFilterProxyModel(parent),
    m_statusFilter(statusFilter)
{}

bool FilteredFileSystemItemModel::filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const
{
    QModelIndex sourceIndex = sourceModel()->index(sourceRow, 0, sourceParent);
    if(sourceIndex.isValid()) {
        if(FileSystemItem *item = reinterpret_cast<FileSystemItem *>(sourceIndex.internalPointer())) {
            if(item->status() == m_statusFilter) {
                return true;
            } else if(item->status() == ItemStatus::Current && !item->counterpart() && !item->note().isEmpty()) {
                return true;
            } else if(item->type() == ItemType::Dir) {
                QModelIndex child = sourceIndex.child(0, 0);
                while(child.isValid()) {
                    if(filterAcceptsRow(child.row(), sourceIndex)) {
                        return true;
                    }
                    child = sourceIndex.child(child.row() + 1, 0);
                }
            }
        }
    }
    return false;
}

bool FilteredFileSystemItemModel::filterAcceptsColumn(int sourceColumn, const QModelIndex &) const
{
    switch(m_statusFilter) {
    case ItemStatus::Current:
        return sourceColumn == 0;
    case ItemStatus::New:
        return sourceColumn == 1 || sourceColumn == 2;
    default:
        return false;
    }
}

}