Database.cpp

#include <iostream>
#include <fstream>
#include <string>
#include <sstream>
#include "Database.h"

Database::Database() {}

static void limparLinha(std::string& linha) {
    while (!linha.empty() && (linha.back() == '\r' || linha.back() == '\n' || linha.back() == ' ')) {
        linha.pop_back();
    }
}

bool Database::readCsv(std::string database) {
    std::ifstream arquivo(database);

    if (!arquivo) return false;
    
    std::string linhaAtual;

    if (std::getline(arquivo, linhaAtual)) {
        limparLinha(linhaAtual);

        std::stringstream linhaAtualStream(linhaAtual);
        std::string coluna;

        std::getline(linhaAtualStream, coluna, ',');

        while (std::getline(linhaAtualStream, coluna, ',')) {
            strNames.push_back(coluna);
        }   
    }

    while (std::getline(arquivo, linhaAtual)) {
        limparLinha(linhaAtual);

        std::stringstream linhaAtualStream(linhaAtual);
        std::string pessoaData;

        Profile pessoa;

        std::getline(linhaAtualStream, pessoaData, ',');

        pessoa.name = pessoaData;

        for (int i = 0; i < (int)strNames.size(); i++) {
            std::getline(linhaAtualStream, pessoaData, ',');
            
            int valor = std::stoi(pessoaData);
            std::string str = strNames[i];

            pessoa.strCounts[str] = valor;
        }

        profiles.push_back(pessoa);
    }
    
    arquivo.close();
    return true;
}

std::vector<std::string> Database::getStrNames() const {
    return strNames;
}

std::string Database::search(const std::map<std::string, int>& profile) const {
    for (const auto& pessoa : profiles) {
        bool combina = true;

        for (const auto& par : pessoa.strCounts) {
            auto desconhecidoPar = profile.find(par.first);

            if (desconhecidoPar == profile.end() || desconhecidoPar->second != par.second) {
                combina = false;

                break;
            }  
        }

        if (combina) return pessoa.name;
    }
    
    return "";
}