interactive_cattle.cpp

//NAME: Ignacio Jimenez
//RED ID: 826019175
//Interactive Cattle Management System
#include "cattle_tree.h"
#include <iostream>
#include <string>
#include <fstream>
#include <sstream>

using namespace std;

void displayMenu() {
    cout << "\n=== CATTLE MANAGEMENT SYSTEM ===" << endl;
    cout << "1. Add Cattle" << endl;
    cout << "2. Query Cattle" << endl;
    cout << "3. Count Cattle" << endl;
    cout << "4. Delete Cattle" << endl;
    cout << "5. Add Notes to Cattle" << endl;
    cout << "6. View Cattle Notes" << endl;
    cout << "7. Load from File" << endl;
    cout << "8. Save to File" << endl;
    cout << "9. Exit" << endl;
    cout << "Choose an option (1-9): ";
}

void addCattle(CattleNode& root) {
    string cattlePath;
    cout << "\nEnter cattle path (e.g., s_0_2_1_0_5): ";
    cin >> cattlePath;
    
    if (root.addCattle(cattlePath.c_str())) {
        cout << "✓ Cattle added successfully!" << endl;
        
        string annotation;
        cout << "Add notes about this cattle (press Enter to skip): ";
        cin.ignore();
        getline(cin, annotation);
        
        if (!annotation.empty()) {
            root.setAnnotation(cattlePath.c_str(), annotation.c_str());
            cout << "✓ Notes added successfully!" << endl;
        }
    } else {
        cout << "✗ Failed to add cattle. Check path format." << endl;
    }
}

void queryCattle(CattleNode& root) {
    string cattlePath;
    cout << "\nEnter cattle path to query (e.g., s_0_2_1_0_5): ";
    cin >> cattlePath;
    
    int count = root.findCattle(cattlePath.c_str());
    string annotation = root.getAnnotation(cattlePath.c_str());
    
    cout << "\nQuery Result:" << endl;
    cout << cattlePath << " " << count;
    if (!annotation.empty()) {
        cout << " [" << annotation << "]";
    }
    cout << endl;
}

void countCattle(CattleNode& root) {
    string cattlePath;
    cout << "\nEnter cattle path to count (e.g., s_0 for all Yacumeño): ";
    cin >> cattlePath;
    
    int count = root.findCattle(cattlePath.c_str());
    cout << "\nCount: " << count << " cattle" << endl;
}

void deleteCattle(CattleNode& root) {
    string cattlePath;
    cout << "\nEnter cattle path to delete (e.g., s_0_2_1_0_5): ";
    cin >> cattlePath;
    
    if (root.removeCattle(cattlePath.c_str())) {
        cout << "✓ Cattle deleted successfully!" << endl;
    } else {
        cout << "✗ Failed to delete cattle. (May not exist or has children)" << endl;
    }
}

void addNotes(CattleNode& root) {
    string cattlePath;
    cout << "\nEnter cattle path to add notes (e.g., s_0_2_1_0_5): ";
    cin >> cattlePath;
    
    string annotation;
    cout << "Enter notes: ";
    cin.ignore();
    getline(cin, annotation);
    
    if (root.setAnnotation(cattlePath.c_str(), annotation.c_str())) {
        cout << "✓ Notes added successfully!" << endl;
    } else {
        cout << "✗ Failed to add notes. Cattle may not exist." << endl;
    }
}

void viewNotes(CattleNode& root) {
    string cattlePath;
    cout << "\nEnter cattle path to view notes (e.g., s_0_2_1_0_5): ";
    cin >> cattlePath;
    
    string annotation = root.getAnnotation(cattlePath.c_str());
    
    if (!annotation.empty()) {
        cout << "\nNotes for " << cattlePath << ": " << annotation << endl;
    } else {
        cout << "\nNo notes found for " << cattlePath << endl;
    }
}

void loadFromFile(CattleNode& root, vector<int>& maxCattlePerLevel) {
    string filename;
    cout << "\nEnter filename to load (e.g., skuchart.txt): ";
    cin >> filename;
    
    ifstream file(filename);
    if (!file) {
        cout << "✗ Unable to open " << filename << endl;
        return;
    }
    
    // Read internal levels
    string line;
    getline(file, line);
    int internalLevels = stoi(line.substr(line.find('=') + 1));
    
    // Read max cattle per level
    getline(file, line);
    stringstream ss(line);
    maxCattlePerLevel.resize(internalLevels);
    for(int i = 0; i < internalLevels; i++){
        ss >> maxCattlePerLevel[i];
    }
    
    // Read cattle paths and annotations
    int count = 0;
    while(getline(file, line) && !line.empty()){
        // Check if line has annotation (contains [ and ])
        size_t startBracket = line.find('[');
        size_t endBracket = line.find(']');
        
        string cattlePath = line;
        string annotation = "";
        
        if (startBracket != string::npos && endBracket != string::npos && endBracket > startBracket) {
            // Extract cattle path (everything before [)
            cattlePath = line.substr(0, startBracket);
            // Remove trailing spaces
            while (!cattlePath.empty() && cattlePath.back() == ' ') {
                cattlePath.pop_back();
            }
            // Extract annotation (everything between [ and ])
            annotation = line.substr(startBracket + 1, endBracket - startBracket - 1);
        }
        
        if (root.addCattle(cattlePath.c_str())) {
            count++;
            // Add annotation if present
            if (!annotation.empty()) {
                root.setAnnotation(cattlePath.c_str(), annotation.c_str());
            }
        }
    }
    
    file.close();
    cout << "✓ Loaded " << count << " cattle from " << filename << endl;
}

void saveToFile(CattleNode& root, const vector<int>& maxCattlePerLevel) {
    string filename;
    cout << "\nEnter filename to save (e.g., my_cattle.txt): ";
    cin >> filename;
    
    ofstream file(filename);
    if (!file) {
        cout << "✗ Unable to create " << filename << endl;
        return;
    }
    
    // Write header
    file << "internalLevels=" << maxCattlePerLevel.size() << endl;
    for(int i = 0; i < maxCattlePerLevel.size(); i++){
        file << maxCattlePerLevel[i];
        if(i < maxCattlePerLevel.size() - 1) file << " ";
    }
    file << endl;
    
    // Note: This is a simplified save - in a real system you'd need to traverse the tree
    // and save all cattle paths with their annotations. For now, this creates a template file.
    file << "# Add your cattle paths here, one per line" << endl;
    file << "# Format: s_0_2_1_0_5 [Optional annotation]" << endl;
    file << "# Example: s_0_2_1_0_5 [Pregnant - Due March 2025]" << endl;
    file << "# Example: s_1_0_0_1_3 [Vaccinated - Last shot Jan 2025]" << endl;
    
    file.close();
    cout << "✓ Template saved to " << filename << endl;
    cout << "  (Note: You'll need to add cattle paths manually)" << endl;
}

int main() {
    cout << "Welcome to the Interactive Cattle Management System!" << endl;
    cout << "Initializing with default settings..." << endl;
    
    // Initialize with default settings
    vector<int> maxCattlePerLevel = {8, 6, 9, 5, 7}; // 8 breeds, 6 age groups, etc.
    CattleNode root("s", 0, maxCattlePerLevel);
    
    int choice;
    do {
        displayMenu();
        cin >> choice;
        
        switch(choice) {
            case 1: addCattle(root); break;
            case 2: queryCattle(root); break;
            case 3: countCattle(root); break;
            case 4: deleteCattle(root); break;
            case 5: addNotes(root); break;
            case 6: viewNotes(root); break;
            case 7: loadFromFile(root, maxCattlePerLevel); break;
            case 8: saveToFile(root, maxCattlePerLevel); break;
            case 9: cout << "Goodbye!" << endl; break;
            default: cout << "Invalid option. Please try again." << endl;
        }
    } while(choice != 9);
    
    return 0;
}