vault.cpp

#include "vault.h"
#include "../security/security.h"
#include <iostream>
#include <fstream>
#include <sstream>
#include <algorithm>

using namespace std;

// ================= START VAULT (MENU) =================
void startVault(string filename) {
    // Load entries from file (encrypted passwords)
    vector<Entry> vault = loadEntries(filename);

    int choice;
    do {
        cout << "\n====== 🔐 CRYPTVAULT ======\n";
        cout << "1. Add Entry\n";
        cout << "2. View Vault\n";
        cout << "3. Search Entry\n";
        cout << "4. Delete Entry\n";
        cout << "5. Exit\n";
        cout << "===========================\n";
        cout << "Choice: ";
        cin >> choice;
        cin.ignore(); // clear newline

        switch(choice) {
            case 1: addEntry(vault); break;
            case 2: viewEntries(vault); break;
            case 3: searchEntry(vault); break;
            case 4: deleteEntry(vault); break;
        }

        // Always save vault after any operation
        saveEntries(filename, vault);

    } while(choice != 5);
}


// ================= LOAD ENTRIES =================
vector<Entry> loadEntries(string filename) {
    vector<Entry> vault;
    ifstream file(filename);
    string line;

    while(getline(file, line)) {
        stringstream ss(line);
        Entry e;
        getline(ss, e.site, '|');
        getline(ss, e.username, '|');
        getline(ss, e.password, '|');

        // Decrypt password before storing in memory
        e.password = decrypt(e.password);

        vault.push_back(e);
    }

    return vault;
}


// ================= SAVE ENTRIES =================
void saveEntries(string filename, vector<Entry>& vault) {
    ofstream file(filename);

    for(auto &e : vault) {
        // Encrypt password before writing to file
        file << e.site << "|" << e.username << "|" << encrypt(e.password) << endl;
    }
}


// ================= ADD ENTRY =================
void addEntry(vector<Entry>& vault) {
    Entry e;
    cout << "Site: "; getline(cin, e.site);
    cout << "Username: "; getline(cin, e.username);
    cout << "Password: "; getline(cin, e.password);

    // Password strength check
    int score = passwordStrength(e.password);
    if(score < 3) {
        cout << "Warning: Password is weak!\n";
    }

    vault.push_back(e);
    cout << "Entry added successfully.\n";
}


// ================= VIEW ENTRIES =================
void viewEntries(vector<Entry>& vault) {
    if(vault.empty()) {
        cout << "Vault is empty.\n";
        return;
    }

    for(size_t i=0; i<vault.size(); i++) {
        cout << i+1 << ". " 
             << vault[i].site << " | " 
             << vault[i].username << " | " 
             << vault[i].password << "\n";
    }
}


// ================= SEARCH ENTRY =================
void searchEntry(vector<Entry>& vault) {
    string site;
    cout << "Enter site to search: ";
    getline(cin, site);

    bool found = false;
    for(auto &e : vault) {
        if(e.site == site) {
            cout << "Found: " << e.site << " | " << e.username << " | " << e.password << "\n";
            found = true;
            break;
        }
    }

    if(!found) cout << "No entry found for site: " << site << "\n";
}


// ================= DELETE ENTRY =================
void deleteEntry(vector<Entry>& vault) {
    string site;
    cout << "Enter site to delete: ";
    getline(cin, site);

    auto it = remove_if(vault.begin(), vault.end(), [&](Entry &e){
        return e.site == site;
    });

    if(it != vault.end()) {
        vault.erase(it, vault.end());
        cout << "Entry deleted.\n";
    } else {
        cout << "No entry found to delete.\n";
    }
}