openmonero/src/YourMoneroRequests.h

//
// Created by mwo on 8/12/16.
//

#ifndef RESTBED_XMR_YOURMONEROREQUESTS_H
#define RESTBED_XMR_YOURMONEROREQUESTS_H

#include <iostream>
#include <functional>

#include "TxSearch.h"
#include "MySqlConnector.h"
#include "tools.h"

#include "../ext/restbed/source/restbed"

namespace xmreg
{

using namespace std;
using namespace restbed;
using namespace nlohmann;


string
get_current_time(const char* format = "%a, %d %b %Y %H:%M:%S %Z")
{

    auto current_time = date::make_zoned(
            date::current_zone(),
            date::floor<chrono::seconds>(std::chrono::system_clock::now())
    );

    return date::format(format, current_time);
}


multimap<string, string>
make_headers(const multimap<string, string>& extra_headers = multimap<string, string>())
{
    multimap<string, string> headers {
            {"Date", get_current_time()},
            {"Access-Control-Allow-Origin",      "http://127.0.0.1"},
            {"access-control-allow-headers",     "*, DNT,X-CustomHeader,Keep-Alive,User-Agent,X-Requested-With,If-Modified-Since,Cache-Control,Content-Type,Set-Cookie"},
            {"access-control-max-age",           "86400, 1728000"},
            {"access-control-allow-methods",     "GET, POST, OPTIONS"},
            {"access-control-allow-credentials", "true"},
            {"Content-Type",                     "application/json"}
    };

    headers.insert(extra_headers.begin(), extra_headers.end());

    return headers;
};

struct handel_
{
    using fetch_func_t = function< void ( const shared_ptr< Session >, const Bytes& ) >;

    fetch_func_t request_callback;

    handel_(const fetch_func_t& callback):
            request_callback {callback}
    {}

    void operator()(const shared_ptr< Session > session)
    {
        const auto request = session->get_request( );
        size_t content_length = request->get_header( "Content-Length", 0);
        session->fetch(content_length, this->request_callback);
    }
};


class YourMoneroRequests
{
    // map that will keep track of search threads. In the
    // map, key is address to which a running thread belongs to.
    // make it static to garantee only one such map exist.
    static map<string, shared_ptr<xmreg::TxSearch>> searching_threads;

    // this manages all mysql queries
   shared_ptr<MySqlAccounts> xmr_accounts;


public:

    static bool show_logs;

    YourMoneroRequests(shared_ptr<MySqlAccounts> _acc):
            xmr_accounts {_acc}
    {}

    /**
     * A login request handler.
     *
     * It takes address and viewkey from the request
     * and check mysql if address/account exist. If yes,
     * it returns this account. If not, it creates new one.
     *
     * Once this complites, a thread is tarted that looks
     * for txs belonging to that account.
     *
     * @param session a Restbed session
     * @param body a POST body, i.e., json string
     */
    void
    login(const shared_ptr<Session> session, const Bytes & body)
    {
        json j_request = body_to_json(body);

        if (show_logs)
            print_json_log("login request: ", j_request);

        string xmr_address  = j_request["address"];

        // a placeholder for exciting or new account data
        xmreg::XmrAccount acc;

        uint64_t acc_id {0};

        json j_response;

        // select this account if its existing one
        if (xmr_accounts->select(xmr_address, acc))
        {
            j_response = {{"new_address", false}};
        }
        else
        {
            // account does not exist, so create new one
            // for this address

            // we will save current blockchain height
            // in mysql, so that we know from what block
            // to start searching txs of this new acount
            // make it 1 block lower than current, just in case.
            // this variable will be our using to initialize
            // `canned_block_height` in mysql Accounts table.
            uint64_t current_blkchain_height = get_current_blockchain_height() - 1;

            if ((acc_id = xmr_accounts->insert(xmr_address, current_blkchain_height)) != 0)
            {
                // select newly created account
                if (xmr_accounts->select(acc_id, acc))
                {
                    j_response = {{"new_address", true}};
                }
            }
        }

        acc.viewkey = j_request["view_key"];

        cout << acc << endl;

        // so we have an account now. Either existing or
        // newly created. Thus, we can start a tread
        // which will scan for transactions belonging to
        // that account, using its address and view key.
        // the thread will scan the blockchain for txs belonging
        // to that account and updated mysql database whenever it
        // will find something.
        //
        // The other client (i.e., a webbrowser) will query other functions to retrieve
        // any belonging transactions in a loop. Thus the thread does not need
        // to do anything except looking for tx and updating mysql
        // with relative tx information

        if (start_tx_search_thread(acc))
        {
            cout << "Search thread started" << endl;
        }

        string response_body = j_response.dump();

        auto response_headers = make_headers({{ "Content-Length", to_string(response_body.size())}});

        session->close( OK, response_body, response_headers);
    }

    void
    get_address_txs(const shared_ptr< Session > session, const Bytes & body)
    {
        json j_request = body_to_json(body);

        if (show_logs)
            print_json_log("get_address_txs request: ", j_request);

        string xmr_address  = j_request["address"];

        // initializa json response
        json j_response {
                { "total_received", "0"},
                { "scanned_height", 0},
                { "scanned_block_height", 0},
                { "start_height", 0},
                { "transaction_height", 0},
                { "blockchain_height", 0}
        };


        // a placeholder for exciting or new account data
        xmreg::XmrAccount acc;

        // select this account if its existing one
        if (xmr_accounts->select(xmr_address, acc))
        {
            j_response["total_received"]       = acc.total_received;
            j_response["scanned_block_height"] = acc.scanned_block_height;
            j_response["blockchain_height"]    = CurrentBlockchainStatus::get_current_blockchain_height();

            vector<XmrTransaction> txs;

            // retrive txs from mysql associated with the given address
            if (xmr_accounts->select_txs(acc.id, txs))
            {
                // we found some txs.

                if (!txs.empty())
                {
                    json j_txs = json::array();

                    for (XmrTransaction tx: txs)
                    {
                        j_txs.push_back(tx.to_json());
                    }

                    j_response["transactions"] = j_txs;
                }
            }
        }


        string response_body = j_response.dump();

        auto response_headers = make_headers({{ "Content-Length", to_string(response_body.size())}});

        session->close( OK, response_body, response_headers);
    }

    void
    get_address_info(const shared_ptr< Session > session, const Bytes & body)
    {
        json j_request = body_to_json(body);

//        if (show_logs)
//            print_json_log("get_address_info request: ", j_request);

        json j_response  {
                {"locked_funds", "0"},
                {"total_received", "0"},
                {"total_sent", "0"},
                {"scanned_height", 2012470},
                {"scanned_block_height", 1195850},
                {"start_height", 2012470},
                {"transaction_height", 2012470},
                {"blockchain_height", 1195850},
                {"spent_outputs", nullptr}
        };

        string response_body = j_response.dump();

        auto response_headers = make_headers({{ "Content-Length", to_string(response_body.size())}});

        session->close( OK, response_body, response_headers);
    }

    void
    import_wallet_request(const shared_ptr< Session > session, const Bytes & body)
    {
        json j_request = body_to_json(body);

        if (show_logs)
            print_json_log("import_wallet_request request: ", j_request);

        json j_response  {
                {"payment_id", "27b64e5edb47f8060cf2648704c8a914ba5657e73cd79cc58a781bc6d21ce5d6"},
                {"import_fee", "1000000000000"},
                {"new_request", true},
                {"request_fulfilled",  false},
                {"payment_address", "44LbNqbmRCmEPxZYmwKw2hbga37svZsHPQ6hLAK4mtApPoWrbpTBiKo6jW452raUXW3M7qUq7yztuchsNYgwYj8S5KQKK43"},
                {"status", "Payment not yet received"}
        };

        string response_body = j_response.dump();

        auto response_headers = make_headers({{ "Content-Length", to_string(response_body.size())}});

        session->close( OK, response_body, response_headers);
    }

    shared_ptr<Resource>
    make_resource(function< void (YourMoneroRequests&, const shared_ptr< Session >, const Bytes& ) > handle_func,
                  const string& path)
    {
        auto a_request = std::bind(handle_func, *this, std::placeholders::_1, std::placeholders::_2);

        shared_ptr<Resource> resource_ptr = make_shared<Resource>();

        resource_ptr->set_path(path);
        resource_ptr->set_method_handler( "OPTIONS", generic_options_handler);
        resource_ptr->set_method_handler( "POST"   , handel_(a_request) );

        return resource_ptr;
    }


    static void
    generic_options_handler( const shared_ptr< Session > session )
    {
        const auto request = session->get_request( );

        size_t content_length = request->get_header( "Content-Length", 0);

        //cout << "generic_options_handler" << endl;

        session->fetch(content_length, [](const shared_ptr< Session > session, const Bytes & body)
        {
            session->close( OK, string{}, make_headers());
        });
    }

    static void
    print_json_log(const string& text, const json& j)
    {
       cout << text << '\n' << j.dump(4) << endl;
    }


    static inline string
    body_to_string(const Bytes & body)
    {
        return string(reinterpret_cast<const char *>(body.data()), body.size());
    }

    static inline json
    body_to_json(const Bytes & body)
    {
        json j = json::parse(body_to_string(body));
        return j;
    }

    bool
    start_tx_search_thread(XmrAccount acc)
    {
        if (searching_threads.count(acc.address) > 0)
        {
            // thread for this address exist, dont make new one
            cout << "Thread exisist, dont make new one" << endl;
            return false;
        }

        // make a tx_search object for the given xmr account
        searching_threads[acc.address] = make_shared<TxSearch>(acc);

        // start the thread for the created object
        std::thread t1 {&TxSearch::search, searching_threads[acc.address].get()};
        t1.detach();

        return true;
    }

    inline uint64_t
    get_current_blockchain_height()
    {
        return CurrentBlockchainStatus::get_current_blockchain_height();
    }

private:



};

// define static variables

bool YourMoneroRequests::show_logs = false;

map<string, shared_ptr<xmreg::TxSearch>> YourMoneroRequests::searching_threads;

}
#endif //RESTBED_XMR_YOURMONEROREQUESTS_H