RRJServer/DB_LMS/DataBaseLMS/databaselms.cpp

#include "databaselms.h"

#include <QtSql>
#include <QSqlDatabase>
#include <QMessageBox>

DataBaseLMS::DataBaseLMS():
    db(nullptr)
{
    createConnection();
}

DataBaseLMS::~DataBaseLMS()
{
    deleteConnection();
}

bool DataBaseLMS::createConnection()
{
    db = new QSqlDatabase(QSqlDatabase::addDatabase(dbType));
    db->setDatabaseName(dbName);
    db->setUserName(dbUserName);
    db->setPassword(dbPassword);
    if(!db->open())
    {
        QMessageBox::critical(nullptr, dbName, "Connection error: " + db->lastError().text());
        return false;
    }
    else
    {
        QMessageBox::information(nullptr, dbName, "Connection is successful!");
        return true;
    }
}

void DataBaseLMS::deleteConnection()
{
    if(db != nullptr)
    {
        if(db->isOpen())
            db->close();
        delete  db;
    }
}

QList<Instructor> DataBaseLMS::selectAllInstructors()
{
    QList<Instructor> listInstructors;

    QString queryStr = QString("SELECT instructor_id, name, login, password, is_admin, archived, logged_in "
                                                 "FROM public.instructors "
                                                 "ORDER BY instructor_id ASC");

    QSqlQuery query(*db);
    if(!query.exec(queryStr))
    {
        messageWarningErrorQuery(queryStr);
    }
    else
    {
        while (query.next())
        {//Инструктор
            Instructor instructor;
            instructor.setID(query.value(0).toInt());
            instructor.setName(query.value(1).toString());
            instructor.setLogin(query.value(2).toString());
            instructor.setPassword(query.value(3).toString());
            instructor.setIsAdmin(query.value(4).toBool());
            instructor.setArchived(query.value(5).toBool());
            instructor.setLoggedIn(query.value(6).toBool());

            listInstructors.append(instructor);
        }
    }

    return listInstructors;
}

QList<Trainee> DataBaseLMS::selectAllTrainees()
{
    QList<Trainee> listTrainees;

    QString queryStr = QString("SELECT trainees.trainee_id, trainees.name, trainees.login, trainees.password, trainees.archived, trainees.logged_in, "
                                                 "groups_of_trainees.name, "
                                                 "educational_classes.name, "
                                                 "computers.name, computers.ip_address "
                                                 "FROM public.trainees JOIN public.groups_of_trainees ON groups_of_trainees.group_id = trainees.group_trainees "
                                                 "LEFT OUTER JOIN public.computers ON computers.computer_id = trainees.computer "
                                                 "LEFT OUTER JOIN public.educational_classes ON educational_classes.class_id = computers.class "
                                                 "ORDER BY groups_of_trainees.name ASC");

    QSqlQuery query(*db);
    if(!query.exec(queryStr))
    {
        messageWarningErrorQuery(queryStr);
    }
    else
    {
        while (query.next())
        {//Обучаемый
            Trainee trainee;

            trainee.setID(query.value(0).toInt());
            trainee.setName(query.value(1).toString());
            trainee.setLogin(query.value(2).toString());
            trainee.setPassword(query.value(3).toString());
            trainee.setArchived(query.value(4).toBool());                      
            trainee.setLoggedIn(query.value(5).toBool());
            trainee.setGroup(query.value(6).toString());

            trainee.setLearnClass(query.value(7).toString());
            trainee.setComputer(query.value(8).toString());
            trainee.setIpAddress(query.value(9).toString());

            trainee.setTasks(selectTasksOfTrainee(trainee.getID()));

            listTrainees.append(trainee);
        }
    }

    return listTrainees;
}

QList<Group> DataBaseLMS::selectAllGroups()
{
    QList<Group> listGroups;

    QString queryStr = QString("SELECT group_id, name, color "
                                                 "FROM public.groups_of_trainees "
                                                 "ORDER BY group_id ASC");

    QSqlQuery query(*db);
    if(!query.exec(queryStr))
    {
        messageWarningErrorQuery(queryStr);
    }
    else
    {
        while (query.next())
        {//Группа
            Group group;
            group.setID(query.value(0).toInt());
            group.setName(query.value(1).toString());
            group.setColor((Group::ColorGroup)query.value(2).toInt());

            listGroups.append(group);
        }
    }

    return listGroups;
}

bool DataBaseLMS::insertInstructor(Instructor instructor)
{
    QString queryStr = QString("INSERT INTO public.instructors (name, login, password, is_admin, archived, logged_in) "
                                                 "VALUES ('%1', '%2', '%3', %4, %5, %6)").arg(
                                                        instructor.getName(),
                                                        instructor.getLogin(),
                                                        instructor.getPassword(),
                                                        instructor.getIsAdmin() ? "true" : "false",
                                                        instructor.getArchived() ? "true" : "false",
                                                        instructor.getLoggedIn() ? "true" : "false");

    return queryExec(queryStr);
}

bool DataBaseLMS::deleteInstructor(int instructor_id)
{
    QString queryStr = QString("DELETE FROM public.instructors WHERE instructor_id = %1").arg(QString::number(instructor_id));

    return queryExec(queryStr);
}

bool DataBaseLMS::updateInstructor(Instructor instructor)
{
    QString queryStr = QString("UPDATE public.instructors "
                                                 "SET name = '%1', login = '%2', password = '%3', is_admin = %4, archived = %5, logged_in = %6 "
                                                 "WHERE instructor_id = %7").arg(
                                                        instructor.getName(),
                                                        instructor.getLogin(),
                                                        instructor.getPassword(),
                                                        instructor.getIsAdmin() ? "true" : "false",
                                                        instructor.getArchived() ? "true" : "false",
                                                        instructor.getLoggedIn() ? "true" : "false",
                                                        QString::number(instructor.getID()) );

    return queryExec(queryStr);
}

bool DataBaseLMS::insertGroup(Group group)
{
    QString queryStr = QString("INSERT INTO public.groups_of_trainees (name, color) "
                                                 "VALUES ('%1', %2)").arg(
                                                        group.getName(),
                                                        QString::number((int)group.getColor()));

    return queryExec(queryStr);
}

bool DataBaseLMS::deleteGroup(int group_id)
{
    QString queryStr = QString("DELETE FROM public.groups_of_trainees WHERE group_id = %1").arg(QString::number(group_id));

    return queryExec(queryStr);
}

bool DataBaseLMS::updateGroup(Group group)
{
    QString queryStr = QString("UPDATE public.groups_of_trainees SET name = '%1', color = %2 WHERE group_id = %3").arg(
                                                        group.getName(),
                                                        QString::number((int)group.getColor()),
                                                        QString::number(group.getID()) );

    return queryExec(queryStr);
}

bool DataBaseLMS::insertTrainee(Trainee trainee)
{
    QString queryStr = QString("INSERT INTO public.trainees (name, login, password, archived, logged_in, group_trainees, computer) "
                                                 "VALUES ('%1', '%2', '%3', %4, %5, "
                                                 "(SELECT group_id FROM public.groups_of_trainees WHERE name = '%6'), "
                                                 "(SELECT computer_id FROM public.computers WHERE name = '%7') "
                                                 ")").arg(
                                                        trainee.getName(),
                                                        trainee.getLogin(),
                                                        trainee.getPassword(),
                                                        trainee.getArchived() ? "true" : "false",
                                                        trainee.getLoggedIn() ? "true" : "false",
                                                        trainee.getGroup(),
                                                        trainee.getComputer());

    return queryExec(queryStr);
}

bool DataBaseLMS::deleteTrainee(int trainee_id)
{
    QString queryStr = QString("DELETE FROM public.trainees WHERE trainee_id = %1").arg(QString::number(trainee_id));

    return queryExec(queryStr);
}

bool DataBaseLMS::updateTrainee(Trainee trainee)
{
    QString queryStr = QString("UPDATE public.trainees "
                                                 "SET name = '%1', login = '%2', password = '%3', archived = %4, logged_in = %5, "
                                                 "group_trainees = "
                                                 "(SELECT group_id FROM public.groups_of_trainees WHERE name = '%6'), "
                                                 "computer = "
                                                 "(SELECT computer_id FROM public.computers WHERE name = '%7') "
                                                 "WHERE trainee_id = %8").arg(
                                                        trainee.getName(),
                                                        trainee.getLogin(),
                                                        trainee.getPassword(),
                                                        trainee.getArchived() ? "true" : "false",
                                                        trainee.getLoggedIn() ? "true" : "false",
                                                        trainee.getGroup(),
                                                        trainee.getComputer(),
                                                        QString::number(trainee.getID()) );

    return queryExec(queryStr);
}

QStringList DataBaseLMS::selectTasksOfTrainee(int trainee_id)
{
    QStringList tasks;

    QString queryStr = QString("SELECT tasks.task_id, tasks.name "
                                "FROM public.trainees "
                                "JOIN public.trainee_tasks ON trainee_tasks.trainee_id = trainees.trainee_id "
                                "JOIN public.tasks ON tasks.task_id = trainee_tasks.task_id "
                                "WHERE trainees.trainee_id = %1 "
                                "ORDER BY tasks.name ASC").arg(
                                                    trainee_id);

    QSqlQuery query(*db);
    if(!query.exec(queryStr))
    {
        messageWarningErrorQuery(queryStr);
    }
    else
    {
        while (query.next())
        {//Задача
            tasks.append(query.value(1).toString());
        }
    }

    return tasks;
}

bool DataBaseLMS::queryExec(QString queryStr)
{
    QSqlQuery query(*db);
    if(!query.exec(queryStr))
    {
        messageWarningErrorQuery(queryStr);
        return false;
    }
    else
        return true;
}

void DataBaseLMS::messageWarningErrorQuery(QString queryStr)
{
    QMessageBox::warning(nullptr, dbName, "Error query: " + queryStr + "\n" + db->lastError().text());
}