// Copyright (c) 2009-present The Bitcoin Core developers

// Distributed under the MIT software license, see the accompanying

// file COPYING or http://www.opensource.org/licenses/mit-license.php.

#include <chain.h>

#include <clientversion.h>

#include <core_io.h>

#include <hash.h>

#include <interfaces/chain.h>

#include <key_io.h>

#include <merkleblock.h>

#include <node/types.h>

#include <rpc/util.h>

#include <script/descriptor.h>

#include <script/script.h>

#include <script/solver.h>

#include <sync.h>

#include <uint256.h>

#include <util/bip32.h>

#include <util/check.h>

#include <util/fs.h>

#include <util/time.h>

#include <util/translation.h>

#include <wallet/export.h>

#include <wallet/imports.h>

#include <wallet/rpc/util.h>

#include <wallet/scan.h>

#include <wallet/wallet.h>

#include <cstdint>

#include <fstream>

#include <tuple>

#include <string>

#include <univalue.h>

using interfaces::FoundBlock;

namespace wallet {

RPCMethod importprunedfunds()

{

    return RPCMethod{

        "importprunedfunds",

        "Imports funds without rescan. Corresponding address or script must previously be included in wallet. Aimed towards pruned wallets. The end-user is responsible to import additional transactions that subsequently spend the imported outputs or rescan after the point in the blockchain the transaction is included.\n",

                {

                    {"rawtransaction", RPCArg::Type::STR_HEX, RPCArg::Optional::NO, "A raw transaction in hex funding an already-existing address in wallet"},

                    {"txoutproof", RPCArg::Type::STR_HEX, RPCArg::Optional::NO, "The hex output from gettxoutproof that contains the transaction"},

                },

                RPCResult{RPCResult::Type::NONE, "", ""},

                RPCExamples{""},

        [](const RPCMethod& self, const JSONRPCRequest& request) -> UniValue

{

    std::shared_ptr<CWallet> const pwallet = GetWalletForJSONRPCRequest(request);

    if (!pwallet) return UniValue::VNULL;

    CMutableTransaction tx;

    if (!DecodeHexTx(tx, request.params[0].get_str())) {

        throw JSONRPCError(RPC_DESERIALIZATION_ERROR, "TX decode failed. Make sure the tx has at least one input.");

    }

    CMerkleBlock merkleBlock;

    SpanReader{ParseHexV(request.params[1], "proof")} >> merkleBlock;

    //Search partial merkle tree in proof for our transaction and index in valid block

    std::vector<Txid> vMatch;

    std::vector<unsigned int> vIndex;

    if (merkleBlock.txn.ExtractMatches(vMatch, vIndex) != merkleBlock.header.hashMerkleRoot) {

        throw JSONRPCError(RPC_INVALID_ADDRESS_OR_KEY, "Something wrong with merkleblock");

    }

    LOCK(pwallet->cs_wallet);

    int height;

    if (!pwallet->chain().findAncestorByHash(pwallet->GetLastBlockHash(), merkleBlock.header.GetHash(), FoundBlock().height(height))) {

        throw JSONRPCError(RPC_INVALID_ADDRESS_OR_KEY, "Block not found in chain");

    }

    std::vector<Txid>::const_iterator it;

    if ((it = std::find(vMatch.begin(), vMatch.end(), tx.GetHash())) == vMatch.end()) {

        throw JSONRPCError(RPC_INVALID_ADDRESS_OR_KEY, "Transaction given doesn't exist in proof");

    }

    unsigned int txnIndex = vIndex[it - vMatch.begin()];

    CTransactionRef tx_ref = MakeTransactionRef(tx);

    if (pwallet->IsMine(*tx_ref)) {

        pwallet->AddToWallet(std::move(tx_ref), TxStateConfirmed{merkleBlock.header.GetHash(), height, static_cast<int>(txnIndex)});

        return UniValue::VNULL;

    }

    throw JSONRPCError(RPC_INVALID_ADDRESS_OR_KEY, "No addresses in wallet correspond to included transaction");

},

    };

}

RPCMethod removeprunedfunds()

{

    return RPCMethod{

        "removeprunedfunds",

        "(DEPRECATED) This feature will be removed in the next major release. Start bitcoind with the `-deprecatedrpc=removeprunedfunds` option in order to use this.\n"

        "Deletes the specified transaction from the wallet. Meant for use with pruned wallets and as a companion to importprunedfunds. This will affect wallet balances.\n",

                {

                    {"txid", RPCArg::Type::STR_HEX, RPCArg::Optional::NO, "The hex-encoded id of the transaction you are deleting"},

                },

                RPCResult{RPCResult::Type::NONE, "", ""},

                RPCExamples{

                    HelpExampleCli("removeprunedfunds", "\"a8d0c0184dde994a09ec054286f1ce581bebf46446a512166eae7628734ea0a5\"") +

            "\nAs a JSON-RPC call\n"

            + HelpExampleRpc("removeprunedfunds", "\"a8d0c0184dde994a09ec054286f1ce581bebf46446a512166eae7628734ea0a5\"")

                },

        [](const RPCMethod& self, const JSONRPCRequest& request) -> UniValue

{

    std::shared_ptr<CWallet> const pwallet = GetWalletForJSONRPCRequest(request);

    if (!pwallet) return UniValue::VNULL;

    if (!pwallet->chain().rpcEnableDeprecated("removeprunedfunds")) {

        throw JSONRPCError(RPC_METHOD_DEPRECATED, "DEPRECATION WARNING: This feature will be removed in the next major release. Start bitcoind with the `-deprecatedrpc=removeprunedfunds` option in order to use this.");

    }

    LOCK(pwallet->cs_wallet);

    Txid hash{Txid::FromUint256(ParseHashV(request.params[0], "txid"))};

    std::vector<Txid> vHash;

    vHash.push_back(hash);

    if (auto res = pwallet->RemoveTxs(vHash); !res) {

        throw JSONRPCError(RPC_WALLET_ERROR, util::ErrorString(res).original);

    }

    return UniValue::VNULL;

},

    };

}

/**

 * Converts the timestamp from UniValue data to an int64_t.

 *

 * @returns The import timestamp in int64_t, or std::nullopt if now was provided.

 */

static std::optional<int64_t> GetImportTimestamp(const UniValue& data)

{

    if (data.exists("timestamp")) {

        const UniValue& timestamp = data["timestamp"];

        if (timestamp.isNum()) {

            const int64_t value{timestamp.getInt<int64_t>()};

            if (value < 0) {

                throw JSONRPCError(RPC_INVALID_PARAMETER, "Timestamp must not be negative");

            }

            return value;

        } else if (timestamp.isStr() && timestamp.get_str() == "now") {

            return std::nullopt; // std::nullopt means use the current best block's MTP time

        }

        throw JSONRPCError(RPC_TYPE_ERROR, strprintf("Expected number or \"now\" timestamp value for key. got type %s", uvTypeName(timestamp.type())));

    }

    throw JSONRPCError(RPC_TYPE_ERROR, "Missing required timestamp field for key");

}

static ImportDescriptorRequest ProcessUniValueDescriptor(const UniValue& data, std::optional<int64_t> timestamp)

{

    ImportDescriptorRequest request;

    if (!data.exists("desc")) {

        throw JSONRPCError(RPC_INVALID_PARAMETER, "Descriptor not found.");

    }

    request.descriptor = data["desc"].get_str();

    request.label = LabelFromValue(data["label"]);

    request.timestamp = timestamp;

    if (data.exists("active")) request.active = data["active"].get_bool();

    if (data.exists("internal")) request.internal = data["internal"].get_bool();

    if (data.exists("range")) request.range = ParseDescriptorRange(data["range"]);

    if (data.exists("next_index")) request.next_index = data["next_index"].getInt<int64_t>();

    return request;

}

RPCMethod importdescriptors()

{

    return RPCMethod{

        "importdescriptors",

        "Import descriptors. This will trigger a rescan of the blockchain based on the earliest timestamp of all descriptors being imported. Requires a new wallet backup.\n"

        "When importing descriptors with multipath key expressions, if the multipath specifier contains exactly two elements, the descriptor produced from the second element will be imported as an internal descriptor.\n"

            "\nNote: This call can take over an hour to complete if using an early timestamp; during that time, other rpc calls\n"

            "may report that the imported keys, addresses or scripts exist but related transactions are still missing.\n"

            "The rescan is significantly faster if block filters are available (using startup option \"-blockfilterindex=1\").\n",

                {

                    {"requests", RPCArg::Type::ARR, RPCArg::Optional::NO, "Data to be imported",

                        {

                            {"", RPCArg::Type::OBJ, RPCArg::Optional::OMITTED, "",

                                {

                                    {"desc", RPCArg::Type::STR, RPCArg::Optional::NO, "Descriptor to import."},

                                    {"active", RPCArg::Type::BOOL, RPCArg::Default{false}, "Set this descriptor to be the active descriptor for the corresponding output type/externality"},

                                    {"range", RPCArg::Type::RANGE, RPCArg::Optional::OMITTED, "If a ranged descriptor is used, this specifies the end or the range (in the form [begin,end]) to import"},

                                    {"next_index", RPCArg::Type::NUM, RPCArg::Optional::OMITTED, "If a ranged descriptor is set to active, this specifies the next index to generate addresses from"},

                                    {"timestamp", RPCArg::Type::NUM, RPCArg::Optional::NO, "Time from which to start rescanning the blockchain for this descriptor, in " + UNIX_EPOCH_TIME + "\n"

                                        "Use the string \"now\" to substitute the current synced blockchain time.\n"

                                        "\"now\" can be specified to bypass scanning, for outputs which are known to never have been used, and\n"

                                        "0 can be specified to scan the entire blockchain. Blocks up to 2 hours before the earliest timestamp\n"

                                        "of all descriptors being imported will be scanned as well as the mempool.",

                                        RPCArgOptions{.type_str={"timestamp | \"now\"", "integer / string"}}

                                    },

                                    {"internal", RPCArg::Type::BOOL, RPCArg::Default{false}, "Whether matching outputs should be treated as not incoming payments (e.g. change)"},

                                    {"label", RPCArg::Type::STR, RPCArg::Default{""}, "Label to assign to the address, only allowed with internal=false. Disabled for ranged descriptors"},

                                },

                            },

                        },

                        RPCArgOptions{.oneline_description="requests"}},

                },

                RPCResult{

                    RPCResult::Type::ARR, "", "Response is an array with the same size as the input that has the execution result",

                    {

                        {RPCResult::Type::OBJ, "", "",

                        {

                            {RPCResult::Type::BOOL, "success", ""},

                            {RPCResult::Type::ARR, "warnings", /*optional=*/true, "",

                            {

                                {RPCResult::Type::STR, "", ""},

                            }},

                            {RPCResult::Type::OBJ, "error", /*optional=*/true, "",

                            {

                                {RPCResult::Type::NUM, "code", "JSONRPC error code"},

                                {RPCResult::Type::STR, "message", "JSONRPC error message"},

                            }},

                        }},

                    }

                },

                RPCExamples{

                    HelpExampleCli("importdescriptors", "'[{ \"desc\": \"<my descriptor>\", \"timestamp\":1455191478, \"internal\": true }, "

                                          "{ \"desc\": \"<my descriptor 2>\", \"label\": \"example 2\", \"timestamp\": 1455191480 }]'") +

                    HelpExampleCli("importdescriptors", "'[{ \"desc\": \"<my descriptor>\", \"timestamp\":1455191478, \"active\": true, \"range\": [0,100], \"label\": \"<my bech32 wallet>\" }]'")

                },

        [](const RPCMethod& self, const JSONRPCRequest& main_request) -> UniValue

{

    std::shared_ptr<CWallet> const pwallet = GetWalletForJSONRPCRequest(main_request);

    if (!pwallet) return UniValue::VNULL;

    CWallet& wallet{*pwallet};

    const UniValue& univalue_requests = main_request.params[0];

    // One result per input, in input order.

    std::vector<UniValue> results(univalue_requests.size());

    // Successfully parsed requests, each with the index of the input it came from.

    struct ParsedRequest {

        size_t input_index;

        ImportDescriptorRequest request;

    };

    std::vector<ParsedRequest> requests;

    // Malformed inputs (e.g. invalid label) are returned as per-item failures.

    for (size_t i = 0; i < univalue_requests.size(); ++i) {

        // Throws a top-level RPC error if "timestamp" is missing or invalid

        std::optional<int64_t> timestamp = GetImportTimestamp(univalue_requests[i]);

        try {

            requests.push_back({i, ProcessUniValueDescriptor(univalue_requests[i], timestamp)});

        } catch (const UniValue& e) {

            results[i] = UniValue(UniValue::VOBJ);

            results[i].pushKV("success", UniValue(false));

            results[i].pushKV("error", e);

        }

    }

    // Hand off the successfully parsed requests to the batch importer, which

    // handles wallet locking, rescanning and rescan-failure error composition.

    std::vector<ImportDescriptorRequest> descriptor_requests;

    descriptor_requests.reserve(requests.size());

    for (auto& parsed : requests) {

        descriptor_requests.push_back(std::move(parsed.request));

    }

    std::vector<ImportResult> import_results{ProcessDescriptorsImport(wallet, descriptor_requests)};

    // Wallet-wide precondition failure (e.g. already rescanning, or locked):

    // surface as a top-level RPC error.

    if (import_results.size() == 1 && import_results[0].has_error() && import_results[0].error->is_general_error) {

        const ImportError& import_error = import_results[0].error.value();

        RPCErrorCode rpc_error_code{HandleWalletErrorCode(import_error.wallet_error.code)};

        throw JSONRPCError(rpc_error_code, import_error.wallet_error.message.original);

    }

    // Translate each ImportResult into the per-input UniValue result. Inputs

    // that failed to parse already hold an error in results[] and are not

    // part of `requests`, so use input_index to map each import_results[k]

    // back to its slot.

    CHECK_NONFATAL(import_results.size() == requests.size());

    for (size_t k = 0; k < requests.size(); ++k) {

        UniValue& result = results[requests[k].input_index];

        const ImportResult& import_result = import_results[k];

        result = UniValue(UniValue::VOBJ);

        UniValue warnings(UniValue::VARR);

        if (import_result.has_error()) {

            const WalletError& error = import_result.error.value().wallet_error;

            auto write_error = [&result, &error](int code) {

                result.pushKV("success", false);

                result.pushKV("error", JSONRPCError(code, error.message.original));

            };

            if (error.code == WalletErrorCode::UnlockNeeded) NONFATAL_UNREACHABLE();

            write_error(HandleWalletErrorCode(error.code));

        } else {

            result.pushKV("success", true);

        }

        for (const auto& w : import_result.warnings) {

            warnings.push_back(w);

        }

        PushWarnings(warnings, result);

    }

    UniValue response(UniValue::VARR);

    for (UniValue& result : results) {

        response.push_back(std::move(result));

    }

    return response;

},

    };

}

RPCMethod listdescriptors()

{

    return RPCMethod{

        "listdescriptors",

        "List all descriptors present in a wallet.\n",

        {

            {"private", RPCArg::Type::BOOL, RPCArg::Default{false}, "Show private descriptors."}

        },

        RPCResult{RPCResult::Type::OBJ, "", "", {

            {RPCResult::Type::STR, "wallet_name", "Name of wallet this operation was performed on"},

            {RPCResult::Type::ARR, "descriptors", "Array of descriptor objects (sorted by descriptor string representation)",

            {

                {RPCResult::Type::OBJ, "", "", {

                    {RPCResult::Type::STR, "desc", "Descriptor string representation"},

                    {RPCResult::Type::NUM, "timestamp", "The creation time of the descriptor"},

                    {RPCResult::Type::BOOL, "active", "Whether this descriptor is currently used to generate new addresses"},

                    {RPCResult::Type::BOOL, "internal", /*optional=*/true, "True if this descriptor is used to generate change addresses. False if this descriptor is used to generate receiving addresses; defined only for active descriptors"},

                    {RPCResult::Type::ARR_FIXED, "range", /*optional=*/true, "Defined only for ranged descriptors", {

                        {RPCResult::Type::NUM, "", "Range start inclusive"},

                        {RPCResult::Type::NUM, "", "Range end inclusive"},

                    }},

                    {RPCResult::Type::NUM, "next", /*optional=*/true, "Same as next_index field. Kept for compatibility reason."},

                    {RPCResult::Type::NUM, "next_index", /*optional=*/true, "The next index to generate addresses from; defined only for ranged descriptors"},

                }},

            }}

        }},

        RPCExamples{

            HelpExampleCli("listdescriptors", "") + HelpExampleRpc("listdescriptors", "")

            + HelpExampleCli("listdescriptors", "true") + HelpExampleRpc("listdescriptors", "true")

        },

        [](const RPCMethod& self, const JSONRPCRequest& request) -> UniValue

{

    const std::shared_ptr<const CWallet> wallet = GetWalletForJSONRPCRequest(request);

    if (!wallet) return UniValue::VNULL;

    const bool priv = !request.params[0].isNull() && request.params[0].get_bool();

    if (wallet->IsWalletFlagSet(WALLET_FLAG_DISABLE_PRIVATE_KEYS) && priv) {

        throw JSONRPCError(RPC_WALLET_ERROR, "Can't get private descriptor string for watch-only wallets");

    }

    if (priv) {

        EnsureWalletIsUnlocked(*wallet);

    }

    LOCK(wallet->cs_wallet);

    util::Expected<std::vector<WalletDescInfo>, std::string> exported = ExportDescriptors(*wallet, priv);

    if (!exported) {

        throw JSONRPCError(RPC_WALLET_ERROR, exported.error());

    }

    std::vector<WalletDescInfo> wallet_descriptors = *exported;

    std::sort(wallet_descriptors.begin(), wallet_descriptors.end(), [](const auto& a, const auto& b) {

        return a.descriptor < b.descriptor;

    });

    UniValue descriptors(UniValue::VARR);

    for (const WalletDescInfo& info : wallet_descriptors) {

        UniValue spk(UniValue::VOBJ);

        spk.pushKV("desc", info.descriptor);

        spk.pushKV("timestamp", info.creation_time);

        spk.pushKV("active", info.active);

        if (info.internal.has_value()) {

            spk.pushKV("internal", info.internal.value());

        }

        if (info.range.has_value()) {

            UniValue range(UniValue::VARR);

            range.push_back(info.range->first);

            range.push_back(info.range->second - 1);

            spk.pushKV("range", std::move(range));

            spk.pushKV("next", info.next_index);

            spk.pushKV("next_index", info.next_index);

        }

        descriptors.push_back(std::move(spk));

    }

    UniValue response(UniValue::VOBJ);

    response.pushKV("wallet_name", wallet->GetName());

    response.pushKV("descriptors", std::move(descriptors));

    return response;

},

    };

}

RPCMethod backupwallet()

{

    return RPCMethod{

        "backupwallet",

        "Safely copies the current wallet file to the specified destination, which can either be a directory or a path with a filename.\n",

                {

                    {"destination", RPCArg::Type::STR, RPCArg::Optional::NO, "The destination directory or file"},

                },

                RPCResult{RPCResult::Type::NONE, "", ""},

                RPCExamples{

                    HelpExampleCli("backupwallet", "\"backup.dat\"")

            + HelpExampleRpc("backupwallet", "\"backup.dat\"")

                },

        [](const RPCMethod& self, const JSONRPCRequest& request) -> UniValue

{

    const std::shared_ptr<const CWallet> pwallet = GetWalletForJSONRPCRequest(request);

    if (!pwallet) return UniValue::VNULL;

    // Make sure the results are valid at least up to the most recent block

    // the user could have gotten from another RPC command prior to now

    pwallet->BlockUntilSyncedToCurrentChain();

    LOCK(pwallet->cs_wallet);

    std::string strDest = request.params[0].get_str();

    if (!pwallet->BackupWallet(strDest)) {

        throw JSONRPCError(RPC_WALLET_ERROR, "Error: Wallet backup failed!");

    }

    return UniValue::VNULL;

},

    };

}

RPCMethod restorewallet()

{

    return RPCMethod{

        "restorewallet",

        "Restores and loads a wallet from backup.\n"

        "\nThe rescan is significantly faster if block filters are available"

        "\n(using startup option \"-blockfilterindex=1\").\n",

        {

            {"wallet_name", RPCArg::Type::STR, RPCArg::Optional::NO, "The name that will be applied to the restored wallet"},

            {"backup_file", RPCArg::Type::STR, RPCArg::Optional::NO, "The backup file that will be used to restore the wallet."},

            {"load_on_startup", RPCArg::Type::BOOL, RPCArg::Optional::OMITTED, "Save wallet name to persistent settings and load on startup. True to add wallet to startup list, false to remove, null to leave unchanged."},

        },

        RPCResult{

            RPCResult::Type::OBJ, "", "",

            {

                {RPCResult::Type::STR, "name", "The wallet name if restored successfully."},

                {RPCResult::Type::ARR, "warnings", /*optional=*/true, "Warning messages, if any, related to restoring and loading the wallet.",

                {

                    {RPCResult::Type::STR, "", ""},

                }},

            }

        },

        RPCExamples{

            HelpExampleCli("restorewallet", "\"testwallet\" \"home\\backups\\backup-file.bak\"")

            + HelpExampleRpc("restorewallet", R"("testwallet", "home\\backups\\backup-file.bak")")

            + HelpExampleCliNamed("restorewallet", {{"wallet_name", "testwallet"}, {"backup_file", "home\\backups\\backup-file.bak"}, {"load_on_startup", true}})

            + HelpExampleRpcNamed("restorewallet", {{"wallet_name", "testwallet"}, {"backup_file", "home\\backups\\backup-file.bak"}, {"load_on_startup", true}})

        },

        [](const RPCMethod& self, const JSONRPCRequest& request) -> UniValue

{

    WalletContext& context = EnsureWalletContext(request.context);

    auto backup_file = fs::u8path(request.params[1].get_str());

    std::string wallet_name = request.params[0].get_str();

    std::optional<bool> load_on_start = request.params[2].isNull() ? std::nullopt : std::optional<bool>(request.params[2].get_bool());

    DatabaseStatus status;

    bilingual_str error;

    std::vector<bilingual_str> warnings;

    const std::shared_ptr<CWallet> wallet = RestoreWallet(context, backup_file, wallet_name, load_on_start, status, error, warnings);

    HandleWalletError(wallet, status, error);

    UniValue obj(UniValue::VOBJ);

    obj.pushKV("name", wallet->GetName());

    PushWarnings(warnings, obj);

    return obj;

},

    };

}

} // namespace wallet