qgsfilewidget.cpp

Go to the documentation of this file.

/***************************************************************************
  qgsfilewidget.cpp
 
 ---------------------
 begin                : 17.12.2015
 copyright            : (C) 2015 by Denis Rouzaud
 email                : [email protected]
 ***************************************************************************
 *                                                                         *
 *   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 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
 
#include "qgsfilewidget.h"
 
#include <QLineEdit>
#include <QToolButton>
#include <QLabel>
#include <QGridLayout>
#include <QUrl>
#include <QDropEvent>
#include <QRegularExpression>
 
#include "qgssettings.h"
#include "qgsfilterlineedit.h"
#include "qgsfocuskeeper.h"
#include "qgslogger.h"
#include "qgsproject.h"
#include "qgsapplication.h"
#include "qgsfileutils.h"
#include "qgsmimedatautils.h"
 
QgsFileWidget::QgsFileWidget( QWidget *parent )
  : QWidget( parent )
{
  mLayout = new QHBoxLayout();
  mLayout->setContentsMargins( 0, 0, 0, 0 );
 
  // If displaying a hyperlink, use a QLabel
  mLinkLabel = new QLabel( this );
  // Make Qt opens the link with the OS defined viewer
  mLinkLabel->setOpenExternalLinks( true );
  // Label should always be enabled to be able to open
  // the link on read only mode.
  mLinkLabel->setEnabled( true );
  mLinkLabel->setSizePolicy( QSizePolicy::Expanding, QSizePolicy::Preferred );
  mLinkLabel->setTextFormat( Qt::RichText );
  mLinkLabel->hide(); // do not show by default
  mLayout->addWidget( mLinkLabel );
 
  // otherwise, use the traditional QLineEdit subclass
  mLineEdit = new QgsFileDropEdit( this );
  mLineEdit->setDragEnabled( true );
  mLineEdit->setToolTip( tr( "Full path to the file(s), including name and extension" ) );
  connect( mLineEdit, &QLineEdit::textChanged, this, &QgsFileWidget::textEdited );
  connect( mLineEdit, &QgsFileDropEdit::fileDropped, this, &QgsFileWidget::fileDropped );
  mLayout->addWidget( mLineEdit );
 
  mLinkEditButton = new QToolButton( this );
  mLinkEditButton->setIcon( QgsApplication::getThemeIcon( QStringLiteral( "/mActionToggleEditing.svg" ) ) );
  mLayout->addWidget( mLinkEditButton );
  connect( mLinkEditButton, &QToolButton::clicked, this, &QgsFileWidget::editLink );
  mLinkEditButton->hide(); // do not show by default
 
  mFileWidgetButton = new QToolButton( this );
  mFileWidgetButton->setText( QChar( 0x2026 ) );
  mFileWidgetButton->setToolTip( tr( "Browse" ) );
  connect( mFileWidgetButton, &QAbstractButton::clicked, this, &QgsFileWidget::openFileDialog );
  mLayout->addWidget( mFileWidgetButton );
 
  setLayout( mLayout );
}
 
QString QgsFileWidget::filePath()
{
  return mFilePath;
}
 
QStringList QgsFileWidget::splitFilePaths( const QString &path )
{
  QStringList paths;
#if QT_VERSION < QT_VERSION_CHECK(5, 15, 0)
  const QStringList pathParts = path.split( QRegExp( "\"\\s+\"" ), QString::SkipEmptyParts );
#else
  const thread_local QRegularExpression partsRegex = QRegularExpression( QStringLiteral( "\"\\s+\"" ) );
  const QStringList pathParts = path.split( partsRegex, Qt::SkipEmptyParts );
#endif
 
  const thread_local QRegularExpression cleanRe( QStringLiteral( "(^\\s*\")|(\"\\s*)" ) );
  paths.reserve( pathParts.size() );
  for ( const QString &pathsPart : pathParts )
  {
    QString cleaned = pathsPart;
    cleaned.remove( cleanRe );
    paths.append( cleaned );
  }
  return paths;
}
 
void QgsFileWidget::setFilePath( const QString &path )
{
  //will trigger textEdited slot
  mLineEdit->setValue( path );
}
 
void QgsFileWidget::setReadOnly( bool readOnly )
{
  if ( mReadOnly == readOnly )
    return;
 
  mReadOnly = readOnly;
 
  updateLayout();
}
 
QString QgsFileWidget::dialogTitle() const
{
  return mDialogTitle;
}
 
void QgsFileWidget::setDialogTitle( const QString &title )
{
  mDialogTitle = title;
}
 
QString QgsFileWidget::filter() const
{
  return mFilter;
}
 
void QgsFileWidget::setFilter( const QString &filters )
{
  mFilter = filters;
  mLineEdit->setFilters( filters );
}
 
QFileDialog::Options QgsFileWidget::options() const
{
  return mOptions;
}
 
void QgsFileWidget::setOptions( QFileDialog::Options options )
{
  mOptions = options;
}
 
bool QgsFileWidget::fileWidgetButtonVisible() const
{
  return mButtonVisible;
}
 
void QgsFileWidget::setFileWidgetButtonVisible( bool visible )
{
  mButtonVisible = visible;
  mFileWidgetButton->setVisible( visible );
}
 
bool QgsFileWidget::isMultiFiles( const QString &path )
{
  return path.contains( QStringLiteral( "\" \"" ) );
}
 
void QgsFileWidget::textEdited( const QString &path )
{
  mFilePath = path;
  mLinkLabel->setText( toUrl( path ) );
  // Show tooltip if multiple files are selected
  if ( isMultiFiles( path ) )
  {
    mLineEdit->setToolTip( tr( "Selected files:<br><ul><li>%1</li></ul><br>" ).arg( splitFilePaths( path ).join( QLatin1String( "</li><li>" ) ) ) );
  }
  else
  {
    mLineEdit->setToolTip( QString() );
  }
  emit fileChanged( mFilePath );
}
 
void QgsFileWidget::editLink()
{
  if ( !mUseLink || mReadOnly )
    return;
 
  mIsLinkEdited = !mIsLinkEdited;
  updateLayout();
}
 
void QgsFileWidget::fileDropped( const QString &filePath )
{
  setSelectedFileNames( QStringList() << filePath );
  mLineEdit->selectAll();
  mLineEdit->setFocus( Qt::MouseFocusReason );
}
 
bool QgsFileWidget::useLink() const
{
  return mUseLink;
}
 
void QgsFileWidget::setUseLink( bool useLink )
{
  if ( mUseLink == useLink )
    return;
 
  mUseLink = useLink;
  updateLayout();
}
 
bool QgsFileWidget::fullUrl() const
{
  return mFullUrl;
}
 
void QgsFileWidget::setFullUrl( bool fullUrl )
{
  mFullUrl = fullUrl;
}
 
QString QgsFileWidget::defaultRoot() const
{
  return mDefaultRoot;
}
 
void QgsFileWidget::setDefaultRoot( const QString &defaultRoot )
{
  mDefaultRoot = defaultRoot;
}
 
QgsFileWidget::StorageMode QgsFileWidget::storageMode() const
{
  return mStorageMode;
}
 
void QgsFileWidget::setStorageMode( QgsFileWidget::StorageMode storageMode )
{
  mStorageMode = storageMode;
  mLineEdit->setStorageMode( storageMode );
}
 
QgsFileWidget::RelativeStorage QgsFileWidget::relativeStorage() const
{
  return mRelativeStorage;
}
 
void QgsFileWidget::setRelativeStorage( QgsFileWidget::RelativeStorage relativeStorage )
{
  mRelativeStorage = relativeStorage;
}
 
QgsFilterLineEdit *QgsFileWidget::lineEdit()
{
  return mLineEdit;
}
 
void QgsFileWidget::updateLayout()
{
  const bool linkVisible = mUseLink && !mIsLinkEdited;
 
  mLineEdit->setVisible( !linkVisible );
  mLinkLabel->setVisible( linkVisible );
  mLinkEditButton->setVisible( mUseLink && !mReadOnly );
 
  mFileWidgetButton->setEnabled( !mReadOnly );
  mLineEdit->setEnabled( !mReadOnly );
 
  mLinkEditButton->setIcon( linkVisible && !mReadOnly ?
                            QgsApplication::getThemeIcon( QStringLiteral( "/mActionToggleEditing.svg" ) ) :
                            QgsApplication::getThemeIcon( QStringLiteral( "/mActionSaveEdits.svg" ) ) );
}
 
void QgsFileWidget::openFileDialog()
{
  QgsSettings settings;
  QString oldPath;
 
  // if we use a relative path option, we need to obtain the full path
  // first choice is the current file path, if one is entered
  if ( !mFilePath.isEmpty() && ( QFile::exists( mFilePath ) || mStorageMode == SaveFile ) )
  {
    oldPath = relativePath( mFilePath, false );
  }
  // If we use fixed default path
  // second choice is the default root
  else if ( !mDefaultRoot.isEmpty() )
  {
    oldPath = QDir::cleanPath( mDefaultRoot );
  }
 
  // If there is no valid value, find a default path to use
  QUrl url = QUrl::fromUserInput( oldPath );
  if ( !url.isValid() )
  {
    QString defPath = QDir::cleanPath( QFileInfo( QgsProject::instance()->absoluteFilePath() ).path() );
    if ( defPath.isEmpty() )
    {
      defPath = QDir::homePath();
    }
    oldPath = settings.value( QStringLiteral( "UI/lastFileNameWidgetDir" ), defPath ).toString();
  }
 
  // Handle Storage
  QString fileName;
  QStringList fileNames;
  QString title;
 
  {
    QgsFocusKeeper focusKeeper;
    switch ( mStorageMode )
    {
      case GetFile:
        title = !mDialogTitle.isEmpty() ? mDialogTitle : tr( "Select a file" );
        fileName = QFileDialog::getOpenFileName( this, title, QFileInfo( oldPath ).absoluteFilePath(), mFilter, &mSelectedFilter, mOptions );
        break;
      case GetMultipleFiles:
        title = !mDialogTitle.isEmpty() ? mDialogTitle : tr( "Select one or more files" );
        fileNames = QFileDialog::getOpenFileNames( this, title, QFileInfo( oldPath ).absoluteFilePath(), mFilter, &mSelectedFilter, mOptions );
        break;
      case GetDirectory:
        title = !mDialogTitle.isEmpty() ? mDialogTitle : tr( "Select a directory" );
        fileName = QFileDialog::getExistingDirectory( this, title, QFileInfo( oldPath ).absoluteFilePath(), mOptions );
        break;
      case SaveFile:
      {
        title = !mDialogTitle.isEmpty() ? mDialogTitle : tr( "Create or select a file" );
        if ( !confirmOverwrite() )
        {
          fileName = QFileDialog::getSaveFileName( this, title, QFileInfo( oldPath ).absoluteFilePath(), mFilter, &mSelectedFilter, mOptions | QFileDialog::DontConfirmOverwrite );
        }
        else
        {
          fileName = QFileDialog::getSaveFileName( this, title, QFileInfo( oldPath ).absoluteFilePath(), mFilter, &mSelectedFilter, mOptions );
        }
 
        // make sure filename ends with filter. This isn't automatically done by
        // getSaveFileName on some platforms (e.g. gnome)
        fileName = QgsFileUtils::addExtensionFromFilter( fileName, mSelectedFilter );
      }
      break;
    }
  }
 
  if ( fileName.isEmpty() && fileNames.isEmpty( ) )
    return;
 
  if ( mStorageMode != GetMultipleFiles )
    fileNames << fileName;
 
  for ( int i = 0; i < fileNames.length(); i++ )
  {
    fileNames.replace( i, QDir::toNativeSeparators( QDir::cleanPath( QFileInfo( fileNames.at( i ) ).absoluteFilePath() ) ) );
  }
 
  // Store the last used path:
  switch ( mStorageMode )
  {
    case GetFile:
    case SaveFile:
    case GetMultipleFiles:
      settings.setValue( QStringLiteral( "UI/lastFileNameWidgetDir" ), QFileInfo( fileNames.first() ).absolutePath() );
      break;
    case GetDirectory:
      settings.setValue( QStringLiteral( "UI/lastFileNameWidgetDir" ), fileNames.first() );
      break;
  }
 
  setSelectedFileNames( fileNames );
}
 
void QgsFileWidget::setSelectedFileNames( QStringList fileNames )
{
  Q_ASSERT( fileNames.count() );
 
  // Handle relative Path storage
  for ( int i = 0; i < fileNames.length(); i++ )
  {
    fileNames.replace( i, relativePath( fileNames.at( i ), true ) );
  }
 
  setFilePaths( fileNames );
}
 
void QgsFileWidget::setFilePaths( const QStringList &filePaths )
{
  if ( mStorageMode != GetMultipleFiles )
  {
    setFilePath( filePaths.first() );
  }
  else
  {
    if ( filePaths.length() > 1 )
    {
      setFilePath( QStringLiteral( "\"%1\"" ).arg( filePaths.join( QLatin1String( "\" \"" ) ) ) );
    }
    else
    {
      setFilePath( filePaths.first( ) );
    }
  }
}
 
QString QgsFileWidget::relativePath( const QString &filePath, bool removeRelative ) const
{
  QString RelativePath;
  if ( mRelativeStorage == RelativeProject )
  {
    RelativePath = QDir::toNativeSeparators( QDir::cleanPath( QFileInfo( QgsProject::instance()->absoluteFilePath() ).path() ) );
  }
  else if ( mRelativeStorage == RelativeDefaultPath && !mDefaultRoot.isEmpty() )
  {
    RelativePath = QDir::toNativeSeparators( QDir::cleanPath( mDefaultRoot ) );
  }
 
  if ( !RelativePath.isEmpty() )
  {
    if ( removeRelative )
    {
      return QDir::cleanPath( QDir( RelativePath ).relativeFilePath( filePath ) );
    }
    else
    {
      return QDir::cleanPath( QDir( RelativePath ).filePath( filePath ) );
    }
  }
 
  return filePath;
}
 
 
QString QgsFileWidget::toUrl( const QString &path ) const
{
  QString rep;
  if ( path.isEmpty() )
  {
    return QgsApplication::nullRepresentation();
  }
 
  if ( isMultiFiles( path ) )
  {
    return QStringLiteral( "<a>%1</a>" ).arg( path );
  }
 
  QString urlStr = relativePath( path, false );
  QUrl url = QUrl::fromUserInput( urlStr );
  if ( !url.isValid() || !url.isLocalFile() )
  {
    QgsDebugMsgLevel( QStringLiteral( "URL: %1 is not valid or not a local file!" ).arg( path ), 2 );
    rep = path;
  }
 
  QString pathStr = url.toString();
  if ( mFullUrl )
  {
    rep = QStringLiteral( "<a href=\"%1\">%2</a>" ).arg( pathStr, path );
  }
  else
  {
    QString fileName = QFileInfo( urlStr ).fileName();
    rep = QStringLiteral( "<a href=\"%1\">%2</a>" ).arg( pathStr, fileName );
  }
 
  return rep;
}
 
 
 
 
QgsFileDropEdit::QgsFileDropEdit( QWidget *parent )
  : QgsHighlightableLineEdit( parent )
{
  setAcceptDrops( true );
}
 
void QgsFileDropEdit::setFilters( const QString &filters )
{
  mAcceptableExtensions.clear();
 
  if ( filters.contains( QStringLiteral( "*.*" ) ) )
    return; // everything is allowed!
 
  QRegularExpression rx( QStringLiteral( "\\*\\.(\\w+)" ) );
  QRegularExpressionMatchIterator i = rx.globalMatch( filters );
  while ( i.hasNext() )
  {
    QRegularExpressionMatch match = i.next();
    if ( match.hasMatch() )
    {
      mAcceptableExtensions << match.captured( 1 ).toLower();
    }
  }
}
 
QStringList QgsFileDropEdit::acceptableFilePaths( QDropEvent *event ) const
{
  QStringList rawPaths;
  QStringList paths;
  if ( event->mimeData()->hasUrls() )
  {
    const QList< QUrl > urls = event->mimeData()->urls();
    rawPaths.reserve( urls.count() );
    for ( const QUrl &url : urls )
    {
      const QString local =  url.toLocalFile();
      if ( !rawPaths.contains( local ) )
        rawPaths.append( local );
    }
  }
 
  QgsMimeDataUtils::UriList lst = QgsMimeDataUtils::decodeUriList( event->mimeData() );
  for ( const QgsMimeDataUtils::Uri &u : std::as_const( lst ) )
  {
    if ( !rawPaths.contains( u.uri ) )
      rawPaths.append( u.uri );
  }
 
  if ( !event->mimeData()->text().isEmpty() && !rawPaths.contains( event->mimeData()->text() ) )
    rawPaths.append( event->mimeData()->text() );
 
  paths.reserve( rawPaths.count() );
  for ( const QString &path : std::as_const( rawPaths ) )
  {
    QFileInfo file( path );
    switch ( mStorageMode )
    {
      case QgsFileWidget::GetFile:
      case QgsFileWidget::GetMultipleFiles:
      case QgsFileWidget::SaveFile:
      {
        if ( file.isFile() && ( mAcceptableExtensions.isEmpty() || mAcceptableExtensions.contains( file.suffix(), Qt::CaseInsensitive ) ) )
          paths.append( file.filePath() );
 
        break;
      }
 
      case QgsFileWidget::GetDirectory:
      {
        if ( file.isDir() )
          paths.append( file.filePath() );
        else if ( file.isFile() )
        {
          // folder mode, but a file dropped. So get folder name from file
          paths.append( file.absolutePath() );
        }
 
        break;
      }
    }
  }
 
  return paths;
}
 
QString QgsFileDropEdit::acceptableFilePath( QDropEvent *event ) const
{
  const QStringList paths = acceptableFilePaths( event );
  if ( paths.size() > 1 )
  {
    return QStringLiteral( "\"%1\"" ).arg( paths.join( QLatin1String( "\" \"" ) ) );
  }
  else if ( paths.size() == 1 )
  {
    return paths.first();
  }
  else
  {
    return QString();
  }
}
 
void QgsFileDropEdit::dragEnterEvent( QDragEnterEvent *event )
{
  QString filePath = acceptableFilePath( event );
  if ( !filePath.isEmpty() )
  {
    event->acceptProposedAction();
    setHighlighted( true );
  }
  else
  {
    event->ignore();
  }
}
 
void QgsFileDropEdit::dragLeaveEvent( QDragLeaveEvent *event )
{
  QgsFilterLineEdit::dragLeaveEvent( event );
  event->accept();
  setHighlighted( false );
}
 
void QgsFileDropEdit::dropEvent( QDropEvent *event )
{
  QString filePath = acceptableFilePath( event );
  if ( !filePath.isEmpty() )
  {
    event->acceptProposedAction();
    emit fileDropped( filePath );
  }
 
  setHighlighted( false );
}