#ifndef DATABASETRAINEES_H
#define DATABASETRAINEES_H

#include "instructorsAndTrainees_global.h"
#include "trainee.h"
#include "group.h"

#include "databaselms.h"

#include <QList>
#include <QColor>
#include <QObject>

class DataBaseTrainees : QObject
{
    Q_OBJECT

public:
    DataBaseTrainees(DataBaseLMS* dbLMS);
    ~DataBaseTrainees();

    void LoadTraineesGroupsPSQL();

    bool AuthorizationTrainee(QString login, QString password, QString classroom_name, QString computer_name);
    bool deAuthorizationTrainee(QString login);

    //void setTasks(QString login, QStringList tasks);
    QStringList getTasks(QString login);

    QString getNameTraineeOnComputer(QString computer_name);
    Trainee getTraineeOnComputer(QString computer_name);

    QString getNameTraineeByLogin(QString login);
    //QColor getColorGroupByLogin(QString login);

    QList<Trainee> getListTraineesInGroup(int id);
    QList<Group> getListGroups();

    Trainee getTrainee(int id);
    Group getGroup(int id);

    int newGroup();
    int deleteGroup(int id);
    int editGroup(Group group);

    int newTrainee(int id_group);
    int deleteTrainee(int id);
    int editTrainee(Trainee trainee);

    bool isArchived(int id);

private:
    QString generateDefaultNameGroup();
    QString generateDefaultNameTrainee();
    bool checkExistNameGroup(QString name);
    bool checkExistNameTrainee(QString name);

    QString generateDefaultLoginTrainee();
    bool checkExistLoginTrainee(QString login);

private:
    QList<Trainee> listOfTrainees;
    QList<Group> listOfGroups;

    QList<QColor> listOfColorGroup;

    DataBaseLMS* dbLMS;
};

#endif // DATABASETRAINEES_H