icarus/src/database/ArtistRepository.cpp

#include "database/ArtistRepository.h"

#include <iostream>
#include <memory>
#include <sstream>
#include <string>
#include <cstring>

database::ArtistRepository::ArtistRepository(const model::BinaryPath& binConf)
    : BaseRepository(binConf)
{
}


model::Artist database::ArtistRepository::retrieveRecord(model::Artist& artist, type::ArtistFilter filter)
{
    std::cout << "retrieving artist record" << std::endl;
    std::stringstream qry;
    auto conn = setupMysqlConnection();
    auto stmt = mysql_stmt_init(conn);
    qry << "SELECT art.* FROM Artist art WHERE ";

    MYSQL_BIND params[1];
    memset(params, 0, sizeof(params));

    //std::unique_ptr<char*> param;
    auto artistLength = artist.artist.size();
    switch (filter) {
        case type::ArtistFilter::id:
            qry << "art.ArtistId = ?";

            params[0].buffer_type = MYSQL_TYPE_LONG;
            params[0].buffer = (char*)&artist.id;
            params[0].length = 0;
            params[0].is_null = 0;
            break;
        case type::ArtistFilter::artist:
            qry << "art.Artist = ?";
            //param = std::make_unique<char*>(new char[artist.artist.size()]);
            //mysql_real_escape_string(conn, *param, artist.artist.c_str(), artist.artist.size());
            params[0].buffer_type = MYSQL_TYPE_STRING;
            params[0].buffer = (char*)artist.artist.c_str();
            params[0].length = &artistLength;
            params[0].is_null = 0;
            break;
        default:
            break;
    }

    qry << " LIMIT 1";

    const auto query = qry.str();
    auto status = mysql_stmt_prepare(stmt, query.c_str(), query.size());
    status = mysql_stmt_bind_param(stmt, params);
    status = mysql_stmt_execute(stmt);

    artist = parseRecord(stmt);

    mysql_stmt_close(stmt);
    mysql_close(conn);

    std::cout << "retrieved record" << std::endl;

    return artist;
}

bool database::ArtistRepository::doesArtistExist(const model::Artist& artist, type::ArtistFilter filter)
{
    auto conn = setupMysqlConnection();
    auto stmt = mysql_stmt_init(conn);

    MYSQL_BIND params[1];
    memset(params, 0, sizeof(params));

    std::stringstream qry;
    qry << "SELECT * FROM Artist WHERE ";

    auto artistLength = artist.artist.size();
    switch (filter) {
        case type::ArtistFilter::id:
            qry << "ArtistId = ?";

            params[0].buffer_type = MYSQL_TYPE_LONG;
            params[0].buffer = (char*)&artist.id;
            params[0].length = 0;
            params[0].is_null = 0;
            break;
        case type::ArtistFilter::artist:
            qry << "Artist = ?";

            params[0].buffer_type = MYSQL_TYPE_STRING;
            params[0].buffer = (char*)artist.artist.c_str();
            params[0].length = &artistLength;
            params[0].is_null = 0;
            break;
        default:
            break;
    }

    qry << " LIMIT 1";

    const auto query = qry.str();
    auto status = mysql_stmt_prepare(stmt, query.c_str(), query.size());
    status = mysql_stmt_bind_param(stmt, params);
    status = mysql_stmt_execute(stmt);

    std::cout << "the query has been performed" << std::endl;

    mysql_stmt_store_result(stmt);
    auto rowCount = mysql_stmt_num_rows(stmt);

    mysql_stmt_close(stmt);
    mysql_close(conn);
    
    return (rowCount > 0) ? true : false;
}

void database::ArtistRepository::saveRecord(const model::Artist& artist)
{
    std::cout << "inserting artist record" << std::endl;

    auto conn = setupMysqlConnection();
    MYSQL_STMT *stmt = mysql_stmt_init(conn);

    const std::string query = "INSERT INTO Artist(Artist) VALUES(?)";

    auto status = mysql_stmt_prepare(stmt, query.c_str(), query.size());

    MYSQL_BIND params[1];
    memset(params, 0, sizeof(params));

    params[0].buffer_type = MYSQL_TYPE_STRING;
    params[0].buffer = (char*)artist.artist.c_str();
    auto artistLength = artist.artist.size();
    params[0].length = &artistLength;
    params[0].is_null = 0;

    status = mysql_stmt_bind_param(stmt, params);
    status = mysql_stmt_execute(stmt);

    mysql_stmt_close(stmt);
    mysql_close(conn);

    std::cout<< "inserted artist record" << std::endl;
}


model::Artist database::ArtistRepository::parseRecord(MYSQL_RES* results)
{
    std::cout << "parsing artist record" << std::endl;
    model::Artist artist;

    auto fieldNum = mysql_num_fields(results);
    auto row = mysql_fetch_row(results);

    for (auto i = 0; i != fieldNum; ++i) {
        const std::string field(mysql_fetch_field(results)->name);

        if (field.compare("ArtistId") == 0) {
            artist.id = std::stoi(row[i]);
        }
        if (field.compare("Artist") == 0) {
            artist.artist = row[i];
        }
    }

    std::cout << "parsed artist record" << std::endl;

    return artist;
}

model::Artist database::ArtistRepository::parseRecord(MYSQL_STMT *stmt)
{
    std::cout << "parsing artist record" << std::endl;
    mysql_stmt_store_result(stmt);

    model::Artist art;
    auto status = 0;
    auto time = 0;
    auto valAmt = 2;
    unsigned long len[valAmt];
    my_bool nullRes[valAmt];

    while (status == 0) {
        if (mysql_stmt_field_count(stmt) > 0) {
            auto res = mysql_stmt_result_metadata(stmt);
            auto fields = mysql_fetch_fields(res);
            auto strLen = 1024;

            MYSQL_BIND val[valAmt];
            memset(val, 0, sizeof(val));

            char artist[strLen];

            val[0].buffer_type = MYSQL_TYPE_LONG;
            val[0].buffer = (char*)&art.id;
            val[0].length = &len[0];
            val[0].is_null = &nullRes[0];

            val[1].buffer_type = MYSQL_TYPE_STRING;
            val[1].buffer = (char*)&artist;
            val[1].buffer_length = strLen;
            val[1].length = &len[1];
            val[1].is_null = &nullRes[1];

            status = mysql_stmt_bind_result(stmt, val);
            mysql_stmt_store_result(stmt);

            status = mysql_stmt_fetch(stmt);

            art.artist = artist;
        }

        status = mysql_stmt_next_result(stmt);
    }

    std::cout << "done parsing artist record" << std::endl;

    return art;
}