#include <common/net/message/storage/attribs/GetEntryInfoRespMsg.h>
#include <common/storage/striping/Raid0Pattern.h>
#include <common/toolkit/MessagingTk.h>
#include <program/Program.h>
#include <storage/MetaStore.h>
#include "GetEntryInfoMsgEx.h"


bool GetEntryInfoMsgEx::processIncoming(ResponseContext& ctx)
{
#ifdef BEEGFS_DEBUG
   const char* logContext = "GetEntryInfoMsg incoming";
#endif // BEEGFS_DEBUG

   LOG_DEBUG(logContext, Log_SPAM,
      "ParentEntryID: " + this->getEntryInfo()->getParentEntryID() + "; "
      "entryID: " + this->getEntryInfo()->getParentEntryID() );

   return BaseType::processIncoming(ctx);
}

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

std::unique_ptr<MirroredMessageResponseState> GetEntryInfoMsgEx::executeLocally(ResponseContext& ctx,
   bool isSecondary)
{

   GetEntryInfoMsgResponseState resp;

   EntryInfo* entryInfo = this->getEntryInfo();
   StripePattern* pattern = NULL;
   FhgfsOpsErr getInfoRes;
   PathInfo pathInfo;
   RemoteStorageTarget rstInfo;
   uint32_t numSessionsRead = 0;
   uint32_t numSessionsWrite = 0;
   uint8_t dataState = 0;

   if (entryInfo->getParentEntryID().empty())
   {
      // special case: get info for root directory
      // no pathInfo here, as this is currently only used for fileInodes
      getInfoRes = getRootInfo(&pattern, &rstInfo);
   }
   else
   {
      getInfoRes = getInfo(entryInfo, &pattern, &pathInfo, &rstInfo, numSessionsRead,
         numSessionsWrite, dataState);
   }

   if (getInfoRes != FhgfsOpsErr_SUCCESS)
   { // error occurred => create a dummy pattern
      UInt16Vector dummyStripeNodes;
      pattern = new Raid0Pattern(1, dummyStripeNodes);
   }

   resp.setGetEntryInfoResult(getInfoRes);
   resp.setStripePattern(pattern);
   resp.setPathInfo(pathInfo);
   resp.setRemoteStorageTarget(rstInfo);
   resp.setNumSessionsRead(numSessionsRead);
   resp.setNumSessionsWrite(numSessionsWrite);
   resp.setFileDataState(dataState);

   App* app = Program::getApp();
   app->getNodeOpStats()->updateNodeOp(ctx.getSocket()->getPeerIP(), MetaOpCounter_GETENTRYINFO,
      getMsgHeaderUserID() );

   return boost::make_unique<ResponseState>(std::move(resp));
}


/**
 * @param outPattern StripePattern clone (in case of success), that must be deleted by the caller
 */
FhgfsOpsErr GetEntryInfoMsgEx::getInfo(EntryInfo* entryInfo, StripePattern** outPattern,
   PathInfo* outPathInfo, RemoteStorageTarget* outRstInfo, uint32_t& outNumReadSessions,
   uint32_t& outNumWriteSessions, uint8_t& outDataState)
{
   MetaStore* metaStore = Program::getApp()->getMetaStore();

   if (entryInfo->getEntryType() == DirEntryType_DIRECTORY)
   { // entry is a directory
      DirInode* dir = metaStore->referenceDir(entryInfo->getEntryID(),
         entryInfo->getIsBuddyMirrored(), true);
      if(dir)
      {
         *outPattern = dir->getStripePatternClone();
         outRstInfo->set(dir->getRemoteStorageTargetInfo());
         metaStore->releaseDir(entryInfo->getEntryID() );
         return FhgfsOpsErr_SUCCESS;
      }
   }
   else
   { // entry is a file
      auto [fileInode, referenceRes] = metaStore->referenceFile(entryInfo);
      if(fileInode)
      {
         *outPattern = fileInode->getStripePattern()->clone();
         outRstInfo->set(fileInode->getRemoteStorageTargetInfo());
         fileInode->getPathInfo(outPathInfo);

         outNumReadSessions = fileInode->getNumSessionsRead();
         outNumWriteSessions = fileInode->getNumSessionsWrite();
         outDataState = fileInode->getFileState().getRawValue();

         metaStore->releaseFile(entryInfo->getParentEntryID(), fileInode);
         return referenceRes;
      }
   }

   return FhgfsOpsErr_PATHNOTEXISTS;
}

/**
 * @param outPattern StripePattern clone (in case of success), that must be deleted by the caller
 */
FhgfsOpsErr GetEntryInfoMsgEx::getRootInfo(StripePattern** outPattern, RemoteStorageTarget* outRstInfo)
{
   App* app = Program::getApp();
   MirrorBuddyGroupMapper* metaBuddyGroupMapper = app->getMetaBuddyGroupMapper();
   DirInode* rootDir = app->getRootDir();

   NumNodeID localNodeID = app->getLocalNodeNumID();

   if (rootDir->getIsBuddyMirrored())
   {
      uint16_t buddyGroupID = metaBuddyGroupMapper->getBuddyGroupID(localNodeID.val());
      if (buddyGroupID != rootDir->getOwnerNodeID().val() )
         return FhgfsOpsErr_NOTOWNER;
   }
   else
   if (localNodeID != rootDir->getOwnerNodeID() )
      return FhgfsOpsErr_NOTOWNER;

   *outPattern = rootDir->getStripePatternClone();
   outRstInfo->set(rootDir->getRemoteStorageTargetInfo());

   return FhgfsOpsErr_SUCCESS;
}