scanworker.cpp

#include "scanworker.h"
#include <QFile>
#include <QFileInfo>
#include <QDateTime>
#include <QTextStream>
#include <QDirIterator>
#include <QDir>
#include <QMap>
#include <QDebug>

ScanWorker::ScanWorker(const QString &src, const QString &out, const std::vector<QString> &words)
    : sourceDir(src), outputDir(out)
{
    // Normalize wordlist to lowercase upfront
    wordlist.reserve(words.size());
    for (const auto &w : words) {
        wordlist.push_back(w.trimmed().toLower());
    }
}

void ScanWorker::requestStop() {
    stopRequested.store(true);
}

void ScanWorker::startScan() {
    foundFiles.clear();

    // Gather all .txt and .ini files
    qDebug() << "Scanning directory:" << sourceDir;
    QDirIterator it(sourceDir, {"*.txt", "*.ini"}, QDir::Files, QDirIterator::Subdirectories);
    int current = 0;
    QStringList allFiles;
    while (it.hasNext()) {
        QString file = it.next();
        allFiles.append(file);
        qDebug() << "Found file:" << file;
    }
    int total = allFiles.size();
    qDebug() << "Total files found:" << total;
    emit totalFilesCounted(total);

    for (const QString &filePath : allFiles) {
        if (stopRequested.load())
            break;

        emit currentFile(filePath);
        emit progress(++current, total);

        QFile file(filePath);
        if (!file.open(QIODevice::ReadOnly | QIODevice::Text)) {
            qDebug() << "Cannot open file:" << filePath;
            continue;
        }

        QTextStream in(&file);
        QStringList matchedWords;
        QMap<QString, QList<int>> wordLineNumbers;  // Track which lines each word appears on

        // Scan file line-by-line
        int lineNumber = 0;
        while (!in.atEnd() && !stopRequested.load()) {
            lineNumber++;
            QString line = in.readLine().toLower();
            for (const auto &word : wordlist) {
                if (line.contains(word)) {
                    if (!matchedWords.contains(word)) {
                        matchedWords.append(word);
                    }
                    wordLineNumbers[word].append(lineNumber);
                    
                    // Emit detailed signal for each word found
                    emit wordFoundInFile(word, filePath, lineNumber);
                    qDebug() << "Match found:" << word << "in" << filePath << "at line" << lineNumber;
                }
            }
        }
        file.close();

        // Copy file if matches are found
        if (!matchedWords.isEmpty()) {
            QFileInfo fi(filePath);
            QString timeStamp = QDateTime::currentDateTime().toString("yyyyMMdd_HHmmss");
            QString destPath = outputDir + "/" + timeStamp + "_" + fi.fileName();
            if (QFile::copy(filePath, destPath)) {
                MatchEntry entry{filePath, destPath, matchedWords.join(", ")};
                foundFiles.append(entry);
                emit matchFound(filePath);
                
                // Emit detailed completion signal
                emit fileMatchComplete(filePath, matchedWords.size(), matchedWords);
            } else {
                qDebug() << "Failed to copy:" << filePath << "to" << destPath;
            }
        }
    }

    // Write results CSV
    QString csvTimeStamp = QDateTime::currentDateTime().toString("yyyyMMdd_HHmmss");
    QString csvPath = outputDir + "/scan_results_" + csvTimeStamp + ".csv";
    QFile csvFile(csvPath);
    if (csvFile.open(QIODevice::WriteOnly | QIODevice::Text)) {
        QTextStream out(&csvFile);
        out << "file_path,copied_to,matched_words\n";
        for (const auto &entry : foundFiles) {
            out << "\"" << entry.originalPath << "\","
                << "\"" << entry.copiedPath << "\","
                << "\"" << entry.matchedWords << "\"\n";
        }
        out << "\"TOTAL_FILES_SCANNED\"," << total << ",\n";

        qDebug() << "CSV written to:" << csvPath << "with" << foundFiles.size() << "entries.";
        emit csvGenerated(csvPath);  // <---- NEW: Notify MainWindow
    } else {
        qDebug() << "Failed to write CSV to:" << csvPath;
    }

    emit finished();
}