Support for user file quota

Author: tomas-soltys

src/cloud/src/file_index.cpp

@@ -108,18 +108,36 @@ qsizetype FileIndex::getSize() const
     return this->index.size();
 }
 
-qint64 FileIndex::findStoreSize() const
+qint64 FileIndex::findStoreSize(const QString &user) const
 {
     qint64 totalSize = 0;
 
     for (auto it = this->index.cbegin(); it != this->index.cend(); ++it)
     {
-        totalSize += it.value().getSize();
+        if (user.isEmpty() || user == it.value().getAccessRights().getOwner().getUser())
+        {
+            totalSize += it.value().getSize();
+        }
     }
 
     return totalSize;
 }
 
+qint64 FileIndex::findStoreCount(const QString &user) const
+{
+    qint64 totalCount = 0;
+
+    for (auto it = this->index.cbegin(); it != this->index.cend(); ++it)
+    {
+        if (user.isEmpty() || user == it.value().getAccessRights().getOwner().getUser())
+        {
+            totalCount++;
+        }
+    }
+
+    return totalCount;
+}
+
 QJsonObject FileIndex::getStatisticsJson() const
 {
     RLogger::debug("[%s] Producting statistics\n",QString("FileIndex").toUtf8().constData());

src/cloud/src/file_index.h

@@ -71,8 +71,11 @@ class FileIndex
         //! Return size of the index (number of entries).
         qsizetype getSize() const;
 
-        //! Find store size.
-        qint64 findStoreSize() const;
+        //! Find store size (total file size).
+        qint64 findStoreSize(const QString &user = QString()) const;
+
+        //! Find store count (total file count).
+        qint64 findStoreCount(const QString &user = QString()) const;
 
         //! Get statistics output in Json form.
         QJsonObject getStatisticsJson() const;

src/cloud/src/file_manager.cpp

@@ -387,6 +387,16 @@ RError::Type FileManager::storeFile(const RUserInfo &executor, const FileObject
         R_LOG_TRACE_RETURN(RError::Unauthorized);
     }
 
+    RFileQuota userStoreQuota(this->fileIndex.findStoreSize(executor.getName())+object.getContent().size(),
+                              object.getContent().size(),
+                              this->fileIndex.findStoreCount(executor.getName())+1);
+
+    if (executor.getFileQuota().quotaExceeded(userStoreQuota))
+    {
+        RLogger::error("[%s] User file quota exceeded.\n",this->settings.getName().toUtf8().constData());
+        R_LOG_TRACE_RETURN(RError::InvalidInput);
+    }
+
     if (this->settings.getMaxFileSize() > 0)
     {
         if (object.getContent().size() > this->settings.getMaxFileSize())
@@ -523,6 +533,16 @@ RError::Type FileManager::updateFile(const RUserInfo &executor, const FileObject
         R_LOG_TRACE_RETURN(RError::Unauthorized);
     }
 
+    RFileQuota userStoreQuota(this->fileIndex.findStoreSize(executor.getName()) + object.getContent().size() - fileInfo.getSize(),
+                              object.getContent().size() - fileInfo.getSize(),
+                              this->fileIndex.findStoreCount(executor.getName()));
+
+    if (executor.getFileQuota().quotaExceeded(userStoreQuota))
+    {
+        RLogger::error("[%s] User file quota exceeded.\n",this->settings.getName().toUtf8().constData());
+        R_LOG_TRACE_RETURN(RError::InvalidInput);
+    }
+
     if (!RFileInfo::isPathValid(object.getInfo().getPath()))
     {
         output = QString("Invalid path \"%1\"").arg(object.getInfo().getPath()).toUtf8();