Программа для поиска введенного слова или фразы во всех файлах каталога - C++ - Ответ 7043280

#include <iostream>
#include <windows.h>
#include <list>
#include <fstream>
#include <time.h>
#include <string>
#include <vector>
#include <stack>
#include "IDZ.h"
 
using namespace std;
 
class poisk
{
protected:
 
    //имя файла, в который записуются результаты поиска
    string Name;
 
    //фраза для поиска
    string Fraz;
 
    //имя конрневого каталога 
    string mainName;
 
 
    bool scan;
    bool ftxt;
    bool fxml;
    bool fhtml;
 
public:
    poisk(string _mainName, bool _scan, string _Fraz, string _Name, bool _txt = 1, bool _xml = 1, bool _html = 1);
 
    int ready();
 
    //главный цикл поиска(перебирает все файлы и ведет статистику)
    void start();
 
};
 
poisk::poisk(string _mainName, bool _scan, string _Fraz, string _Name, bool _ftxt, bool _fxml, bool _fhtml)
{
 
    mainName = _mainName;
    scan = _scan;
    Fraz = _Fraz;
    Name = _Name;
    ftxt = _ftxt;
    fxml = _fxml;
    fhtml = _fhtml;
 
}
 
int poisk::ready()
{
    //если не нужно искать ни 1 из типов, то тогда нет смысла начинать:)
    if (!(ftxt||fxml||fhtml)) return 1;
    
    //проверяем каталог для поиска на существование и доступа чтения
    if (_access(mainName.c_str(), 4) == -1)
        return 2;
 
    //находим имя папки, в которую будем писать результаты
    string logDir(Name, 0, Name.find_last_of('\\')+1);
    //проверяем папку на возможность записи
    if (_access(logDir.c_str(), 2) == -1)
        return 3;
 
    return 0;
}
 
void poisk::start(){
 
    //не запускаем,если не готовы
    if (ready() != 0) return;
 
 
    //создаем массив(массив указателей на класс Results), который будет содержать лог
    vector<Result *> LOG;
    //создавать обьекты такого класса(абстрактного) нельзя, но можно создавать на него ссылки
    //этими ссылками так же можно указывать на все классы, что унаследованы от класса Result
    //т.е. у нас массив с ссылками, что указывают на разные классы, но отец у них 1-Result
    //у него есть все нужные методы, кроме search, их реализует каждый потомок самостоятельно
 
    //vector-это такой динамический класс, который при переполнении сам выделяет себе память
    //его можно использовать как обычный массив, но кроме этого у него есть еще полезные методы типа size и push_back
 
    //в этом стеке будут все имена каталогов, которые нужно будет проверить
    stack<string> dirs;
 
    //изначально в стеке путь к папке, которую задал пользователь
    //дальше, если задан поиск в подкоталогах, в стек будут добавляться имена подкаталогов
    dirs.push(mainName);
 
    //засекаем время
    clock_t startT = clock();
    
    //количество сканированых файлов, совпадений
    int fCount(0), mCount(0);
    //строк
    long lCount(0);
 
 
    //главный цикл поиска
    bool cEnd = false;
 
    //пока есть папки, в которых можно производить поиск
    while (!dirs.empty())
    {
 
        //находим файл//////////////////
        string fullFileName;
        string fileName;
 
        string curDir;
        //вытаскиваем из стека полное имя каталога
        curDir = dirs.top();
        dirs.pop();
 
        WIN32_FIND_DATA dir;
        string buf;
 
        //string(curDir+"\\").c_str() - создаем анонимный класс и передаем указатель на массив символов
        HANDLE han = FindFirstFile(string(curDir+"*").c_str(), &dir);
 
        if (han != INVALID_HANDLE_VALUE)
        {
            
            do
            {
                fileName=dir.cFileName;
                //выводим статистику
 
                //очищаем командную строку
                system("cls");
                cout << "В процессе проверки:" << fileName << endl;
                cout << "Файлов проверено:" << fCount << endl;
                cout << "Строк проверено:" << lCount << endl;
                cout << "Совпадений найдено:" << mCount << endl;
 
                //вычисляем время со старта
                clock_t currT = clock();
                cout << "Время с начала старта поиска: " << (currT - startT) / 1000. << " сек." << endl;
 
                //проверем "файл ли это?"
                if (fileName.find('.') != -1)
                {
                    //полное имя файла
                    fullFileName.clear();
                    fullFileName = curDir + fileName;
 
                    //проверяем окончание////////////////////////
 
                    //находим номер символа "точка"
                    int offs = fullFileName.find_last_of('.') + 1;
 
                    //копируем адресс названия файла со смещением(копируем символы после точки)
                    string format(fullFileName, offs, fullFileName.size() - offs);
 
 
                    //поскольку string не подходит для switch используем if
                    //тк расширение только 1, то и выполниться только 1 из следующих вариантов
 
                    if (format == "txt" && ftxt)
                    {
                        LOG.push_back(new TxtResult(fullFileName, Fraz));
                        LOG[LOG.size() - 1]->search();
                        //запись статистики/////////////////////
                        lCount += LOG[LOG.size() - 1]->getStrCount();
                        mCount += LOG[LOG.size() - 1]->getMatchesCount();
                        fCount++;
                    }
 
                    if (format == "xml" && fxml)
                    {
                        LOG.push_back(new XmlResult(fullFileName, Fraz));
                        LOG[LOG.size() - 1]->search();
                        //запись статистики/////////////////////
                        lCount += LOG[LOG.size() - 1]->getStrCount();
                        mCount += LOG[LOG.size() - 1]->getMatchesCount();
                        fCount++;
                    }
 
                    if (format == "html" && fhtml)
                    {
                        LOG.push_back(new HtmlResult(fullFileName, Fraz));
                        LOG[LOG.size() - 1]->search();
                        //запись статистики/////////////////////
                        lCount += LOG[LOG.size() - 1]->getStrCount();
                        mCount += LOG[LOG.size() - 1]->getMatchesCount();
                        fCount++;                       
                    }
 
                    
                }
                else
                {//если в имени файла нет точки, то в fileName имя каталога
                
                    //если включен поиск в подкаталогах, то добавляем в стек папку для дальнейшей проверки
                    if (scan) dirs.push(curDir + fileName+"\\");
                }
            } while (FindNextFile(han, &dir));
 
 
        }   
    }
    
    //очищаем данные прошлоо поиска
    ofstream f(Name, ios::out);
    f.close();
 
    //для каждого обьекта вызываем метод printIfSucces, который запишет данные своего обьекта
    //если он что-то нашел
    for (int i = 0; i < LOG.size(); i++)
    {
        LOG[i]->printIfSucces(Name);
    }
 
    //выводим статистику
 
    //очищаем командную строку
    system("cls");
 
    cout << "Файлов проверено:" << fCount << endl;
    cout << "Строк проверено:" << lCount << endl;
    cout << "Совпадений найдено:" << mCount << endl;
 
    //вычисляем время со старта
    clock_t currT = clock();
    cout << "Время с начала старта поиска: " << (currT - startT) / 1000. << " сек." << endl;
    
    cout << "!!!!!!!!!!!!!!!!Search was finishd!!!!!!!!!!!!!!!!!" << endl;
 
 
}
int main()
{
    setlocale(0, "rus");
    
    //буфер для коректного переноса русских символов в string
    char OEMc[256] = "";
 
    bool allFine = false;
 
    //переенные для запоминания параметров ss-поиск в подкаталогах
    bool txt(false), html(false), xml(false), v(false);
    string root, Fraz,logDir;
 
    while (!allFine)
    {
        cout << "Vvedite put: ";
            
 
        //вводим строку
        cin.getline(OEMc,256);
        
        //перекодирываем из кодировки консоли в Ansi
        OemToCharA(OEMc, OEMc);
 
        //определяем параметры
        string buf(OEMc);
 
        //диапазоны параметров в строке
        int a, b;
 
        //в вырезаем в root название каталога(без [])
        a = buf.find('[');
        b = buf.find(']');
        //проверяем коректность ввода пользователя
        if (a>=0 && b>0 && b>a)
        {
            root = string(buf, a + 1, b - a - 1);
            buf.erase(0, b + 1);
        }
        else
        {
            allFine = false;
        }
    
 
        //вырезаем фразу для поиска
        a = buf.find('[');
        b = buf.find(']');
        if (a >= 0 && b>0 && b>a)
        {
            Fraz = string(buf, a + 1, b - a - 1);
            buf.erase(0, b + 1);
        }
        else
        {
            allFine = false;
        }
 
        //вырезаем имя директории для создания файла с результатами
        a = buf.find('[');
        b = buf.find(']');
        if (a >= 0 && b>0 && b>a)
        {
            logDir = string(buf, a + 1, b - a - 1);
            buf.erase(0, b + 1);
        }
        else
        {
            allFine = false;
        }
 
        //такой вариант был бы неуместный, если бы мы могли искать в файлах типа xhtml,css или тд
        //но в нашем случае так делать можно
        if (buf.find("txt")!=-1) txt = true;
        if (buf.find("xml") != -1) xml = true;
        if (buf.find("html") != -1) html = true;
        if (buf.find("ss") != -1) v = true;
        
        
        if (!allFine) cout << "Был выполнен некорректный ввод!\n";
 
        //создаем обьект поискового класса
        poisk asd(root, v, Fraz,logDir, txt, xml, html);
 
        allFine = false;
 
        switch (asd.ready())
        {
        case 0: {
                    asd.start();
                    allFine = true;
        }break;
        case 1: cout << " Укажите хотя бы 1 тип для поиска!\n";
            break;
        case 2: cout << " Указаная папка для поиска не доступна для чтения или не существует!\n";
            break;
        case 3: cout << " Невозможно создать файл по указанному пути!\n";
            break;
        default:
            break;
        }
 
 
        
    }
 
    
    
    system("pause");
}