Поиск файлов по маске

/**
 * find.exe <directory> -name "file.*"
 * -abc сортирует по возрастанию, -cba по убывания
 * Пример find.exe C:\test\ -name "*.txt" -abc
 * Что выведет:
 *    C:\test\folder2\doc.txt
 *    C:\test\folder2\doc1.txt
 *    C:\test\folder3\folder2\doc1.txt
 */

#include <iostream> // Для cin, cout, endl
#include <string> // для string
#include <vector> // для vector
#include <windows.h> // для поиска файлов - WIN32_FIND_DATA, HANDLE, FindFirstFile, ...
#include <algorithm> // Для сортировки sort()

using namespace std;

// Структура данных о файле.
// Содержит название, аттрибуты, и прочее
WIN32_FIND_DATA fileData;
HANDLE handle; // Поисковой дискриптор. Нужен для функции FindFirstFile

// Функция нужна для поиска папок в заданной папке
// Например На диске C:\ есть папки dir1 и dir2
// Тогда вызвав findSubdirectory("C:\") получим вектор с "dir1" и "dir2"
vector<string> findSubdirectory(string &dir) {
    vector<string> subdirectories; // Массив с названием папок

    // Создаем поисковой дискриптор
    handle = FindFirstFile(dir.c_str(), &fileData);

    // INVALID_HANDLE_VALUE возникает когда не найден путь/директория
    if (handle != INVALID_HANDLE_VALUE) {
        // Выполняем цикл, пока существуют файлы
        do {
            // Если найденная сущность является папкой, добавляем название в массив папок
            if ((fileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0) {
                subdirectories.push_back(fileData.cFileName);
            }
        } while (FindNextFile(handle, &fileData) != 0);
    }
    FindClose(handle);
    return subdirectories;
}

// Функция нужна для поиска файлов.
// На вход подается строка, директория + название файла. Поддерживает маску.
// Например C:\test.* или C:\* или C:\test\dir\file.doc
// Возвращаемое значение - вектор с названиями файлов (без директории)
vector<string> findFiles(string &filePatch) {
    vector<string> result;

    handle = FindFirstFile(filePatch.c_str(), &fileData);

    if (handle != INVALID_HANDLE_VALUE) {
        do {
            //если найденный файл не является папкой добавляем его в вектор
            if ((fileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == 0) {
                result.push_back(fileData.cFileName);
            }
        } while (FindNextFile(handle, &fileData) != 0);
        FindClose(handle);
    }
    return result;
}

// Поиск файла
// 1. Ищем файл в текущей директории
// 2. Ищем все субдиректории
// 3. Для каждой субдиректории выполняем эту же функцию
//
// Если нам нужно сортировать результат, то сначала мы должны создать
// vector<string> из найденных файлов, а потом отсортировать их
// А если нам не нужно сортировать - то мы можем сразу выводить найденные файлы
//
// dir - директория, например C:\ или C:\Users\Alex
// fileName - название файла, myFile.cpp или *file.*
// result - результат. Сюда будем записывать найденные файлы.
void find(vector<string> &result, string &dir, string &fileName, bool needSave) {
    // 1. Ищем файлы в текущей директории
    vector<string> files = findFiles(dir + fileName);
    // Если не нужно сохранять, то выводим найденные файлы
    if (!needSave) {
        for (auto findFileName : files) {
            cout << dir << findFileName << endl;
        }
    // Если нужно сохранять, то добавляем найденные файлы в result
    } else {
        for (auto fileName : files) {
            result.push_back(dir + fileName);
        }
    }
    // 2. Ищем все папки в текущей директории
    vector<string> dirs = findSubdirectory(dir + "*");

    // 3. Для всех найденных папок выполням эту же функцию
    // Указатели на текущую директорию "." и на директорию выше ".." не учитываем
    for (auto find_dir : dirs) {
        if (find_dir != "." && find_dir != "..") {
            // cout << "Выплняем поиск в папке " << (dir + find_dir) << endl;
            find(result, dir + find_dir + "\\", fileName, needSave);
        }
    }
}

// argc - кол-во аргументов, переданное программе
// argv - сами аргументы
int main(int argc, char* argv[]) {
    setlocale(LC_ALL, "Russian");
    string directory;
    string fileName;
    int sort = 0; // "1" - abc, "-1" - cba, "0" - нет сортировки
    // Для всех аргументов
    for (int i = 0; i < argc; ++i) {
        string arg = argv[i];
        // Первым аргументом задаётся директория
        if (i == 1) directory = arg;
        else if (arg == "-abc") sort = 1;
        else if (arg == "-cba") sort = -1;
        else if (arg == "-name") {
            fileName = argv[i + 1];
            i++;
        }
    }

    // Если не задана директория или название файла - выходим из программы
    if (directory.size() == 0 || fileName.size() == 0) {
        cout << "Используйте find.exe <directory> -name <fileName> [-abc/-cab]" << endl;
        return 0;
    }

    vector<string> result;

    // Вызываем функцию для рекурсивного поиска файлов
    find(result, directory, fileName, (sort != 0));

    if (sort != 0) {
        // Сортируем от A до Z
        std::sort(result.begin(), result.end());
        // Если нужно соритровать в обратном порядке, реверсим массив
        if (sort == -1) std::reverse(result.begin(), result.end());

        for (auto fileName : result) {
            cout << fileName << endl;
        }
    }

    return 0;
}