File indexing completed on 2024-04-14 03:49:45

 /*
     SPDX-FileCopyrightText: 2014 Vishesh Handa <me@vhanda.in>
 
     SPDX-License-Identifier: LGPL-2.1-or-later
 */
 
 #include "filtereddiriterator.h"
 #include "fileindexerconfig.h"
 
 using namespace Baloo;
 
 FilteredDirIterator::FilteredDirIterator(const FileIndexerConfig* config, const QString& folder, Filter filter)
     : m_config(config)
     , m_currentIter(nullptr)
     , m_filters(QDir::NoDotAndDotDot | QDir::Readable | QDir::NoSymLinks | QDir::Hidden)
     , m_firstItem(false)
 {
     if (filter == DirsOnly) {
         m_filters |= QDir::Dirs;
     } else if (filter == FilesAndDirs) {
         m_filters |= (QDir::Files | QDir::Dirs);
     }
 
     if (!m_config || m_config->shouldFolderBeIndexed(folder)) {
         m_currentIter = new QDirIterator(folder, m_filters);
         m_firstItem = true;
     }
 }
 
 FilteredDirIterator::~FilteredDirIterator()
 {
     delete m_currentIter;
 }
 
 QString FilteredDirIterator::next()
 {
     if (m_firstItem) {
         m_firstItem = false;
         m_filePath = m_currentIter->path();
         m_fileInfo = QFileInfo(m_filePath);
         return m_filePath;
     }
 
     m_filePath.clear();
     if (!m_currentIter) {
         m_fileInfo = QFileInfo();
         return QString();
     }
 
     bool shouldIndexHidden = false;
     if (m_config) {
         shouldIndexHidden = m_config->indexHiddenFilesAndFolders();
     }
 
     while (true) {
         // Last entry in the current directory found, try the next
         // directory from the stack
         // Loop until the directory is non-empty, or the stack is empty
         while (!m_currentIter->hasNext()) {
             delete m_currentIter;
             m_currentIter = nullptr;
 
             if (m_paths.isEmpty()) {
                 m_fileInfo = QFileInfo();
                 return QString();
             }
 
             const QString path = m_paths.pop();
             m_currentIter = new QDirIterator(path, m_filters);
         }
 
         m_filePath = m_currentIter->next();
         m_fileInfo = m_currentIter->fileInfo();
 
         auto shouldIndexFolder = [&] () -> bool {
             if (!m_config) {
                 return !m_fileInfo.isHidden();
             }
             QString folder;
             if (!m_config->folderInFolderList(m_filePath, folder)) {
                 return false;
             }
         if ((folder == m_filePath) || (folder == QString(m_filePath).append(QLatin1Char('/')))) {
                 return true;
             }
             if (!shouldIndexHidden && m_fileInfo.isHidden()) {
                 return false;
             }
             return m_config->shouldFileBeIndexed(m_fileInfo.fileName());
         };
 
         if (m_fileInfo.isDir()) {
             if (shouldIndexFolder()) {
                 // Push the current item to the directory stack
                 m_paths.push(m_filePath);
                 return m_filePath;
             }
         }
         else if (m_fileInfo.isFile()) {
             if ((!shouldIndexHidden) && m_fileInfo.isHidden()) {
                 continue;
             }
             if ((!m_config) || m_config->shouldFileBeIndexed(m_fileInfo.fileName())) {
                 return m_filePath;
             }
         }
     }
 }
 
 QString FilteredDirIterator::filePath() const
 {
     return m_filePath;
 }
 
 QFileInfo FilteredDirIterator::fileInfo() const
 {
     return m_fileInfo;
 }