Directory.cpp

#include "Directory.h"
#include <iostream>
#include <fstream>
#include <algorithm>
#include <iomanip>
#include <filesystem>
#include <string_view>
#include "UIUtils.h"

using namespace std;

bool Directory::idExists(string_view id) const noexcept {
    return any_of(students.begin(), students.end(), [id](const Student& s) {
        return s.getId() == id;
    });
}

void Directory::insertStudent(Student s) {
    if (idExists(s.getId())) {
        UI::printError("Student with ID " + string(s.getId()) + " already exists!");
        return;
    }
    students.push_back(move(s));
    UI::printSuccess("Student added successfully.");
}

void Directory::updateStudent(string_view id) {
    auto it = find_if(students.begin(), students.end(), [id](const Student& s) {
        return s.getId() == id;
    });

    if (it != students.end()) {
        string temp;
        cout << "  Enter new address (leave blank to keep current): ";
        getline(cin, temp);
        if (!temp.empty()) it->setAddress(temp);

        cout << "  Enter new phone number (leave blank to keep current): ";
        getline(cin, temp);
        if (!temp.empty()) it->setPhone(temp);

        UI::printSuccess("Record updated.");
    } else {
        UI::printError("Student with ID " + string(id) + " not found.");
    }
}

void Directory::deleteStudent(string_view id) {
    auto it = remove_if(students.begin(), students.end(), [id](const Student& s) {
        return s.getId() == id;
    });

    if (it != students.end()) {
        students.erase(it, students.end());
        UI::printSuccess("Record deleted.");
    } else {
        UI::printError("Student with ID " + string(id) + " not found.");
    }
}

void Directory::searchStudent(string_view query) const {
    bool found = false;
    cout << UI::CYAN << "\n  [ Search Results for: " << string(query) << " ]" << UI::RESET << endl;
    for (const auto& s : students) {
        if (s.getName().find(query) != string_view::npos || s.getId() == query) {
            cout << UI::CYAN << "  ┌──────────────────────────────────┐" << endl;
            cout << "  │ " << UI::BOLD << "Name:    " << UI::RESET << setw(24) << string(s.getName()) << UI::CYAN << " │" << endl;
            cout << "  │ " << UI::BOLD << "ID:      " << UI::RESET << setw(24) << string(s.getId()) << UI::CYAN << " │" << endl;
            cout << "  │ " << UI::BOLD << "Phone:   " << UI::RESET << setw(24) << string(s.getPhone()) << UI::CYAN << " │" << endl;
            cout << "  └──────────────────────────────────┘" << UI::RESET << endl;
            found = true;
        }
    }
    if (!found) UI::printError("No matching records found.");
}

void Directory::listDirectory() const {
    if (students.empty()) {
        UI::printError("Registry is empty.");
        return;
    }

    cout << "\n" << UI::CYAN << UI::BOLD << "  " << left << setw(10) << "ID" << setw(20) << "Name" << setw(15) << "Phone" << "Address" << UI::RESET << endl;
    cout << UI::CYAN << "  " << string(65, '-') << UI::RESET << endl;
    
    for (const auto& s : students) {
        cout << "  " << left << setw(10) << string(s.getId()) << setw(20) << string(s.getName()) 
             << setw(15) << string(s.getPhone()) << string(s.getAddress()) << endl;
    }
    cout << UI::CYAN << "  " << string(65, '-') << UI::RESET << endl;
    cout << UI::MAGENTA << "  Total Records: " << students.size() << UI::RESET << endl;
}

void Directory::sortByName() {
    sort(students.begin(), students.end(), [](const Student& a, const Student& b) {
        return a.getName() < b.getName();
    });
    UI::printSuccess("Registry sorted by Name.");
}

void Directory::sortByID() {
    sort(students.begin(), students.end(), [](const Student& a, const Student& b) {
        return a.getId() < b.getId();
    });
    UI::printSuccess("Registry sorted by ID.");
}

void Directory::saveToFile(string_view filename) const {
    ofstream file((string(filename)));
    if (!file) {
        UI::printError("Error opening file for saving.");
        return;
    }
    for (const auto& s : students) {
        file << string(s.getName()) << "|" << string(s.getId()) << "|" << string(s.getAddress()) << "|" << string(s.getPhone()) << "\n";
    }
    UI::printSuccess("Saved " + to_string(students.size()) + " records to " + string(filename));
}

void Directory::loadFromFile(string_view filename) {
    if (!filesystem::exists(filename)) return;

    ifstream file((string(filename)));
    students.clear();
    string line;
    while (getline(file, line)) {
        if (line.empty()) continue;
        
        size_t p1 = line.find('|');
        size_t p2 = line.find('|', p1 + 1);
        size_t p3 = line.find('|', p2 + 1);

        if (p1 != string::npos && p2 != string::npos && p3 != string::npos) {
            students.emplace_back(
                line.substr(0, p1),
                line.substr(p1 + 1, p2 - p1 - 1),
                line.substr(p2 + 1, p3 - p2 - 1),
                line.substr(p3 + 1)
            );
        }
    }
    UI::printSuccess("Loaded " + to_string(students.size()) + " records from " + string(filename));
}