File indexing completed on 2024-04-28 15:15:40

 // SPDX-License-Identifier: LGPL-2.1-or-later
 //
 // SPDX-FileCopyrightText: 2007 Tobias Koenig <tokoe@kde.org>
 //
 
 
 // Own
 #include "FileStoragePolicy.h"
 
 // Qt
 #include <QDir>
 #include <QDirIterator>
 #include <QFile>
 #include <QFileInfo>
 
 // Marble
 #include "MarbleDebug.h"
 #include "MarbleGlobal.h"
 #include "MarbleDirs.h"
 
 using namespace Marble;
 
 FileStoragePolicy::FileStoragePolicy( const QString &dataDirectory, QObject *parent )
     : StoragePolicy( parent ),
       m_dataDirectory( dataDirectory )
 {
     if ( m_dataDirectory.isEmpty() )
         m_dataDirectory = MarbleDirs::localPath() + QLatin1String("/cache/");
 
     if ( !QDir( m_dataDirectory ).exists() ) 
         QDir::root().mkpath( m_dataDirectory );
 }
 
 FileStoragePolicy::~FileStoragePolicy()
 {
 }
 
 bool FileStoragePolicy::fileExists( const QString &fileName ) const
 {
     const QString fullName = m_dataDirectory + QLatin1Char('/') + fileName;
     return QFile::exists( fullName );
 }
 
 bool FileStoragePolicy::updateFile( const QString &fileName, const QByteArray &data )
 {
     QFileInfo const dirInfo( fileName );
     QString const fullName = dirInfo.isAbsolute() ? fileName : m_dataDirectory + QLatin1Char('/') + fileName;
 
     // Create directory if it doesn't exist yet...
     QFileInfo info( fullName );
 
     const QDir localFileDir = info.dir();
     const QString localFileDirPath = localFileDir.absolutePath();
 
     if ( !QDir( localFileDirPath ).exists() )
         QDir::root().mkpath( localFileDirPath );
 
     // ... and save the file content
     QFile file( fullName );
     if ( !file.open( QIODevice::WriteOnly ) ) {
         m_errorMsg = fullName + QLatin1String(": ") + file.errorString();
         qCritical() << "file.open" << m_errorMsg;
         return false;
     }
 
     quint64 oldSize = file.size();
 
     if ( !file.write( data ) ) {
         m_errorMsg = fullName + QLatin1String(": ") + file.errorString();
         qCritical() << "file.write" << m_errorMsg;
         emit sizeChanged( file.size() - oldSize );
         return false;
     }
 
     emit sizeChanged( file.size() - oldSize );
     file.close();
 
     return true;
 }
 
 void FileStoragePolicy::clearCache()
 {
     mDebug();
     if ( m_dataDirectory.isEmpty()
     || !(m_dataDirectory.endsWith(QLatin1String( "data" ))   // on Windows
     ||   m_dataDirectory.endsWith(QLatin1String( "marble" )) // on all other OSes
     ))
     {
         mDebug() << "Data Directory:" << m_dataDirectory;
         mDebug() << "Error: Refusing to erase files under unknown conditions for safety reasons!";
         return;
     }
 
     const QString cachedMapsDirectory = m_dataDirectory + QLatin1String("/maps");
 
     QDirIterator it( cachedMapsDirectory, QDir::NoDotAndDotDot | QDir::Dirs );
     mDebug() << cachedMapsDirectory;
     while (it.hasNext()) {
         it.next();
         QString planetDirectory = it.filePath();
         QDirIterator itPlanet( planetDirectory, QDir::NoDotAndDotDot | QDir::Dirs );
         while (itPlanet.hasNext()) {
             itPlanet.next();
             QString themeDirectory = itPlanet.filePath();
             QDirIterator itTheme( themeDirectory, QDir::NoDotAndDotDot | QDir::Dirs );
             while (itTheme.hasNext()) {
                 itTheme.next();
                 QString tileDirectory = itTheme.filePath();
 
                 if ( itTheme.fileName().toInt() <= maxBaseTileLevel ) {
                     continue;
                 }
 
                 QDirIterator itTile( tileDirectory, QDir::Files | QDir::NoSymLinks, QDirIterator::Subdirectories );
                 while (itTile.hasNext()) {
                     itTile.next();
                     QString filePath = itTile.filePath();
                     QString lowerCase = filePath.toLower();
 
                     // We try to be very careful and just delete images
                     // FIXME, when vectortiling I suppose also vector tiles will have
                     // to be deleted
                     if ( lowerCase.endsWith( QLatin1String( ".jpg" ) ) 
                       || lowerCase.endsWith( QLatin1String( ".png" ) )
                       || lowerCase.endsWith( QLatin1String( ".gif" ) )
                       || lowerCase.endsWith( QLatin1String( ".svg" ) )
                       || lowerCase.endsWith( QLatin1String( ".o5m" ) )
                     )
                     {
                         // We cannot emit clear, because we don't make a full clear
                         QFile file( filePath );
                         emit sizeChanged( -file.size() );
                         file.remove();
                     }
                 }
             }
         }
     }
 }
 
 QString FileStoragePolicy::lastErrorMessage() const
 {
     return m_errorMsg;
 }
 
 #include "moc_FileStoragePolicy.cpp"