core-legacy/models/relay_dbo.cc

#include <cstdio>
#include <cstring>
#include <trantor/utils/Logger.h>
#include <helpers.h>
#include "relay_dbo.h"
#include "globals.h"
#include "controller_dbo.h"
#include "schedule_dbo.h"

static bool relay_db_update_insert(relay_dbo *relay, sqlite3_stmt *stmt)
{
    int rc;

    sqlite3_bind_int(stmt, 1, relay->id);
    sqlite3_bind_int(stmt, 2, relay->number);
    sqlite3_bind_text(stmt, 3, relay->name, -1, SQLITE_STATIC);
    sqlite3_bind_text(stmt, 4, relay->active_schedule_id, -1, SQLITE_STATIC);
    sqlite3_bind_text(stmt, 5, relay->controller_id, -1, SQLITE_STATIC);
    sqlite3_bind_text(stmt, 6, relay->tag, -1, SQLITE_STATIC);

    rc = sqlite3_step(stmt);
    if (rc != SQLITE_DONE)
    {
        printf("ERROR inserting data: %s\n", sqlite3_errmsg(globals::db));
        return false;
    }

    sqlite3_finalize(stmt);

    return true;
}

static relay_dbo*
relay_db_select_mapper(sqlite3_stmt *stmt)
{
    auto *new_relay = new relay_dbo();
    const char* new_tag;
    for(int i = 0; i < sqlite3_column_count(stmt); i++)
    {
        const char *name = sqlite3_column_name(stmt, i);
        switch(name[0])
        {
            case 'a': // active_schedule_id
                strncpy(new_relay->active_schedule_id, (const char*)sqlite3_column_text(stmt, i), 32);
                break;
            case 'c': // controller_id
                strncpy(new_relay->controller_id, (const char*)sqlite3_column_text(stmt, i), 32);
                break;
            case 'i':
                new_relay->id = sqlite3_column_int(stmt, i);
                break;
            case 'n':
                switch(name[1])
                {
                    case 'a': // name
                        strncpy(new_relay->name, (const char*)sqlite3_column_text(stmt, i), 127);
                        break;
                    case 'u': // number
                        new_relay->number = sqlite3_column_int(stmt, i);
                        break;
                    default:
                        break;
                }
                break;
            case 't': // tag
                new_tag = (const char*)sqlite3_column_text(stmt, i);
                new_relay->tag[0] = '\0';
                if(new_tag)
                {
                    strncpy(new_relay->tag, (const char*)sqlite3_column_text(stmt, i), 63);
                }
                break;
            default: // ignore columns not implemented
                break;
        }
    }
    return new_relay;
}

static relay_dbo**
relay_db_select(sqlite3_stmt *stmt)
{
    auto **all_relays = (relay_dbo**)malloc(sizeof(relay_dbo*));

    int row = 0;

    while(true)
    {
        int s;

        s = sqlite3_step(stmt);
        if (s == SQLITE_ROW)
        {
            relay_dbo *new_relay = relay_db_select_mapper(stmt);

            new_relay->reload_active_schedule();

            row++;

            all_relays = (relay_dbo**)realloc(all_relays, sizeof(relay_dbo*) * (row + 1));
            all_relays[row - 1] = new_relay;
        }
        else
        {
            if (s == SQLITE_DONE)
            {
                break;
            }
            else
            {
                LOG_ERROR << "Error Selecting relays from database: " << sqlite3_errstr(s);
                break;
            }
        }
    }

    sqlite3_finalize(stmt);
    all_relays[row] = nullptr;

    return all_relays;
}

void
relay_dbo::reload_active_schedule()
{
    schedule_dbo **schedules = schedule_dbo::get_by_simple("id", this->active_schedule_id, (intptr_t)&sqlite3_bind_text);

    if(!schedules[0])
    {
        free(schedules);
        schedules = schedule_dbo::get_by_simple("id", "off", (intptr_t)&sqlite3_bind_text);
        strcpy(this->active_schedule_id, "off");
    }

    this->active_schedule = schedules[0];

    free(schedules);
}

bool
relay_dbo::update()
{
    sqlite3_stmt *stmt;

    sqlite3_prepare_v2(globals::db, "UPDATE relays set number = ?2, name = ?3, active_schedule_id = ?4, controller_id = ?5, tag = ?6 WHERE id = ?1;", -1, &stmt, nullptr);

    return relay_db_update_insert(this, stmt);
}

bool
relay_dbo::insert()
{
    sqlite3_stmt *stmt;

    sqlite3_prepare_v2(globals::db, "INSERT INTO relays(number, name, active_schedule_id, controller_id, tag) values (?2, ?3, ?4, ?5, ?6);", -1, &stmt, nullptr);

    return relay_db_update_insert(this, stmt);
}

bool
relay_dbo::remove()
{
    sqlite3_stmt *stmt;
    int rc;

    sqlite3_prepare_v2(globals::db, "DELETE FROM relays WHERE id=?1;", -1, &stmt, nullptr);
    sqlite3_bind_int(stmt, 1, this->id);
    rc = sqlite3_step(stmt);
    sqlite3_finalize(stmt);

    return rc == SQLITE_DONE;

}

Json::Value
relay_dbo::to_json()
{
    Json::Value relay_json;
    relay_json["name"] = this->name;
    relay_json["number"] = this->number;
    relay_json["active_schedule_id"] = this->active_schedule_id;
    relay_json["controller_id"] = this->controller_id;
    relay_json["active_schedule"] = this->active_schedule->to_json();
    relay_json["tag"] = this->tag;

    return relay_json;
}

relay_dbo**
relay_dbo::get_all()
{
    sqlite3_stmt *stmt;

    sqlite3_prepare_v2(globals::db, "SELECT * FROM relays;", -1, &stmt, nullptr);

    return relay_db_select(stmt);
}

relay_dbo**
relay_dbo::get_by_simple(const char *key, const void *value, intptr_t bind_func)
{
    helpers::sql_filter_builder *filters[1];
    helpers::sql_filter_builder filter
    {
        key,
        value,
        bind_func,
        ";"
    };
    filters[0] = &filter;
    sqlite3_stmt *stmt = helpers::create_sql_filtered_query("SELECT * FROM relays WHERE", filters);

    return relay_db_select(stmt);
}

relay_dbo**
relay_dbo::get_by(helpers::sql_filter_builder **filters)
{
    sqlite3_stmt *stmt = helpers::create_sql_filtered_query("SELECT * FROM relays WHERE", filters);
    return relay_db_select(stmt);
}

relay_dbo*
relay_dbo::get_relay_for_controller(const char *controller_id, int relay_num)
{
    helpers::sql_filter_builder *filters[2];
    helpers::sql_filter_builder filter(
            "number",
            (void*)(intptr_t)relay_num,
            (intptr_t)&sqlite3_bind_int,
            "AND"
    );
    helpers::sql_filter_builder filter2(
            "controller_id",
            (void*)controller_id,
            (intptr_t)sqlite3_bind_text,
            ";"
    );
    filters[0] = &filter;
    filters[1] = &filter2;

    auto relays = relay_dbo::get_by(filters);

    relay_dbo *relay = relays[0];
    free(relays);
    return relay;
}

bool
relay_dbo::valid_num_for_controller(const char *search_controller_id, int relay_num)
{
    controller_dbo **controllers = controller_dbo::get_by_simple("id", search_controller_id, (intptr_t)&sqlite3_bind_text);

    bool valid_id_and_num = controllers[0] && controllers[0]->relay_count > relay_num;
    controller_dbo::free_list(controllers);

    return valid_id_and_num;
}

void
relay_dbo::free_list(relay_dbo **relays_list)
{
    for(int i = 0; relays_list[i] != nullptr; i++)
    {
        delete relays_list[i];
    }
    free(relays_list);
}