/*
 * @file ContactRecord.cpp
 * @author Mateusz Piesta (mateusz.piesta@mudita.com)
 * @date 29.05.19
 * @brief
 * @copyright Copyright (C) 2019 mudita.com
 * @details
 */

#include "ContactRecord.hpp"
#include <Utils.hpp>
#include <log/log.hpp>

ContactRecordInterface::ContactRecordInterface(ContactsDB *db) : contactDB(db)
{}

ContactRecordInterface::~ContactRecordInterface()
{}

bool ContactRecordInterface::Add(const ContactRecord &rec)
{

    bool ret = contactDB->contacts.Add(ContactsTableRow{.type           = rec.contactType,
                                                        .isOnWhitelist  = rec.isOnBlacklist,
                                                        .isOnBlacklist  = rec.isOnBlacklist,
                                                        .isOnFavourites = rec.isOnFavourites,
                                                        .speedDial      = rec.speeddial,
                                                        .namePrimary    = rec.primaryName});

    if (!ret) {
        return ret;
    }

    uint32_t contactID = contactDB->GetLastInsertRowID();
    LOG_DEBUG("New contact with ID %u created", contactID);

    ret = contactDB->name.Add(ContactsNameTableRow{.contactID       = contactID,
                                                   .namePrimary     = rec.primaryName,
                                                   .nameAlternative = rec.alternativeName,
                                                   .favourite       = rec.isOnFavourites});

    if (!ret) {
        return ret;
    }

    auto contactNameID = contactDB->GetLastInsertRowID();

    for (auto a : rec.numbers) {
        ret = contactDB->number.Add(ContactsNumberTableRow{.contactID  = contactID,
                                                           .numberUser = a.numberUser.c_str(),
                                                           .numbere164 = a.numberE164.c_str(),
                                                           .type       = a.numberType});
    }

    if (!ret) {
        return ret;
    }

    auto contactNumberID = contactDB->GetLastInsertRowID();

    ret = contactDB->ringtones.Add(
        ContactsRingtonesTableRow{.ID = DB_ID_NONE, .contactID = contactID, .assetPath = rec.assetPath});

    if (!ret) {
        return ret;
    }

    auto contactRingID = contactDB->GetLastInsertRowID();

    // TODO: add missing feature: multiple addresses per contact
    ret = contactDB->address.Add(ContactsAddressTableRow{.contactID = contactID,
                                                         .country   = rec.country,
                                                         .city      = rec.city,
                                                         .street    = rec.street,
                                                         .number    = rec.number,
                                                         .type      = rec.addressType,
                                                         .note      = rec.note,
                                                         .mail      = rec.mail});

    if (!ret) {
        return ret;
    }

    auto contactAddressID = contactDB->GetLastInsertRowID();

    ret = contactDB->contacts.Update(ContactsTableRow{.ID         = contactID,
                                                      .nameID     = contactNameID,
                                                      .numbersID  = std::to_string(contactNumberID),
                                                      .ringID     = contactRingID,
                                                      .addressIDs = std::to_string(contactAddressID),

                                                      .type           = rec.contactType,
                                                      .isOnWhitelist  = rec.isOnBlacklist,
                                                      .isOnBlacklist  = rec.isOnBlacklist,
                                                      .isOnFavourites = rec.isOnFavourites,
                                                      .speedDial      = rec.speeddial});

    return ret;
}

bool ContactRecordInterface::BlockByID(uint32_t id, const bool shouldBeBlocked)
{
    return contactDB->contacts.BlockByID(id, shouldBeBlocked);
}

bool ContactRecordInterface::RemoveByID(uint32_t id)
{

    auto contact = contactDB->contacts.GetByID(id);
    if (contact.ID == 0) {
        return false;
    }

    if (contactDB->name.RemoveByID(contact.nameID) == false) {
        return false;
    }

    if (contactDB->address.RemoveByID(std::stoul(contact.addressIDs)) == false) {
        return false;
    }

    if (contactDB->number.RemoveByID(std::stoul(contact.numbersID)) == false) {
        return false;
    }

    if (contactDB->ringtones.RemoveByID(contact.ringID) == false) {
        return false;
    }

    if (contactDB->contacts.RemoveByID(id) == false) {
        return false;
    }

    return true;
}

bool ContactRecordInterface::Update(const ContactRecord &rec)
{
    // dummy not needed
    return true;
}

ContactRecord ContactRecordInterface::GetByID(uint32_t id)
{

    ContactRecord rec = ContactRecord();

    auto contact = contactDB->contacts.GetByID(id);
    if (contact.ID == 0) {
        return rec;
    }

    std::vector<ContactRecord::Number> nrs = getNumbers(contact.numbersID);
    if (nrs.size() == 0) {
        return rec;
    }
    auto ring = contactDB->ringtones.GetByID(contact.ringID);
    if (ring.ID == 0) {
        return rec;
    }

    auto address = contactDB->address.GetByID(std::stoul(contact.addressIDs));
    if (address.ID == 0) {
        return rec;
    }

    auto name = contactDB->name.GetByID(contact.nameID);
    if (name.ID == 0) {
        return rec;
    }

    rec.ID              = contact.ID;
    rec.primaryName     = name.namePrimary;
    rec.alternativeName = name.nameAlternative;
    rec.numbers         = nrs;
    rec.contactType     = contact.type;
    rec.country         = address.country;
    rec.city            = address.city;
    rec.street          = address.street;
    rec.number          = address.number;
    rec.note            = address.note;
    rec.mail            = address.mail;
    rec.addressType     = address.type;
    rec.assetPath       = ring.assetPath;
    rec.isOnWhitelist   = contact.isOnWhitelist;
    rec.isOnBlacklist   = contact.isOnBlacklist;
    rec.isOnFavourites  = contact.isOnFavourites;
    rec.speeddial       = contact.speedDial;

    return rec;
}

uint32_t ContactRecordInterface::GetCount()
{

    return contactDB->contacts.GetCount();
}

uint32_t ContactRecordInterface::GetCountFavourites()
{
    return contactDB->contacts.GetCountByFieldID("favourites", 1);
}

std::unique_ptr<std::vector<ContactRecord>> ContactRecordInterface::GetLimitOffset(uint32_t offset, uint32_t limit)
{
    auto records = std::make_unique<std::vector<ContactRecord>>();

    auto ret = contactDB->name.GetLimitOffset(offset, limit);

    for (const auto &w : ret) {

        auto contact = contactDB->contacts.GetByID(w.ID);
        if (contact.ID == 0) {
            return records;
        }

        auto name = contactDB->name.GetByID(contact.nameID);
        if (name.ID == 0) {
            return records;
        }

        auto nrs = getNumbers(contact.numbersID);
        if (nrs.size() == 0) {
            return records;
        }

        auto ring = contactDB->ringtones.GetByID(contact.ringID);
        if (ring.ID == 0) {
            return records;
        }

        auto address = contactDB->address.GetByID(std::stoul(contact.addressIDs));
        if (address.ID == 0) {
            return records;
        }

        records->push_back(ContactRecord{.ID              = contact.ID,
                                         .primaryName     = name.namePrimary,
                                         .alternativeName = name.nameAlternative,
                                         .contactType     = contact.type,
                                         .numbers         = nrs,
                                         .country         = address.country,
                                         .city            = address.city,
                                         .street          = address.street,
                                         .number          = address.number,
                                         .note            = address.note,
                                         .mail            = address.mail,
                                         .addressType     = address.type,
                                         .assetPath       = ring.assetPath,
                                         .isOnWhitelist   = contact.isOnWhitelist,
                                         .isOnBlacklist   = contact.isOnBlacklist,
                                         .isOnFavourites  = contact.isOnFavourites,
                                         .speeddial       = static_cast<uint8_t>(contact.speedDial)});
    }

    return records;
}

std::unique_ptr<std::vector<ContactRecord>> ContactRecordInterface::GetLimitOffsetByField(uint32_t offset,
                                                                                          uint32_t limit,
                                                                                          ContactRecordField field,
                                                                                          const char *str)
{
    auto records = std::make_unique<std::vector<ContactRecord>>();

    switch (field) {
    case ContactRecordField ::PrimaryName: {
        auto ret = contactDB->name.GetLimitOffsetByField(offset, limit, ContactNameTableFields::NamePrimary, str);

        for (const auto &w : ret) {

            auto contact = contactDB->contacts.GetByID(w.contactID);
            if (contact.ID == 0) {
                return records;
            }

            auto nrs = getNumbers(contact.numbersID);
            if (nrs.size() == 0) {
                return records;
            }

            auto ring = contactDB->ringtones.GetByID(contact.ringID);
            if (ring.ID == 0) {
                return records;
            }

            auto address = contactDB->address.GetByID(std::stoul(contact.addressIDs));
            if (address.ID == 0) {
                return records;
            }

            records->push_back(ContactRecord{.ID              = w.ID,
                                             .primaryName     = w.namePrimary,
                                             .alternativeName = w.nameAlternative,
                                             .contactType     = contact.type,
                                             .numbers         = nrs,
                                             .country         = address.country,
                                             .city            = address.city,
                                             .street          = address.street,
                                             .number          = address.number,
                                             .note            = address.note,
                                             .mail            = address.mail,
                                             .addressType     = address.type,
                                             .assetPath       = ring.assetPath,
                                             .isOnWhitelist   = contact.isOnWhitelist,
                                             .isOnBlacklist   = contact.isOnBlacklist,
                                             .isOnFavourites  = contact.isOnFavourites,
                                             .speeddial       = static_cast<uint8_t>(contact.speedDial)});
        }
    }

    break;

    case ContactRecordField ::NumberE164: {
        auto ret = contactDB->number.GetLimitOffsetByField(offset, limit, ContactNumberTableFields ::NumberE164, str);

        for (const auto &w : ret) {

            auto contact = contactDB->contacts.GetByID(w.contactID);
            if (contact.ID == 0) {
                return records;
            }

            auto nrs = getNumbers(contact.numbersID);
            if (nrs.size() == 0) {
                return records;
            }

            auto name = contactDB->name.GetByID(contact.nameID);
            if (name.ID == 0) {
                return records;
            }

            auto ring = contactDB->ringtones.GetByID(contact.ringID);
            if (ring.ID == 0) {
                return records;
            }

            auto address = contactDB->address.GetByID(std::stoul(contact.addressIDs));
            if (address.ID == 0) {
                return records;
            }

            records->push_back(ContactRecord{.ID              = w.ID,
                                             .primaryName     = name.namePrimary,
                                             .alternativeName = name.nameAlternative,
                                             .contactType     = contact.type,
                                             .numbers         = nrs,
                                             .country         = address.country,
                                             .city            = address.city,
                                             .street          = address.street,
                                             .number          = address.number,
                                             .note            = address.note,
                                             .mail            = address.mail,
                                             .addressType     = address.type,
                                             .assetPath       = ring.assetPath,
                                             .isOnWhitelist   = contact.isOnWhitelist,
                                             .isOnBlacklist   = contact.isOnBlacklist,
                                             .isOnFavourites  = contact.isOnFavourites,
                                             .speeddial       = static_cast<uint8_t>(contact.speedDial)

            });
        }
    } break;
    case ContactRecordField::SpeedDial: {
        auto ret = contactDB->contacts.GetLimitOffsetByField(0, 1, ContactTableFields::SpeedDial, str);

        for (const auto &w : ret) {

            auto contact = contactDB->contacts.GetByID(w.ID);
            if (contact.ID == 0) {
                return records;
            }

            auto name = contactDB->name.GetByID(contact.nameID);
            if (name.ID == 0) {
                return records;
            }

            auto nrs = getNumbers(contact.numbersID);
            if (nrs.size() == 0) {
                return records;
            }

            auto ring = contactDB->ringtones.GetByID(contact.ringID);
            if (ring.ID == 0) {
                return records;
            }

            auto address = contactDB->address.GetByID(std::stoul(contact.addressIDs));
            if (address.ID == 0) {
                return records;
            }

            records->push_back(ContactRecord{.ID              = contact.ID,
                                             .primaryName     = name.namePrimary,
                                             .alternativeName = name.nameAlternative,
                                             .contactType     = contact.type,
                                             .numbers         = nrs,
                                             .country         = address.country,
                                             .city            = address.city,
                                             .street          = address.street,
                                             .number          = address.number,
                                             .note            = address.note,
                                             .mail            = address.mail,
                                             .addressType     = address.type,
                                             .assetPath       = ring.assetPath,
                                             .isOnWhitelist   = contact.isOnWhitelist,
                                             .isOnBlacklist   = contact.isOnBlacklist,
                                             .isOnFavourites  = contact.isOnFavourites,
                                             .speeddial       = static_cast<uint8_t>(contact.speedDial)});
        }
    } break;

    case ContactRecordField::Favourite: {
        auto ret = contactDB->name.GetLimitOffsetByField(offset, limit, ContactNameTableFields::Favourite, str);

        for (const auto &w : ret) {

            auto contact = contactDB->contacts.GetByID(w.ID);
            if (contact.ID == 0) {
                return records;
            }

            auto name = contactDB->name.GetByID(contact.nameID);
            if (name.ID == 0) {
                return records;
            }

            auto nrs = getNumbers(contact.numbersID);
            if (nrs.size() == 0) {
                return records;
            }

            auto ring = contactDB->ringtones.GetByID(contact.ringID);
            if (ring.ID == 0) {
                return records;
            }

            auto address = contactDB->address.GetByID(std::stoul(contact.addressIDs));
            if (address.ID == 0) {
                return records;
            }

            records->push_back(ContactRecord{.ID              = contact.ID,
                                             .primaryName     = name.namePrimary,
                                             .alternativeName = name.nameAlternative,
                                             .contactType     = contact.type,
                                             .numbers         = nrs,
                                             .country         = address.country,
                                             .city            = address.city,
                                             .street          = address.street,
                                             .number          = address.number,
                                             .note            = address.note,
                                             .mail            = address.mail,
                                             .addressType     = address.type,
                                             .assetPath       = ring.assetPath,
                                             .isOnWhitelist   = contact.isOnWhitelist,
                                             .isOnBlacklist   = contact.isOnBlacklist,
                                             .isOnFavourites  = contact.isOnFavourites,
                                             .speeddial       = static_cast<uint8_t>(contact.speedDial)});
        }
    } break;
    }

    return records;
}

std::unique_ptr<std::vector<ContactRecord>> ContactRecordInterface::GetByName(UTF8 primaryName, UTF8 alternativeName)
{

    auto records = std::make_unique<std::vector<ContactRecord>>();

    auto ret = contactDB->name.GetByName(primaryName.c_str(), alternativeName.c_str());

    for (const auto &w : ret) {

        auto contact = contactDB->contacts.GetByID(w.contactID);
        if (contact.ID == 0) {
            return records;
        }

        auto nrs = getNumbers(contact.numbersID);
        if (nrs.size() == 0) {
            return records;
        }

        auto ring = contactDB->ringtones.GetByID(contact.ringID);
        if (ring.ID == 0) {
            return records;
        }

        auto address = contactDB->address.GetByID(std::stoul(contact.addressIDs));
        if (address.ID == 0) {
            return records;
        }

        records->push_back(ContactRecord{.ID              = w.ID,
                                         .primaryName     = w.namePrimary,
                                         .alternativeName = w.nameAlternative,
                                         .contactType     = contact.type,
                                         .numbers         = nrs,
                                         .country         = address.country,
                                         .city            = address.city,
                                         .street          = address.street,
                                         .number          = address.number,
                                         .note            = address.note,
                                         .mail            = address.mail,
                                         .addressType     = address.type,
                                         .assetPath       = ring.assetPath,
                                         .isOnWhitelist   = contact.isOnWhitelist,
                                         .isOnBlacklist   = contact.isOnBlacklist,
                                         .isOnFavourites  = contact.isOnFavourites,
                                         .speeddial       = static_cast<uint8_t>(contact.speedDial)});
    }

    return records;
}

std::unique_ptr<std::vector<ContactRecord>> ContactRecordInterface::Search(const char *primaryName,
                                                                           const char *alternativeName,
                                                                           const char *number)
{
    auto records = std::make_unique<std::vector<ContactRecord>>();

    auto ret = contactDB->contacts.Search(primaryName, alternativeName, number);

    for (const auto &w : ret) {
        auto contact = contactDB->contacts.GetByID(w.ID);
        if (contact.ID == 0) {
            return records;
        }

        auto nrs = getNumbers(contact.numbersID);
        if (nrs.size() == 0) {
            return records;
        }

        auto ring = contactDB->ringtones.GetByID(contact.ringID);
        if (ring.ID == 0) {
            return records;
        }

        auto address = contactDB->address.GetByID(std::stoul(contact.addressIDs));
        if (address.ID == 0) {
            return records;
        }

        records->push_back(ContactRecord{.ID              = w.ID,
                                         .primaryName     = w.namePrimary,
                                         .alternativeName = w.nameAlternative,
                                         .contactType     = contact.type,
                                         .numbers         = nrs,
                                         .country         = address.country,
                                         .city            = address.city,
                                         .street          = address.street,
                                         .number          = address.number,
                                         .note            = address.note,
                                         .mail            = address.mail,
                                         .addressType     = address.type,
                                         .assetPath       = ring.assetPath,
                                         .isOnWhitelist   = contact.isOnWhitelist,
                                         .isOnBlacklist   = contact.isOnBlacklist,
                                         .isOnFavourites  = contact.isOnFavourites,
                                         .speeddial       = static_cast<uint8_t>(contact.speedDial)});
    }

    return records;
}

std::unique_ptr<std::vector<ContactRecord>> ContactRecordInterface::GetContactByNumber(const UTF8 &number)
{
    return GetLimitOffsetByField(0, 1, ContactRecordField::NumberE164, number.c_str());
}

std::unique_ptr<std::vector<ContactRecord>> ContactRecordInterface::GetByNumber(const UTF8 &number,
                                                                                CreateTempContact createTempContact)
{
    auto ret = GetContactByNumber(number);
    if (ret->size() > 0) {
        return ret;
    }

    // Contact not found, create temporary one
    if (createTempContact == CreateTempContact::True) {
        LOG_INFO("Cannot find contact for number %s, creating temporary one", number.c_str());
        if (!Add(ContactRecord{
                .contactType = ContactType::TEMPORARY,
                .numbers = std::vector<ContactRecord::Number>{ContactRecord::Number(number.c_str(), number.c_str())},
            })) {
            LOG_ERROR("Cannot add contact record");
            return ret;
        }

        ret->push_back(GetByID(contactDB->GetLastInsertRowID()));
    }

    return ret;
}

std::unique_ptr<std::vector<ContactRecord>> ContactRecordInterface::GetBySpeedDial(uint8_t speedDial)
{
    return GetLimitOffsetByField(0, 1, ContactRecordField::SpeedDial, std::to_string(speedDial).c_str());
}

std::vector<ContactRecord::Number> ContactRecordInterface::getNumbers(const std::string &numbers_id)
{
    std::vector<ContactRecord::Number> nrs;
    for (auto nr_str : utils::split(numbers_id, ' ')) {
        auto nr = contactDB->number.GetByID(std::stol(nr_str));
        if (nr.ID == 0) {
            return nrs;
        }
        nrs.push_back(ContactRecord::Number(nr.numberUser, nr.numbere164, nr.type));
    }
    return nrs;
}