/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the examples of the Qt Toolkit.

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include <QtGui>

#include "addressbook.h"

AddressBook::AddressBook(QWidget *parent)

    : QWidget(parent)

{

    QLabel *nameLabel = new QLabel(tr("Name:"));

    nameLine = new QLineEdit;

    nameLine->setReadOnly(true);

    QLabel *addressLabel = new QLabel(tr("Address:"));

    addressText = new QTextEdit;

    addressText->setReadOnly(true);

    addButton = new QPushButton(tr("&Add"));

    editButton = new QPushButton(tr("&Edit"));

    editButton->setEnabled(false);

    removeButton = new QPushButton(tr("&Remove"));

    removeButton->setEnabled(false);

    findButton = new QPushButton(tr("&Find"));

    findButton->setEnabled(false);

    submitButton = new QPushButton(tr("&Submit"));

    submitButton->hide();

    cancelButton = new QPushButton(tr("&Cancel"));

    cancelButton->hide();

    nextButton = new QPushButton(tr("&Next"));

    nextButton->setEnabled(false);

    previousButton = new QPushButton(tr("&Previous"));

    previousButton->setEnabled(false);

    loadButton = new QPushButton(tr("&Load..."));

    loadButton->setToolTip(tr("Load contacts from a file"));

    saveButton = new QPushButton(tr("Sa&ve..."));

    saveButton->setToolTip(tr("Save contacts to a file"));

    saveButton->setEnabled(false);

    exportButton = new QPushButton(tr("E&xport"));

    exportButton->setToolTip(tr("Export as vCard"));

    exportButton->setEnabled(false);

    dialog = new FindDialog;

    connect(addButton, SIGNAL(clicked()), this, SLOT(addContact()));

    connect(submitButton, SIGNAL(clicked()), this, SLOT(submitContact()));

    connect(editButton, SIGNAL(clicked()), this, SLOT(editContact()));

    connect(cancelButton, SIGNAL(clicked()), this, SLOT(cancel()));

    connect(removeButton, SIGNAL(clicked()), this, SLOT(removeContact()));

    connect(findButton, SIGNAL(clicked()), this, SLOT(findContact()));

    connect(nextButton, SIGNAL(clicked()), this, SLOT(next()));

    connect(previousButton, SIGNAL(clicked()), this, SLOT(previous()));

    connect(loadButton, SIGNAL(clicked()), this, SLOT(loadFromFile()));

    connect(saveButton, SIGNAL(clicked()), this, SLOT(saveToFile()));

    connect(exportButton, SIGNAL(clicked()), this, SLOT(exportAsVCard()));

    QVBoxLayout *buttonLayout1 = new QVBoxLayout;

    buttonLayout1->addWidget(addButton);

    buttonLayout1->addWidget(editButton);

    buttonLayout1->addWidget(removeButton);

    buttonLayout1->addWidget(findButton);

    buttonLayout1->addWidget(submitButton);

    buttonLayout1->addWidget(cancelButton);

    buttonLayout1->addWidget(loadButton);

    buttonLayout1->addWidget(saveButton);

    buttonLayout1->addWidget(exportButton);

    buttonLayout1->addStretch();

    QHBoxLayout *buttonLayout2 = new QHBoxLayout;

    buttonLayout2->addWidget(previousButton);

    buttonLayout2->addWidget(nextButton);

    QGridLayout *mainLayout = new QGridLayout;

    mainLayout->addWidget(nameLabel, 0, 0);

    mainLayout->addWidget(nameLine, 0, 1);

    mainLayout->addWidget(addressLabel, 1, 0, Qt::AlignTop);

    mainLayout->addWidget(addressText, 1, 1);

    mainLayout->addLayout(buttonLayout1, 1, 2);

    mainLayout->addLayout(buttonLayout2, 2, 1);

    setLayout(mainLayout);

    setWindowTitle(tr("Simple Address Book"));

}

void AddressBook::addContact()

{

    oldName = nameLine->text();

    oldAddress = addressText->toPlainText();

    nameLine->clear();

    addressText->clear();

    updateInterface(AddingMode);

}

void AddressBook::editContact()

{

    oldName = nameLine->text();

    oldAddress = addressText->toPlainText();

    updateInterface(EditingMode);

}

void AddressBook::submitContact()

{

    QString name = nameLine->text();

    QString address = addressText->toPlainText();

    if (name == "" || address == "") {

        QMessageBox::information(this, tr("Empty Field"),

            tr("Please enter a name and address."));

    }

    if (currentMode == AddingMode) {

        if (!contacts.contains(name)) {

            contacts.insert(name, address);

            QMessageBox::information(this, tr("Add Successful"),

                tr("\"%1\" has been added to your address book.").arg(name));

        } else {

            QMessageBox::information(this, tr("Add Unsuccessful"),

                tr("Sorry, \"%1\" is already in your address book.").arg(name));

        }

    } else if (currentMode == EditingMode) {

        if (oldName != name) {

            if (!contacts.contains(name)) {

                QMessageBox::information(this, tr("Edit Successful"),

                    tr("\"%1\" has been edited in your address book.").arg(oldName));

                contacts.remove(oldName);

                contacts.insert(name, address);

            } else {

                QMessageBox::information(this, tr("Edit Unsuccessful"),

                    tr("Sorry, \"%1\" is already in your address book.").arg(name));

            }

        } else if (oldAddress != address) {

            QMessageBox::information(this, tr("Edit Successful"),

                tr("\"%1\" has been edited in your address book.").arg(name));

            contacts[name] = address;

        }

    }

    updateInterface(NavigationMode);

}

void AddressBook::cancel()

{

    nameLine->setText(oldName);

    addressText->setText(oldAddress);

    updateInterface(NavigationMode);

}

void AddressBook::removeContact()

{

    QString name = nameLine->text();

    QString address = addressText->toPlainText();

    if (contacts.contains(name)) {

        int button = QMessageBox::question(this,

            tr("Confirm Remove"),

            tr("Are you sure you want to remove \"%1\"?").arg(name),

            QMessageBox::Yes | QMessageBox::No);

        if (button == QMessageBox::Yes) {

            previous();

            contacts.remove(name);

            QMessageBox::information(this, tr("Remove Successful"),

                tr("\"%1\" has been removed from your address book.").arg(name));

        }

    }

    updateInterface(NavigationMode);

}

void AddressBook::next()

{

    QString name = nameLine->text();

    QMap<QString, QString>::iterator i = contacts.find(name);

    if (i != contacts.end())

        i++;

    if (i == contacts.end())

        i = contacts.begin();

    nameLine->setText(i.key());

    addressText->setText(i.value());

}

void AddressBook::previous()

{

    QString name = nameLine->text();

    QMap<QString, QString>::iterator i = contacts.find(name);

    if (i == contacts.end()) {

        nameLine->clear();

        addressText->clear();

        return;

    }

    if (i == contacts.begin())

        i = contacts.end();

    i--;

    nameLine->setText(i.key());

    addressText->setText(i.value());

}

void AddressBook::findContact()

{

    dialog->show();

    if (dialog->exec() == 1) {

        QString contactName = dialog->getFindText();

        if (contacts.contains(contactName)) {

            nameLine->setText(contactName);

            addressText->setText(contacts.value(contactName));

        } else {

            QMessageBox::information(this, tr("Contact Not Found"),

                tr("Sorry, \"%1\" is not in your address book.").arg(contactName));

            return;

        }

    }

    updateInterface(NavigationMode);

}

void AddressBook::updateInterface(Mode mode)

{

    currentMode = mode;

    switch (currentMode) {

    case AddingMode:

    case EditingMode:

        nameLine->setReadOnly(false);

        nameLine->setFocus(Qt::OtherFocusReason);

        addressText->setReadOnly(false);

        addButton->setEnabled(false);

        editButton->setEnabled(false);

        removeButton->setEnabled(false);

        nextButton->setEnabled(false);

        previousButton->setEnabled(false);

        submitButton->show();

        cancelButton->show();

        loadButton->setEnabled(false);

        saveButton->setEnabled(false);

        exportButton->setEnabled(false);

        break;

    case NavigationMode:

        if (contacts.isEmpty()) {

            nameLine->clear();

            addressText->clear();

        }

        nameLine->setReadOnly(true);

        addressText->setReadOnly(true);

        addButton->setEnabled(true);

        int number = contacts.size();

        editButton->setEnabled(number >= 1);

        removeButton->setEnabled(number >= 1);

        findButton->setEnabled(number > 2);

        nextButton->setEnabled(number > 1);

        previousButton->setEnabled(number > 1);

        submitButton->hide();

        cancelButton->hide();

        exportButton->setEnabled(number >= 1);

        loadButton->setEnabled(true);

        saveButton->setEnabled(number >= 1);

        break;

    }

}

void AddressBook::saveToFile()

{

    QString fileName = QFileDialog::getSaveFileName(this,

        tr("Save Address Book"), "",

        tr("Address Book (*.abk);;All Files (*)"));

    if (fileName.isEmpty())

        return;

    else {

        QFile file(fileName);

        if (!file.open(QIODevice::WriteOnly)) {

            QMessageBox::information(this, tr("Unable to open file"),

                file.errorString());

            return;

        }

        QDataStream out(&file);

        out.setVersion(QDataStream::Qt_4_3);

        out << contacts;

    }

    updateInterface(NavigationMode);

}

void AddressBook::loadFromFile()

{

    QString fileName = QFileDialog::getOpenFileName(this,

        tr("Open Address Book"), "",

        tr("Address Book (*.abk);;All Files (*)"));

    if (fileName.isEmpty())

        return;

    else {

        QFile file(fileName);

        if (!file.open(QIODevice::ReadOnly)) {

            QMessageBox::information(this, tr("Unable to open file"),

                file.errorString());

            return;

        }

        QDataStream in(&file);

        in.setVersion(QDataStream::Qt_4_3);

        contacts.empty();   // empty existing contacts

        in >> contacts;

        QMap<QString, QString>::iterator i = contacts.begin();

        nameLine->setText(i.key());

        addressText->setText(i.value());

    }

    updateInterface(NavigationMode);

}

//! [export function part1]

void AddressBook::exportAsVCard()

{

    QString name = nameLine->text();

    QString address = addressText->toPlainText();

    QString firstName;

    QString lastName;

    QStringList nameList;

    int index = name.indexOf(" ");

    if (index != -1) {

        nameList = name.split(QRegExp("\\s+"), QString::SkipEmptyParts);

        firstName = nameList.first();

        lastName = nameList.last();

    } else {

        firstName = name;

        lastName = "";

    }

    QString fileName = QFileDialog::getSaveFileName(this,

        tr("Export Contact"), "",

        tr("vCard Files (*.vcf);;All Files (*)"));

    if (fileName.isEmpty())

        return;

    QFile file(fileName);

//! [export function part1]

//! [export function part2]    

    if (!file.open(QIODevice::WriteOnly)) {

        QMessageBox::information(this, tr("Unable to open file"),

            file.errorString());

        return;

    }

    QTextStream out(&file);

//! [export function part2]

//! [export function part3]

    out << "BEGIN:VCARD" << "\n";

    out << "VERSION:2.1" << "\n";

    out << "N:" << lastName << ";" << firstName << "\n";

    if (!nameList.isEmpty())            

       out << "FN:" << nameList.join(" ") << "\n";

    else

       out << "FN:" << firstName << "\n";

//! [export function part3] 

//! [export function part4]

    address.replace(";", "\\;", Qt::CaseInsensitive);

    address.replace("\n", ";", Qt::CaseInsensitive);

    address.replace(",", " ", Qt::CaseInsensitive);

    out << "ADR;HOME:;" << address << "\n";

    out << "END:VCARD" << "\n";

    QMessageBox::information(this, tr("Export Successful"),

        tr("\"%1\" has been exported as a vCard.").arg(name));

}

//! [export function part4]