RyXéo » ryxeo-WebPhotoAlbum

/** Classe RyXeo WebPhotoAlbum: fichier principal de l'application

  * @see https://redmine.ryxeo.com/projects/

  * @author 2012 Eric Seigne <eric.seigne@ryxeo.com>

  * @see The GNU Public License (GNU/GPL) v3

  *

  *

  *

  * This program is free software; you can redistribute it and/or modify

  * it under the terms of the GNU General Public License as published by

  * the Free Software Foundation; either version 3 of the License, or

  * (at your option) any later version.

  *

  * This program is distributed in the hope that it will be useful, but

  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY

  * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License

  * for more details.

  *

  * You should have received a copy of the GNU General Public License along

  * with this program. If not, see <http://www.gnu.org/licenses/>.

  */

#include "webphotoalbum.h"

#include "ui_webphotoalbum.h"

//---------------------------------------------------------------------------

QString formatSize(qint64 num)

{

    QString total;

    const qint64 kb = 1024;

    const qint64 mb = 1024 * kb;

    const qint64 gb = 1024 * mb;

    const qint64 tb = 1024 * gb;

    if (num >= tb) total = QString("%1TB").arg(QString::number(qreal(num) / tb, 'f', 2));

    else if(num >= gb) total = QString("%1GB").arg(QString::number(qreal(num) / gb, 'f', 2));

    else if(num >= mb) total = QString("%1MB").arg(QString::number(qreal(num) / mb, 'f', 1));

    else if(num >= kb) total = QString("%1KB").arg(QString::number(qreal(num) / kb,'f', 1));

    else total = QString("%1 bytes").arg(num);

    return total;

}

// =========================================================================================================================

WebPhotoAlbum::WebPhotoAlbum(QWidget *parent) :

    QMainWindow(parent),

    ui(new Ui::WebPhotoAlbum)

{

    ui->setupUi(this);

    m_currentProjectPictureNumber = 0;

    //Hop pour que le splitter soit a peu pres a la bonne taille dès le début

    QList<int> sizes;

    sizes.append(200);

    sizes.append(500);

    ui->splitter->setSizes(sizes);

#ifdef Q_OS_WIN32

    QSettings settings(QSettings::UserScope, "Microsoft", "Windows");

    settings.beginGroup("CurrentVersion/Explorer/Shell Folders");

    m_baseDir = QDir(settings.value("Personal").toString() + "/RyXeo-Applications/WebPhotoAlbum");

    qDebug() << "Sous windows " << m_baseDir;

#else

    m_baseDir = QDir(QDir::homePath() + "/RyXeo-Applications/WebPhotoAlbum");

#endif

    m_baseDir.setFilter(QDir::Dirs | QDir::NoSymLinks | QDir::NoDotAndDotDot);

    QFileInfoList list = m_baseDir.entryInfoList();

    for(int i = 0; i < list.count(); i++) {

        QTreeWidgetItem *item = new QTreeWidgetItem(ui->treeWidget);

        item->setText(0, list.at(i).fileName());

        item->setData(1,0,list.at(i).absoluteFilePath());

    }

    ui->dateEdit->setDate(QDate::currentDate());

    m_tools = new tools(m_baseDir, this);

    connect(m_tools, SIGNAL(signalUploadEnd()), this, SLOT(uploadEnd()));

    //============== l'étiquette invitant a déposer des images

    m_labelDeposez = new QLabel(trUtf8("Glissez-déposez ici les photos que vous voulez ajouter à cet album ..."),ui->lwPictures);

    m_labelDeposez->setStyleSheet("color: #aaa;\nfont-size: 24pt;\nfont-weight: bold;\n");

    m_labelDeposez->setScaledContents(true);

    m_labelDeposez->setWordWrap(true);

    m_labelDeposez->setAlignment(Qt::AlignCenter | Qt::AlignHCenter);

    m_labelDeposez->hide();

    //============== le menu contextuel du treewidget (gauche)

    m_menu = new QMenu(ui->treeWidget);

    QList<QAction*> actions;

    QAction *a_nouveau = new QAction(QIcon(":/images/folder_add.png"),trUtf8("&Nouvel album"),m_menu);

    a_nouveau->setIconVisibleInMenu(true);

    a_nouveau->connect(a_nouveau, SIGNAL(triggered()), this, SLOT(on_action_Nouvel_album_triggered()));

    QAction *a_renommer = new QAction(QIcon(":/images/folder_edit.png"),trUtf8("&Renommer..."),m_menu);

    a_renommer->setIconVisibleInMenu(true);

    //a_renommer->connect(a_renommer, SIGNAL(triggered()), this, SLOT(on_action_Nouvel_album_triggered()));

    QAction *a_supprimer = new QAction(QIcon(":/images/folder_delete.png"),trUtf8("&Supprimer..."),m_menu);

    a_supprimer->setIconVisibleInMenu(true);

    //a_supprimer->connect(a_supprimer, SIGNAL(triggered()), this, SLOT(on_action_Nouvel_album_triggered()));

    actions << a_nouveau << a_renommer << a_supprimer;

    m_menu->addActions(actions);

    //============== Accepte le drag & drop

    ui->lwPictures->setDragEnabled(true);

    ui->lwPictures->setDropIndicatorShown(true);

}

       WebPhotoAlbum::~WebPhotoAlbum()

{

    delete ui;

}

void WebPhotoAlbum::on_action_Nouvel_album_triggered()

{

    m_currentProjectPictureNumber = 0;

    bool ok;

    QString text = NewAlbumDialog::getValue(this);

    //Si ce nom n'existe pas déjà ...

    QTreeWidgetItem *i = new QTreeWidgetItem(ui->treeWidget);

    i->setText(0, text);

    m_currentProjectDir = m_baseDir.absolutePath() + "/" + text;

    i->setData(1,0,m_currentProjectDir);

    //Creation de la destination

    QDir rep;

    rep.mkpath(m_currentProjectDir);

    rep.mkpath(m_currentProjectDir + "/hq");

    rep.mkpath(m_currentProjectDir + "/lq");

    rep.mkpath(m_currentProjectDir + "/mq");

    rep.mkpath(m_currentProjectDir + "/thumbs");

    rep.mkpath(m_currentProjectDir + "/zip");

    rep.mkpath(m_currentProjectDir + "/comments");

}

void WebPhotoAlbum::on_treeWidget_itemClicked(QTreeWidgetItem *item, int column)

{

    QFileInfo fi(ui->treeWidget->currentItem()->data(1,0).toString()+"/info.txt");

    qDebug() << "Lecture de " << fi.absoluteFilePath();

    //Remise a zero

    ui->leTitre->clear();

    ui->leAuthor->clear();

    ui->leDesc->clear();

    ui->dateEdit->clear();

    ui->leLogin->clear();

    ui->lePasswd->clear();

    if(fi.exists()) {

        QFile file(fi.absoluteFilePath());

        if (!file.open(QIODevice::ReadOnly | QIODevice::Text))

            return;

        /* Exemple de fichier

                name|Sortie Plongée SAGC. (Cala Montjoi, Spain) 28.05.2012

                author|Éric Seigne

                description|Sortie Plongée SAGC. (Cala Montjoi, Spain)

                date|28.05.2012

                restricted_user|sagc

                restricted_password|motdepasse

                  */

        while (!file.atEnd()) {

            QByteArray line = file.readLine();

            QStringList tab = QString::fromUtf8(line.constData()).split('|');

            if(tab.at(0) == "name")

                ui->leTitre->setText(tab.at(1));

            if(tab.at(0) == "author")

                ui->leAuthor->setText(tab.at(1));

            if(tab.at(0) == "description")

                ui->leDesc->setText(tab.at(1));

            if(tab.at(0) == "date") {

                qDebug() << "**************** DATE : " << tab.at(1).trimmed();

                ui->dateEdit->setDate(QDate::fromString(tab.at(1).trimmed(),"dd.MM.yyyy"));

            }

            if(tab.at(0) == "restricted_user")

                ui->leLogin->setText(tab.at(1));

            if(tab.at(0) == "restricted_password")

                ui->lePasswd->setText(tab.at(1));

        }

    }

    m_currentProjectPictureNumber++;

    refreshPictureList();

    qDebug() << "On continue la numerotation a " << m_currentProjectPictureNumber;

    qDebug() << "Espace de stockage du projet " << m_currentProjectDir;

}

void WebPhotoAlbum::on_btnSave_clicked()

{

    QByteArray data;

    data.append(QString("name|%1\n").arg(ui->leTitre->text().trimmed()));

    data.append(QString("author|%1\n").arg(ui->leAuthor->text().trimmed()));

    data.append(QString("description|%1\n").arg(ui->leDesc->text().trimmed()));

    data.append(QString("date|%1\n").arg(ui->dateEdit->date().toString("dd.MM.yyyy")));

    if(ui->leLogin->text().trimmed() != "") {

        data.append(QString("restricted_user|%1\n").arg(ui->leLogin->text().trimmed()));

        data.append(QString("restricted_password|%1\n").arg(ui->lePasswd->text().trimmed()));

    }

    QFileInfo fi(ui->treeWidget->currentItem()->data(1,0).toString()+"/info.txt");

    qDebug() << "Ecriture de " << fi.absoluteFilePath();

    QFile file(fi.absoluteFilePath());

    if (!file.open(QIODevice::WriteOnly | QIODevice::Text))

        return;

    QTextStream out(&file);

    out.setCodec("UTF-8");

    out << data;

    file.close();

}

void WebPhotoAlbum::on_treeWidget_customContextMenuRequested(const QPoint &pos)

{

    m_menu->exec(ui->treeWidget->mapToGlobal(pos));

}

void WebPhotoAlbum::dragEnterEvent(QDragEnterEvent *event)

{

    if (event->mimeData()->hasUrls())

    {

        event->acceptProposedAction();

    }

    else

    {

        event->ignore();

    }

}

void WebPhotoAlbum::dropEvent(QDropEvent *event)

{

    qDebug() << "Accepte le drop d'une image";

    if(event->mimeData()->hasUrls())

    {

        for(int i = 0; i < event->mimeData()->urls().count(); i++) {

            //recopier l'image dans le projet

            qDebug() << "Windows d&drop : " << event->mimeData()->urls().at(i).toString();

            //Attention sous windows le drag & drop est un file:/// sous linux file:// et osx ?

#ifdef Q_OS_WIN32

            addPictureToProject(event->mimeData()->urls().at(i).toString().remove("file:///"));

#endif

#ifdef Q_OS_LINUX

            addPictureToProject(event->mimeData()->urls().at(i).toString().remove("file://"));

#endif

            qDebug() << "Accepte le drop d'une image";

            event->acceptProposedAction();

        }

    }

}

void WebPhotoAlbum::addPictureToProject(QString fic)

{

    ui->statusBar->clearMessage();

    ui->statusBar->showMessage(trUtf8("Fichier en cours: %1").arg(fic),3000);

    qApp->processEvents();

    if(m_currentProjectPictureNumber == 0)

        m_currentProjectPictureNumber = 1;

    qDebug() << "Processing : " << fic;

    QPixmap pixmap(fic);

    pixmap.save(QString("%1/hq/img-%2.jpg").arg(m_currentProjectDir).arg(m_currentProjectPictureNumber));

    qDebug() << "Save to : " << QString("%1/hq/img-%2.jpg").arg(m_currentProjectDir).arg(m_currentProjectPictureNumber);

    ui->statusBar->showMessage(trUtf8("Fichier en cours: %1 [hq]").arg(fic),3000);

    QPixmap pixmapThumbs(pixmap.scaled(QSize(120,120), Qt::KeepAspectRatio, Qt::SmoothTransformation));

    pixmapThumbs.save(QString("%1/thumbs/img-%2.jpg").arg(m_currentProjectDir).arg(m_currentProjectPictureNumber));

    ui->statusBar->showMessage(trUtf8("Fichier en cours: %1 [hq] [mq]").arg(fic),3000);

    QPixmap pixmapMQ(pixmap.scaled(QSize(800,600), Qt::KeepAspectRatio, Qt::SmoothTransformation));

    pixmapMQ.save(QString("%1/mq/img-%2.jpg").arg(m_currentProjectDir).arg(m_currentProjectPictureNumber));

    ui->statusBar->showMessage(trUtf8("Fichier en cours: %1 [hq] [mq] [lq]").arg(fic),3000);

    QPixmap pixmapLQ(pixmap.scaled(QSize(640,480), Qt::KeepAspectRatio, Qt::SmoothTransformation));

    pixmapLQ.save(QString("%1/lq/img-%2.jpg").arg(m_currentProjectDir).arg(m_currentProjectPictureNumber));

    QFile file(QString("%1/comments/%2.txt").arg(m_currentProjectDir).arg(m_currentProjectPictureNumber));

    if (!file.open(QIODevice::WriteOnly | QIODevice::Text))

        return;

    QTextStream out(&file);

    out << "<span>Photo " + QString::number(m_currentProjectPictureNumber) + "</span>\n";

    file.close();

    ui->statusBar->showMessage(trUtf8("Fichier en cours: %1 [hq] [mq] [lq] [meta]").arg(fic),3000);

    //================== on copie les metadata

    try {

        Exiv2::Image::AutoPtr image = Exiv2::ImageFactory::open(fic.toStdString());

        Exiv2::Image::AutoPtr imageLQ = Exiv2::ImageFactory::open(QString("%1/lq/img-%2.jpg").arg(m_currentProjectDir).arg(m_currentProjectPictureNumber).toStdString());

        Exiv2::Image::AutoPtr imageMQ = Exiv2::ImageFactory::open(QString("%1/mq/img-%2.jpg").arg(m_currentProjectDir).arg(m_currentProjectPictureNumber).toStdString());

        Exiv2::Image::AutoPtr imageHQ = Exiv2::ImageFactory::open(QString("%1/hq/img-%2.jpg").arg(m_currentProjectDir).arg(m_currentProjectPictureNumber).toStdString());

        Exiv2::Image::AutoPtr imageThumb = Exiv2::ImageFactory::open(QString("%1/thumbs/img-%2.jpg").arg(m_currentProjectDir).arg(m_currentProjectPictureNumber).toStdString());

        if (image.get() != 0)

        {

            image->readMetadata();

            Exiv2::ExifData &exifData = image->exifData();

            if (!exifData.empty())

            {

                bool setArtist = false;

                bool setCopyright = false;

                bool setSoftware = false;

                Exiv2::ExifData::const_iterator end = exifData.end();

                //                qDebug() << "============================ DEBUT ===============================";

                for (Exiv2::ExifData::const_iterator i = exifData.begin(); i != end; ++i)

                {

                    //                    qDebug() << i->key().c_str() << " = " << i->value().toString().c_str();

                    if (i->key() == "Exif.Image.Artist") {

                        exifData["Exif.Image.Artist"]    = ui->leAuthor->text().trimmed().toStdString();

                        setArtist = true;

                    }

                    if (i->key() == "Exif.Image.Copyright") {

                        exifData["Exif.Image.Copyright"]    = ui->leAuthor->text().trimmed().toStdString();

                        setCopyright = true;

                    }

                    if (i->key() == "Exif.Image.Software") {

                        exifData["Exif.Image.Software"]  = "RyXeo WebPhotoAlbum - http://www.ryxeo.com/";

                        setSoftware = true;

                    }

                }

                //Si on n'a pas les meta il faut les ajouter

                //Mise à jour de l'auteur

                if(!setArtist) {

                    Exiv2::ExifKey k("Exif.Image.Artist");

                    Exiv2::Value::AutoPtr v = Exiv2::Value::create(Exiv2::asciiString);

                    v->read(ui->leAuthor->text().trimmed().toStdString());

                    exifData.add(k,v.get());

                }

                if(!setCopyright) {

                    Exiv2::ExifKey k2("Exif.Image.Copyright");

                    Exiv2::Value::AutoPtr v2 = Exiv2::Value::create(Exiv2::asciiString);

                    v2->read(ui->leAuthor->text().trimmed().toStdString());

                    exifData.add(k2,v2.get());

                }

                if(!setSoftware) {

                    Exiv2::ExifKey k3("Exif.Image.Software");

                    Exiv2::Value::AutoPtr v3 = Exiv2::Value::create(Exiv2::asciiString);

                    v3->read("RyXéo WebPhotoAlbum - http://www.ryxeo.com/");

                    exifData.add(k3,v3.get());

                }

                //                qDebug() << "============================ APRES ===============================";

                Exiv2::ExifData::const_iterator end2 = exifData.end();

                //                for (Exiv2::ExifData::const_iterator i2 = exifData.begin(); i2 != end2; ++i2)

                //                {

                //                    qDebug() << i2->key().c_str() << " = " << i2->value().toString().c_str();

                //                }

                //                qDebug() << "============================ FIN ===============================";

                imageLQ->setExifData(exifData);

                imageLQ->writeMetadata();

                imageMQ->setExifData(exifData);

                imageMQ->writeMetadata();

                imageHQ->setExifData(exifData);

                imageHQ->writeMetadata();

                imageThumb->setExifData(exifData);

                imageThumb->writeMetadata();

            }

        }

    }

    catch(...) {

        qDebug() << "Erreur EXIV2";

    }

    ui->statusBar->showMessage(trUtf8("Fichier en cours: %1 [hq] [mq] [lq] [meta] ... terminé").arg(fic),3000);

    m_currentProjectPictureNumber++;

    refreshPictureList();

}

void WebPhotoAlbum::refreshPictureList()

{

    QTreeWidgetItem *item = ui->treeWidget->currentItem();

    ui->lwPictures->clear();

    m_currentProjectDir = item->data(1,0).toString();

    int currentProjectPictureNumber=0;

    QDir dir(item->data(1,0).toString()+"/thumbs/");

    dir.setFilter(QDir::Files);

    dir.setSorting(QDir::Name | QDir::IgnoreCase | QDir::LocaleAware);

    QFileInfoList list = dir.entryInfoList();

    for(int i = 0; i < list.count(); i++) {

        qDebug() << list.at(i).fileName();

        QListWidgetItem *newitem = new QListWidgetItem();

        QIcon icone(list.at(i).absoluteFilePath());//pour la mettre  à coté de l'item

        newitem->setIcon(icone); // ajout de la petite icone sur l'item

        newitem->setText(list.at(i).fileName());

        ui->lwPictures->insertItem(i,newitem);

        currentProjectPictureNumber = i+1;

    }

    if(currentProjectPictureNumber == 0) {

        qDebug() << "Pas d'image ... on affiche le message drag & drop";

        m_labelDeposez->resize(ui->lwPictures->size());

        m_labelDeposez->move(ui->lwPictures->pos());

        m_labelDeposez->show();

        qDebug() << m_labelDeposez->pos();

        qDebug() << m_labelDeposez->size();

    }

    else {

        m_labelDeposez->hide();

    }

    m_currentProjectPictureNumber = currentProjectPictureNumber+1;

}

void WebPhotoAlbum::on_btnUpload_clicked()

{

    qDebug() << "on_btnUpload_clicked : " << ui->treeWidget->currentItem()->data(1,0).toString();

    makeZip(m_baseDir.absolutePath() + "/" + ui->treeWidget->currentItem()->text(0) + "/hq",

            m_baseDir.absolutePath() + "/" + ui->treeWidget->currentItem()->text(0) + "/zip/hq.zip");

    makeZip(m_baseDir.absolutePath() + "/" + ui->treeWidget->currentItem()->text(0) + "/mq",

            m_baseDir.absolutePath() + "/" + ui->treeWidget->currentItem()->text(0) + "/zip/mq.zip");

    makeZip(m_baseDir.absolutePath() + "/" + ui->treeWidget->currentItem()->text(0) + "/lq",

            m_baseDir.absolutePath() + "/" + ui->treeWidget->currentItem()->text(0) + "/zip/lq.zip");

    makeZip(m_baseDir.absolutePath() + "/" + ui->treeWidget->currentItem()->text(0) + "/thumbs",

            m_baseDir.absolutePath() + "/" + ui->treeWidget->currentItem()->text(0) + "/zip/thumbs.zip");

    m_tools->clear();

    m_tools->ftpConnect(ui->leFTPServeur->text(),

                        ui->leFTPLogin->text(),

                        ui->leFTPPass->text());

    m_tools->ftpMkdir(ui->leFTPDirectory->text());

    m_tools->ftpMkdir(ui->leFTPDirectory->text() + "/galleries");

    m_tools->ftpMkdir(ui->leFTPDirectory->text() + "/galleries/" + ui->treeWidget->currentItem()->text(0));

    //On uploade la gallerie

    //m_ftp->cd(ui->leFTPDirectory->text());

    QFileInfo fi(ui->treeWidget->currentItem()->data(1,0).toString());

    QDir dir(fi.absoluteFilePath());

    m_tools->parcoursRecursif(dir.absolutePath(),ui->treeWidget->currentItem()->text(0));

    //qDebug() << m_fileListToUpload;

    m_tools->uploadRecursifGallery(ui->leFTPDirectory->text());

    connect(m_tools, SIGNAL(signalUpload(int,int,QString)), this, SLOT(updateProgressFichier(int,int,QString)));

    connect(m_tools, SIGNAL(signalUploadData(int,int)), this, SLOT(updateProgressData(int,int)));

}

void WebPhotoAlbum::on_action_Assistant_d_installation_triggered()

{

    Assistant *a = new Assistant(this);

    a->show();

}

void WebPhotoAlbum::updateProgressFichier(int total, int current, QString fileName)

{

    ui->pbFTPFichier->setMaximum(total);

    ui->pbFTPFichier->setValue(current);

    ui->textBrowser->append(fileName + "\n");

}

void WebPhotoAlbum::updateProgressData(int total, int current)

{

    ui->pbFTPProgress->setMaximum(total);

    ui->pbFTPProgress->setValue(current);

}

void WebPhotoAlbum::makeZip(QString repSource, QString destFileName)

{

    QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));

    ui->statusBar->showMessage(QString::fromUtf8("Création du fichier ZIP %1 en cours").arg(destFileName), 5000);

    qApp->processEvents();

    qDebug() << "WebPhotoAlbum::makeZip " << repSource << " :: " << destFileName;

    //Creation d'une archive ZIP dans tous les cas

    QuaZip zipDestination(destFileName);

    if(zipDestination.open(QuaZip::mdCreate)) {

        //On sauvegarde tout dans ce fichier

        QuaZipFile fichierDehors(&zipDestination);

        //Les droits par defaut en 644 pour les fichiers contenus dans le .zip

        QuaZipNewInfo newInfo("");

        int permission644 = 33188;

        newInfo.externalAttr = (permission644 << 16L);

        QDir dir(repSource);

        dir.setFilter(QDir::Files);

        QFileInfoList fileList = dir.entryInfoList();

        qDebug() << "avant " << fileList.count();

        for(int i = 0; i < fileList.count(); i++) {

            QString fileName = fileList.at(i).fileName();

            QFile fic(fileList.at(i).absoluteFilePath());

            qDebug() << "On ajoute le fichier " << fileName;

            newInfo.name = fileName;

            if(fic.open(QIODevice::ReadOnly) && fichierDehors.open(QIODevice::WriteOnly,newInfo)) {

                fichierDehors.write(fic.read(fic.size()));

                fichierDehors.close();

                fic.close();

                qDebug() << " ajout ok";

            }

        }

        zipDestination.close();

    }

    QApplication::restoreOverrideCursor();

}

void WebPhotoAlbum::uploadEnd()

{

    ui->textBrowser->append(trUtf8("Téléchargement terminé\n"));

}