/*
    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/>.

    Copyright 2017-2020 Telegram Systems LLP
*/
#include "HighloadWallet.h"
#include "GenericAccount.h"
#include "SmartContractCode.h"

#include "vm/boc.h"
#include "vm/cells/CellString.h"
#include "td/utils/base64.h"

#include <limits>

namespace ton {

td::Result<td::Ref<vm::Cell>> HighloadWallet::make_a_gift_message(const td::Ed25519::PrivateKey& private_key,
                                                                  td::uint32 valid_until, td::Span<Gift> gifts) const {
  TRY_RESULT(wallet_id, get_wallet_id());
  TRY_RESULT(seqno, get_seqno());
  CHECK(gifts.size() <= get_max_gifts_size());
  vm::Dictionary messages(16);
  for (size_t i = 0; i < gifts.size(); i++) {
    auto& gift = gifts[i];
    td::int32 send_mode = 3;
    if (gift.gramms == -1) {
      send_mode += 128;
    }
    auto message_inner = create_int_message(gift);
    vm::CellBuilder cb;
    cb.store_long(send_mode, 8).store_ref(message_inner);
    auto key = messages.integer_key(td::make_refint(i), 16, false);
    messages.set_builder(key.bits(), 16, cb);
  }

  vm::CellBuilder cb;
  cb.store_long(wallet_id, 32).store_long(valid_until, 32).store_long(seqno, 32);
  CHECK(cb.store_maybe_ref(messages.get_root_cell()));
  auto message_outer = cb.finalize();
  auto signature = private_key.sign(message_outer->get_hash().as_slice()).move_as_ok();
  return vm::CellBuilder().store_bytes(signature).append_cellslice(vm::load_cell_slice(message_outer)).finalize();
}

td::Ref<vm::Cell> HighloadWallet::get_init_data(const InitData& init_data) noexcept {
  return vm::CellBuilder()
      .store_long(init_data.seqno, 32)
      .store_long(init_data.wallet_id, 32)
      .store_bytes(init_data.public_key)
      .finalize();
}

td::Result<td::uint32> HighloadWallet::get_wallet_id() const {
  return TRY_VM([&]() -> td::Result<td::uint32> {
    if (state_.data.is_null()) {
      return 0;
    }
    auto cs = vm::load_cell_slice(state_.data);
    cs.skip_first(32);
    return static_cast<td::uint32>(cs.fetch_ulong(32));
  }());
}

td::Result<td::Ed25519::PublicKey> HighloadWallet::get_public_key() const {
  return TRY_VM([&]() -> td::Result<td::Ed25519::PublicKey> {
    if (state_.data.is_null()) {
      return td::Status::Error("data is null");
    }
    auto cs = vm::load_cell_slice(state_.data);
    cs.skip_first(64);
    td::SecureString res(td::Ed25519::PublicKey::LENGTH);
    cs.fetch_bytes(res.as_mutable_slice().ubegin(), td::narrow_cast<td::int32>(res.size()));
    return td::Ed25519::PublicKey(std::move(res));
  }());
}

}  // namespace ton