#include <dht/publishservicejob.hpp>

#include <dht/context.hpp>
#include <dht/messages/pubintro.hpp>
#include <dht/messages/gotintro.hpp>
#include <path/path_context.hpp>
#include <routing/dht_message.hpp>
#include <router/abstractrouter.hpp>

#include <utility>
namespace llarp
{
  namespace dht
  {
    PublishServiceJob::PublishServiceJob(
        const TXOwner &asker, const service::EncryptedIntroSet &introset_,
        AbstractContext *ctx, uint64_t relayOrder_)
        : TX< TXOwner, service::EncryptedIntroSet >(asker, asker, ctx)
        , relayOrder(relayOrder_)
        , introset(introset_)
    {
    }

    bool
    PublishServiceJob::Validate(const service::EncryptedIntroSet &value) const
    {
      if(value.derivedSigningKey != introset.derivedSigningKey)
      {
        llarp::LogWarn(
            "publish introset acknowledgement acked a different service");
        return false;
      }
      const llarp_time_t now = llarp::time_now_ms();
      return value.Verify(now);
    }

    void
    PublishServiceJob::Start(const TXOwner &peer)
    {
      parent->DHTSendTo(
          peer.node.as_array(),
          new PublishIntroMessage(introset, peer.txid, false, relayOrder));
    }

    void
    PublishServiceJob::SendReply()
    {
      parent->DHTSendTo(whoasked.node.as_array(),
                        new GotIntroMessage({introset}, whoasked.txid));
    }

    LocalPublishServiceJob::LocalPublishServiceJob(
        const TXOwner &peer, const PathID_t &fromID, uint64_t _txid,
        const service::EncryptedIntroSet &introset, AbstractContext *ctx,
        uint64_t relayOrder)
        : PublishServiceJob(peer, introset, ctx, relayOrder)
        , localPath(fromID)
        , txid(_txid)
    {
    }

    void
    LocalPublishServiceJob::SendReply()
    {
      auto path = parent->GetRouter()->pathContext().GetByUpstream(
          parent->OurKey().as_array(), localPath);
      if(!path)
      {
        llarp::LogWarn(
            "did not send reply for relayed dht request, no such local path "
            "for pathid=",
            localPath);
        return;
      }
      routing::DHTMessage msg;
      msg.M.emplace_back(new GotIntroMessage({introset}, whoasked.txid));
      if(!path->SendRoutingMessage(msg, parent->GetRouter()))
      {
        llarp::LogWarn(
            "failed to send routing message when informing result of dht "
            "request, pathid=",
            localPath);
      }
    }
  }  // namespace dht
}  // namespace llarp