#include "irda-app-brute-force.hpp"

void IrdaAppBruteForce::add_record(int index, const char* name) {
    records[name].index = index;
    records[name].amount = 0;
}

bool IrdaAppBruteForce::calculate_messages() {
    bool fs_res = false;
    fs_res = file_parser.get_fs_api().file.open(
        &file, universal_db_filename, FSAM_READ, FSOM_OPEN_EXISTING);
    if(!fs_res) {
        file_parser.get_sd_api().show_error(file_parser.get_sd_api().context, "Can't open file");
        return false;
    }

    file_parser.reset();
    while(1) {
        auto message = file_parser.read_signal(&file);
        if(!message) break;
        auto element = records.find(message->name);
        if(element != records.cend()) {
            ++element->second.amount;
        }
    }

    file_parser.get_fs_api().file.close(&file);

    return true;
}

void IrdaAppBruteForce::stop_bruteforce() {
    if(current_record.size()) {
        file_parser.get_fs_api().file.close(&file);
        current_record.clear();
    }
}

// TODO: [FL-1418] replace with timer-chained consequence of messages.
bool IrdaAppBruteForce::send_next_bruteforce(void) {
    furi_assert(current_record.size());

    std::unique_ptr<IrdaAppFileParser::IrdaFileSignal> file_signal;

    do {
        file_signal = file_parser.read_signal(&file);
    } while(file_signal && current_record.compare(file_signal->name));

    if(file_signal) {
        file_signal->signal.transmit();
    }
    return !!file_signal;
}

bool IrdaAppBruteForce::start_bruteforce(int index, int& record_amount) {
    file_parser.reset();
    for(const auto& it : records) {
        if(it.second.index == index) {
            record_amount = it.second.amount;
            current_record = it.first;
            break;
        }
    }

    if(record_amount) {
        bool fs_res = file_parser.get_fs_api().file.open(
            &file, universal_db_filename, FSAM_READ, FSOM_OPEN_EXISTING);
        if(fs_res) {
            return true;
        } else {
            file_parser.get_sd_api().show_error(
                file_parser.get_sd_api().context, "Can't open file");
        }
    }

    return false;
}