#include <iostream>
#include <fstream>
#include <ostream>
#include <string>
#include <vector>
#include <regex>

#include "log_parsing.h"

parsed_logs::parsed_logs(std::string log_path) {
    std::ifstream file(log_path);
    if (!file.is_open()) {
        std::cerr << "Error loading " << log_path << std::endl;
        return;
    }

    std::regex log_pattern(R"((\S+)\s+-\s+-\s+\[([^\]]+)\]\s+\"([^\"]+)\"\s+(\d+)\s+(\d+|-)\s+\"([^\"]*)\"\s+\"([^\"]*)\")");
    
    std::string line;
    std::smatch match;

    while (std::getline(file, line)) {
        if (std::regex_search(line, match, log_pattern)) {
            
            Entry current_entry;
            current_entry.ip        = match[1].str();
            current_entry.timestamp = match[2].str();
            current_entry.request   = match[3].str();
            current_entry.status    = match[4].str();
            current_entry.bytes     = match[5].str();
            current_entry.referer   = match[6].str();

            std::string raw_ua = match[7].str();

            if (raw_ua.find("Windows NT 10.0") != std::string::npos)       current_entry.os = "Windows 10/11";
            else if (raw_ua.find("Windows NT 6.1") != std::string::npos)   current_entry.os = "Windows 7";
            else if (raw_ua.find("iPhone") != std::string::npos)           current_entry.os = "iOS (iPhone)";
            else if (raw_ua.find("iPad") != std::string::npos)             current_entry.os = "iOS (iPad)";
            else if (raw_ua.find("Android") != std::string::npos)          current_entry.os = "Android";
            else if (raw_ua.find("Macintosh") != std::string::npos)        current_entry.os = "macOS";
            else if (raw_ua.find("Linux") != std::string::npos)            current_entry.os = "Linux";
            else                                                           current_entry.os = "Unknown OS";

            if (raw_ua.find("Edg/") != std::string::npos)                  current_entry.browser = "Microsoft Edge";
            else if (raw_ua.find("OPR/") != std::string::npos)             current_entry.browser = "Opera";
            else if (raw_ua.find("Chrome/") != std::string::npos)          current_entry.browser = "Google Chrome";
            else if (raw_ua.find("Safari/") != std::string::npos)          current_entry.browser = "Apple Safari";
            else if (raw_ua.find("Firefox/") != std::string::npos)         current_entry.browser = "Mozilla Firefox";
            else if (raw_ua.find("curl/") != std::string::npos)            current_entry.browser = "curl (CLI Tool)";
            else                                                           current_entry.browser = "Unknown Browser/Bot";

            logs.push_back(current_entry);
        }
    }
    file.close();
}

std::string parsed_logs::entryx_ip(int x){
    return logs[x].ip;
}

std::string parsed_logs::entryx_timestamp(int x){
    return logs[x].timestamp;
}

std::string parsed_logs::entryx_request(int x){
    return logs[x].request;
}

std::string parsed_logs::entryx_status(int x){
    return logs[x].status;
}

std::string parsed_logs::entryx_bytes(int x){
    return logs[x].bytes;
}

std::string parsed_logs::entryx_referer(int x){
    return logs[x].referer;
}

std::string parsed_logs::entryx_os(int x){
    return logs[x].os;
}

std::string parsed_logs::entryx_browser(int x){
    return logs[x].browser;
}