cattle_manager.cpp

//NAME: Ignacio Jimenez
//RED ID: 826019175
//Simple Cattle Manager for Farmers
#include "cattle_tree.h"
#include <iostream>
#include <string>
#include <fstream>
#include <sstream>

using namespace std;

void showMenu() {
    cout << "\n🐄 CATTLE MANAGER 🐄" << endl;
    cout << "===================" << endl;
    cout << "1. Add Cow" << endl;
    cout << "2. Find Cow" << endl;
    cout << "3. Count Cows" << endl;
    cout << "4. View All Cattle" << endl;
    cout << "5. Remove Cow" << endl;
    cout << "6. Add Note" << endl;
    cout << "7. See Note" << endl;
    cout << "8. Load Data" << endl;
    cout << "9. Save Data" << endl;
    cout << "10. Exit" << endl;
    cout << "\nWhat do you want to do? (1-10): ";
}

void addCow(CattleNode& root) {
    cout << "\n🐮 ADD NEW COW" << endl;
    cout << "==============" << endl;
    
    cout << "Breed (0=Yacumeño, 1=Chaqueño, 2=Chusco, 3=Saavedreño, 4=Holstein, 5=Brahman, 6=Angus, 7=Hereford): ";
    int breed;
    cin >> breed;
    
    cout << "Age (0=Calf, 1=Young, 2=Mature): ";
    int age;
    cin >> age;
    
    cout << "Health (0=Excellent, 1=Good, 2=Fair, 3=Sick, 4=Quarantine, 5=Pregnant): ";
    int health;
    cin >> health;
    
    cout << "Purpose (0=Breeding, 1=Dairy, 2=Meat, 3=Replacement, 4=Cull, 5=Show): ";
    int purpose;
    cin >> purpose;
    
    cout << "Cow Number (0-9): ";
    int cowNum;
    cin >> cowNum;
    
    // Build the cattle path
    string cattlePath = "s_" + to_string(breed) + "_" + to_string(age) + "_" + 
                       to_string(health) + "_" + to_string(purpose) + "_" + to_string(cowNum);
    
    if (root.addCattle(cattlePath.c_str())) {
        cout << "✅ Cow added successfully!" << endl;
        cout << "Cow ID: " << cattlePath << endl;
        
        cout << "\nAdd a note about this cow (or press Enter to skip): ";
        cin.ignore();
        string note;
        getline(cin, note);
        
        if (!note.empty()) {
            root.setAnnotation(cattlePath.c_str(), note.c_str());
            cout << "✅ Note added!" << endl;
        }
    } else {
        cout << "❌ Failed to add cow. Please check your numbers." << endl;
    }
}

void findCow(CattleNode& root) {
    cout << "\n🔍 FIND COW" << endl;
    cout << "===========" << endl;
    
    cout << "Enter cow ID (like s_0_2_1_0_5): ";
    string cowId;
    cin >> cowId;
    
    int count = root.findCattle(cowId.c_str());
    string note = root.getAnnotation(cowId.c_str());
    
    if (count > 0) {
        cout << "✅ Found your cow!" << endl;
        cout << "Cow: " << cowId << " (Count: " << count << ")" << endl;
        if (!note.empty()) {
            cout << "Note: " << note << endl;
        }
    } else {
        cout << "❌ Cow not found. Check the ID." << endl;
    }
}

void countCows(CattleNode& root) {
    cout << "\n📊 COUNT COWS" << endl;
    cout << "=============" << endl;
    
    cout << "What do you want to count?" << endl;
    cout << "1. All cows of a breed" << endl;
    cout << "2. All cows of an age group" << endl;
    cout << "3. All cows with same health" << endl;
    cout << "4. All cows with same purpose" << endl;
    cout << "5. Custom search" << endl;
    cout << "Choose (1-5): ";
    
    int choice;
    cin >> choice;
    
    string searchPath;
    
    switch(choice) {
        case 1: {
            cout << "Breed (0-7): ";
            int breed;
            cin >> breed;
            searchPath = "s_" + to_string(breed);
            break;
        }
        case 2: {
            cout << "Breed (0-7): ";
            int breed;
            cin >> breed;
            cout << "Age (0-2): ";
            int age;
            cin >> age;
            searchPath = "s_" + to_string(breed) + "_" + to_string(age);
            break;
        }
        case 3: {
            cout << "Breed (0-7): ";
            int breed;
            cin >> breed;
            cout << "Age (0-2): ";
            int age;
            cin >> age;
            cout << "Health (0-5): ";
            int health;
            cin >> health;
            searchPath = "s_" + to_string(breed) + "_" + to_string(age) + "_" + to_string(health);
            break;
        }
        case 4: {
            cout << "Breed (0-7): ";
            int breed;
            cin >> breed;
            cout << "Age (0-2): ";
            int age;
            cin >> age;
            cout << "Health (0-5): ";
            int health;
            cin >> health;
            cout << "Purpose (0-5): ";
            int purpose;
            cin >> purpose;
            searchPath = "s_" + to_string(breed) + "_" + to_string(age) + "_" + to_string(health) + "_" + to_string(purpose);
            break;
        }
        case 5: {
            cout << "Enter search path (like s_0 for all Yacumeño): ";
            cin >> searchPath;
            break;
        }
        default:
            cout << "Invalid choice." << endl;
            return;
    }
    
    int count = root.findCattle(searchPath.c_str());
    cout << "\n📈 RESULT: " << count << " cows found" << endl;
}

void removeCow(CattleNode& root) {
    cout << "\n🗑️ REMOVE COW" << endl;
    cout << "=============" << endl;
    
    cout << "Enter cow ID to remove (like s_0_2_1_0_5): ";
    string cowId;
    cin >> cowId;
    
    if (root.removeCattle(cowId.c_str())) {
        cout << "✅ Cow removed successfully!" << endl;
    } else {
        cout << "❌ Could not remove cow. It might not exist or have children." << endl;
    }
}

void addNote(CattleNode& root) {
    cout << "\n📝 ADD NOTE" << endl;
    cout << "===========" << endl;
    
    cout << "Enter cow ID (like s_0_2_1_0_5): ";
    string cowId;
    cin >> cowId;
    
    cout << "Enter your note: ";
    cin.ignore();
    string note;
    getline(cin, note);
    
    if (root.setAnnotation(cowId.c_str(), note.c_str())) {
        cout << "✅ Note added successfully!" << endl;
    } else {
        cout << "❌ Could not add note. Cow might not exist." << endl;
    }
}

void seeNote(CattleNode& root) {
    cout << "\n👀 SEE NOTE" << endl;
    cout << "===========" << endl;
    
    cout << "Enter cow ID (like s_0_2_1_0_5): ";
    string cowId;
    cin >> cowId;
    
    string note = root.getAnnotation(cowId.c_str());
    
    if (!note.empty()) {
        cout << "📄 Note for " << cowId << ": " << note << endl;
    } else {
        cout << "❌ No note found for this cow." << endl;
    }
}

void viewAllCattle(CattleNode& root) {
    cout << "\n📋 ALL YOUR CATTLE" << endl;
    cout << "=================" << endl;
    
    // Show counts by breed
    cout << "🐄 CATTLE SUMMARY:" << endl;
    cout << "==================" << endl;
    
    string breeds[] = {"Yacumeño", "Chaqueño", "Chusco", "Saavedreño", "Holstein", "Brahman", "Angus", "Hereford"};
    
    for(int i = 0; i < 8; i++) {
        string breedPath = "s_" + to_string(i);
        int count = root.findCattle(breedPath.c_str());
        if(count > 0) {
            cout << breeds[i] << ": " << count << " cows" << endl;
        }
    }
    
    cout << "\n💡 TIP: Use option 2 (Find Cow) to see details about specific cows." << endl;
    cout << "   Use option 3 (Count Cows) for more detailed counts." << endl;
}

void loadData(CattleNode& root, vector<int>& maxCattlePerLevel) {
    cout << "\n📂 LOAD DATA" << endl;
    cout << "============" << endl;
    
    cout << "Enter filename (like my_cattle.txt): ";
    string filename;
    cin >> filename;
    
    ifstream file(filename);
    if (!file) {
        cout << "❌ Could not open file: " << filename << endl;
        return;
    }
    
    // Read header
    string line;
    getline(file, line);
    int internalLevels = stoi(line.substr(line.find('=') + 1));
    
    getline(file, line);
    stringstream ss(line);
    maxCattlePerLevel.resize(internalLevels);
    for(int i = 0; i < internalLevels; i++){
        ss >> maxCattlePerLevel[i];
    }
    
    // Read cattle
    int count = 0;
    while(getline(file, line) && !line.empty()){
        size_t startBracket = line.find('[');
        size_t endBracket = line.find(']');
        
        string cattlePath = line;
        string annotation = "";
        
        if (startBracket != string::npos && endBracket != string::npos && endBracket > startBracket) {
            cattlePath = line.substr(0, startBracket);
            while (!cattlePath.empty() && cattlePath.back() == ' ') {
                cattlePath.pop_back();
            }
            annotation = line.substr(startBracket + 1, endBracket - startBracket - 1);
        }
        
        if (root.addCattle(cattlePath.c_str())) {
            count++;
            if (!annotation.empty()) {
                root.setAnnotation(cattlePath.c_str(), annotation.c_str());
            }
        }
    }
    
    file.close();
    cout << "✅ Loaded " << count << " cows from " << filename << endl;
    
    // Automatically show the cattle summary
    cout << "\n📋 YOUR CATTLE:" << endl;
    cout << "===============" << endl;
    
    string breeds[] = {"Yacumeño", "Chaqueño", "Chusco", "Saavedreño", "Holstein", "Brahman", "Angus", "Hereford"};
    
    for(int i = 0; i < 8; i++) {
        string breedPath = "s_" + to_string(i);
        int breedCount = root.findCattle(breedPath.c_str());
        if(breedCount > 0) {
            cout << breeds[i] << ": " << breedCount << " cows" << endl;
        }
    }
}

void saveData(CattleNode& root, const vector<int>& maxCattlePerLevel) {
    cout << "\n💾 SAVE DATA" << endl;
    cout << "============" << endl;
    
    cout << "Enter filename (like my_cattle.txt): ";
    string filename;
    cin >> filename;
    
    ofstream file(filename);
    if (!file) {
        cout << "❌ Could not create file: " << 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 creates a template - in a real system you'd traverse the tree
    file << "# Add your cattle here, one per line" << endl;
    file << "# Format: s_breed_age_health_purpose_number [note]" << endl;
    file << "# Example: s_0_2_1_0_5 [Pregnant - Due March 2025]" << endl;
    
    file.close();
    cout << "✅ Template saved to " << filename << endl;
    cout << "   (Add your cattle data to this file)" << endl;
}

int main() {
    cout << "🐄 Welcome to Cattle Manager! 🐄" << endl;
    cout << "================================" << endl;
    cout << "A simple way to track your cattle." << endl;
    cout << "Just follow the menu and type numbers!" << endl;
    
    // Initialize with default settings
    vector<int> maxCattlePerLevel = {8, 6, 9, 5, 7};
    CattleNode root("s", 0, maxCattlePerLevel);
    
    // Try to load current cattle data automatically
    cout << "\n🔄 Loading current cattle data..." << endl;
    ifstream file("current_cattle.txt");
    if (file) {
        // Read header
        string line;
        getline(file, line);
        int internalLevels = stoi(line.substr(line.find('=') + 1));
        
        getline(file, line);
        stringstream ss(line);
        maxCattlePerLevel.resize(internalLevels);
        for(int i = 0; i < internalLevels; i++){
            ss >> maxCattlePerLevel[i];
        }
        
        // Read cattle
        int count = 0;
        while(getline(file, line) && !line.empty()){
            size_t startBracket = line.find('[');
            size_t endBracket = line.find(']');
            
            string cattlePath = line;
            string annotation = "";
            
            if (startBracket != string::npos && endBracket != string::npos && endBracket > startBracket) {
                cattlePath = line.substr(0, startBracket);
                while (!cattlePath.empty() && cattlePath.back() == ' ') {
                    cattlePath.pop_back();
                }
                annotation = line.substr(startBracket + 1, endBracket - startBracket - 1);
            }
            
            if (root.addCattle(cattlePath.c_str())) {
                count++;
                if (!annotation.empty()) {
                    root.setAnnotation(cattlePath.c_str(), annotation.c_str());
                }
            }
        }
        file.close();
        cout << "✅ Loaded " << count << " cows from current_cattle.txt" << endl;
    } else {
        cout << "ℹ️  No sample data found. Use option 8 to load your own cattle data." << endl;
    }
    
    int choice;
    do {
        showMenu();
        cin >> choice;
        
        switch(choice) {
            case 1: addCow(root); break;
            case 2: findCow(root); break;
            case 3: countCows(root); break;
            case 4: viewAllCattle(root); break;
            case 5: removeCow(root); break;
            case 6: addNote(root); break;
            case 7: seeNote(root); break;
            case 8: loadData(root, maxCattlePerLevel); break;
            case 9: saveData(root, maxCattlePerLevel); break;
            case 10: cout << "\n👋 Thanks for using Cattle Manager! Goodbye!" << endl; break;
            default: cout << "❌ Please choose 1-10." << endl;
        }
    } while(choice != 10);
    
    return 0;
}