#pragma once

#include <common/storage/StorageErrors.h>
#include <common/net/message/storage/moving/MovingDirInsertMsg.h>
#include <common/net/message/storage/moving/MovingDirInsertRespMsg.h>
#include <storage/MetaStore.h>
#include <net/message/MirroredMessage.h>

// Move directory to another meta-data server

class MovingDirInsertMsgEx : public MirroredMessage<MovingDirInsertMsg,
   std::tuple<FileIDLock, ParentNameLock>>
{
   public:
      typedef ErrorCodeResponseState<MovingDirInsertRespMsg, NETMSGTYPE_MovingDirInsert>
         ResponseState;

      virtual bool processIncoming(ResponseContext& ctx) override;

      std::tuple<FileIDLock, ParentNameLock> lock(EntryLockStore& store) override
      {
         // we must not lock the directory if it is owned by the current node. if it is, the
         // current message was also sent by the local node, specifically by a RmDirMsgEx, which
         // also locks the directory for write
         if (rctx->isLocallyGenerated())
            return {};

         FileIDLock dirLock(&store, getToDirInfo()->getEntryID(), true);
         ParentNameLock nameLock(&store, getToDirInfo()->getEntryID(), getNewName());

         return std::make_tuple(std::move(dirLock), std::move(nameLock));
      }

      bool isMirrored() override { return getToDirInfo()->getIsBuddyMirrored(); }

   private:
      ResponseContext* rctx;

      std::unique_ptr<MirroredMessageResponseState> executeLocally(ResponseContext& ctx,
         bool isSecondary) override;

      void forwardToSecondary(ResponseContext& ctx) override;

      FhgfsOpsErr processSecondaryResponse(NetMessage& resp) override
      {
         return static_cast<MovingDirInsertRespMsg&>(resp).getResult();
      }

      const char* mirrorLogContext() const override { return "MovingDirInsertMsgEx/forward"; }
};