/*
    This file is part of TON Blockchain Library.

    TON Blockchain Library is free software: you can redistribute it and/or modify
    it under the terms of the GNU Lesser General Public License as published by
    the Free Software Foundation, either version 2 of the License, or
    (at your option) any later version.

    TON Blockchain Library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License
    along with TON Blockchain Library.  If not, see <http://www.gnu.org/licenses/>.
*/
#pragma once
#include "vm/cells.h"
#include "ton/ton-types.h"
#include "auto/tl/ton_api.h"
#include "interfaces/out-msg-queue-proof.h"
#include "td/actor/actor.h"
#include "interfaces/shard.h"
#include "validator.h"

namespace ton {

namespace validator {
using td::Ref;

class ValidatorManager;
class ValidatorManagerInterface;

class OutMsgQueueImporter : public td::actor::Actor {
 public:
  OutMsgQueueImporter(td::actor::ActorId<ValidatorManager> manager, td::Ref<ValidatorManagerOptions> opts,
                      td::Ref<MasterchainState> last_masterchain_state)
      : manager_(manager), opts_(opts), last_masterchain_state_(last_masterchain_state) {
  }

  void new_masterchain_block_notification(td::Ref<MasterchainState> state, std::set<ShardIdFull> collating_shards);
  void get_neighbor_msg_queue_proofs(ShardIdFull dst_shard, std::vector<BlockIdExt> blocks, td::Timestamp timeout,
                                     td::Promise<std::map<BlockIdExt, td::Ref<OutMsgQueueProof>>> promise);

  void update_options(td::Ref<ValidatorManagerOptions> opts) {
    opts_ = std::move(opts);
  }

  void alarm() override;

 private:
  td::actor::ActorId<ValidatorManager> manager_;
  td::Ref<ValidatorManagerOptions> opts_;
  td::Ref<MasterchainState> last_masterchain_state_;

  struct CacheEntry {
    ShardIdFull dst_shard;
    std::vector<BlockIdExt> blocks;
    bool done = false;
    std::map<BlockIdExt, td::Ref<OutMsgQueueProof>> result;
    std::vector<std::pair<td::Promise<std::map<BlockIdExt, td::Ref<OutMsgQueueProof>>>, td::Timestamp>> promises;
    td::Timestamp timeout = td::Timestamp::never();
    td::Timer timer;
    size_t pending = 0;
  };
  std::map<std::pair<ShardIdFull, std::vector<BlockIdExt>>, std::shared_ptr<CacheEntry>> cache_;

  void get_proof_local(std::shared_ptr<CacheEntry> entry, BlockIdExt block);
  void get_proof_import(std::shared_ptr<CacheEntry> entry, std::vector<BlockIdExt> blocks,
                        block::ImportedMsgQueueLimits limits);
  void got_proof(std::shared_ptr<CacheEntry> entry, std::vector<td::Ref<OutMsgQueueProof>> proofs);
  void finish_query(std::shared_ptr<CacheEntry> entry);
  bool check_timeout(std::shared_ptr<CacheEntry> entry);

  constexpr static const double CACHE_TTL = 60.0;
};

class BuildOutMsgQueueProof : public td::actor::Actor {
 public:
  BuildOutMsgQueueProof(ShardIdFull dst_shard, std::vector<BlockIdExt> blocks, block::ImportedMsgQueueLimits limits,
                        td::actor::ActorId<ValidatorManagerInterface> manager,
                        td::Promise<tl_object_ptr<ton_api::tonNode_outMsgQueueProof>> promise)
      : dst_shard_(dst_shard), limits_(limits), manager_(manager), promise_(std::move(promise)) {
    blocks_.resize(blocks.size());
    for (size_t i = 0; i < blocks_.size(); ++i) {
      blocks_[i].id = blocks[i];
    }
  }

  void abort_query(td::Status reason);
  void start_up() override;
  void got_state_root(size_t i, Ref<vm::Cell> root);
  void got_block_root(size_t i, Ref<vm::Cell> root);
  void build_proof();

 private:
  ShardIdFull dst_shard_;
  std::vector<OutMsgQueueProof::OneBlock> blocks_;
  block::ImportedMsgQueueLimits limits_;

  td::actor::ActorId<ValidatorManagerInterface> manager_;
  td::Promise<tl_object_ptr<ton_api::tonNode_outMsgQueueProof>> promise_;

  size_t pending = 0;
};

}  // namespace validator
}  // namespace ton