Chris@88: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@88: 
Chris@88: /*
Chris@88:     EasyMercurial
Chris@88: 
Chris@88:     Based on HgExplorer by Jari Korhonen
Chris@88:     Copyright (c) 2010 Jari Korhonen
Chris@88:     Copyright (c) 2010 Chris Cannam
Chris@88:     Copyright (c) 2010 Queen Mary, University of London
Chris@88:     
Chris@88:     This program is free software; you can redistribute it and/or
Chris@88:     modify it under the terms of the GNU General Public License as
Chris@88:     published by the Free Software Foundation; either version 2 of the
Chris@88:     License, or (at your option) any later version.  See the file
Chris@88:     COPYING included with this distribution for more information.
Chris@88: */
Chris@88: 
Chris@88: #include "filestatuswidget.h"
Chris@88: 
Chris@88: #include <QLabel>
Chris@88: #include <QListWidget>
Chris@88: #include <QGridLayout>
Chris@93: #include <QFileInfo>
Chris@93: #include <QApplication>
Chris@93: #include <QDateTime>
Chris@93: 
Chris@93: #include "debug.h"
Chris@88: 
Chris@88: FileStatusWidget::FileStatusWidget(QWidget *parent) :
Chris@93:     QWidget(parent),
Chris@93:     m_dateReference(0)
Chris@88: {
Chris@88:     QGridLayout *layout = new QGridLayout;
Chris@88:     setLayout(layout);
Chris@88: 
Chris@88:     int row = 0;
Chris@88:     
Chris@88:     layout->addWidget(new QLabel(tr("Local:")), row, 0);
Chris@88:     m_localPathLabel = new QLabel;
Chris@89:     QFont f(m_localPathLabel->font());
Chris@89:     f.setBold(true);
Chris@89:     m_localPathLabel->setFont(f);
Chris@88:     layout->addWidget(m_localPathLabel, row, 1);
Chris@88: 
Chris@88:     ++row;
Chris@88:     layout->addWidget(new QLabel(tr("Remote:")), row, 0);
Chris@88:     m_remoteURLLabel = new QLabel;
Chris@88:     layout->addWidget(m_remoteURLLabel, row, 1);
Chris@88: 
Chris@89:     layout->setColumnStretch(1, 20);
Chris@88: 
Chris@99:     layout->addItem(new QSpacerItem(5, 8), ++row, 0);
Chris@99: 
Chris@94:     QMap<FileStates::State, QString> labels;
Chris@94:     labels[FileStates::Clean] = tr("Unmodified files:");
Chris@94:     labels[FileStates::Modified] = tr("Modified files:");
Chris@94:     labels[FileStates::Added] = tr("Added files:");
Chris@94:     labels[FileStates::Removed] = tr("Removed files:");
Chris@94:     labels[FileStates::Missing] = tr("Missing files:");
Chris@94:     labels[FileStates::Unknown] = tr("Untracked files:");
Chris@89: 
Chris@94:     for (int i = int(FileStates::FirstState);
Chris@94:              i <= int(FileStates::LastState); ++i) {
Chris@89: 
Chris@94:         FileStates::State s = FileStates::State(i);
Chris@89: 
Chris@89:         QWidget *box = new QWidget;
Chris@89:         QGridLayout *boxlayout = new QGridLayout;
Chris@99:         boxlayout->setMargin(0);
Chris@89:         box->setLayout(boxlayout);
Chris@89: 
Chris@94:         boxlayout->addWidget(new QLabel(labels[s]), 0, 0);
Chris@89: 
Chris@94:         QListWidget *w = new QListWidget;
Chris@94:         m_stateListMap[s] = w;
Chris@94:         w->setSelectionMode(QListWidget::ExtendedSelection);
Chris@94:         boxlayout->addWidget(w, 1, 0);
Chris@89: 
Chris@95:         connect(w, SIGNAL(itemSelectionChanged()),
Chris@95:                 this, SLOT(itemSelectionChanged()));
Chris@95: 
Chris@89:         layout->addWidget(box, ++row, 0, 1, 2);
Chris@89:         box->hide();
Chris@89:     }
Chris@89: 
Chris@89:     layout->setRowStretch(++row, 20);
Chris@88: }
Chris@88: 
Chris@93: FileStatusWidget::~FileStatusWidget()
Chris@93: {
Chris@93:     delete m_dateReference;
Chris@93: }
Chris@93: 
Chris@95: void FileStatusWidget::itemSelectionChanged()
Chris@95: {
Chris@95:     m_selectedFiles.clear();
Chris@95: 
Chris@95:     DEBUG << "FileStatusWidget::itemSelectionChanged" << endl;
Chris@95: 
Chris@95:     foreach (QListWidget *w, m_stateListMap) {
Chris@95:         QList<QListWidgetItem *> sel = w->selectedItems();
Chris@95:         foreach (QListWidgetItem *i, sel) {
Chris@95:             m_selectedFiles.push_back(i->text());
Chris@95:             DEBUG << "file " << i->text() << " is selected" << endl;
Chris@95:         }
Chris@95:     }
Chris@95: 
Chris@95:     emit selectionChanged();
Chris@95: }
Chris@95: 
Chris@94: void FileStatusWidget::clearSelections()
Chris@94: {
Chris@95:     m_selectedFiles.clear();
Chris@94:     foreach (QListWidget *w, m_stateListMap) {
Chris@94:         w->clearSelection();
Chris@94:     }
Chris@94: }
Chris@94: 
Chris@95: bool FileStatusWidget::haveChangesToCommit() const
Chris@95: {
Chris@95:     return !m_fileStates.added().empty() ||
Chris@95:            !m_fileStates.removed().empty() ||
Chris@95:            !m_fileStates.modified().empty();
Chris@95: }
Chris@95: 
Chris@95: bool FileStatusWidget::haveSelection() const
Chris@95: {
Chris@95:     return !m_selectedFiles.empty();
Chris@95: }
Chris@95: 
Chris@95: QStringList FileStatusWidget::getAllSelectedFiles() const
Chris@95: {
Chris@95:     return m_selectedFiles;
Chris@95: }
Chris@95: 
Chris@95: QStringList FileStatusWidget::getSelectedCommittableFiles() const
Chris@95: {
Chris@95:     QStringList files;
Chris@95:     foreach (QString f, m_selectedFiles) {
Chris@95:         switch (m_fileStates.getStateOfFile(f)) {
Chris@95:         case FileStates::Added:
Chris@95:         case FileStates::Modified:
Chris@95:         case FileStates::Removed:
Chris@95:             files.push_back(f);
Chris@95:             break;
Chris@95:         default: break;
Chris@95:         }
Chris@95:     }
Chris@95:     return files;
Chris@95: }
Chris@95: 
Chris@95: QStringList FileStatusWidget::getSelectedAddableFiles() const
Chris@95: {
Chris@95:     QStringList files;
Chris@95:     foreach (QString f, m_selectedFiles) {
Chris@95:         switch (m_fileStates.getStateOfFile(f)) {
Chris@95:         case FileStates::Unknown:
Chris@95:         case FileStates::Removed:
Chris@95:             files.push_back(f);
Chris@95:             break;
Chris@95:         default: break;
Chris@95:         }
Chris@95:     }
Chris@95:     return files;
Chris@95: }
Chris@95: 
Chris@95: QStringList FileStatusWidget::getSelectedRemovableFiles() const
Chris@95: {
Chris@95:     QStringList files;
Chris@95:     foreach (QString f, m_selectedFiles) {
Chris@95:         switch (m_fileStates.getStateOfFile(f)) {
Chris@95:         case FileStates::Clean:
Chris@95:         case FileStates::Added:
Chris@95:         case FileStates::Modified:
Chris@95:         case FileStates::Missing:
Chris@95:             files.push_back(f);
Chris@95:             break;
Chris@95:         default: break;
Chris@95:         }
Chris@95:     }
Chris@95:     return files;
Chris@95: }
Chris@95: 
Chris@88: void
Chris@88: FileStatusWidget::setLocalPath(QString p)
Chris@88: {
Chris@88:     m_localPath = p;
Chris@88:     m_localPathLabel->setText(p);
Chris@93:     delete m_dateReference;
Chris@93:     m_dateReference = new QFileInfo(p + "/.hg/dirstate");
Chris@93:     if (!m_dateReference->exists() ||
Chris@93:         !m_dateReference->isFile() ||
Chris@93:         !m_dateReference->isReadable()) {
Chris@93:         DEBUG << "FileStatusWidget::setLocalPath: date reference file "
Chris@93:                 << m_dateReference->absoluteFilePath()
Chris@93:                 << " does not exist, is not a file, or cannot be read"
Chris@93:                 << endl;
Chris@93:         delete m_dateReference;
Chris@93:         m_dateReference = 0;
Chris@93:     }
Chris@88: }
Chris@88: 
Chris@88: void
Chris@88: FileStatusWidget::setRemoteURL(QString r)
Chris@88: {
Chris@88:     m_remoteURL = r;
Chris@88:     m_remoteURLLabel->setText(r);
Chris@88: }
Chris@88: 
Chris@88: void
Chris@92: FileStatusWidget::setFileStates(FileStates p)
Chris@88: {
Chris@92:     m_fileStates = p;
Chris@88:     updateWidgets();
Chris@88: }
Chris@88: 
Chris@88: void
Chris@88: FileStatusWidget::updateWidgets()
Chris@88: {
Chris@95:     QDateTime lastInteractionTime;
Chris@95:     if (m_dateReference) {
Chris@95:         lastInteractionTime = m_dateReference->lastModified();
Chris@95:         DEBUG << "reference time: " << lastInteractionTime << endl;
Chris@95:     }
Chris@95: 
Chris@95:     QSet<QString> selectedFiles;
Chris@95:     foreach (QString f, m_selectedFiles) selectedFiles.insert(f);
Chris@95: 
Chris@94:     foreach (FileStates::State s, m_stateListMap.keys()) {
Chris@95: 
Chris@94:         QListWidget *w = m_stateListMap[s];
Chris@94:         w->clear();
Chris@95:         QStringList files = m_fileStates.getFilesInState(s);
Chris@93: 
Chris@95:         QStringList highPriority, lowPriority;
Chris@95: 
Chris@95:         foreach (QString file, files) {
Chris@95: 
Chris@95:             bool highlighted = false;
Chris@95: 
Chris@95:             if (s == FileStates::Unknown) {
Chris@95:                 // We want to highlight untracked files that have appeared
Chris@95:                 // since the last interaction with the repo
Chris@95:                 QString fn(m_localPath + "/" + file);
Chris@95:                 DEBUG << "comparing with " << fn << endl;
Chris@95:                 QFileInfo fi(fn);
Chris@95:                 if (fi.exists() && fi.lastModified() > lastInteractionTime) {
Chris@95:                     DEBUG << "file " << fn << " is newer (" << fi.lastModified()
Chris@95:                             << ") than reference" << endl;
Chris@95:                     highlighted = true;
Chris@95:                 }
Chris@95:             }
Chris@95: 
Chris@95:             if (highlighted) {
Chris@95:                 highPriority.push_back(file);
Chris@95:             } else {
Chris@95:                 lowPriority.push_back(file);
Chris@93:             }
Chris@93:         }
Chris@95: 
Chris@95:         foreach (QString file, highPriority) {
Chris@95:             QListWidgetItem *item = new QListWidgetItem(file);
Chris@95:             w->addItem(item);
Chris@95:             item->setForeground(Qt::red); //!!! and a nice gold star
Chris@95:             item->setSelected(selectedFiles.contains(file));
Chris@95:         }
Chris@95: 
Chris@95:         foreach (QString file, lowPriority) {
Chris@95:             QListWidgetItem *item = new QListWidgetItem(file);
Chris@95:             w->addItem(item);
Chris@95:             item->setSelected(selectedFiles.contains(file));
Chris@95:         }
Chris@95: 
Chris@95:         w->parentWidget()->setVisible(!files.empty());
Chris@93:     }
Chris@88: }
Chris@88: