LumberJack_TUI/ip_to_geo/ip_to_geo.cpp

#include "ip_to_geo.hpp"
#include <exception>
#include <iostream>
#include <fstream>
#include <cstdlib>
#include <map>
#include <string>
#include <GeoLite2PP.hpp>

void load_env_file(const std::string& env_path) {
    std::ifstream file(env_path);
    if (!file.is_open()) {
        std::cerr << "Warning: Could not open " << env_path << " file." << std::endl;
        return;
    }

    std::string line;
    while (std::getline(file, line)) {
        // Skip empty lines or comments
        if (line.empty() || line[0] == '#') continue;

        std::size_t delimiter_pos = line.find('=');
        if (delimiter_pos != std::string::npos) {
            std::string key = line.substr(0, delimiter_pos);
            std::string value = line.substr(delimiter_pos + 1);
            
            // Set the variable globally in the execution environment
            setenv(key.c_str(), value.c_str(), 1); 
        }
    }
}

std::string iplookup(const std::string& ip) {

    // 1. Load variables from the local .env file
    load_env_file();

    // 2. Fetch the path out of the environment variables
    const char* env_db_path = std::getenv("DB_PATH");
    // Fallback to a default path if the environment variable wasn't set
    std::string db_path = (env_db_path != nullptr) ? env_db_path : "data/GeoLite2-City.mmdb";


    std::string geo_string = "";

    try {
        GeoLite2PP::DB db(db_path);
        std::map<std::string, std::string> geo_data = db.get_all_fields(ip, "en");
    
         geo_string = geo_data["country"] + "," + geo_data["subdivision"] + "," + geo_data["city"] + "," + geo_data["latitude"] + "," + geo_data["longitude"] + "\n";
    }
    catch (const std::exception& e) {
        std::cerr << "Database failed to load: " << e.what() << std::endl;
    }
    return geo_string;
};