OSDN Git Service

Added the copy_file() function + some improvements to directory clean-up code.
[mutilities/MUtilities.git] / src / Global.cpp
index b560432..17db27c 100644 (file)
 #define _CRT_RAND_S 1
 #endif
 
+//MUtils
 #include <MUtils/Global.h>
 #include <MUtils/OSSupport.h>
 
+//Internal
+#include "DirLocker.h"
+#include "3rd_party/strnatcmp/include/strnatcmp.h"
+
 //Qt
 #include <QDir>
 #include <QReadWriteLock>
 #include <QProcess>
+#include <QTextCodec>
 
 //CRT
 #include <cstdlib>
 #include <ctime>
 #include <process.h>
 
+//VLD
+#ifdef _MSC_VER
+#include <vld.h>
+#endif
+
 ///////////////////////////////////////////////////////////////////////////////
 // Random Support
 ///////////////////////////////////////////////////////////////////////////////
@@ -102,9 +113,8 @@ QString MUtils::rand_str(const bool &bLong)
 // TEMP FOLDER
 ///////////////////////////////////////////////////////////////////////////////
 
-static QScopedPointer<QFile>   g_temp_folder_file;
-static QScopedPointer<QString> g_temp_folder_path;
-static QReadWriteLock          g_temp_folder_lock;
+static QScopedPointer<MUtils::Internal::DirLock> g_temp_folder_file;
+static QReadWriteLock                            g_temp_folder_lock;
 
 static QString try_create_subfolder(const QString &baseDir, const QString &postfix)
 {
@@ -120,29 +130,37 @@ static QString try_create_subfolder(const QString &baseDir, const QString &postf
        return QString();
 }
 
-static QString try_init_temp_folder(const QString &baseDir)
+static MUtils::Internal::DirLock *try_init_temp_folder(const QString &baseDir)
 {
-       static const char *TEST_DATA = "Lorem ipsum dolor sit amet, consectetur, adipisci velit!";
-       
-       QString tempPath = try_create_subfolder(baseDir, MUtils::rand_str());
+       const QString tempPath = try_create_subfolder(baseDir, MUtils::rand_str());
        if(!tempPath.isEmpty())
        {
-               const QByteArray testData = QByteArray(TEST_DATA);
                for(int i = 0; i < 32; i++)
                {
-                       g_temp_folder_file.reset(new QFile(QString("%1/~%2.lck").arg(tempPath, MUtils::rand_str())));
-                       if(g_temp_folder_file->open(QIODevice::ReadWrite | QIODevice::Truncate))
+                       MUtils::Internal::DirLock *lockFile = NULL;
+                       try
                        {
-                               if(g_temp_folder_file->write(testData) >= testData.size())
-                               {
-                                       return tempPath;
-                               }
-                               g_temp_folder_file->remove();
+                               lockFile = new MUtils::Internal::DirLock(tempPath);
+                               return lockFile;
+                       }
+                       catch(MUtils::Internal::DirLockException&)
+                       {
+                               /*ignore error and try again*/
                        }
                }
        }
+       return NULL;
+}
 
-       return QString();
+static void temp_folder_cleaup(void)
+{
+       QWriteLocker writeLock(&g_temp_folder_lock);
+       
+       //Clean the directory
+       while(!g_temp_folder_file.isNull())
+       {
+               g_temp_folder_file.reset(NULL);
+       }
 }
 
 const QString &MUtils::temp_folder(void)
@@ -150,9 +168,9 @@ const QString &MUtils::temp_folder(void)
        QReadLocker readLock(&g_temp_folder_lock);
 
        //Already initialized?
-       if((!g_temp_folder_path.isNull()) && (!g_temp_folder_path->isEmpty()))
+       if(!g_temp_folder_file.isNull())
        {
-               return (*g_temp_folder_path.data());
+               return g_temp_folder_file->getPath();
        }
 
        //Obtain the write lock to initilaize
@@ -160,17 +178,17 @@ const QString &MUtils::temp_folder(void)
        QWriteLocker writeLock(&g_temp_folder_lock);
        
        //Still uninitilaized?
-       if((!g_temp_folder_path.isNull()) && (!g_temp_folder_path->isEmpty()))
+       if(!g_temp_folder_file.isNull())
        {
-               return (*g_temp_folder_path.data());
+               return g_temp_folder_file->getPath();
        }
 
        //Try the %TMP% or %TEMP% directory first
-       QString tempPath = try_init_temp_folder(QDir::tempPath());
-       if(!tempPath.isEmpty())
+       if(MUtils::Internal::DirLock *lockFile = try_init_temp_folder(QDir::tempPath()))
        {
-               g_temp_folder_path.reset(new QString(tempPath));
-               return (*g_temp_folder_path.data());
+               g_temp_folder_file.reset(lockFile);
+               atexit(temp_folder_cleaup);
+               return lockFile->getPath();
        }
 
        qWarning("%%TEMP%% directory not found -> trying fallback mode now!");
@@ -183,11 +201,11 @@ const QString &MUtils::temp_folder(void)
                        const QString tempRoot = try_create_subfolder(knownFolder, QLatin1String("TEMP"));
                        if(!tempRoot.isEmpty())
                        {
-                               tempPath = try_init_temp_folder(tempRoot);
-                               if(!tempPath.isEmpty())
+                               if(MUtils::Internal::DirLock *lockFile = try_init_temp_folder(tempRoot))
                                {
-                                       g_temp_folder_path.reset(new QString(tempPath));
-                                       return (*g_temp_folder_path.data());
+                                       g_temp_folder_file.reset(lockFile);
+                                       atexit(temp_folder_cleaup);
+                                       return lockFile->getPath();
                                }
                        }
                }
@@ -198,6 +216,98 @@ const QString &MUtils::temp_folder(void)
 }
 
 ///////////////////////////////////////////////////////////////////////////////
+// REMOVE DIRECTORY / FILE
+///////////////////////////////////////////////////////////////////////////////
+
+bool MUtils::remove_file(const QString &fileName)
+{
+       QFileInfo fileInfo(fileName);
+       if(!(fileInfo.exists() && fileInfo.isFile()))
+       {
+               return true;
+       }
+
+       for(int i = 0; i < 32; i++)
+       {
+               QFile file(fileName);
+               file.setPermissions(QFile::ReadOther | QFile::WriteOther);
+               if((!(fileInfo.exists() && fileInfo.isFile())) || file.remove())
+               {
+                       return true;
+               }
+               fileInfo.refresh();
+       }
+
+       qWarning("Could not delete \"%s\"", MUTILS_UTF8(fileName));
+       return false;
+}
+
+static bool remove_directory_helper(QDir folder)
+{
+       if(!folder.exists())
+       {
+               return true;
+       }
+       
+       const QString dirName = folder.dirName();
+       if(dirName.isEmpty() || (!folder.cdUp()))
+       {
+               return false;
+       }
+
+       return folder.rmdir(dirName);
+}
+
+bool MUtils::remove_directory(const QString &folderPath, const bool &recursive)
+{
+       QDir folder(folderPath);
+       if(!folder.exists())
+       {
+               return true;
+       }
+
+       if(recursive)
+       {
+               const QFileInfoList entryList = folder.entryInfoList(QDir::AllEntries | QDir::NoDotAndDotDot | QDir::Hidden);
+               for(QFileInfoList::ConstIterator iter = entryList.constBegin(); iter != entryList.constEnd(); iter++)
+               {
+                       if(iter->isDir())
+                       {
+                               remove_directory(iter->canonicalFilePath(), true);
+                       }
+                       else if(iter->isFile())
+                       {
+                               remove_file(iter->canonicalFilePath());
+                       }
+               }
+       }
+
+       for(int i = 0; i < 32; i++)
+       {
+               if(!folder.exists())
+               {
+                       return true;
+               }
+               const QString dirName = folder.dirName();
+               if(!dirName.isEmpty())
+               {
+                       QDir parent(folder);
+                       if(parent.cdUp())
+                       {
+                               if(parent.rmdir(dirName))
+                               {
+                                       return true;
+                               }
+                       }
+               }
+               folder.refresh();
+       }
+       
+       qWarning("Could not rmdir \"%s\"", MUTILS_UTF8(folderPath));
+       return false;
+}
+
+///////////////////////////////////////////////////////////////////////////////
 // PROCESS UTILS
 ///////////////////////////////////////////////////////////////////////////////
 
@@ -247,30 +357,152 @@ void MUtils::init_process(QProcess &process, const QString &wokringDir, const bo
 }
 
 ///////////////////////////////////////////////////////////////////////////////
-// LIB VERSION
+// NATURAL ORDER STRING COMPARISON
+///////////////////////////////////////////////////////////////////////////////
+
+static bool natural_string_sort_helper(const QString &str1, const QString &str2)
+{
+       return (MUtils::Internal::NaturalSort::strnatcmp(MUTILS_WCHR(str1), MUTILS_WCHR(str2)) < 0);
+}
+
+static bool natural_string_sort_helper_fold_case(const QString &str1, const QString &str2)
+{
+       return (MUtils::Internal::NaturalSort::strnatcasecmp(MUTILS_WCHR(str1), MUTILS_WCHR(str2)) < 0);
+}
+
+void MUtils::natural_string_sort(QStringList &list, const bool bIgnoreCase)
+{
+       qSort(list.begin(), list.end(), bIgnoreCase ? natural_string_sort_helper_fold_case : natural_string_sort_helper);
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// CLEAN FILE PATH
 ///////////////////////////////////////////////////////////////////////////////
 
-const char* MUtils::mutils_build_date(void)
+static const struct
 {
-       static const char *const BUILD_DATE = __DATE__;
-       return BUILD_DATE;
+       const char *const search;
+       const char *const replace;
+}
+CLEAN_FILE_NAME[] =
+{
+       { "\\",  "-"  },
+       { " / ", ", " },
+       { "/",   ","  },
+       { ":",   "-"  },
+       { "*",   "x"  },
+       { "?",   "!"  },
+       { "<",   "["  },
+       { ">",   "]"  },
+       { "|",   "!"  },
+       { "\"",  "'"  },
+       { NULL,  NULL }
+};
+
+QString MUtils::clean_file_name(const QString &name)
+{
+       QString str = name.simplified();
+
+       for(size_t i = 0; CLEAN_FILE_NAME[i].search; i++) 
+       {
+               str.replace(CLEAN_FILE_NAME[i].search, CLEAN_FILE_NAME[i].replace);
+       }
+       
+       QRegExp regExp("\"(.+)\"");
+       regExp.setMinimal(true);
+       str.replace(regExp, "`\\1ยด");
+       
+       return str.simplified();
 }
 
-const char* MUtils::mutils_build_time(void)
+QString MUtils::clean_file_path(const QString &path)
 {
-       static const char *const BUILD_TIME = __TIME__;
-       return BUILD_TIME;
+       QStringList parts = path.simplified().replace("\\", "/").split("/", QString::SkipEmptyParts);
+
+       for(int i = 0; i < parts.count(); i++)
+       {
+               parts[i] = MUtils::clean_file_name(parts[i]);
+       }
+
+       return parts.join("/");
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// REGULAR EXPESSION HELPER
+///////////////////////////////////////////////////////////////////////////////
+
+bool MUtils::regexp_parse_uint32(const QRegExp &regexp, quint32 &value)
+{
+       return regexp_parse_uint32(regexp, &value, 1);
+}
+
+bool MUtils::regexp_parse_uint32(const QRegExp &regexp, quint32 *values, const size_t &count)
+{
+       const QStringList caps = regexp.capturedTexts();
+       
+       if(caps.isEmpty() || (quint32(caps.count()) <= count))
+       {
+               return false;
+       }
+
+       for(size_t i = 0; i < count; i++)
+       {
+               bool ok = false;
+               values[i] = caps[i+1].toUInt(&ok);
+               if(!ok)
+               {
+                       return false;
+               }
+       }
+
+       return true;
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// AVAILABLE CODEPAGES
+///////////////////////////////////////////////////////////////////////////////
+
+QStringList MUtils::available_codepages(const bool &noAliases)
+{
+       QStringList codecList;
+       QList<QByteArray> availableCodecs = QTextCodec::availableCodecs();
+
+       while(!availableCodecs.isEmpty())
+       {
+               const QByteArray current = availableCodecs.takeFirst();
+               if(!current.toLower().startsWith("system"))
+               {
+                       codecList << QString::fromLatin1(current.constData(), current.size());
+                       if(noAliases)
+                       {
+                               if(QTextCodec *const currentCodec = QTextCodec::codecForName(current.constData()))
+                               {
+                                       const QList<QByteArray> aliases = currentCodec->aliases();
+                                       for(QList<QByteArray>::ConstIterator iter = aliases.constBegin(); iter != aliases.constEnd(); iter++)
+                                       {
+                                               availableCodecs.removeAll(*iter);
+                                       }
+                               }
+                       }
+               }
+       }
+
+       return codecList;
 }
 
 ///////////////////////////////////////////////////////////////////////////////
 // SELF-TEST
 ///////////////////////////////////////////////////////////////////////////////
 
-int MUtils::Internal::selfTest(const char *const date, const bool debug)
+int MUtils::Internal::selfTest(const char *const buildKey, const bool debug)
 {
-       if(strcmp(date, __DATE__) || (MUTILS_DEBUG != debug))
+       static const bool MY_DEBUG_FLAG = MUTILS_DEBUG;
+       static const char *const MY_BUILD_KEY = __DATE__"@"__TIME__;
+
+       if(strncmp(buildKey, MY_BUILD_KEY, 14) || (MY_DEBUG_FLAG != debug))
        {
                MUtils::OS::system_message_err(L"MUtils", L"FATAL ERROR: MUtils library version mismatch detected!");
+               MUtils::OS::system_message_wrn(L"MUtils", L"Please re-build the complete solution in order to fix this issue!");
                abort();
        }
        return 0;