#include <common/app/log/LogContext.h>
#include <common/net/message/storage/quota/SetExceededQuotaRespMsg.h>
#include <common/storage/StoragePool.h>
#include <program/Program.h>

#include "SetExceededQuotaMsgEx.h"

bool SetExceededQuotaMsgEx::processIncoming(ResponseContext& ctx)
{
   LogContext log("SetExceededQuotaMsgEx incoming");

   bool retVal = true;
   FhgfsOpsErr errorCode = FhgfsOpsErr_SUCCESS;

   if(Program::getApp()->getConfig()->getQuotaEnableEnforcement() )
   {
      // get the storage pool for which quota is exceeded
      StoragePoolPtr storagePool =
         Program::getApp()->getStoragePoolStore()->getPool(getStoragePoolId());

      if (!storagePool)
      {
         LOG(QUOTA, WARNING, "Couldn't set exceeded quota, "
                      "because requested storage pool doesn't exist on metadata server.",
                      ("storagePoolId", getStoragePoolId()));

         errorCode = FhgfsOpsErr_UNKNOWNPOOL;

         goto send_response;
      }

      // set exceeded quota info for all of its targets
      UInt16Set targetIds = storagePool->getTargets();

      for (auto  targetId : targetIds)
      {
         // update exceeded quota
         ExceededQuotaStorePtr exQuotaStore =
               Program::getApp()->getExceededQuotaStores()->get(targetId);
         if (!exQuotaStore)
         {
            LOG(QUOTA, ERR, "Could not access exceeded quota store.", targetId);
            errorCode = FhgfsOpsErr_UNKNOWNTARGET;
            goto send_response;
         }
         exQuotaStore->updateExceededQuota(getExceededQuotaIDs(), getQuotaDataType(),
            getExceededType());
      }
   }
   else
   {
      log.log(Log_ERR, "Unable to set exceeded quota IDs. Configuration problem detected. "
         "The management daemon on " + ctx.peerName() + " has quota enforcement enabled, "
         "but not this storage daemon. Fix this configuration problem or quota enforcement will "
         "not work correctly. If quota enforcement settings have changed recently in the "
         "mgmtd configuration, please restart all BeeGFS services.");

      errorCode = FhgfsOpsErr_INTERNAL;
   }

send_response:
   ctx.sendResponse(SetExceededQuotaRespMsg(errorCode) );

   return retVal;
}