#include <common/net/message/control/GenericResponseMsg.h>
#include <common/net/message/storage/TruncFileRespMsg.h>
#include <common/toolkit/MessagingTk.h>
#include <components/FileEventLogger.h>
#include <net/msghelpers/MsgHelperTrunc.h>
#include "TruncFileMsgEx.h"
#include <program/Program.h>

FileIDLock TruncFileMsgEx::lock(EntryLockStore& store)
{
   return {&store, getEntryInfo()->getEntryID(), true};
}

bool TruncFileMsgEx::processIncoming(ResponseContext& ctx)
{
   return BaseType::processIncoming(ctx);
}

std::unique_ptr<MirroredMessageResponseState> TruncFileMsgEx::executeLocally(ResponseContext& ctx,
   bool isSecondary)
{
   App* app = Program::getApp();
   // update operation counters
   updateNodeOp(ctx, MetaOpCounter_TRUNCATE);

   MetaStore* metaStore = Program::getApp()->getMetaStore();

   if (!isSecondary)
   {
      FhgfsOpsErr truncRes = MsgHelperTrunc::truncFile(getEntryInfo(), getFilesize(),
            isMsgHeaderFeatureFlagSet(TRUNCFILEMSG_FLAG_USE_QUOTA), getMsgHeaderUserID(),
            dynAttribs);
      if (truncRes == FhgfsOpsErr_SUCCESS && (shouldFixTimestamps() || getFileEvent()))
      {
         auto [inode, referenceRes] = metaStore->referenceFile(getEntryInfo());
         unsigned numHardlinks = 0;

         if (likely(inode))
         {
            if (shouldFixTimestamps())
               fixInodeTimestamp(*inode, mirroredTimestamps, nullptr);

            numHardlinks = inode->getNumHardlinks();
            metaStore->releaseFile(getEntryInfo()->getParentEntryID(), inode);
         }

         if (app->getFileEventLogger() && getFileEvent())
         {
            EventContext eventCtx = makeEventContext(
               getEntryInfo(),
               getEntryInfo()->getParentEntryID(),
               getMsgHeaderUserID(),
               "",
               numHardlinks,
               isSecondary
            );

            logEvent(app->getFileEventLogger(), *getFileEvent(), eventCtx);
         }
      }
      return boost::make_unique<ResponseState>(truncRes);
   }

   auto [inode, referenceRes] = metaStore->referenceFile(getEntryInfo());
   if(!inode)
      return boost::make_unique<ResponseState>(referenceRes);

   inode->setDynAttribs(dynAttribs);
   if (shouldFixTimestamps())
      fixInodeTimestamp(*inode, mirroredTimestamps, nullptr);
   inode->updateInodeOnDisk(getEntryInfo());

   metaStore->releaseFile(getEntryInfo()->getParentEntryID(), inode);

   return boost::make_unique<ResponseState>(FhgfsOpsErr_SUCCESS);
}

void TruncFileMsgEx::forwardToSecondary(ResponseContext& ctx)
{
   sendToSecondary(ctx, *this, NETMSGTYPE_TruncFileResp);
}