scanner.cpp

#include "scanner.h"
#include "bip39_sequence.h"
#include "bip39_checksum.h"
#include "bip39_wordlist_std.h"
#include <QDirIterator>
#include <QFile>
#include <QFileInfo>
#include <QDir>
#include <QTextStream>
#include <QDateTime>
#include <QDebug>

Scanner::Scanner() {}

void Scanner::setWordlist(const std::vector<QString> &words) {
    wordlist = words;
    wordSet  = Bip39Sequence::buildWordSet(words);
}

bool Scanner::containsBip39Word(const QString &text) {
    for (const auto &word : wordlist) {
        if (text.contains(word, Qt::CaseInsensitive))
            return true;
    }
    return false;
}

void Scanner::scanDirectory(const QString &directoryPath,
                            const QString &outputPath,
                            std::function<void(const QString&)> onMatch,
                            std::function<void(int, int, const QString&)> onProgress,
                            std::function<bool()> shouldStop) {
    QDirIterator it(directoryPath, QStringList() << "*.txt" << "*.ini",
                    QDir::Files, QDirIterator::Subdirectories);

    QStringList allFiles;
    while (it.hasNext()) {
        allFiles << it.next();
    }

    int total = allFiles.size();
    int current = 0;
    QDir().mkpath(outputPath);

    for (const auto &filePath : allFiles) {
        if (shouldStop()) break;
        current++;
        onProgress(current, total, filePath); // Report progress and current file

        QFile file(filePath);
        if (!file.open(QIODevice::ReadOnly | QIODevice::Text))
            continue;

        QTextStream in(&file);
        QString content = in.readAll();

        // Quick pre-filter: any BIP-39 word present at all?
        if (!containsBip39Word(content))
            continue;

        // Full filter: at least one valid-length sequence must pass checksum.
        bool validated = false;
        for (const Bip39Sequence::Match &seq : Bip39Sequence::extract(content, wordSet)) {
            std::vector<std::string> stdWords;
            stdWords.reserve(seq.words.size());
            for (const QString &w : seq.words)
                stdWords.push_back(w.toStdString());
            if (BIP39Checksum::validate(stdWords, Bip39WordsStd::wordlist)) {
                validated = true;
                break;
            }
        }

        if (validated) {
            QString timeStamp = QDateTime::currentDateTime().toString("yyyyMMdd_HHmmss");
            QFileInfo fi(filePath);
            QString destPath = outputPath + "/" + timeStamp + "_" + fi.fileName();
            QFile::copy(filePath, destPath);
            onMatch(filePath);
        }
    }
}