source: SMSSender/src/ui/VCAddressBook/vcaddressbook.cpp @ 71:415b45c71663

#include "vcaddressbook.h"

#include <QMessageBox>
#include <QDebug>

#include "../completers/bettercompleter.h"

#include "../VCEditContact/vceditcontact.h"
#include "../VCEditGroup/vceditgroup.h"

#include "../UIHelper.h"

#include <scontact.h>
#include <sgroup.h>
#include <snumber.h>
#include "../../business/BCContactManager.h"
#include "../../business/BCGroupManager.h"

#include <typeconvert.h>


const int CONTACT_COL_INDEX_IMAGE  = 0;
const int CONTACT_COL_INDEX_NAME   = 1;
const int CONTACT_COL_INDEX_NUMBER = 2;

const int GROUP_COL_INDEX_IMAGE    = 0;
const int GROUP_COL_INDEX_NAME     = 1;
const int GROUP_COL_INDEX_MEMBERS  = 2;


VCAddressBook::VCAddressBook(QWidget *parent) : QDialog(parent) {
        ui.setupUi(this);

        /* Contacts */
    contactModel_      = new ContactTableModel(this);
    contactProxyModel_ = new BetterCompletionModel(this);
    contactProxyModel_->setFilterKeyColumn(-1); // Filter from all
    contactProxyModel_->setSourceModel(contactModel_);
    connect(ui.edtContactFilter, SIGNAL(textChanged(QString)),
            contactProxyModel_, SLOT(setFilterFixedString(QString)));

    ui.tblViewContacts->setModel(contactProxyModel_);
    ui.tblViewContacts->resizeRowsToContents();
    ui.tblViewContacts->horizontalHeader()->setResizeMode(QHeaderView::ResizeToContents);
    ui.tblViewContacts->horizontalHeader()->setResizeMode(ContactTableModel::ColAliases, QHeaderView::Stretch);
    ui.tblViewContacts->sortByColumn(ContactTableModel::ColName, Qt::AscendingOrder);
    ui.tblViewContacts->resizeColumnsToContents();
    ui.tblViewContacts->verticalHeader()->setResizeMode(QHeaderView::ResizeToContents);

    connect(ui.tblViewContacts, SIGNAL(doubleClicked(const QModelIndex&)),
            this, SLOT(on_btnEditContact_clicked()));
    connect(ui.tblViewContacts->selectionModel(), SIGNAL(selectionChanged(const QItemSelection&, const QItemSelection&)),
            this, SLOT(on_tblViewContacts_selectionChanged()));

    on_tblViewContacts_selectionChanged(); // Set initial state


    /* Groups */
    groupModel_      = new GroupTableModel(this);
    groupProxyModel_ = new BetterCompletionModel(this);
    groupProxyModel_->setFilterKeyColumn(-1); // Filter from all
    groupProxyModel_->setSourceModel(groupModel_);
    connect(ui.edtGroupFilter, SIGNAL(textChanged(QString)),
            groupProxyModel_, SLOT(setFilterFixedString(QString)));

    ui.tblViewGroups->setModel(groupProxyModel_);
    ui.tblViewGroups->resizeRowsToContents();
    ui.tblViewGroups->horizontalHeader()->setResizeMode(QHeaderView::ResizeToContents);
    ui.tblViewGroups->horizontalHeader()->setResizeMode(GroupTableModel::ColMembers, QHeaderView::Stretch);
    ui.tblViewGroups->sortByColumn(ContactTableModel::ColName, Qt::AscendingOrder);
    ui.tblViewGroups->resizeColumnsToContents();
    ui.tblViewGroups->verticalHeader()->setResizeMode(QHeaderView::ResizeToContents);

    connect(ui.tblViewGroups, SIGNAL(doubleClicked(const QModelIndex&)),
            this, SLOT(on_btnEditGroup_clicked()));
    connect(ui.tblViewGroups->selectionModel(), SIGNAL(selectionChanged(const QItemSelection&, const QItemSelection&)),
            this, SLOT(on_tblViewGroups_selectionChanged()));

    on_tblViewGroups_selectionChanged(); // Set initial state
}

/*QSet<IContact*> VCAddressBook::getFilteredContactList(){
    if (actualContactFilterStr_ == ""){
        return contactList_;
    }

    QSet<IContact*> contactList;
    foreach (IContact* contact, contactList_) {
        bool containsFilter = false;

        containsFilter = containsFilter || contact->name().contains(actualContactFilterStr_, Qt::CaseInsensitive);

        SNumber number = contact->number();
        QString numberStr;
        // +41791234567
        numberStr = number.toString("+CCAAAu");
        numberStr.remove(" ");
        containsFilter = containsFilter || numberStr.contains(actualContactFilterStr_, Qt::CaseInsensitive);
        // +410791234567
        numberStr = number.toString("+ccaaau");
        containsFilter = containsFilter || numberStr.contains(actualContactFilterStr_, Qt::CaseInsensitive);

        QStringList aliases = contact->aliases();
        for (int x = 0; x < aliases.size() && !containsFilter; ++x){
            QString alias = aliases.at(x);
            alias.remove(" ");
            containsFilter = containsFilter || alias.contains(actualContactFilterStr_, Qt::CaseInsensitive);
        }

        if (containsFilter){
            contactList.insert(contact);
        }
    }

    return contactList;
}*/

/* Contacts */
IContact* VCAddressBook::getContactFromRow(int rowId) const {
    QModelIndex idx = contactProxyModel_->mapToSource(contactProxyModel_->index(rowId, 0));
    return contactModel_->dataObject(idx);
}

IContact* VCAddressBook::getSelectedContact() const {
    QItemSelectionModel* ism = ui.tblViewContacts->selectionModel();
    if (!ism->hasSelection())
        return NULL;

    int row = ism->selectedRows().first().row();
    return getContactFromRow(row);
}


void VCAddressBook::editContact(IContact* contact){
    VCEditContact* ec = new VCEditContact(contact, this);
    if (ec->exec() == QDialog::Accepted){ // Wait for return
        BCContactManager::instance()->saveContact(contact);
    }
}


/* Groups */
IGroup* VCAddressBook::getGroupFromRow(int rowId) const {
    QModelIndex idx = groupProxyModel_->mapToSource(groupProxyModel_->index(rowId, 0));
    return groupModel_->dataObject(idx);
}

IGroup* VCAddressBook::getSelectedGroup() const {
    QItemSelectionModel* ism = ui.tblViewGroups->selectionModel();
    if (!ism->hasSelection())
        return NULL;

    int row = ism->selectedRows().first().row();
    return getGroupFromRow(row);
}


void VCAddressBook::editGroup(IGroup* group){
    VCEditGroup* eg = new VCEditGroup(group, this);
    if (eg->exec() == QDialog::Accepted){ // Wait for return
        BCGroupManager::instance()->saveGroup(group);
    }
}


// Slots...

/* Contacts */
void VCAddressBook::on_btnAddContact_clicked(){
    editContact(new SContact());
}

void VCAddressBook::on_btnEditContact_clicked(){
    IContact* contact = getSelectedContact();
    if (contact == NULL)
        return;

    editContact(contact);
}

void VCAddressBook::on_btnRemoveContact_clicked(){
    IContact* contact = getSelectedContact();
    if (contact == NULL)
        return;

    if (QMessageBox::Yes == QMessageBox::warning(this, tr("SMSSender"),
                                                 tr("Do you really want to remove this contact?"),
                                                 QMessageBox::Yes,
                                                 QMessageBox::No | QMessageBox::Default | QMessageBox::Escape)){

        //TODO: do this eventually over the model...
        BCContactManager::instance()->removeContact(contact->id());
    }
}

void VCAddressBook::on_tblViewContacts_selectionChanged() {
    ui.btnEditContact->setEnabled(ui.tblViewContacts->selectionModel()->hasSelection());
    ui.btnRemoveContact->setEnabled(ui.tblViewContacts->selectionModel()->hasSelection());
}


/* Groups */

void VCAddressBook::on_btnAddGroup_clicked(){
    editGroup(new SGroup());
}

void VCAddressBook::on_btnEditGroup_clicked(){
    IGroup* group = getSelectedGroup();
    if (group == NULL)
        return;

    editGroup(group);
}

void VCAddressBook::on_btnRemoveGroup_clicked(){
    IGroup* group = getSelectedGroup();
    if (group == NULL)
        return;

    if (QMessageBox::Yes == QMessageBox::warning(this, tr("SMSSender"),
                                                 tr("Do you really want to remove this group?"),
                                                 QMessageBox::Yes,
                                                 QMessageBox::No | QMessageBox::Default | QMessageBox::Escape)){

        //TODO: do this eventually over the model...
        BCGroupManager::instance()->removeGroup(group->id());
    }
}

void VCAddressBook::on_tblViewGroups_selectionChanged() {
    ui.btnEditGroup->setEnabled(ui.tblViewGroups->selectionModel()->hasSelection());
    ui.btnRemoveGroup->setEnabled(ui.tblViewGroups->selectionModel()->hasSelection());
}