#include <iostream>
#include <vector>
#include "user.h"
#include "auth.h"
#include "db.h"

int main(){
    // Liste des étudiants
    std::vector<Student> students = {
        {"student1", {15,12,18}},
        {"student2", {14,10,16}},
        {"student3", {13,11,17}}
    };

    // Liste des utilisateurs
    std::vector<User> users = {
        {"admin","1234",ADMIN},
        {"prof1","abcd",PROF},
        {"student1","pass",STUDENT},
        {"student2","123",STUDENT},
        {"student3","abc",STUDENT}
    };

    // Login
    std::string login, password;
    std::cout << "Login : "; std::cin >> login;
    std::cout << "Mot de passe : "; std::cin >> password;

    int idx = authentifier(login,password,users);
    if(idx == -1){
        std::cout << "Identifiants incorrects.\n";
        return 1;
    }

    User currentUser = users[idx];
    std::string roleStr = (currentUser.role == ADMIN ? "ADMIN" : currentUser.role == PROF ? "PROF" : "STUDENT");

    // Vues selon rôle
    if(currentUser.role == ADMIN){
        std::cout << "--- ADMIN ---\n";
        for(const auto& s : students){
            std::cout << s.nom << " : ";
            for(int n : s.notes) std::cout << n << " ";
            std::cout << "\n";
        }
    } else if(currentUser.role == PROF){
        std::cout << "--- PROF ---\n";
        for(int i=0;i<students.size();i++){
            std::cout << i << ". " << students[i].nom << " : ";
            for(int n : students[i].notes) std::cout << n << " ";
            std::cout << "\n";
        }
        int id,noteIdx,newNote;
        std::cout << "ID étudiant à modifier (-1 pour quitter) : ";
        std::cin >> id;
        if(id>=0){
            std::cout << "Note (0-2) : "; std::cin >> noteIdx;
            std::cout << "Nouvelle note : "; std::cin >> newNote;
            students[id].notes[noteIdx] = newNote;
        }
    } else { // STUDENT
        std::cout << "--- STUDENT ---\n";
        for(const auto& s : students){
            if(s.nom == currentUser.login){
                std::cout << s.nom << " : ";
                for(int n : s.notes) std::cout << n << " ";
                std::cout << "\n";
            }
        }
    }

    // Export / Import
    exportData("data.txt", students, roleStr, currentUser.login);
    importData("data.txt", students, roleStr);

    return 0;
}