mirror of
				https://git.wownero.com/wownero/onion-wownero-blockchain-explorer.git
				synced 2024-08-15 00:33:12 +00:00 
			
		
		
		
	first commit
This commit is contained in:
		
						commit
						98661d9b67
					
				
					 65 changed files with 16558 additions and 0 deletions
				
			
		
							
								
								
									
										4
									
								
								.gitignore
									
										
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										4
									
								
								.gitignore
									
										
									
									
										vendored
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,4 @@
 | 
			
		|||
.DS_Store
 | 
			
		||||
.sass-cache
 | 
			
		||||
*.*~
 | 
			
		||||
.idea/
 | 
			
		||||
							
								
								
									
										101
									
								
								CMakeLists.txt
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										101
									
								
								CMakeLists.txt
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,101 @@
 | 
			
		|||
cmake_minimum_required(VERSION 3.5)
 | 
			
		||||
 | 
			
		||||
set(PROJECT_NAME
 | 
			
		||||
        crowxmr)
 | 
			
		||||
 | 
			
		||||
project(${PROJECT_NAME})
 | 
			
		||||
 | 
			
		||||
set(CMAKE_CXX_FLAGS
 | 
			
		||||
        "${CMAKE_CXX_FLAGS} -std=c++14")
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
# find boost
 | 
			
		||||
find_package(Boost COMPONENTS
 | 
			
		||||
        system
 | 
			
		||||
        filesystem
 | 
			
		||||
        thread
 | 
			
		||||
        date_time
 | 
			
		||||
        chrono
 | 
			
		||||
        regex
 | 
			
		||||
        serialization
 | 
			
		||||
        program_options
 | 
			
		||||
        date_time
 | 
			
		||||
        REQUIRED)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
# set location of monero static libraries
 | 
			
		||||
set(MONERO_LIBS_DIR
 | 
			
		||||
        /opt/bitmonero-dev/libs)
 | 
			
		||||
 | 
			
		||||
# set location of moneroheaders
 | 
			
		||||
set(MONERO_HEADERS_DIR
 | 
			
		||||
        /opt/bitmonero-dev/headers)
 | 
			
		||||
 | 
			
		||||
# include monero headers
 | 
			
		||||
include_directories(
 | 
			
		||||
        ${MONERO_HEADERS_DIR}/src
 | 
			
		||||
        ${MONERO_HEADERS_DIR}/external
 | 
			
		||||
        ${MONERO_HEADERS_DIR}/contrib/epee/include
 | 
			
		||||
        ${MONERO_HEADERS_DIR}/external/db_drivers/liblmdb)
 | 
			
		||||
 | 
			
		||||
# get individual monero static libraries
 | 
			
		||||
# that are needed in this project
 | 
			
		||||
 | 
			
		||||
add_library(common STATIC IMPORTED)
 | 
			
		||||
set_property(TARGET common PROPERTY IMPORTED_LOCATION ${MONERO_LIBS_DIR}/libcommon.a)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
add_library(blocks STATIC IMPORTED)
 | 
			
		||||
set_property(TARGET blocks PROPERTY IMPORTED_LOCATION ${MONERO_LIBS_DIR}/libblocks.a)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
add_library(crypto STATIC IMPORTED)
 | 
			
		||||
set_property(TARGET crypto
 | 
			
		||||
        PROPERTY IMPORTED_LOCATION ${MONERO_LIBS_DIR}/libcrypto.a)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
add_library(cryptonote_core STATIC IMPORTED)
 | 
			
		||||
set_property(TARGET cryptonote_core
 | 
			
		||||
        PROPERTY IMPORTED_LOCATION ${MONERO_LIBS_DIR}/libcryptonote_core.a)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
add_library(blockchain_db STATIC IMPORTED)
 | 
			
		||||
set_property(TARGET blockchain_db
 | 
			
		||||
        PROPERTY IMPORTED_LOCATION ${MONERO_LIBS_DIR}/libblockchain_db.a)
 | 
			
		||||
 | 
			
		||||
add_library(lmdb STATIC IMPORTED)
 | 
			
		||||
set_property(TARGET lmdb
 | 
			
		||||
        PROPERTY IMPORTED_LOCATION ${MONERO_LIBS_DIR}/liblmdb.a)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
# include boost headers
 | 
			
		||||
include_directories(${Boost_INCLUDE_DIRS})
 | 
			
		||||
include_directories("ext/mstch/include")
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
# add ext/ subfolder
 | 
			
		||||
add_subdirectory(ext/)
 | 
			
		||||
 | 
			
		||||
# add src/ subfolder
 | 
			
		||||
add_subdirectory(src/)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
set(SOURCE_FILES
 | 
			
		||||
        main.cpp)
 | 
			
		||||
 | 
			
		||||
add_executable(${PROJECT_NAME}
 | 
			
		||||
        ${SOURCE_FILES})
 | 
			
		||||
 | 
			
		||||
target_link_libraries(${PROJECT_NAME}
 | 
			
		||||
        myxrm
 | 
			
		||||
        myext
 | 
			
		||||
        mstch
 | 
			
		||||
        cryptonote_core
 | 
			
		||||
        blockchain_db
 | 
			
		||||
        crypto
 | 
			
		||||
        blocks
 | 
			
		||||
        common
 | 
			
		||||
        lmdb
 | 
			
		||||
        ${Boost_LIBRARIES}
 | 
			
		||||
        pthread
 | 
			
		||||
        unbound)
 | 
			
		||||
							
								
								
									
										1
									
								
								README.md
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								README.md
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1 @@
 | 
			
		|||
# crow-monero-test
 | 
			
		||||
							
								
								
									
										24
									
								
								ext/CMakeLists.txt
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										24
									
								
								ext/CMakeLists.txt
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,24 @@
 | 
			
		|||
# first build mstch template library
 | 
			
		||||
add_subdirectory("mstch")
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
# now build myext library from other files
 | 
			
		||||
project(myext)
 | 
			
		||||
 | 
			
		||||
set(SOURCE_HEADERS
 | 
			
		||||
        minicsv.h
 | 
			
		||||
        format.h
 | 
			
		||||
        dateparser.h)
 | 
			
		||||
 | 
			
		||||
set(SOURCE_FILES
 | 
			
		||||
        format.cc
 | 
			
		||||
        dateparser.cpp)
 | 
			
		||||
 | 
			
		||||
# make static library called libmyxrm
 | 
			
		||||
# that we are going to link to
 | 
			
		||||
# in the root CMakeLists.txt file
 | 
			
		||||
add_library(myext
 | 
			
		||||
        STATIC
 | 
			
		||||
        ${SOURCE_FILES})
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										34
									
								
								ext/crow/ci_map.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										34
									
								
								ext/crow/ci_map.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,34 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <boost/algorithm/string/predicate.hpp>
 | 
			
		||||
#include <boost/functional/hash.hpp>
 | 
			
		||||
#include <unordered_map>
 | 
			
		||||
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
    struct ci_hash
 | 
			
		||||
    {
 | 
			
		||||
        size_t operator()(const std::string& key) const
 | 
			
		||||
        {
 | 
			
		||||
            std::size_t seed = 0;
 | 
			
		||||
            std::locale locale;
 | 
			
		||||
 | 
			
		||||
            for(auto c : key)
 | 
			
		||||
            {
 | 
			
		||||
                boost::hash_combine(seed, std::toupper(c, locale));
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            return seed;
 | 
			
		||||
        }
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    struct ci_key_eq
 | 
			
		||||
    {
 | 
			
		||||
        bool operator()(const std::string& l, const std::string& r) const
 | 
			
		||||
        {
 | 
			
		||||
            return boost::iequals(l, r);
 | 
			
		||||
        }
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    using ci_map = std::unordered_multimap<std::string, std::string, ci_hash, ci_key_eq>;
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										126
									
								
								ext/crow/common.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										126
									
								
								ext/crow/common.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,126 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <vector>
 | 
			
		||||
#include <string>
 | 
			
		||||
#include <stdexcept>
 | 
			
		||||
#include <iostream>
 | 
			
		||||
#include "utility.h"
 | 
			
		||||
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
    enum class HTTPMethod
 | 
			
		||||
    {
 | 
			
		||||
        DELETE,
 | 
			
		||||
        GET,
 | 
			
		||||
        HEAD,
 | 
			
		||||
        POST,
 | 
			
		||||
        PUT,
 | 
			
		||||
        CONNECT,
 | 
			
		||||
        OPTIONS,
 | 
			
		||||
        TRACE,
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    inline std::string method_name(HTTPMethod method)
 | 
			
		||||
    {
 | 
			
		||||
        switch(method)
 | 
			
		||||
        {
 | 
			
		||||
            case HTTPMethod::DELETE:
 | 
			
		||||
                return "DELETE";
 | 
			
		||||
            case HTTPMethod::GET:
 | 
			
		||||
                return "GET";
 | 
			
		||||
            case HTTPMethod::HEAD:
 | 
			
		||||
                return "HEAD";
 | 
			
		||||
            case HTTPMethod::POST:
 | 
			
		||||
                return "POST";
 | 
			
		||||
            case HTTPMethod::PUT:
 | 
			
		||||
                return "PUT";
 | 
			
		||||
            case HTTPMethod::CONNECT:
 | 
			
		||||
                return "CONNECT";
 | 
			
		||||
            case HTTPMethod::OPTIONS:
 | 
			
		||||
                return "OPTIONS";
 | 
			
		||||
            case HTTPMethod::TRACE:
 | 
			
		||||
                return "TRACE";
 | 
			
		||||
        }
 | 
			
		||||
        return "invalid";
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    enum class ParamType
 | 
			
		||||
    {
 | 
			
		||||
        INT,
 | 
			
		||||
        UINT,
 | 
			
		||||
        DOUBLE,
 | 
			
		||||
        STRING,
 | 
			
		||||
        PATH,
 | 
			
		||||
 | 
			
		||||
        MAX
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    struct routing_params
 | 
			
		||||
    {
 | 
			
		||||
        std::vector<int64_t> int_params;
 | 
			
		||||
        std::vector<uint64_t> uint_params;
 | 
			
		||||
        std::vector<double> double_params;
 | 
			
		||||
        std::vector<std::string> string_params;
 | 
			
		||||
 | 
			
		||||
        void debug_print() const
 | 
			
		||||
        {
 | 
			
		||||
            std::cerr << "routing_params" << std::endl;
 | 
			
		||||
            for(auto i:int_params)
 | 
			
		||||
                std::cerr<<i <<", " ;
 | 
			
		||||
            std::cerr<<std::endl;
 | 
			
		||||
            for(auto i:uint_params)
 | 
			
		||||
                std::cerr<<i <<", " ;
 | 
			
		||||
            std::cerr<<std::endl;
 | 
			
		||||
            for(auto i:double_params)
 | 
			
		||||
                std::cerr<<i <<", " ;
 | 
			
		||||
            std::cerr<<std::endl;
 | 
			
		||||
            for(auto& i:string_params)
 | 
			
		||||
                std::cerr<<i <<", " ;
 | 
			
		||||
            std::cerr<<std::endl;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <typename T>
 | 
			
		||||
        T get(unsigned) const;
 | 
			
		||||
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    template<>
 | 
			
		||||
    inline int64_t routing_params::get<int64_t>(unsigned index) const
 | 
			
		||||
    {
 | 
			
		||||
        return int_params[index];
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    template<>
 | 
			
		||||
    inline uint64_t routing_params::get<uint64_t>(unsigned index) const
 | 
			
		||||
    {
 | 
			
		||||
        return uint_params[index];
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    template<>
 | 
			
		||||
    inline double routing_params::get<double>(unsigned index) const
 | 
			
		||||
    {
 | 
			
		||||
        return double_params[index];
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    template<>
 | 
			
		||||
    inline std::string routing_params::get<std::string>(unsigned index) const
 | 
			
		||||
    {
 | 
			
		||||
        return string_params[index];
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#ifndef CROW_MSVC_WORKAROUND
 | 
			
		||||
constexpr crow::HTTPMethod operator "" _method(const char* str, size_t len)
 | 
			
		||||
{
 | 
			
		||||
    return
 | 
			
		||||
        crow::black_magic::is_equ_p(str, "GET", 3) ? crow::HTTPMethod::GET :
 | 
			
		||||
        crow::black_magic::is_equ_p(str, "DELETE", 6) ? crow::HTTPMethod::DELETE :
 | 
			
		||||
        crow::black_magic::is_equ_p(str, "HEAD", 4) ? crow::HTTPMethod::HEAD :
 | 
			
		||||
        crow::black_magic::is_equ_p(str, "POST", 4) ? crow::HTTPMethod::POST :
 | 
			
		||||
        crow::black_magic::is_equ_p(str, "PUT", 3) ? crow::HTTPMethod::PUT :
 | 
			
		||||
        crow::black_magic::is_equ_p(str, "OPTIONS", 7) ? crow::HTTPMethod::OPTIONS :
 | 
			
		||||
        crow::black_magic::is_equ_p(str, "CONNECT", 7) ? crow::HTTPMethod::CONNECT :
 | 
			
		||||
        crow::black_magic::is_equ_p(str, "TRACE", 5) ? crow::HTTPMethod::TRACE :
 | 
			
		||||
        throw std::runtime_error("invalid http method");
 | 
			
		||||
};
 | 
			
		||||
#endif
 | 
			
		||||
							
								
								
									
										199
									
								
								ext/crow/crow.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										199
									
								
								ext/crow/crow.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,199 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <string>
 | 
			
		||||
#include <functional>
 | 
			
		||||
#include <memory>
 | 
			
		||||
#include <future>
 | 
			
		||||
#include <cstdint>
 | 
			
		||||
#include <type_traits>
 | 
			
		||||
#include <thread>
 | 
			
		||||
 | 
			
		||||
#include "settings.h"
 | 
			
		||||
#include "logging.h" 
 | 
			
		||||
#include "utility.h"
 | 
			
		||||
#include "routing.h"
 | 
			
		||||
#include "middleware_context.h"
 | 
			
		||||
#include "http_request.h"
 | 
			
		||||
#include "http_server.h"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
#ifdef CROW_MSVC_WORKAROUND
 | 
			
		||||
#define CROW_ROUTE(app, url) app.route_dynamic(url)
 | 
			
		||||
#else
 | 
			
		||||
#define CROW_ROUTE(app, url) app.route<crow::black_magic::get_parameter_tag(url)>(url)
 | 
			
		||||
#endif
 | 
			
		||||
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
#ifdef CROW_ENABLE_SSL
 | 
			
		||||
    using ssl_context_t = boost::asio::ssl::context;
 | 
			
		||||
#endif
 | 
			
		||||
    template <typename ... Middlewares>
 | 
			
		||||
    class Crow
 | 
			
		||||
    {
 | 
			
		||||
    public:
 | 
			
		||||
        using self_t = Crow;
 | 
			
		||||
        using server_t = Server<Crow, SocketAdaptor, Middlewares...>;
 | 
			
		||||
#ifdef CROW_ENABLE_SSL
 | 
			
		||||
        using ssl_server_t = Server<Crow, SSLAdaptor, Middlewares...>;
 | 
			
		||||
#endif
 | 
			
		||||
        Crow()
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void handle(const request& req, response& res)
 | 
			
		||||
        {
 | 
			
		||||
            router_.handle(req, res);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        DynamicRule& route_dynamic(std::string&& rule)
 | 
			
		||||
        {
 | 
			
		||||
            return router_.new_rule_dynamic(std::move(rule));
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <uint64_t Tag>
 | 
			
		||||
        auto route(std::string&& rule)
 | 
			
		||||
            -> typename std::result_of<decltype(&Router::new_rule_tagged<Tag>)(Router, std::string&&)>::type
 | 
			
		||||
        {
 | 
			
		||||
            return router_.new_rule_tagged<Tag>(std::move(rule));
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        self_t& port(std::uint16_t port)
 | 
			
		||||
        {
 | 
			
		||||
            port_ = port;
 | 
			
		||||
            return *this;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        self_t& multithreaded()
 | 
			
		||||
        {
 | 
			
		||||
            return concurrency(std::thread::hardware_concurrency());
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        self_t& concurrency(std::uint16_t concurrency)
 | 
			
		||||
        {
 | 
			
		||||
            if (concurrency < 1)
 | 
			
		||||
                concurrency = 1;
 | 
			
		||||
            concurrency_ = concurrency;
 | 
			
		||||
            return *this;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void validate()
 | 
			
		||||
        {
 | 
			
		||||
            router_.validate();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void run()
 | 
			
		||||
        {
 | 
			
		||||
            validate();
 | 
			
		||||
#ifdef CROW_ENABLE_SSL
 | 
			
		||||
            if (use_ssl_)
 | 
			
		||||
            {
 | 
			
		||||
                ssl_server_t server(this, port_, &middlewares_, concurrency_, &ssl_context_);
 | 
			
		||||
                server.run();
 | 
			
		||||
            }
 | 
			
		||||
            else
 | 
			
		||||
#endif
 | 
			
		||||
            {
 | 
			
		||||
                server_t server(this, port_, &middlewares_, concurrency_, nullptr);
 | 
			
		||||
                server.run();
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void debug_print()
 | 
			
		||||
        {
 | 
			
		||||
            CROW_LOG_DEBUG << "Routing:";
 | 
			
		||||
            router_.debug_print();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
#ifdef CROW_ENABLE_SSL
 | 
			
		||||
        self_t& ssl_file(const std::string& crt_filename, const std::string& key_filename)
 | 
			
		||||
        {
 | 
			
		||||
            use_ssl_ = true;
 | 
			
		||||
            ssl_context_.set_verify_mode(boost::asio::ssl::verify_peer);
 | 
			
		||||
            ssl_context_.use_certificate_file(crt_filename, ssl_context_t::pem);
 | 
			
		||||
            ssl_context_.use_private_key_file(key_filename, ssl_context_t::pem);
 | 
			
		||||
            ssl_context_.set_options(
 | 
			
		||||
                    boost::asio::ssl::context::default_workarounds
 | 
			
		||||
                          | boost::asio::ssl::context::no_sslv2
 | 
			
		||||
                          | boost::asio::ssl::context::no_sslv3
 | 
			
		||||
                    );
 | 
			
		||||
            return *this;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        self_t& ssl_file(const std::string& pem_filename)
 | 
			
		||||
        {
 | 
			
		||||
            use_ssl_ = true;
 | 
			
		||||
            ssl_context_.set_verify_mode(boost::asio::ssl::verify_peer);
 | 
			
		||||
            ssl_context_.load_verify_file(pem_filename);
 | 
			
		||||
            ssl_context_.set_options(
 | 
			
		||||
                    boost::asio::ssl::context::default_workarounds
 | 
			
		||||
                          | boost::asio::ssl::context::no_sslv2
 | 
			
		||||
                          | boost::asio::ssl::context::no_sslv3
 | 
			
		||||
                    );
 | 
			
		||||
            return *this;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        self_t& ssl(boost::asio::ssl::context&& ctx)
 | 
			
		||||
        {
 | 
			
		||||
            use_ssl_ = true;
 | 
			
		||||
            ssl_context_ = std::move(ctx);
 | 
			
		||||
            return *this;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        bool use_ssl_{false};
 | 
			
		||||
        ssl_context_t ssl_context_{boost::asio::ssl::context::sslv23};
 | 
			
		||||
 | 
			
		||||
#else
 | 
			
		||||
        template <typename T, typename ... Remain>
 | 
			
		||||
        self_t& ssl_file(T&& t, Remain&&...)
 | 
			
		||||
        {
 | 
			
		||||
            // We can't call .ssl() member function unless CROW_ENABLE_SSL is defined.
 | 
			
		||||
            static_assert(
 | 
			
		||||
                    // make static_assert dependent to T; always false
 | 
			
		||||
                    std::is_base_of<T, void>::value, 
 | 
			
		||||
                    "Define CROW_ENABLE_SSL to enable ssl support.");
 | 
			
		||||
            return *this;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <typename T>
 | 
			
		||||
        self_t& ssl(T&& ctx)
 | 
			
		||||
        {
 | 
			
		||||
            // We can't call .ssl() member function unless CROW_ENABLE_SSL is defined.
 | 
			
		||||
            static_assert(
 | 
			
		||||
                    // make static_assert dependent to T; always false
 | 
			
		||||
                    std::is_base_of<T, void>::value, 
 | 
			
		||||
                    "Define CROW_ENABLE_SSL to enable ssl support.");
 | 
			
		||||
            return *this;
 | 
			
		||||
        }
 | 
			
		||||
#endif
 | 
			
		||||
 | 
			
		||||
        // middleware
 | 
			
		||||
        using context_t = detail::context<Middlewares...>;
 | 
			
		||||
        template <typename T>
 | 
			
		||||
        typename T::context& get_context(const request& req)
 | 
			
		||||
        {
 | 
			
		||||
            static_assert(black_magic::contains<T, Middlewares...>::value, "App doesn't have the specified middleware type.");
 | 
			
		||||
            auto& ctx = *reinterpret_cast<context_t*>(req.middleware_context);
 | 
			
		||||
            return ctx.template get<T>();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <typename T>
 | 
			
		||||
        T& get_middleware()
 | 
			
		||||
        {
 | 
			
		||||
            return utility::get_element_by_type<T, Middlewares...>(middlewares_);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
    private:
 | 
			
		||||
        uint16_t port_ = 80;
 | 
			
		||||
        uint16_t concurrency_ = 1;
 | 
			
		||||
 | 
			
		||||
        Router router_;
 | 
			
		||||
 | 
			
		||||
        std::tuple<Middlewares...> middlewares_;
 | 
			
		||||
    };
 | 
			
		||||
    template <typename ... Middlewares>
 | 
			
		||||
    using App = Crow<Middlewares...>;
 | 
			
		||||
    using SimpleApp = Crow<>;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										81
									
								
								ext/crow/dumb_timer_queue.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										81
									
								
								ext/crow/dumb_timer_queue.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,81 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <boost/asio.hpp>
 | 
			
		||||
#include <deque>
 | 
			
		||||
#include <functional>
 | 
			
		||||
#include <chrono>
 | 
			
		||||
#include <thread>
 | 
			
		||||
 | 
			
		||||
#include "logging.h"
 | 
			
		||||
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
    namespace detail 
 | 
			
		||||
    {
 | 
			
		||||
        // fast timer queue for fixed tick value.
 | 
			
		||||
        class dumb_timer_queue
 | 
			
		||||
        {
 | 
			
		||||
        public:
 | 
			
		||||
            using key = std::pair<dumb_timer_queue*, int>;
 | 
			
		||||
 | 
			
		||||
            void cancel(key& k)
 | 
			
		||||
            {
 | 
			
		||||
                auto self = k.first;
 | 
			
		||||
                k.first = nullptr;
 | 
			
		||||
                if (!self)
 | 
			
		||||
                    return;
 | 
			
		||||
 | 
			
		||||
                unsigned int index = (unsigned int)(k.second - self->step_);
 | 
			
		||||
                if (index < self->dq_.size())
 | 
			
		||||
                    self->dq_[index].second = nullptr;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            key add(std::function<void()> f)
 | 
			
		||||
            {
 | 
			
		||||
                dq_.emplace_back(std::chrono::steady_clock::now(), std::move(f));
 | 
			
		||||
                int ret = step_+dq_.size()-1;
 | 
			
		||||
 | 
			
		||||
                CROW_LOG_DEBUG << "timer add inside: " << this << ' ' << ret ;
 | 
			
		||||
                return {this, ret};
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            void process()
 | 
			
		||||
            {
 | 
			
		||||
                if (!io_service_)
 | 
			
		||||
                    return;
 | 
			
		||||
 | 
			
		||||
                auto now = std::chrono::steady_clock::now();
 | 
			
		||||
                while(!dq_.empty())
 | 
			
		||||
                {
 | 
			
		||||
                    auto& x = dq_.front();
 | 
			
		||||
                    if (now - x.first < std::chrono::seconds(tick))
 | 
			
		||||
                        break;
 | 
			
		||||
                    if (x.second)
 | 
			
		||||
                    {
 | 
			
		||||
                        CROW_LOG_DEBUG << "timer call: " << this << ' ' << step_;
 | 
			
		||||
                        // we know that timer handlers are very simple currenty; call here
 | 
			
		||||
                        x.second();
 | 
			
		||||
                    }
 | 
			
		||||
                    dq_.pop_front();
 | 
			
		||||
                    step_++;
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            void set_io_service(boost::asio::io_service& io_service)
 | 
			
		||||
            {
 | 
			
		||||
                io_service_ = &io_service;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            dumb_timer_queue() noexcept
 | 
			
		||||
            {
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
        private:
 | 
			
		||||
 | 
			
		||||
            int tick{5};
 | 
			
		||||
            boost::asio::io_service* io_service_{};
 | 
			
		||||
            std::deque<std::pair<decltype(std::chrono::steady_clock::now()), std::function<void()>>> dq_;
 | 
			
		||||
            int step_{};
 | 
			
		||||
        };
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										575
									
								
								ext/crow/http_connection.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										575
									
								
								ext/crow/http_connection.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,575 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
#include <boost/asio.hpp>
 | 
			
		||||
#include <boost/algorithm/string/predicate.hpp>
 | 
			
		||||
#include <boost/lexical_cast.hpp>
 | 
			
		||||
#include <boost/array.hpp>
 | 
			
		||||
#include <atomic>
 | 
			
		||||
#include <chrono>
 | 
			
		||||
#include <vector>
 | 
			
		||||
 | 
			
		||||
#include "http_parser_merged.h"
 | 
			
		||||
 | 
			
		||||
#include "parser.h"
 | 
			
		||||
#include "http_response.h"
 | 
			
		||||
#include "logging.h"
 | 
			
		||||
#include "settings.h"
 | 
			
		||||
#include "dumb_timer_queue.h"
 | 
			
		||||
#include "middleware_context.h"
 | 
			
		||||
#include "socket_adaptors.h"
 | 
			
		||||
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
    using namespace boost;
 | 
			
		||||
    using tcp = asio::ip::tcp;
 | 
			
		||||
 | 
			
		||||
    namespace detail
 | 
			
		||||
    {
 | 
			
		||||
        template <typename MW>
 | 
			
		||||
        struct check_before_handle_arity_3_const
 | 
			
		||||
        {
 | 
			
		||||
            template <typename T,
 | 
			
		||||
                void (T::*)(request&, response&, typename MW::context&) const = &T::before_handle
 | 
			
		||||
            >
 | 
			
		||||
            struct get
 | 
			
		||||
            { };
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <typename MW>
 | 
			
		||||
        struct check_before_handle_arity_3
 | 
			
		||||
        {
 | 
			
		||||
            template <typename T,
 | 
			
		||||
                void (T::*)(request&, response&, typename MW::context&) = &T::before_handle
 | 
			
		||||
            >
 | 
			
		||||
            struct get
 | 
			
		||||
            { };
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <typename MW>
 | 
			
		||||
        struct check_after_handle_arity_3_const
 | 
			
		||||
        {
 | 
			
		||||
            template <typename T,
 | 
			
		||||
                void (T::*)(request&, response&, typename MW::context&) const = &T::after_handle
 | 
			
		||||
            >
 | 
			
		||||
            struct get
 | 
			
		||||
            { };
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <typename MW>
 | 
			
		||||
        struct check_after_handle_arity_3
 | 
			
		||||
        {
 | 
			
		||||
            template <typename T,
 | 
			
		||||
                void (T::*)(request&, response&, typename MW::context&) = &T::after_handle
 | 
			
		||||
            >
 | 
			
		||||
            struct get
 | 
			
		||||
            { };
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <typename T>
 | 
			
		||||
        struct is_before_handle_arity_3_impl
 | 
			
		||||
        {
 | 
			
		||||
            template <typename C>
 | 
			
		||||
            static std::true_type f(typename check_before_handle_arity_3_const<T>::template get<C>*);
 | 
			
		||||
 | 
			
		||||
            template <typename C>
 | 
			
		||||
            static std::true_type f(typename check_before_handle_arity_3<T>::template get<C>*);
 | 
			
		||||
 | 
			
		||||
            template <typename C>
 | 
			
		||||
            static std::false_type f(...);
 | 
			
		||||
 | 
			
		||||
        public:
 | 
			
		||||
            static const bool value = decltype(f<T>(nullptr))::value;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <typename T>
 | 
			
		||||
        struct is_after_handle_arity_3_impl
 | 
			
		||||
        {
 | 
			
		||||
            template <typename C>
 | 
			
		||||
            static std::true_type f(typename check_after_handle_arity_3_const<T>::template get<C>*);
 | 
			
		||||
 | 
			
		||||
            template <typename C>
 | 
			
		||||
            static std::true_type f(typename check_after_handle_arity_3<T>::template get<C>*);
 | 
			
		||||
 | 
			
		||||
            template <typename C>
 | 
			
		||||
            static std::false_type f(...);
 | 
			
		||||
 | 
			
		||||
        public:
 | 
			
		||||
            static const bool value = decltype(f<T>(nullptr))::value;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <typename MW, typename Context, typename ParentContext>
 | 
			
		||||
        typename std::enable_if<!is_before_handle_arity_3_impl<MW>::value>::type
 | 
			
		||||
        before_handler_call(MW& mw, request& req, response& res, Context& ctx, ParentContext& parent_ctx)
 | 
			
		||||
        {
 | 
			
		||||
            mw.before_handle(req, res, ctx.template get<MW>(), ctx);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <typename MW, typename Context, typename ParentContext>
 | 
			
		||||
        typename std::enable_if<is_before_handle_arity_3_impl<MW>::value>::type
 | 
			
		||||
        before_handler_call(MW& mw, request& req, response& res, Context& ctx, ParentContext& parent_ctx)
 | 
			
		||||
        {
 | 
			
		||||
            mw.before_handle(req, res, ctx.template get<MW>());
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <typename MW, typename Context, typename ParentContext>
 | 
			
		||||
        typename std::enable_if<!is_after_handle_arity_3_impl<MW>::value>::type
 | 
			
		||||
        after_handler_call(MW& mw, request& req, response& res, Context& ctx, ParentContext& parent_ctx)
 | 
			
		||||
        {
 | 
			
		||||
            mw.after_handle(req, res, ctx.template get<MW>(), ctx);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <typename MW, typename Context, typename ParentContext>
 | 
			
		||||
        typename std::enable_if<is_after_handle_arity_3_impl<MW>::value>::type
 | 
			
		||||
        after_handler_call(MW& mw, request& req, response& res, Context& ctx, ParentContext& parent_ctx)
 | 
			
		||||
        {
 | 
			
		||||
            mw.after_handle(req, res, ctx.template get<MW>());
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <int N, typename Context, typename Container, typename CurrentMW, typename ... Middlewares>
 | 
			
		||||
        bool middleware_call_helper(Container& middlewares, request& req, response& res, Context& ctx)
 | 
			
		||||
        {
 | 
			
		||||
            using parent_context_t = typename Context::template partial<N-1>;
 | 
			
		||||
            before_handler_call<CurrentMW, Context, parent_context_t>(std::get<N>(middlewares), req, res, ctx, static_cast<parent_context_t&>(ctx));
 | 
			
		||||
 | 
			
		||||
            if (res.is_completed())
 | 
			
		||||
            {
 | 
			
		||||
                after_handler_call<CurrentMW, Context, parent_context_t>(std::get<N>(middlewares), req, res, ctx, static_cast<parent_context_t&>(ctx));
 | 
			
		||||
                return true;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            if (middleware_call_helper<N+1, Context, Container, Middlewares...>(middlewares, req, res, ctx))
 | 
			
		||||
            {
 | 
			
		||||
                after_handler_call<CurrentMW, Context, parent_context_t>(std::get<N>(middlewares), req, res, ctx, static_cast<parent_context_t&>(ctx));
 | 
			
		||||
                return true;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <int N, typename Context, typename Container>
 | 
			
		||||
        bool middleware_call_helper(Container& middlewares, request& req, response& res, Context& ctx)
 | 
			
		||||
        {
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <int N, typename Context, typename Container>
 | 
			
		||||
        typename std::enable_if<(N<0)>::type 
 | 
			
		||||
        after_handlers_call_helper(Container& middlewares, Context& context, request& req, response& res)
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <int N, typename Context, typename Container>
 | 
			
		||||
        typename std::enable_if<(N==0)>::type after_handlers_call_helper(Container& middlewares, Context& ctx, request& req, response& res)
 | 
			
		||||
        {
 | 
			
		||||
            using parent_context_t = typename Context::template partial<N-1>;
 | 
			
		||||
            using CurrentMW = typename std::tuple_element<N, typename std::remove_reference<Container>::type>::type;
 | 
			
		||||
            after_handler_call<CurrentMW, Context, parent_context_t>(std::get<N>(middlewares), req, res, ctx, static_cast<parent_context_t&>(ctx));
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <int N, typename Context, typename Container>
 | 
			
		||||
        typename std::enable_if<(N>0)>::type after_handlers_call_helper(Container& middlewares, Context& ctx, request& req, response& res)
 | 
			
		||||
        {
 | 
			
		||||
            using parent_context_t = typename Context::template partial<N-1>;
 | 
			
		||||
            using CurrentMW = typename std::tuple_element<N, typename std::remove_reference<Container>::type>::type;
 | 
			
		||||
            after_handler_call<CurrentMW, Context, parent_context_t>(std::get<N>(middlewares), req, res, ctx, static_cast<parent_context_t&>(ctx));
 | 
			
		||||
            after_handlers_call_helper<N-1, Context, Container>(middlewares, ctx, req, res);
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
#ifdef CROW_ENABLE_DEBUG
 | 
			
		||||
    static int connectionCount;
 | 
			
		||||
#endif
 | 
			
		||||
    template <typename Adaptor, typename Handler, typename ... Middlewares>
 | 
			
		||||
    class Connection
 | 
			
		||||
    {
 | 
			
		||||
    public:
 | 
			
		||||
        Connection(
 | 
			
		||||
            boost::asio::io_service& io_service, 
 | 
			
		||||
            Handler* handler, 
 | 
			
		||||
            const std::string& server_name,
 | 
			
		||||
            std::tuple<Middlewares...>* middlewares,
 | 
			
		||||
            std::function<std::string()>& get_cached_date_str_f,
 | 
			
		||||
            detail::dumb_timer_queue& timer_queue,
 | 
			
		||||
            typename Adaptor::context* adaptor_ctx_
 | 
			
		||||
            ) 
 | 
			
		||||
            : adaptor_(io_service, adaptor_ctx_), 
 | 
			
		||||
            handler_(handler), 
 | 
			
		||||
            parser_(this), 
 | 
			
		||||
            server_name_(server_name),
 | 
			
		||||
            middlewares_(middlewares),
 | 
			
		||||
            get_cached_date_str(get_cached_date_str_f),
 | 
			
		||||
            timer_queue(timer_queue)
 | 
			
		||||
        {
 | 
			
		||||
#ifdef CROW_ENABLE_DEBUG
 | 
			
		||||
            connectionCount ++;
 | 
			
		||||
            CROW_LOG_DEBUG << "Connection open, total " << connectionCount << ", " << this;
 | 
			
		||||
#endif
 | 
			
		||||
        }
 | 
			
		||||
        
 | 
			
		||||
        ~Connection()
 | 
			
		||||
        {
 | 
			
		||||
            res.complete_request_handler_ = nullptr;
 | 
			
		||||
            cancel_deadline_timer();
 | 
			
		||||
#ifdef CROW_ENABLE_DEBUG
 | 
			
		||||
            connectionCount --;
 | 
			
		||||
            CROW_LOG_DEBUG << "Connection closed, total " << connectionCount << ", " << this;
 | 
			
		||||
#endif
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        decltype(std::declval<Adaptor>().raw_socket())& socket()
 | 
			
		||||
        {
 | 
			
		||||
            return adaptor_.raw_socket();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void start()
 | 
			
		||||
        {
 | 
			
		||||
            adaptor_.start([this](const boost::system::error_code& ec) {
 | 
			
		||||
                if (!ec)
 | 
			
		||||
                {
 | 
			
		||||
                    start_deadline();
 | 
			
		||||
 | 
			
		||||
                    do_read();
 | 
			
		||||
                }
 | 
			
		||||
                else
 | 
			
		||||
                {
 | 
			
		||||
                    check_destroy();
 | 
			
		||||
                }
 | 
			
		||||
            });
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void handle_header()
 | 
			
		||||
        {
 | 
			
		||||
            // HTTP 1.1 Expect: 100-continue
 | 
			
		||||
            if (parser_.check_version(1, 1) && parser_.headers.count("expect") && get_header_value(parser_.headers, "expect") == "100-continue")
 | 
			
		||||
            {
 | 
			
		||||
                buffers_.clear();
 | 
			
		||||
                static std::string expect_100_continue = "HTTP/1.1 100 Continue\r\n\r\n";
 | 
			
		||||
                buffers_.emplace_back(expect_100_continue.data(), expect_100_continue.size());
 | 
			
		||||
                do_write();
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void handle()
 | 
			
		||||
        {
 | 
			
		||||
            cancel_deadline_timer();
 | 
			
		||||
            bool is_invalid_request = false;
 | 
			
		||||
            add_keep_alive_ = false;
 | 
			
		||||
 | 
			
		||||
            req_ = std::move(parser_.to_request());
 | 
			
		||||
            request& req = req_;
 | 
			
		||||
            if (parser_.check_version(1, 0))
 | 
			
		||||
            {
 | 
			
		||||
                // HTTP/1.0
 | 
			
		||||
                if (req.headers.count("connection"))
 | 
			
		||||
                {
 | 
			
		||||
                    if (boost::iequals(req.get_header_value("connection"),"Keep-Alive"))
 | 
			
		||||
                        add_keep_alive_ = true;
 | 
			
		||||
                }
 | 
			
		||||
                else
 | 
			
		||||
                    close_connection_ = true;
 | 
			
		||||
            }
 | 
			
		||||
            else if (parser_.check_version(1, 1))
 | 
			
		||||
            {
 | 
			
		||||
                // HTTP/1.1
 | 
			
		||||
                if (req.headers.count("connection"))
 | 
			
		||||
                {
 | 
			
		||||
                    if (req.get_header_value("connection") == "close")
 | 
			
		||||
                        close_connection_ = true;
 | 
			
		||||
                    else if (boost::iequals(req.get_header_value("connection"),"Keep-Alive"))
 | 
			
		||||
                        add_keep_alive_ = true;
 | 
			
		||||
                }
 | 
			
		||||
                if (!req.headers.count("host"))
 | 
			
		||||
                {
 | 
			
		||||
                    is_invalid_request = true;
 | 
			
		||||
                    res = response(400);
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            CROW_LOG_INFO << "Request: " << boost::lexical_cast<std::string>(adaptor_.remote_endpoint()) << " " << this << " HTTP/" << parser_.http_major << "." << parser_.http_minor << ' '
 | 
			
		||||
             << method_name(req.method) << " " << req.url;
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
            need_to_call_after_handlers_ = false;
 | 
			
		||||
            if (!is_invalid_request)
 | 
			
		||||
            {
 | 
			
		||||
                res.complete_request_handler_ = []{};
 | 
			
		||||
                res.is_alive_helper_ = [this]()->bool{ return adaptor_.is_open(); };
 | 
			
		||||
 | 
			
		||||
                ctx_ = detail::context<Middlewares...>();
 | 
			
		||||
                req.middleware_context = (void*)&ctx_;
 | 
			
		||||
                detail::middleware_call_helper<0, decltype(ctx_), decltype(*middlewares_), Middlewares...>(*middlewares_, req, res, ctx_);
 | 
			
		||||
 | 
			
		||||
                if (!res.completed_)
 | 
			
		||||
                {
 | 
			
		||||
                    res.complete_request_handler_ = [this]{ this->complete_request(); };
 | 
			
		||||
                    need_to_call_after_handlers_ = true;
 | 
			
		||||
                    handler_->handle(req, res);
 | 
			
		||||
                    if (add_keep_alive_)
 | 
			
		||||
                        res.set_header("connection", "Keep-Alive");
 | 
			
		||||
                }
 | 
			
		||||
                else
 | 
			
		||||
                {
 | 
			
		||||
                    complete_request();
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            else
 | 
			
		||||
            {
 | 
			
		||||
                complete_request();
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void complete_request()
 | 
			
		||||
        {
 | 
			
		||||
            CROW_LOG_INFO << "Response: " << this << ' ' << req_.raw_url << ' ' << res.code << ' ' << close_connection_;
 | 
			
		||||
 | 
			
		||||
            if (need_to_call_after_handlers_)
 | 
			
		||||
            {
 | 
			
		||||
                need_to_call_after_handlers_ = false;
 | 
			
		||||
 | 
			
		||||
                // call all after_handler of middlewares
 | 
			
		||||
                detail::after_handlers_call_helper<
 | 
			
		||||
                    ((int)sizeof...(Middlewares)-1),
 | 
			
		||||
                    decltype(ctx_),
 | 
			
		||||
                    decltype(*middlewares_)> 
 | 
			
		||||
                (*middlewares_, ctx_, req_, res);
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            //auto self = this->shared_from_this();
 | 
			
		||||
            res.complete_request_handler_ = nullptr;
 | 
			
		||||
            
 | 
			
		||||
            if (!adaptor_.is_open())
 | 
			
		||||
            {
 | 
			
		||||
                //CROW_LOG_DEBUG << this << " delete (socket is closed) " << is_reading << ' ' << is_writing;
 | 
			
		||||
                //delete this;
 | 
			
		||||
                return;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            static std::unordered_map<int, std::string> statusCodes = {
 | 
			
		||||
                {200, "HTTP/1.1 200 OK\r\n"},
 | 
			
		||||
                {201, "HTTP/1.1 201 Created\r\n"},
 | 
			
		||||
                {202, "HTTP/1.1 202 Accepted\r\n"},
 | 
			
		||||
                {204, "HTTP/1.1 204 No Content\r\n"},
 | 
			
		||||
 | 
			
		||||
                {300, "HTTP/1.1 300 Multiple Choices\r\n"},
 | 
			
		||||
                {301, "HTTP/1.1 301 Moved Permanently\r\n"},
 | 
			
		||||
                {302, "HTTP/1.1 302 Moved Temporarily\r\n"},
 | 
			
		||||
                {304, "HTTP/1.1 304 Not Modified\r\n"},
 | 
			
		||||
 | 
			
		||||
                {400, "HTTP/1.1 400 Bad Request\r\n"},
 | 
			
		||||
                {401, "HTTP/1.1 401 Unauthorized\r\n"},
 | 
			
		||||
                {403, "HTTP/1.1 403 Forbidden\r\n"},
 | 
			
		||||
                {404, "HTTP/1.1 404 Not Found\r\n"},
 | 
			
		||||
 | 
			
		||||
                {500, "HTTP/1.1 500 Internal Server Error\r\n"},
 | 
			
		||||
                {501, "HTTP/1.1 501 Not Implemented\r\n"},
 | 
			
		||||
                {502, "HTTP/1.1 502 Bad Gateway\r\n"},
 | 
			
		||||
                {503, "HTTP/1.1 503 Service Unavailable\r\n"},
 | 
			
		||||
            };
 | 
			
		||||
 | 
			
		||||
            static std::string seperator = ": ";
 | 
			
		||||
            static std::string crlf = "\r\n";
 | 
			
		||||
 | 
			
		||||
            buffers_.clear();
 | 
			
		||||
            buffers_.reserve(4*(res.headers.size()+5)+3);
 | 
			
		||||
 | 
			
		||||
            if (res.body.empty() && res.json_value.t() == json::type::Object)
 | 
			
		||||
            {
 | 
			
		||||
                res.body = json::dump(res.json_value);
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            if (!statusCodes.count(res.code))
 | 
			
		||||
                res.code = 500;
 | 
			
		||||
            {
 | 
			
		||||
                auto& status = statusCodes.find(res.code)->second;
 | 
			
		||||
                buffers_.emplace_back(status.data(), status.size());
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            if (res.code >= 400 && res.body.empty())
 | 
			
		||||
                res.body = statusCodes[res.code].substr(9);
 | 
			
		||||
 | 
			
		||||
            for(auto& kv : res.headers)
 | 
			
		||||
            {
 | 
			
		||||
                buffers_.emplace_back(kv.first.data(), kv.first.size());
 | 
			
		||||
                buffers_.emplace_back(seperator.data(), seperator.size());
 | 
			
		||||
                buffers_.emplace_back(kv.second.data(), kv.second.size());
 | 
			
		||||
                buffers_.emplace_back(crlf.data(), crlf.size());
 | 
			
		||||
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            if (!res.headers.count("content-length"))
 | 
			
		||||
            {
 | 
			
		||||
                content_length_ = std::to_string(res.body.size());
 | 
			
		||||
                static std::string content_length_tag = "Content-Length: ";
 | 
			
		||||
                buffers_.emplace_back(content_length_tag.data(), content_length_tag.size());
 | 
			
		||||
                buffers_.emplace_back(content_length_.data(), content_length_.size());
 | 
			
		||||
                buffers_.emplace_back(crlf.data(), crlf.size());
 | 
			
		||||
            }
 | 
			
		||||
            if (!res.headers.count("server"))
 | 
			
		||||
            {
 | 
			
		||||
                static std::string server_tag = "Server: ";
 | 
			
		||||
                buffers_.emplace_back(server_tag.data(), server_tag.size());
 | 
			
		||||
                buffers_.emplace_back(server_name_.data(), server_name_.size());
 | 
			
		||||
                buffers_.emplace_back(crlf.data(), crlf.size());
 | 
			
		||||
            }
 | 
			
		||||
            if (!res.headers.count("date"))
 | 
			
		||||
            {
 | 
			
		||||
                static std::string date_tag = "Date: ";
 | 
			
		||||
                date_str_ = get_cached_date_str();
 | 
			
		||||
                buffers_.emplace_back(date_tag.data(), date_tag.size());
 | 
			
		||||
                buffers_.emplace_back(date_str_.data(), date_str_.size());
 | 
			
		||||
                buffers_.emplace_back(crlf.data(), crlf.size());
 | 
			
		||||
            }
 | 
			
		||||
            if (add_keep_alive_)
 | 
			
		||||
            {
 | 
			
		||||
                static std::string keep_alive_tag = "Connection: Keep-Alive";
 | 
			
		||||
                buffers_.emplace_back(keep_alive_tag.data(), keep_alive_tag.size());
 | 
			
		||||
                buffers_.emplace_back(crlf.data(), crlf.size());
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            buffers_.emplace_back(crlf.data(), crlf.size());
 | 
			
		||||
            res_body_copy_.swap(res.body);
 | 
			
		||||
            buffers_.emplace_back(res_body_copy_.data(), res_body_copy_.size());
 | 
			
		||||
 | 
			
		||||
            do_write();
 | 
			
		||||
 | 
			
		||||
            if (need_to_start_read_after_complete_)
 | 
			
		||||
            {
 | 
			
		||||
                need_to_start_read_after_complete_ = false;
 | 
			
		||||
                start_deadline();
 | 
			
		||||
                do_read();
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
    private:
 | 
			
		||||
        void do_read()
 | 
			
		||||
        {
 | 
			
		||||
            //auto self = this->shared_from_this();
 | 
			
		||||
            is_reading = true;
 | 
			
		||||
            adaptor_.socket().async_read_some(boost::asio::buffer(buffer_), 
 | 
			
		||||
                [this](const boost::system::error_code& ec, std::size_t bytes_transferred)
 | 
			
		||||
                {
 | 
			
		||||
                    bool error_while_reading = true;
 | 
			
		||||
                    if (!ec)
 | 
			
		||||
                    {
 | 
			
		||||
                        bool ret = parser_.feed(buffer_.data(), bytes_transferred);
 | 
			
		||||
                        if (ret && adaptor_.is_open() && !close_connection_)
 | 
			
		||||
                        {
 | 
			
		||||
                            error_while_reading = false;
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
 | 
			
		||||
                    if (error_while_reading)
 | 
			
		||||
                    {
 | 
			
		||||
                        cancel_deadline_timer();
 | 
			
		||||
                        parser_.done();
 | 
			
		||||
                        adaptor_.close();
 | 
			
		||||
                        is_reading = false;
 | 
			
		||||
                        CROW_LOG_DEBUG << this << " from read(1)";
 | 
			
		||||
                        check_destroy();
 | 
			
		||||
                    }
 | 
			
		||||
                    else if (!need_to_call_after_handlers_)
 | 
			
		||||
                    {
 | 
			
		||||
                        start_deadline();
 | 
			
		||||
                        do_read();
 | 
			
		||||
                    }
 | 
			
		||||
                    else
 | 
			
		||||
                    {
 | 
			
		||||
                        // res will be completed later by user
 | 
			
		||||
                        need_to_start_read_after_complete_ = true;
 | 
			
		||||
                    }
 | 
			
		||||
                });
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void do_write()
 | 
			
		||||
        {
 | 
			
		||||
            //auto self = this->shared_from_this();
 | 
			
		||||
            is_writing = true;
 | 
			
		||||
            boost::asio::async_write(adaptor_.socket(), buffers_, 
 | 
			
		||||
                [&](const boost::system::error_code& ec, std::size_t bytes_transferred)
 | 
			
		||||
                {
 | 
			
		||||
                    is_writing = false;
 | 
			
		||||
                    res.clear();
 | 
			
		||||
                    res_body_copy_.clear();
 | 
			
		||||
                    if (!ec)
 | 
			
		||||
                    {
 | 
			
		||||
                        if (close_connection_)
 | 
			
		||||
                        {
 | 
			
		||||
                            adaptor_.close();
 | 
			
		||||
                            CROW_LOG_DEBUG << this << " from write(1)";
 | 
			
		||||
                            check_destroy();
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                    else
 | 
			
		||||
                    {
 | 
			
		||||
                        CROW_LOG_DEBUG << this << " from write(2)";
 | 
			
		||||
                        check_destroy();
 | 
			
		||||
                    }
 | 
			
		||||
                });
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void check_destroy()
 | 
			
		||||
        {
 | 
			
		||||
            CROW_LOG_DEBUG << this << " is_reading " << is_reading << " is_writing " << is_writing;
 | 
			
		||||
            if (!is_reading && !is_writing)
 | 
			
		||||
            {
 | 
			
		||||
                CROW_LOG_DEBUG << this << " delete (idle) ";
 | 
			
		||||
                delete this;
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void cancel_deadline_timer()
 | 
			
		||||
        {
 | 
			
		||||
            CROW_LOG_DEBUG << this << " timer cancelled: " << timer_cancel_key_.first << ' ' << timer_cancel_key_.second;
 | 
			
		||||
            timer_queue.cancel(timer_cancel_key_);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void start_deadline(int timeout = 5)
 | 
			
		||||
        {
 | 
			
		||||
            cancel_deadline_timer();
 | 
			
		||||
            
 | 
			
		||||
            timer_cancel_key_ = timer_queue.add([this]
 | 
			
		||||
            {
 | 
			
		||||
                if (!adaptor_.is_open())
 | 
			
		||||
                {
 | 
			
		||||
                    return;
 | 
			
		||||
                }
 | 
			
		||||
                adaptor_.close();
 | 
			
		||||
            });
 | 
			
		||||
            CROW_LOG_DEBUG << this << " timer added: " << timer_cancel_key_.first << ' ' << timer_cancel_key_.second;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
    private:
 | 
			
		||||
        Adaptor adaptor_;
 | 
			
		||||
        Handler* handler_;
 | 
			
		||||
 | 
			
		||||
        boost::array<char, 4096> buffer_;
 | 
			
		||||
 | 
			
		||||
        HTTPParser<Connection> parser_;
 | 
			
		||||
        request req_;
 | 
			
		||||
        response res;
 | 
			
		||||
 | 
			
		||||
        bool close_connection_ = false;
 | 
			
		||||
 | 
			
		||||
        const std::string& server_name_;
 | 
			
		||||
        std::vector<boost::asio::const_buffer> buffers_;
 | 
			
		||||
 | 
			
		||||
        std::string content_length_;
 | 
			
		||||
        std::string date_str_;
 | 
			
		||||
        std::string res_body_copy_;
 | 
			
		||||
 | 
			
		||||
        //boost::asio::deadline_timer deadline_;
 | 
			
		||||
        detail::dumb_timer_queue::key timer_cancel_key_;
 | 
			
		||||
 | 
			
		||||
        bool is_reading{};
 | 
			
		||||
        bool is_writing{};
 | 
			
		||||
        bool need_to_call_after_handlers_{};
 | 
			
		||||
        bool need_to_start_read_after_complete_{};
 | 
			
		||||
        bool add_keep_alive_{};
 | 
			
		||||
 | 
			
		||||
        std::tuple<Middlewares...>* middlewares_;
 | 
			
		||||
        detail::context<Middlewares...> ctx_;
 | 
			
		||||
 | 
			
		||||
        std::function<std::string()>& get_cached_date_str;
 | 
			
		||||
        detail::dumb_timer_queue& timer_queue;
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										2640
									
								
								ext/crow/http_parser_merged.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										2640
									
								
								ext/crow/http_parser_merged.h
									
										
									
									
									
										Normal file
									
								
							
										
											
												File diff suppressed because it is too large
												Load diff
											
										
									
								
							
							
								
								
									
										52
									
								
								ext/crow/http_request.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										52
									
								
								ext/crow/http_request.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,52 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include "common.h"
 | 
			
		||||
#include "ci_map.h"
 | 
			
		||||
#include "query_string.h"
 | 
			
		||||
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
    template <typename T>
 | 
			
		||||
    inline const std::string& get_header_value(const T& headers, const std::string& key)
 | 
			
		||||
    {
 | 
			
		||||
        if (headers.count(key))
 | 
			
		||||
        {
 | 
			
		||||
            return headers.find(key)->second;
 | 
			
		||||
        }
 | 
			
		||||
        static std::string empty;
 | 
			
		||||
        return empty;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    struct request
 | 
			
		||||
    {
 | 
			
		||||
        HTTPMethod method;
 | 
			
		||||
        std::string raw_url;
 | 
			
		||||
        std::string url;
 | 
			
		||||
        query_string url_params;
 | 
			
		||||
        ci_map headers;
 | 
			
		||||
        std::string body;
 | 
			
		||||
 | 
			
		||||
        void* middleware_context{};
 | 
			
		||||
 | 
			
		||||
        request()
 | 
			
		||||
            : method(HTTPMethod::GET)
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        request(HTTPMethod method, std::string raw_url, std::string url, query_string url_params, ci_map headers, std::string body)
 | 
			
		||||
            : method(method), raw_url(std::move(raw_url)), url(std::move(url)), url_params(std::move(url_params)), headers(std::move(headers)), body(std::move(body))
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void add_header(std::string key, std::string value)
 | 
			
		||||
        {
 | 
			
		||||
            headers.emplace(std::move(key), std::move(value));
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        const std::string& get_header_value(const std::string& key) const
 | 
			
		||||
        {
 | 
			
		||||
            return crow::get_header_value(headers, key);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
    };
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										128
									
								
								ext/crow/http_response.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										128
									
								
								ext/crow/http_response.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,128 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
#include <string>
 | 
			
		||||
#include <unordered_map>
 | 
			
		||||
#include "json.h"
 | 
			
		||||
#include "http_request.h"
 | 
			
		||||
#include "ci_map.h"
 | 
			
		||||
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
    template <typename Adaptor, typename Handler, typename ... Middlewares>
 | 
			
		||||
    class Connection;
 | 
			
		||||
    struct response
 | 
			
		||||
    {
 | 
			
		||||
        template <typename Adaptor, typename Handler, typename ... Middlewares>
 | 
			
		||||
        friend class crow::Connection;
 | 
			
		||||
 | 
			
		||||
        std::string body;
 | 
			
		||||
        json::wvalue json_value;
 | 
			
		||||
        int code{200};
 | 
			
		||||
 | 
			
		||||
        // `headers' stores HTTP headers.
 | 
			
		||||
        ci_map headers;
 | 
			
		||||
 | 
			
		||||
        void set_header(std::string key, std::string value)
 | 
			
		||||
        {
 | 
			
		||||
            headers.erase(key);
 | 
			
		||||
            headers.emplace(std::move(key), std::move(value));
 | 
			
		||||
        }
 | 
			
		||||
        void add_header(std::string key, std::string value)
 | 
			
		||||
        {
 | 
			
		||||
            headers.emplace(std::move(key), std::move(value));
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        const std::string& get_header_value(const std::string& key)
 | 
			
		||||
        {
 | 
			
		||||
            return crow::get_header_value(headers, key);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        response() {}
 | 
			
		||||
        explicit response(int code) : code(code) {}
 | 
			
		||||
        response(std::string body) : body(std::move(body)) {}
 | 
			
		||||
        response(json::wvalue&& json_value) : json_value(std::move(json_value)) 
 | 
			
		||||
        {
 | 
			
		||||
            json_mode();    
 | 
			
		||||
        }
 | 
			
		||||
        response(int code, std::string body) : body(std::move(body)), code(code) {}
 | 
			
		||||
        response(const json::wvalue& json_value) : body(json::dump(json_value)) 
 | 
			
		||||
        {
 | 
			
		||||
            json_mode();
 | 
			
		||||
        }
 | 
			
		||||
        response(int code, const json::wvalue& json_value) : code(code), body(json::dump(json_value))
 | 
			
		||||
        {
 | 
			
		||||
            json_mode();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        response(response&& r)
 | 
			
		||||
        {
 | 
			
		||||
            *this = std::move(r);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        response& operator = (const response& r) = delete;
 | 
			
		||||
 | 
			
		||||
        response& operator = (response&& r) noexcept
 | 
			
		||||
        {
 | 
			
		||||
            body = std::move(r.body);
 | 
			
		||||
            json_value = std::move(r.json_value);
 | 
			
		||||
            code = r.code;
 | 
			
		||||
            headers = std::move(r.headers);
 | 
			
		||||
            completed_ = r.completed_;
 | 
			
		||||
            return *this;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        bool is_completed() const noexcept
 | 
			
		||||
        {
 | 
			
		||||
            return completed_;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void clear()
 | 
			
		||||
        {
 | 
			
		||||
            body.clear();
 | 
			
		||||
            json_value.clear();
 | 
			
		||||
            code = 200;
 | 
			
		||||
            headers.clear();
 | 
			
		||||
            completed_ = false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void write(const std::string& body_part)
 | 
			
		||||
        {
 | 
			
		||||
            body += body_part;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void end()
 | 
			
		||||
        {
 | 
			
		||||
            if (!completed_)
 | 
			
		||||
            {
 | 
			
		||||
                completed_ = true;
 | 
			
		||||
                
 | 
			
		||||
                if (complete_request_handler_)
 | 
			
		||||
                {
 | 
			
		||||
                    complete_request_handler_();
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void end(const std::string& body_part)
 | 
			
		||||
        {
 | 
			
		||||
            body += body_part;
 | 
			
		||||
            end();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        bool is_alive()
 | 
			
		||||
        {
 | 
			
		||||
            return is_alive_helper_ && is_alive_helper_();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        private:
 | 
			
		||||
            bool completed_{};
 | 
			
		||||
            std::function<void()> complete_request_handler_;
 | 
			
		||||
            std::function<bool()> is_alive_helper_;
 | 
			
		||||
            
 | 
			
		||||
            //In case of a JSON object, set the Content-Type header
 | 
			
		||||
            void json_mode()
 | 
			
		||||
            {
 | 
			
		||||
                set_header("Content-Type", "application/json");
 | 
			
		||||
            }
 | 
			
		||||
    };
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										184
									
								
								ext/crow/http_server.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										184
									
								
								ext/crow/http_server.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,184 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <boost/date_time/posix_time/posix_time.hpp>
 | 
			
		||||
#include <boost/asio.hpp>
 | 
			
		||||
#ifdef CROW_ENABLE_SSL
 | 
			
		||||
#include <boost/asio/ssl.hpp>
 | 
			
		||||
#endif
 | 
			
		||||
#include <cstdint>
 | 
			
		||||
#include <atomic>
 | 
			
		||||
#include <future>
 | 
			
		||||
#include <vector>
 | 
			
		||||
 | 
			
		||||
#include <memory>
 | 
			
		||||
 | 
			
		||||
#include "http_connection.h"
 | 
			
		||||
#include "logging.h"
 | 
			
		||||
#include "dumb_timer_queue.h"
 | 
			
		||||
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
    using namespace boost;
 | 
			
		||||
    using tcp = asio::ip::tcp;
 | 
			
		||||
    
 | 
			
		||||
    template <typename Handler, typename Adaptor = SocketAdaptor, typename ... Middlewares>
 | 
			
		||||
    class Server
 | 
			
		||||
    {
 | 
			
		||||
    public:
 | 
			
		||||
        Server(Handler* handler, uint16_t port, std::tuple<Middlewares...>* middlewares = nullptr, uint16_t concurrency = 1, typename Adaptor::context* adaptor_ctx = nullptr)
 | 
			
		||||
            : acceptor_(io_service_, tcp::endpoint(asio::ip::address(), port)), 
 | 
			
		||||
            signals_(io_service_, SIGINT, SIGTERM),
 | 
			
		||||
            handler_(handler), 
 | 
			
		||||
            concurrency_(concurrency),
 | 
			
		||||
            port_(port),
 | 
			
		||||
            middlewares_(middlewares),
 | 
			
		||||
            adaptor_ctx_(adaptor_ctx)
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void run()
 | 
			
		||||
        {
 | 
			
		||||
            if (concurrency_ < 0)
 | 
			
		||||
                concurrency_ = 1;
 | 
			
		||||
 | 
			
		||||
            for(int i = 0; i < concurrency_;  i++)
 | 
			
		||||
                io_service_pool_.emplace_back(new boost::asio::io_service());
 | 
			
		||||
            get_cached_date_str_pool_.resize(concurrency_);
 | 
			
		||||
            timer_queue_pool_.resize(concurrency_);
 | 
			
		||||
 | 
			
		||||
            std::vector<std::future<void>> v;
 | 
			
		||||
            for(uint16_t i = 0; i < concurrency_; i ++)
 | 
			
		||||
                v.push_back(
 | 
			
		||||
                        std::async(std::launch::async, [this, i]{
 | 
			
		||||
 | 
			
		||||
                            // thread local date string get function
 | 
			
		||||
                            auto last = std::chrono::steady_clock::now();
 | 
			
		||||
 | 
			
		||||
                            std::string date_str;
 | 
			
		||||
                            auto update_date_str = [&]
 | 
			
		||||
                            {
 | 
			
		||||
                                auto last_time_t = time(0);
 | 
			
		||||
                                tm my_tm;
 | 
			
		||||
 | 
			
		||||
#ifdef _MSC_VER
 | 
			
		||||
                                gmtime_s(&my_tm, &last_time_t);
 | 
			
		||||
#else
 | 
			
		||||
                                gmtime_r(&last_time_t, &my_tm);
 | 
			
		||||
#endif
 | 
			
		||||
                                date_str.resize(100);
 | 
			
		||||
                                size_t date_str_sz = strftime(&date_str[0], 99, "%a, %d %b %Y %H:%M:%S GMT", &my_tm);
 | 
			
		||||
                                date_str.resize(date_str_sz);
 | 
			
		||||
                            };
 | 
			
		||||
                            update_date_str();
 | 
			
		||||
                            get_cached_date_str_pool_[i] = [&]()->std::string
 | 
			
		||||
                            {
 | 
			
		||||
                                if (std::chrono::steady_clock::now() - last >= std::chrono::seconds(1))
 | 
			
		||||
                                {
 | 
			
		||||
                                    last = std::chrono::steady_clock::now();
 | 
			
		||||
                                    update_date_str();
 | 
			
		||||
                                }
 | 
			
		||||
                                return date_str;
 | 
			
		||||
                            };
 | 
			
		||||
 | 
			
		||||
                            // initializing timer queue
 | 
			
		||||
                            detail::dumb_timer_queue timer_queue;
 | 
			
		||||
                            timer_queue_pool_[i] = &timer_queue;
 | 
			
		||||
 | 
			
		||||
                            timer_queue.set_io_service(*io_service_pool_[i]);
 | 
			
		||||
                            boost::asio::deadline_timer timer(*io_service_pool_[i]);
 | 
			
		||||
                            timer.expires_from_now(boost::posix_time::seconds(1));
 | 
			
		||||
 | 
			
		||||
                            std::function<void(const boost::system::error_code& ec)> handler;
 | 
			
		||||
                            handler = [&](const boost::system::error_code& ec){
 | 
			
		||||
                                if (ec)
 | 
			
		||||
                                    return;
 | 
			
		||||
                                timer_queue.process();
 | 
			
		||||
                                timer.expires_from_now(boost::posix_time::seconds(1));
 | 
			
		||||
                                timer.async_wait(handler);
 | 
			
		||||
                            };
 | 
			
		||||
                            timer.async_wait(handler);
 | 
			
		||||
 | 
			
		||||
                            io_service_pool_[i]->run();
 | 
			
		||||
                        }));
 | 
			
		||||
            CROW_LOG_INFO << server_name_ << " server is running, local port " << port_;
 | 
			
		||||
 | 
			
		||||
            signals_.async_wait(
 | 
			
		||||
                [&](const boost::system::error_code& error, int signal_number){
 | 
			
		||||
                    stop();
 | 
			
		||||
                });
 | 
			
		||||
 | 
			
		||||
            for (int i = 0; i < concurrency_; i++)
 | 
			
		||||
            {
 | 
			
		||||
                while (timer_queue_pool_[i] == nullptr)
 | 
			
		||||
                    std::this_thread::yield();
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            do_accept();
 | 
			
		||||
 | 
			
		||||
            std::thread([this]{
 | 
			
		||||
                io_service_.run();
 | 
			
		||||
                CROW_LOG_INFO << "Exiting.";
 | 
			
		||||
            }).join();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void stop()
 | 
			
		||||
        {
 | 
			
		||||
            io_service_.stop();
 | 
			
		||||
            for(auto& io_service:io_service_pool_)
 | 
			
		||||
                io_service->stop();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
    private:
 | 
			
		||||
        asio::io_service& pick_io_service()
 | 
			
		||||
        {
 | 
			
		||||
            // TODO load balancing
 | 
			
		||||
            roundrobin_index_++;
 | 
			
		||||
            if (roundrobin_index_ >= io_service_pool_.size())
 | 
			
		||||
                roundrobin_index_ = 0;
 | 
			
		||||
            return *io_service_pool_[roundrobin_index_];
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void do_accept()
 | 
			
		||||
        {
 | 
			
		||||
            asio::io_service& is = pick_io_service();
 | 
			
		||||
            auto p = new Connection<Adaptor, Handler, Middlewares...>(
 | 
			
		||||
                is, handler_, server_name_, middlewares_,
 | 
			
		||||
                get_cached_date_str_pool_[roundrobin_index_], *timer_queue_pool_[roundrobin_index_],
 | 
			
		||||
                adaptor_ctx_);
 | 
			
		||||
            acceptor_.async_accept(p->socket(), 
 | 
			
		||||
                [this, p, &is](boost::system::error_code ec)
 | 
			
		||||
                {
 | 
			
		||||
                    if (!ec)
 | 
			
		||||
                    {
 | 
			
		||||
                        is.post([p]
 | 
			
		||||
                        {
 | 
			
		||||
                            p->start();
 | 
			
		||||
                        });
 | 
			
		||||
                    }
 | 
			
		||||
                    do_accept();
 | 
			
		||||
                });
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
    private:
 | 
			
		||||
        asio::io_service io_service_;
 | 
			
		||||
        std::vector<std::unique_ptr<asio::io_service>> io_service_pool_;
 | 
			
		||||
        std::vector<detail::dumb_timer_queue*> timer_queue_pool_;
 | 
			
		||||
        std::vector<std::function<std::string()>> get_cached_date_str_pool_;
 | 
			
		||||
        tcp::acceptor acceptor_;
 | 
			
		||||
        boost::asio::signal_set signals_;
 | 
			
		||||
 | 
			
		||||
        Handler* handler_;
 | 
			
		||||
        uint16_t concurrency_{1};
 | 
			
		||||
        std::string server_name_ = "Crow/0.1";
 | 
			
		||||
        uint16_t port_;
 | 
			
		||||
        unsigned int roundrobin_index_{};
 | 
			
		||||
 | 
			
		||||
        std::tuple<Middlewares...>* middlewares_;
 | 
			
		||||
 | 
			
		||||
#ifdef CROW_ENABLE_SSL
 | 
			
		||||
        bool use_ssl_{false};
 | 
			
		||||
        boost::asio::ssl::context ssl_context_{boost::asio::ssl::context::sslv23};
 | 
			
		||||
#endif
 | 
			
		||||
        typename Adaptor::context* adaptor_ctx_;
 | 
			
		||||
    };
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										1422
									
								
								ext/crow/json.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										1422
									
								
								ext/crow/json.h
									
										
									
									
									
										Normal file
									
								
							
										
											
												File diff suppressed because it is too large
												Load diff
											
										
									
								
							
							
								
								
									
										134
									
								
								ext/crow/logging.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										134
									
								
								ext/crow/logging.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,134 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <string>
 | 
			
		||||
#include <cstdio>
 | 
			
		||||
#include <cstdlib>
 | 
			
		||||
#include <ctime>
 | 
			
		||||
#include <iostream>
 | 
			
		||||
#include <sstream>
 | 
			
		||||
 | 
			
		||||
#include "settings.h"
 | 
			
		||||
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
    enum class LogLevel
 | 
			
		||||
    {
 | 
			
		||||
        DEBUG,
 | 
			
		||||
        INFO,
 | 
			
		||||
        WARNING,
 | 
			
		||||
        ERROR,
 | 
			
		||||
        CRITICAL,
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    class ILogHandler {
 | 
			
		||||
        public:
 | 
			
		||||
            virtual void log(std::string message, LogLevel level) = 0;
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    class CerrLogHandler : public ILogHandler {
 | 
			
		||||
        public:
 | 
			
		||||
            void log(std::string message, LogLevel level) override {
 | 
			
		||||
                std::cerr << message;
 | 
			
		||||
            }
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    class logger {
 | 
			
		||||
 | 
			
		||||
        private:
 | 
			
		||||
            //
 | 
			
		||||
            static std::string timestamp()
 | 
			
		||||
            {
 | 
			
		||||
                char date[32];
 | 
			
		||||
                time_t t = time(0);
 | 
			
		||||
 | 
			
		||||
                tm my_tm;
 | 
			
		||||
 | 
			
		||||
#ifdef _MSC_VER
 | 
			
		||||
                gmtime_s(&my_tm, &t);
 | 
			
		||||
#else
 | 
			
		||||
                gmtime_r(&t, &my_tm);
 | 
			
		||||
#endif
 | 
			
		||||
 | 
			
		||||
                size_t sz = strftime(date, sizeof(date), "%Y-%m-%d %H:%M:%S", &my_tm);
 | 
			
		||||
                return std::string(date, date+sz);
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
        public:
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
            logger(std::string prefix, LogLevel level) : level_(level) {
 | 
			
		||||
    #ifdef CROW_ENABLE_LOGGING
 | 
			
		||||
                    stringstream_ << "(" << timestamp() << ") [" << prefix << "] ";
 | 
			
		||||
    #endif
 | 
			
		||||
 | 
			
		||||
            }
 | 
			
		||||
            ~logger() {
 | 
			
		||||
    #ifdef CROW_ENABLE_LOGGING
 | 
			
		||||
                if(level_ >= get_current_log_level()) {
 | 
			
		||||
                    stringstream_ << std::endl;
 | 
			
		||||
                    get_handler_ref()->log(stringstream_.str(), level_);
 | 
			
		||||
                }
 | 
			
		||||
    #endif
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            //
 | 
			
		||||
            template <typename T>
 | 
			
		||||
            logger& operator<<(T const &value) {
 | 
			
		||||
 | 
			
		||||
    #ifdef CROW_ENABLE_LOGGING
 | 
			
		||||
                if(level_ >= get_current_log_level()) {
 | 
			
		||||
                    stringstream_ << value;
 | 
			
		||||
                }
 | 
			
		||||
    #endif
 | 
			
		||||
                return *this;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            //
 | 
			
		||||
            static void setLogLevel(LogLevel level) {
 | 
			
		||||
                get_log_level_ref() = level;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            static void setHandler(ILogHandler* handler) {
 | 
			
		||||
                get_handler_ref() = handler;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            static LogLevel get_current_log_level() {
 | 
			
		||||
                return get_log_level_ref();
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
        private:
 | 
			
		||||
            //
 | 
			
		||||
            static LogLevel& get_log_level_ref()
 | 
			
		||||
            {
 | 
			
		||||
                static LogLevel current_level = (LogLevel)CROW_LOG_LEVEL;
 | 
			
		||||
                return current_level;
 | 
			
		||||
            }
 | 
			
		||||
            static ILogHandler*& get_handler_ref()
 | 
			
		||||
            {
 | 
			
		||||
                static CerrLogHandler default_handler;
 | 
			
		||||
                static ILogHandler* current_handler = &default_handler;
 | 
			
		||||
                return current_handler;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            //
 | 
			
		||||
            std::ostringstream stringstream_;
 | 
			
		||||
            LogLevel level_;
 | 
			
		||||
    };
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#define CROW_LOG_CRITICAL   \
 | 
			
		||||
        if (crow::logger::get_current_log_level() <= crow::LogLevel::CRITICAL) \
 | 
			
		||||
            crow::logger("CRITICAL", crow::LogLevel::CRITICAL)
 | 
			
		||||
#define CROW_LOG_ERROR      \
 | 
			
		||||
        if (crow::logger::get_current_log_level() <= crow::LogLevel::ERROR) \
 | 
			
		||||
            crow::logger("ERROR   ", crow::LogLevel::ERROR)
 | 
			
		||||
#define CROW_LOG_WARNING    \
 | 
			
		||||
        if (crow::logger::get_current_log_level() <= crow::LogLevel::WARNING) \
 | 
			
		||||
            crow::logger("WARNING ", crow::LogLevel::WARNING)
 | 
			
		||||
#define CROW_LOG_INFO       \
 | 
			
		||||
        if (crow::logger::get_current_log_level() <= crow::LogLevel::INFO) \
 | 
			
		||||
            crow::logger("INFO    ", crow::LogLevel::INFO)
 | 
			
		||||
#define CROW_LOG_DEBUG      \
 | 
			
		||||
        if (crow::logger::get_current_log_level() <= crow::LogLevel::DEBUG) \
 | 
			
		||||
            crow::logger("DEBUG   ", crow::LogLevel::DEBUG)
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										174
									
								
								ext/crow/middleware.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										174
									
								
								ext/crow/middleware.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,174 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
#include <boost/algorithm/string/trim.hpp>
 | 
			
		||||
#include "http_request.h"
 | 
			
		||||
#include "http_response.h"
 | 
			
		||||
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
    // Any middleware requires following 3 members:
 | 
			
		||||
 | 
			
		||||
    // struct context;
 | 
			
		||||
    //      storing data for the middleware; can be read from another middleware or handlers
 | 
			
		||||
 | 
			
		||||
    // before_handle
 | 
			
		||||
    //      called before handling the request.
 | 
			
		||||
    //      if res.end() is called, the operation is halted. 
 | 
			
		||||
    //      (still call after_handle of this middleware)
 | 
			
		||||
    //      2 signatures:
 | 
			
		||||
    //      void before_handle(request& req, response& res, context& ctx)
 | 
			
		||||
    //          if you only need to access this middlewares context.
 | 
			
		||||
    //      template <typename AllContext>
 | 
			
		||||
    //      void before_handle(request& req, response& res, context& ctx, AllContext& all_ctx)
 | 
			
		||||
    //          you can access another middlewares' context by calling `all_ctx.template get<MW>()'
 | 
			
		||||
    //          ctx == all_ctx.template get<CurrentMiddleware>()
 | 
			
		||||
 | 
			
		||||
    // after_handle
 | 
			
		||||
    //      called after handling the request.
 | 
			
		||||
    //      void after_handle(request& req, response& res, context& ctx)
 | 
			
		||||
    //      template <typename AllContext>
 | 
			
		||||
    //      void after_handle(request& req, response& res, context& ctx, AllContext& all_ctx)
 | 
			
		||||
 | 
			
		||||
    struct CookieParser
 | 
			
		||||
    {
 | 
			
		||||
        struct context
 | 
			
		||||
        {
 | 
			
		||||
            std::unordered_map<std::string, std::string> jar;
 | 
			
		||||
            std::unordered_map<std::string, std::string> cookies_to_add;
 | 
			
		||||
 | 
			
		||||
            std::string get_cookie(const std::string& key)
 | 
			
		||||
            {
 | 
			
		||||
                if (jar.count(key))
 | 
			
		||||
                    return jar[key];
 | 
			
		||||
                return {};
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            void set_cookie(const std::string& key, const std::string& value)
 | 
			
		||||
            {
 | 
			
		||||
                cookies_to_add.emplace(key, value);
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        void before_handle(request& req, response& res, context& ctx)
 | 
			
		||||
        {
 | 
			
		||||
            int count = req.headers.count("Cookie");
 | 
			
		||||
            if (!count)
 | 
			
		||||
                return;
 | 
			
		||||
            if (count > 1)
 | 
			
		||||
            {
 | 
			
		||||
                res.code = 400;
 | 
			
		||||
                res.end();
 | 
			
		||||
                return;
 | 
			
		||||
            }
 | 
			
		||||
            std::string cookies = req.get_header_value("Cookie");
 | 
			
		||||
            size_t pos = 0;
 | 
			
		||||
            while(pos < cookies.size())
 | 
			
		||||
            {
 | 
			
		||||
                size_t pos_equal = cookies.find('=', pos);
 | 
			
		||||
                if (pos_equal == cookies.npos)
 | 
			
		||||
                    break;
 | 
			
		||||
                std::string name = cookies.substr(pos, pos_equal-pos);
 | 
			
		||||
                boost::trim(name);
 | 
			
		||||
                pos = pos_equal+1;
 | 
			
		||||
                while(pos < cookies.size() && cookies[pos] == ' ') pos++;
 | 
			
		||||
                if (pos == cookies.size())
 | 
			
		||||
                    break;
 | 
			
		||||
 | 
			
		||||
                std::string value;
 | 
			
		||||
 | 
			
		||||
                if (cookies[pos] == '"')
 | 
			
		||||
                {
 | 
			
		||||
                    int dquote_meet_count = 0;
 | 
			
		||||
                    pos ++;
 | 
			
		||||
                    size_t pos_dquote = pos-1;
 | 
			
		||||
                    do
 | 
			
		||||
                    {
 | 
			
		||||
                        pos_dquote = cookies.find('"', pos_dquote+1);
 | 
			
		||||
                        dquote_meet_count ++;
 | 
			
		||||
                    } while(pos_dquote < cookies.size() && cookies[pos_dquote-1] == '\\');
 | 
			
		||||
                    if (pos_dquote == cookies.npos)
 | 
			
		||||
                        break;
 | 
			
		||||
 | 
			
		||||
                    if (dquote_meet_count == 1)
 | 
			
		||||
                        value = cookies.substr(pos, pos_dquote - pos);
 | 
			
		||||
                    else
 | 
			
		||||
                    {
 | 
			
		||||
                        value.clear();
 | 
			
		||||
                        value.reserve(pos_dquote-pos);
 | 
			
		||||
                        for(size_t p = pos; p < pos_dquote; p++)
 | 
			
		||||
                        {
 | 
			
		||||
                            // FIXME minimal escaping
 | 
			
		||||
                            if (cookies[p] == '\\' && p + 1 < pos_dquote)
 | 
			
		||||
                            {
 | 
			
		||||
                                p++;
 | 
			
		||||
                                if (cookies[p] == '\\' || cookies[p] == '"')
 | 
			
		||||
                                    value += cookies[p];
 | 
			
		||||
                                else
 | 
			
		||||
                                {
 | 
			
		||||
                                    value += '\\';
 | 
			
		||||
                                    value += cookies[p];
 | 
			
		||||
                                }
 | 
			
		||||
                            }
 | 
			
		||||
                            else
 | 
			
		||||
                                value += cookies[p];
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
 | 
			
		||||
                    ctx.jar.emplace(std::move(name), std::move(value));
 | 
			
		||||
                    pos = cookies.find(";", pos_dquote+1);
 | 
			
		||||
                    if (pos == cookies.npos)
 | 
			
		||||
                        break;
 | 
			
		||||
                    pos++;
 | 
			
		||||
                    while(pos < cookies.size() && cookies[pos] == ' ') pos++;
 | 
			
		||||
                    if (pos == cookies.size())
 | 
			
		||||
                        break;
 | 
			
		||||
                }
 | 
			
		||||
                else
 | 
			
		||||
                {
 | 
			
		||||
                    size_t pos_semicolon = cookies.find(';', pos);
 | 
			
		||||
                    value = cookies.substr(pos, pos_semicolon - pos);
 | 
			
		||||
                    boost::trim(value);
 | 
			
		||||
                    ctx.jar.emplace(std::move(name), std::move(value));
 | 
			
		||||
                    pos = pos_semicolon;
 | 
			
		||||
                    if (pos == cookies.npos)
 | 
			
		||||
                        break;
 | 
			
		||||
                    pos ++;
 | 
			
		||||
                    while(pos < cookies.size() && cookies[pos] == ' ') pos++;
 | 
			
		||||
                    if (pos == cookies.size())
 | 
			
		||||
                        break;
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void after_handle(request& req, response& res, context& ctx)
 | 
			
		||||
        {
 | 
			
		||||
            for(auto& cookie:ctx.cookies_to_add)
 | 
			
		||||
            {
 | 
			
		||||
                res.add_header("Set-Cookie", cookie.first + "=" + cookie.second);
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    /*
 | 
			
		||||
    App<CookieParser, AnotherJarMW> app;
 | 
			
		||||
    A B C
 | 
			
		||||
    A::context
 | 
			
		||||
        int aa;
 | 
			
		||||
 | 
			
		||||
    ctx1 : public A::context
 | 
			
		||||
    ctx2 : public ctx1, public B::context
 | 
			
		||||
    ctx3 : public ctx2, public C::context
 | 
			
		||||
 | 
			
		||||
    C depends on A
 | 
			
		||||
 | 
			
		||||
    C::handle
 | 
			
		||||
        context.aaa
 | 
			
		||||
 | 
			
		||||
    App::context : private CookieParser::contetx, ... 
 | 
			
		||||
    {
 | 
			
		||||
        jar
 | 
			
		||||
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    SimpleApp
 | 
			
		||||
    */
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										59
									
								
								ext/crow/middleware_context.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										59
									
								
								ext/crow/middleware_context.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,59 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include "utility.h"
 | 
			
		||||
#include "http_request.h"
 | 
			
		||||
#include "http_response.h"
 | 
			
		||||
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
    namespace detail
 | 
			
		||||
    {
 | 
			
		||||
        template <typename ... Middlewares>
 | 
			
		||||
        struct partial_context
 | 
			
		||||
            : public black_magic::pop_back<Middlewares...>::template rebind<partial_context>
 | 
			
		||||
            , public black_magic::last_element_type<Middlewares...>::type::context
 | 
			
		||||
        {
 | 
			
		||||
            using parent_context = typename black_magic::pop_back<Middlewares...>::template rebind<::crow::detail::partial_context>;
 | 
			
		||||
            template <int N>
 | 
			
		||||
            using partial = typename std::conditional<N == sizeof...(Middlewares)-1, partial_context, typename parent_context::template partial<N>>::type;
 | 
			
		||||
 | 
			
		||||
            template <typename T> 
 | 
			
		||||
            typename T::context& get()
 | 
			
		||||
            {
 | 
			
		||||
                return static_cast<typename T::context&>(*this);
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <>
 | 
			
		||||
        struct partial_context<>
 | 
			
		||||
        {
 | 
			
		||||
            template <int>
 | 
			
		||||
            using partial = partial_context;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <int N, typename Context, typename Container, typename CurrentMW, typename ... Middlewares>
 | 
			
		||||
        bool middleware_call_helper(Container& middlewares, request& req, response& res, Context& ctx);
 | 
			
		||||
 | 
			
		||||
        template <typename ... Middlewares>
 | 
			
		||||
        struct context : private partial_context<Middlewares...>
 | 
			
		||||
        //struct context : private Middlewares::context... // simple but less type-safe
 | 
			
		||||
        {
 | 
			
		||||
            template <int N, typename Context, typename Container>
 | 
			
		||||
            friend typename std::enable_if<(N==0)>::type after_handlers_call_helper(Container& middlewares, Context& ctx, request& req, response& res);
 | 
			
		||||
            template <int N, typename Context, typename Container>
 | 
			
		||||
            friend typename std::enable_if<(N>0)>::type after_handlers_call_helper(Container& middlewares, Context& ctx, request& req, response& res);
 | 
			
		||||
 | 
			
		||||
            template <int N, typename Context, typename Container, typename CurrentMW, typename ... Middlewares2>
 | 
			
		||||
            friend bool middleware_call_helper(Container& middlewares, request& req, response& res, Context& ctx);
 | 
			
		||||
 | 
			
		||||
            template <typename T> 
 | 
			
		||||
            typename T::context& get()
 | 
			
		||||
            {
 | 
			
		||||
                return static_cast<typename T::context&>(*this);
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            template <int N>
 | 
			
		||||
            using partial = typename partial_context<Middlewares...>::template partial<N>;
 | 
			
		||||
        };
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										559
									
								
								ext/crow/mustache.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										559
									
								
								ext/crow/mustache.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,559 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
#include <string>
 | 
			
		||||
#include <vector>
 | 
			
		||||
#include <fstream>
 | 
			
		||||
#include <iterator>
 | 
			
		||||
#include <functional>
 | 
			
		||||
#include "json.h"
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
    namespace mustache
 | 
			
		||||
    {
 | 
			
		||||
        using context = json::wvalue;
 | 
			
		||||
 | 
			
		||||
        template_t load(const std::string& filename);
 | 
			
		||||
 | 
			
		||||
        class invalid_template_exception : public std::exception
 | 
			
		||||
        {
 | 
			
		||||
            public:
 | 
			
		||||
            invalid_template_exception(const std::string& msg)
 | 
			
		||||
                : msg("crow::mustache error: " + msg)
 | 
			
		||||
            {
 | 
			
		||||
            }
 | 
			
		||||
            virtual const char* what() const throw()
 | 
			
		||||
            {
 | 
			
		||||
                return msg.c_str();
 | 
			
		||||
            }
 | 
			
		||||
            std::string msg;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        enum class ActionType
 | 
			
		||||
        {
 | 
			
		||||
            Ignore,
 | 
			
		||||
            Tag,
 | 
			
		||||
            UnescapeTag,
 | 
			
		||||
            OpenBlock,
 | 
			
		||||
            CloseBlock,
 | 
			
		||||
            ElseBlock,
 | 
			
		||||
            Partial,
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        struct Action
 | 
			
		||||
        {
 | 
			
		||||
            int start;
 | 
			
		||||
            int end;
 | 
			
		||||
            int pos;
 | 
			
		||||
            ActionType t;
 | 
			
		||||
            Action(ActionType t, int start, int end, int pos = 0) 
 | 
			
		||||
                : start(start), end(end), pos(pos), t(t)
 | 
			
		||||
            {}
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        class template_t 
 | 
			
		||||
        {
 | 
			
		||||
        public:
 | 
			
		||||
            template_t(std::string body)
 | 
			
		||||
                : body_(std::move(body))
 | 
			
		||||
            {
 | 
			
		||||
                // {{ {{# {{/ {{^ {{! {{> {{=
 | 
			
		||||
                parse();
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
        private:
 | 
			
		||||
            std::string tag_name(const Action& action)
 | 
			
		||||
            {
 | 
			
		||||
                return body_.substr(action.start, action.end - action.start);
 | 
			
		||||
            }
 | 
			
		||||
            auto find_context(const std::string& name, const std::vector<context*>& stack)->std::pair<bool, context&>
 | 
			
		||||
            {
 | 
			
		||||
                if (name == ".")
 | 
			
		||||
                {
 | 
			
		||||
                    return {true, *stack.back()};
 | 
			
		||||
                }
 | 
			
		||||
                int dotPosition = name.find(".");
 | 
			
		||||
                if (dotPosition == (int)name.npos)
 | 
			
		||||
                {
 | 
			
		||||
                    for(auto it = stack.rbegin(); it != stack.rend(); ++it)
 | 
			
		||||
                    {
 | 
			
		||||
                        if ((*it)->t() == json::type::Object)
 | 
			
		||||
                        {
 | 
			
		||||
                            if ((*it)->count(name))
 | 
			
		||||
                                return {true, (**it)[name]};
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
                else
 | 
			
		||||
                {
 | 
			
		||||
                    std::vector<int> dotPositions;
 | 
			
		||||
                    dotPositions.push_back(-1);
 | 
			
		||||
                    while(dotPosition != (int)name.npos)
 | 
			
		||||
                    {
 | 
			
		||||
                        dotPositions.push_back(dotPosition);
 | 
			
		||||
                        dotPosition = name.find(".", dotPosition+1);
 | 
			
		||||
                    }
 | 
			
		||||
                    dotPositions.push_back(name.size());
 | 
			
		||||
                    std::vector<std::string> names;
 | 
			
		||||
                    names.reserve(dotPositions.size()-1);
 | 
			
		||||
                    for(int i = 1; i < (int)dotPositions.size(); i ++)
 | 
			
		||||
                        names.emplace_back(name.substr(dotPositions[i-1]+1, dotPositions[i]-dotPositions[i-1]-1));
 | 
			
		||||
 | 
			
		||||
                    for(auto it = stack.rbegin(); it != stack.rend(); ++it)
 | 
			
		||||
                    {
 | 
			
		||||
                        context* view = *it;
 | 
			
		||||
                        bool found = true;
 | 
			
		||||
                        for(auto jt = names.begin(); jt != names.end(); ++jt)
 | 
			
		||||
                        {
 | 
			
		||||
                            if (view->t() == json::type::Object &&
 | 
			
		||||
                                view->count(*jt))
 | 
			
		||||
                            {
 | 
			
		||||
                                view = &(*view)[*jt];
 | 
			
		||||
                            }
 | 
			
		||||
                            else
 | 
			
		||||
                            {
 | 
			
		||||
                                found = false;
 | 
			
		||||
                                break;
 | 
			
		||||
                            }
 | 
			
		||||
                        }
 | 
			
		||||
                        if (found)
 | 
			
		||||
                            return {true, *view};
 | 
			
		||||
                    }
 | 
			
		||||
 | 
			
		||||
                }
 | 
			
		||||
 | 
			
		||||
                static json::wvalue empty_str;
 | 
			
		||||
                empty_str = "";
 | 
			
		||||
                return {false, empty_str};
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            void escape(const std::string& in, std::string& out)
 | 
			
		||||
            {
 | 
			
		||||
                out.reserve(out.size() + in.size());
 | 
			
		||||
                for(auto it = in.begin(); it != in.end(); ++it)
 | 
			
		||||
                {
 | 
			
		||||
                    switch(*it)
 | 
			
		||||
                    {
 | 
			
		||||
                        case '&': out += "&"; break;
 | 
			
		||||
                        case '<': out += "<"; break;
 | 
			
		||||
                        case '>': out += ">"; break;
 | 
			
		||||
                        case '"': out += """; break;
 | 
			
		||||
                        case '\'': out += "'"; break;
 | 
			
		||||
                        case '/': out += "/"; break;
 | 
			
		||||
                        default: out += *it; break;
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            void render_internal(int actionBegin, int actionEnd, std::vector<context*>& stack, std::string& out, int indent)
 | 
			
		||||
            {
 | 
			
		||||
                int current = actionBegin;
 | 
			
		||||
 | 
			
		||||
                if (indent)
 | 
			
		||||
                    out.insert(out.size(), indent, ' ');
 | 
			
		||||
 | 
			
		||||
                while(current < actionEnd)
 | 
			
		||||
                {
 | 
			
		||||
                    auto& fragment = fragments_[current];
 | 
			
		||||
                    auto& action = actions_[current];
 | 
			
		||||
                    render_fragment(fragment, indent, out);
 | 
			
		||||
                    switch(action.t)
 | 
			
		||||
                    {
 | 
			
		||||
                        case ActionType::Ignore:
 | 
			
		||||
                            // do nothing
 | 
			
		||||
                            break;
 | 
			
		||||
                        case ActionType::Partial:
 | 
			
		||||
                            {
 | 
			
		||||
                                std::string partial_name = tag_name(action);
 | 
			
		||||
                                auto partial_templ = load(partial_name);
 | 
			
		||||
                                int partial_indent = action.pos;
 | 
			
		||||
                                partial_templ.render_internal(0, partial_templ.fragments_.size()-1, stack, out, partial_indent?indent+partial_indent:0);
 | 
			
		||||
                            }
 | 
			
		||||
                            break;
 | 
			
		||||
                        case ActionType::UnescapeTag:
 | 
			
		||||
                        case ActionType::Tag:
 | 
			
		||||
                            {
 | 
			
		||||
                                auto optional_ctx = find_context(tag_name(action), stack);
 | 
			
		||||
                                auto& ctx = optional_ctx.second;
 | 
			
		||||
                                switch(ctx.t())
 | 
			
		||||
                                {
 | 
			
		||||
                                    case json::type::Number:
 | 
			
		||||
                                        out += json::dump(ctx);
 | 
			
		||||
                                        break;
 | 
			
		||||
                                    case json::type::String:
 | 
			
		||||
                                        if (action.t == ActionType::Tag)
 | 
			
		||||
                                            escape(ctx.s, out);
 | 
			
		||||
                                        else
 | 
			
		||||
                                            out += ctx.s;
 | 
			
		||||
                                        break;
 | 
			
		||||
                                    default:
 | 
			
		||||
                                        throw std::runtime_error("not implemented tag type" + boost::lexical_cast<std::string>((int)ctx.t()));
 | 
			
		||||
                                }
 | 
			
		||||
                            }
 | 
			
		||||
                            break;
 | 
			
		||||
                        case ActionType::ElseBlock:
 | 
			
		||||
                            {
 | 
			
		||||
                                static context nullContext;
 | 
			
		||||
                                auto optional_ctx = find_context(tag_name(action), stack);
 | 
			
		||||
                                if (!optional_ctx.first)
 | 
			
		||||
                                {
 | 
			
		||||
                                    stack.emplace_back(&nullContext);
 | 
			
		||||
                                    break;
 | 
			
		||||
                                }
 | 
			
		||||
 | 
			
		||||
                                auto& ctx = optional_ctx.second;
 | 
			
		||||
                                switch(ctx.t())
 | 
			
		||||
                                {
 | 
			
		||||
                                    case json::type::List:
 | 
			
		||||
                                        if (ctx.l && !ctx.l->empty())
 | 
			
		||||
                                            current = action.pos;
 | 
			
		||||
                                        else
 | 
			
		||||
                                            stack.emplace_back(&nullContext);
 | 
			
		||||
                                        break;
 | 
			
		||||
                                    case json::type::False:
 | 
			
		||||
                                    case json::type::Null:
 | 
			
		||||
                                        stack.emplace_back(&nullContext);
 | 
			
		||||
                                        break;
 | 
			
		||||
                                    default:
 | 
			
		||||
                                        current = action.pos;
 | 
			
		||||
                                        break;
 | 
			
		||||
                                }
 | 
			
		||||
                                break;
 | 
			
		||||
                            }
 | 
			
		||||
                        case ActionType::OpenBlock:
 | 
			
		||||
                            {
 | 
			
		||||
                                auto optional_ctx = find_context(tag_name(action), stack);
 | 
			
		||||
                                if (!optional_ctx.first)
 | 
			
		||||
                                {
 | 
			
		||||
                                    current = action.pos;
 | 
			
		||||
                                    break;
 | 
			
		||||
                                }
 | 
			
		||||
 | 
			
		||||
                                auto& ctx = optional_ctx.second;
 | 
			
		||||
                                switch(ctx.t())
 | 
			
		||||
                                {
 | 
			
		||||
                                    case json::type::List:
 | 
			
		||||
                                        if (ctx.l)
 | 
			
		||||
                                            for(auto it = ctx.l->begin(); it != ctx.l->end(); ++it)
 | 
			
		||||
                                            {
 | 
			
		||||
                                                stack.push_back(&*it);
 | 
			
		||||
                                                render_internal(current+1, action.pos, stack, out, indent);
 | 
			
		||||
                                                stack.pop_back();
 | 
			
		||||
                                            }
 | 
			
		||||
                                        current = action.pos;
 | 
			
		||||
                                        break;
 | 
			
		||||
                                    case json::type::Number:
 | 
			
		||||
                                    case json::type::String:
 | 
			
		||||
                                    case json::type::Object:
 | 
			
		||||
                                    case json::type::True:
 | 
			
		||||
                                        stack.push_back(&ctx);
 | 
			
		||||
                                        break;
 | 
			
		||||
                                    case json::type::False:
 | 
			
		||||
                                    case json::type::Null:
 | 
			
		||||
                                        current = action.pos;
 | 
			
		||||
                                        break;
 | 
			
		||||
                                    default:
 | 
			
		||||
                                        throw std::runtime_error("{{#: not implemented context type: " + boost::lexical_cast<std::string>((int)ctx.t()));
 | 
			
		||||
                                        break;
 | 
			
		||||
                                }
 | 
			
		||||
                                break;
 | 
			
		||||
                            }
 | 
			
		||||
                        case ActionType::CloseBlock:
 | 
			
		||||
                            stack.pop_back();
 | 
			
		||||
                            break;
 | 
			
		||||
                        default:
 | 
			
		||||
                            throw std::runtime_error("not implemented " + boost::lexical_cast<std::string>((int)action.t));
 | 
			
		||||
                    }
 | 
			
		||||
                    current++;
 | 
			
		||||
                }
 | 
			
		||||
                auto& fragment = fragments_[actionEnd];
 | 
			
		||||
                render_fragment(fragment, indent, out);
 | 
			
		||||
            }
 | 
			
		||||
            void render_fragment(const std::pair<int, int> fragment, int indent, std::string& out)
 | 
			
		||||
            {
 | 
			
		||||
                if (indent)
 | 
			
		||||
                {
 | 
			
		||||
                    for(int i = fragment.first; i < fragment.second; i ++)
 | 
			
		||||
                    {
 | 
			
		||||
                        out += body_[i];
 | 
			
		||||
                        if (body_[i] == '\n' && i+1 != (int)body_.size())
 | 
			
		||||
                            out.insert(out.size(), indent, ' ');
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
                else
 | 
			
		||||
                    out.insert(out.size(), body_, fragment.first, fragment.second-fragment.first);
 | 
			
		||||
            }
 | 
			
		||||
        public:
 | 
			
		||||
            std::string render()
 | 
			
		||||
            {
 | 
			
		||||
                context empty_ctx;
 | 
			
		||||
                std::vector<context*> stack;
 | 
			
		||||
                stack.emplace_back(&empty_ctx);
 | 
			
		||||
 | 
			
		||||
                std::string ret;
 | 
			
		||||
                render_internal(0, fragments_.size()-1, stack, ret, 0);
 | 
			
		||||
                return ret;
 | 
			
		||||
            }
 | 
			
		||||
            std::string render(context& ctx)
 | 
			
		||||
            {
 | 
			
		||||
                std::vector<context*> stack;
 | 
			
		||||
                stack.emplace_back(&ctx);
 | 
			
		||||
 | 
			
		||||
                std::string ret;
 | 
			
		||||
                render_internal(0, fragments_.size()-1, stack, ret, 0);
 | 
			
		||||
                return ret;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
        private:
 | 
			
		||||
 | 
			
		||||
            void parse()
 | 
			
		||||
            {
 | 
			
		||||
                std::string tag_open = "{{";
 | 
			
		||||
                std::string tag_close = "}}";
 | 
			
		||||
 | 
			
		||||
                std::vector<int> blockPositions;
 | 
			
		||||
                
 | 
			
		||||
                size_t current = 0;
 | 
			
		||||
                while(1)
 | 
			
		||||
                {
 | 
			
		||||
                    size_t idx = body_.find(tag_open, current);
 | 
			
		||||
                    if (idx == body_.npos)
 | 
			
		||||
                    {
 | 
			
		||||
                        fragments_.emplace_back(current, body_.size());
 | 
			
		||||
                        actions_.emplace_back(ActionType::Ignore, 0, 0);
 | 
			
		||||
                        break;
 | 
			
		||||
                    }
 | 
			
		||||
                    fragments_.emplace_back(current, idx);
 | 
			
		||||
 | 
			
		||||
                    idx += tag_open.size();
 | 
			
		||||
                    size_t endIdx = body_.find(tag_close, idx);
 | 
			
		||||
                    if (endIdx == idx)
 | 
			
		||||
                    {
 | 
			
		||||
                        throw invalid_template_exception("empty tag is not allowed");
 | 
			
		||||
                    }
 | 
			
		||||
                    if (endIdx == body_.npos)
 | 
			
		||||
                    {
 | 
			
		||||
                        // error, no matching tag
 | 
			
		||||
                        throw invalid_template_exception("not matched opening tag");
 | 
			
		||||
                    }
 | 
			
		||||
                    current = endIdx + tag_close.size();
 | 
			
		||||
                    switch(body_[idx])
 | 
			
		||||
                    {
 | 
			
		||||
                        case '#':
 | 
			
		||||
                            idx++;
 | 
			
		||||
                            while(body_[idx] == ' ') idx++;
 | 
			
		||||
                            while(body_[endIdx-1] == ' ') endIdx--;
 | 
			
		||||
                            blockPositions.emplace_back(actions_.size());
 | 
			
		||||
                            actions_.emplace_back(ActionType::OpenBlock, idx, endIdx);
 | 
			
		||||
                            break;
 | 
			
		||||
                        case '/':
 | 
			
		||||
                            idx++;
 | 
			
		||||
                            while(body_[idx] == ' ') idx++;
 | 
			
		||||
                            while(body_[endIdx-1] == ' ') endIdx--;
 | 
			
		||||
                            {
 | 
			
		||||
                                auto& matched = actions_[blockPositions.back()];
 | 
			
		||||
                                if (body_.compare(idx, endIdx-idx, 
 | 
			
		||||
                                        body_, matched.start, matched.end - matched.start) != 0)
 | 
			
		||||
                                {
 | 
			
		||||
                                    throw invalid_template_exception("not matched {{# {{/ pair: " + 
 | 
			
		||||
                                        body_.substr(matched.start, matched.end - matched.start) + ", " + 
 | 
			
		||||
                                        body_.substr(idx, endIdx-idx));
 | 
			
		||||
                                }
 | 
			
		||||
                                matched.pos = actions_.size();
 | 
			
		||||
                            }
 | 
			
		||||
                            actions_.emplace_back(ActionType::CloseBlock, idx, endIdx, blockPositions.back());
 | 
			
		||||
                            blockPositions.pop_back();
 | 
			
		||||
                            break;
 | 
			
		||||
                        case '^':
 | 
			
		||||
                            idx++;
 | 
			
		||||
                            while(body_[idx] == ' ') idx++;
 | 
			
		||||
                            while(body_[endIdx-1] == ' ') endIdx--;
 | 
			
		||||
                            blockPositions.emplace_back(actions_.size());
 | 
			
		||||
                            actions_.emplace_back(ActionType::ElseBlock, idx, endIdx);
 | 
			
		||||
                            break;
 | 
			
		||||
                        case '!':
 | 
			
		||||
                            // do nothing action
 | 
			
		||||
                            actions_.emplace_back(ActionType::Ignore, idx+1, endIdx);
 | 
			
		||||
                            break;
 | 
			
		||||
                        case '>': // partial
 | 
			
		||||
                            idx++;
 | 
			
		||||
                            while(body_[idx] == ' ') idx++;
 | 
			
		||||
                            while(body_[endIdx-1] == ' ') endIdx--;
 | 
			
		||||
                            actions_.emplace_back(ActionType::Partial, idx, endIdx);
 | 
			
		||||
                            break;
 | 
			
		||||
                        case '{':
 | 
			
		||||
                            if (tag_open != "{{" || tag_close != "}}")
 | 
			
		||||
                                throw invalid_template_exception("cannot use triple mustache when delimiter changed");
 | 
			
		||||
 | 
			
		||||
                            idx ++;
 | 
			
		||||
                            if (body_[endIdx+2] != '}')
 | 
			
		||||
                            {
 | 
			
		||||
                                throw invalid_template_exception("{{{: }}} not matched");
 | 
			
		||||
                            }
 | 
			
		||||
                            while(body_[idx] == ' ') idx++;
 | 
			
		||||
                            while(body_[endIdx-1] == ' ') endIdx--;
 | 
			
		||||
                            actions_.emplace_back(ActionType::UnescapeTag, idx, endIdx);
 | 
			
		||||
                            current++;
 | 
			
		||||
                            break;
 | 
			
		||||
                        case '&':
 | 
			
		||||
                            idx ++;
 | 
			
		||||
                            while(body_[idx] == ' ') idx++;
 | 
			
		||||
                            while(body_[endIdx-1] == ' ') endIdx--;
 | 
			
		||||
                            actions_.emplace_back(ActionType::UnescapeTag, idx, endIdx);
 | 
			
		||||
                            break;
 | 
			
		||||
                        case '=':
 | 
			
		||||
                            // tag itself is no-op
 | 
			
		||||
                            idx ++;
 | 
			
		||||
                            actions_.emplace_back(ActionType::Ignore, idx, endIdx);
 | 
			
		||||
                            endIdx --;
 | 
			
		||||
                            if (body_[endIdx] != '=')
 | 
			
		||||
                                throw invalid_template_exception("{{=: not matching = tag: "+body_.substr(idx, endIdx-idx));
 | 
			
		||||
                            endIdx --;
 | 
			
		||||
                            while(body_[idx] == ' ') idx++;
 | 
			
		||||
                            while(body_[endIdx] == ' ') endIdx--;
 | 
			
		||||
                            endIdx++;
 | 
			
		||||
                            {
 | 
			
		||||
                                bool succeeded = false;
 | 
			
		||||
                                for(size_t i = idx; i < endIdx; i++)
 | 
			
		||||
                                {
 | 
			
		||||
                                    if (body_[i] == ' ')
 | 
			
		||||
                                    {
 | 
			
		||||
                                        tag_open = body_.substr(idx, i-idx);
 | 
			
		||||
                                        while(body_[i] == ' ') i++;
 | 
			
		||||
                                        tag_close = body_.substr(i, endIdx-i);
 | 
			
		||||
                                        if (tag_open.empty())
 | 
			
		||||
                                            throw invalid_template_exception("{{=: empty open tag");
 | 
			
		||||
                                        if (tag_close.empty())
 | 
			
		||||
                                            throw invalid_template_exception("{{=: empty close tag");
 | 
			
		||||
 | 
			
		||||
                                        if (tag_close.find(" ") != tag_close.npos)
 | 
			
		||||
                                            throw invalid_template_exception("{{=: invalid open/close tag: "+tag_open+" " + tag_close);
 | 
			
		||||
                                        succeeded = true;
 | 
			
		||||
                                        break;
 | 
			
		||||
                                    }
 | 
			
		||||
                                }
 | 
			
		||||
                                if (!succeeded)
 | 
			
		||||
                                    throw invalid_template_exception("{{=: cannot find space between new open/close tags");
 | 
			
		||||
                            }
 | 
			
		||||
                            break;
 | 
			
		||||
                        default:
 | 
			
		||||
                            // normal tag case;
 | 
			
		||||
                            while(body_[idx] == ' ') idx++;
 | 
			
		||||
                            while(body_[endIdx-1] == ' ') endIdx--;
 | 
			
		||||
                            actions_.emplace_back(ActionType::Tag, idx, endIdx);
 | 
			
		||||
                            break;
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
 | 
			
		||||
                // removing standalones
 | 
			
		||||
                for(int i = actions_.size()-2; i >= 0; i --)
 | 
			
		||||
                {
 | 
			
		||||
                    if (actions_[i].t == ActionType::Tag || actions_[i].t == ActionType::UnescapeTag)
 | 
			
		||||
                        continue;
 | 
			
		||||
                    auto& fragment_before = fragments_[i];
 | 
			
		||||
                    auto& fragment_after = fragments_[i+1];
 | 
			
		||||
                    bool is_last_action = i == (int)actions_.size()-2;
 | 
			
		||||
                    bool all_space_before = true;
 | 
			
		||||
                    int j, k;
 | 
			
		||||
                    for(j = fragment_before.second-1;j >= fragment_before.first;j--)
 | 
			
		||||
                    {
 | 
			
		||||
                        if (body_[j] != ' ')
 | 
			
		||||
                        {
 | 
			
		||||
                            all_space_before = false;
 | 
			
		||||
                            break;
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                    if (all_space_before && i > 0)
 | 
			
		||||
                        continue;
 | 
			
		||||
                    if (!all_space_before && body_[j] != '\n')
 | 
			
		||||
                        continue;
 | 
			
		||||
                    bool all_space_after = true;
 | 
			
		||||
                    for(k = fragment_after.first; k < (int)body_.size() && k < fragment_after.second; k ++)
 | 
			
		||||
                    {
 | 
			
		||||
                        if (body_[k] != ' ')
 | 
			
		||||
                        {
 | 
			
		||||
                            all_space_after = false;
 | 
			
		||||
                            break;
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                    if (all_space_after && !is_last_action)
 | 
			
		||||
                        continue;
 | 
			
		||||
                    if (!all_space_after && 
 | 
			
		||||
                            !(
 | 
			
		||||
                                body_[k] == '\n' 
 | 
			
		||||
                            || 
 | 
			
		||||
                                (body_[k] == '\r' && 
 | 
			
		||||
                                k + 1 < (int)body_.size() && 
 | 
			
		||||
                                body_[k+1] == '\n')))
 | 
			
		||||
                        continue;
 | 
			
		||||
                    if (actions_[i].t == ActionType::Partial)
 | 
			
		||||
                    {
 | 
			
		||||
                        actions_[i].pos = fragment_before.second - j - 1;
 | 
			
		||||
                    }
 | 
			
		||||
                    fragment_before.second = j+1;
 | 
			
		||||
                    if (!all_space_after)
 | 
			
		||||
                    {
 | 
			
		||||
                        if (body_[k] == '\n')
 | 
			
		||||
                            k++;
 | 
			
		||||
                        else 
 | 
			
		||||
                            k += 2;
 | 
			
		||||
                        fragment_after.first = k;
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            
 | 
			
		||||
            std::vector<std::pair<int,int>> fragments_;
 | 
			
		||||
            std::vector<Action> actions_;
 | 
			
		||||
            std::string body_;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        inline template_t compile(const std::string& body)
 | 
			
		||||
        {
 | 
			
		||||
            return template_t(body);
 | 
			
		||||
        }
 | 
			
		||||
        namespace detail
 | 
			
		||||
        {
 | 
			
		||||
            inline std::string& get_template_base_directory_ref()
 | 
			
		||||
            {
 | 
			
		||||
                static std::string template_base_directory = "templates";
 | 
			
		||||
                return template_base_directory;
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        inline std::string default_loader(const std::string& filename)
 | 
			
		||||
        {
 | 
			
		||||
            std::ifstream inf(detail::get_template_base_directory_ref() + filename);
 | 
			
		||||
            if (!inf)
 | 
			
		||||
                return {};
 | 
			
		||||
            return {std::istreambuf_iterator<char>(inf), std::istreambuf_iterator<char>()};
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        namespace detail
 | 
			
		||||
        {
 | 
			
		||||
            inline std::function<std::string (std::string)>& get_loader_ref()
 | 
			
		||||
            {
 | 
			
		||||
                static std::function<std::string (std::string)> loader = default_loader;
 | 
			
		||||
                return loader;
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        inline void set_base(const std::string& path)
 | 
			
		||||
        {
 | 
			
		||||
            auto& base = detail::get_template_base_directory_ref();
 | 
			
		||||
            base = path;
 | 
			
		||||
            if (base.back() != '\\' && 
 | 
			
		||||
                base.back() != '/')
 | 
			
		||||
            {
 | 
			
		||||
                base += '/';
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        inline void set_loader(std::function<std::string(std::string)> loader)
 | 
			
		||||
        {
 | 
			
		||||
            detail::get_loader_ref() = std::move(loader);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        inline template_t load(const std::string& filename)
 | 
			
		||||
        {
 | 
			
		||||
            return compile(detail::get_loader_ref()(filename));
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										163
									
								
								ext/crow/parser.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										163
									
								
								ext/crow/parser.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,163 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <string>
 | 
			
		||||
#include <unordered_map>
 | 
			
		||||
#include <boost/algorithm/string.hpp>
 | 
			
		||||
#include <boost/tokenizer.hpp>
 | 
			
		||||
#include <algorithm>
 | 
			
		||||
 | 
			
		||||
#include "http_parser_merged.h"
 | 
			
		||||
#include "http_request.h"
 | 
			
		||||
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
    template <typename Handler>
 | 
			
		||||
    struct HTTPParser : public http_parser
 | 
			
		||||
    {
 | 
			
		||||
        static int on_message_begin(http_parser* self_)
 | 
			
		||||
        {
 | 
			
		||||
            HTTPParser* self = static_cast<HTTPParser*>(self_);
 | 
			
		||||
            self->clear();
 | 
			
		||||
            return 0;
 | 
			
		||||
        }
 | 
			
		||||
        static int on_url(http_parser* self_, const char* at, size_t length)
 | 
			
		||||
        {
 | 
			
		||||
            HTTPParser* self = static_cast<HTTPParser*>(self_);
 | 
			
		||||
            self->raw_url.insert(self->raw_url.end(), at, at+length);
 | 
			
		||||
            return 0;
 | 
			
		||||
        }
 | 
			
		||||
        static int on_header_field(http_parser* self_, const char* at, size_t length)
 | 
			
		||||
        {
 | 
			
		||||
            HTTPParser* self = static_cast<HTTPParser*>(self_);
 | 
			
		||||
            switch (self->header_building_state)
 | 
			
		||||
            {
 | 
			
		||||
                case 0:
 | 
			
		||||
                    if (!self->header_value.empty())
 | 
			
		||||
                    {
 | 
			
		||||
                        self->headers.emplace(std::move(self->header_field), std::move(self->header_value));
 | 
			
		||||
                    }
 | 
			
		||||
                    self->header_field.assign(at, at+length);
 | 
			
		||||
                    self->header_building_state = 1;
 | 
			
		||||
                    break;
 | 
			
		||||
                case 1:
 | 
			
		||||
                    self->header_field.insert(self->header_field.end(), at, at+length);
 | 
			
		||||
                    break;
 | 
			
		||||
            }
 | 
			
		||||
            return 0;
 | 
			
		||||
        }
 | 
			
		||||
        static int on_header_value(http_parser* self_, const char* at, size_t length)
 | 
			
		||||
        {
 | 
			
		||||
            HTTPParser* self = static_cast<HTTPParser*>(self_);
 | 
			
		||||
            switch (self->header_building_state)
 | 
			
		||||
            {
 | 
			
		||||
                case 0:
 | 
			
		||||
                    self->header_value.insert(self->header_value.end(), at, at+length);
 | 
			
		||||
                    break;
 | 
			
		||||
                case 1:
 | 
			
		||||
                    self->header_building_state = 0;
 | 
			
		||||
                    self->header_value.assign(at, at+length);
 | 
			
		||||
                    break;
 | 
			
		||||
            }
 | 
			
		||||
            return 0;
 | 
			
		||||
        }
 | 
			
		||||
        static int on_headers_complete(http_parser* self_)
 | 
			
		||||
        {
 | 
			
		||||
            HTTPParser* self = static_cast<HTTPParser*>(self_);
 | 
			
		||||
            if (!self->header_field.empty())
 | 
			
		||||
            {
 | 
			
		||||
                self->headers.emplace(std::move(self->header_field), std::move(self->header_value));
 | 
			
		||||
            }
 | 
			
		||||
            self->process_header();
 | 
			
		||||
            return 0;
 | 
			
		||||
        }
 | 
			
		||||
        static int on_body(http_parser* self_, const char* at, size_t length)
 | 
			
		||||
        {
 | 
			
		||||
            HTTPParser* self = static_cast<HTTPParser*>(self_);
 | 
			
		||||
            self->body.insert(self->body.end(), at, at+length);
 | 
			
		||||
            return 0;
 | 
			
		||||
        }
 | 
			
		||||
        static int on_message_complete(http_parser* self_)
 | 
			
		||||
        {
 | 
			
		||||
            HTTPParser* self = static_cast<HTTPParser*>(self_);
 | 
			
		||||
 | 
			
		||||
            // url params
 | 
			
		||||
            self->url = self->raw_url.substr(0, self->raw_url.find("?"));
 | 
			
		||||
            self->url_params = query_string(self->raw_url);
 | 
			
		||||
 | 
			
		||||
            self->process_message();
 | 
			
		||||
            return 0;
 | 
			
		||||
        }
 | 
			
		||||
        HTTPParser(Handler* handler) :
 | 
			
		||||
            handler_(handler)
 | 
			
		||||
        {
 | 
			
		||||
            http_parser_init(this, HTTP_REQUEST);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        // return false on error
 | 
			
		||||
        bool feed(const char* buffer, int length)
 | 
			
		||||
        {
 | 
			
		||||
            const static http_parser_settings settings_{
 | 
			
		||||
                on_message_begin,
 | 
			
		||||
                on_url,
 | 
			
		||||
                nullptr,
 | 
			
		||||
                on_header_field,
 | 
			
		||||
                on_header_value,
 | 
			
		||||
                on_headers_complete,
 | 
			
		||||
                on_body,
 | 
			
		||||
                on_message_complete,
 | 
			
		||||
            };
 | 
			
		||||
 | 
			
		||||
            int nparsed = http_parser_execute(this, &settings_, buffer, length);
 | 
			
		||||
            return nparsed == length;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        bool done()
 | 
			
		||||
        {
 | 
			
		||||
            return feed(nullptr, 0);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void clear()
 | 
			
		||||
        {
 | 
			
		||||
            url.clear();
 | 
			
		||||
            raw_url.clear();
 | 
			
		||||
            header_building_state = 0;
 | 
			
		||||
            header_field.clear();
 | 
			
		||||
            header_value.clear();
 | 
			
		||||
            headers.clear();
 | 
			
		||||
            url_params.clear();
 | 
			
		||||
            body.clear();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void process_header()
 | 
			
		||||
        {
 | 
			
		||||
            handler_->handle_header();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void process_message()
 | 
			
		||||
        {
 | 
			
		||||
            handler_->handle();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        request to_request() const
 | 
			
		||||
        {
 | 
			
		||||
            return request{(HTTPMethod)method, std::move(raw_url), std::move(url), std::move(url_params), std::move(headers), std::move(body)};
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        bool check_version(int major, int minor) const
 | 
			
		||||
        {
 | 
			
		||||
            return http_major == major && http_minor == minor;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        std::string raw_url;
 | 
			
		||||
        std::string url;
 | 
			
		||||
 | 
			
		||||
        int header_building_state = 0;
 | 
			
		||||
        std::string header_field;
 | 
			
		||||
        std::string header_value;
 | 
			
		||||
        ci_map headers;
 | 
			
		||||
        query_string url_params;
 | 
			
		||||
        std::string body;
 | 
			
		||||
 | 
			
		||||
        Handler* handler_;
 | 
			
		||||
    };
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										342
									
								
								ext/crow/query_string.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										342
									
								
								ext/crow/query_string.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,342 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <stdio.h>
 | 
			
		||||
#include <string.h>
 | 
			
		||||
#include <string>
 | 
			
		||||
#include <vector>
 | 
			
		||||
#include <iostream>
 | 
			
		||||
 | 
			
		||||
// ----------------------------------------------------------------------------
 | 
			
		||||
// qs_parse (modified)
 | 
			
		||||
// https://github.com/bartgrantham/qs_parse
 | 
			
		||||
// ----------------------------------------------------------------------------
 | 
			
		||||
/*  Similar to strncmp, but handles URL-encoding for either string  */
 | 
			
		||||
int qs_strncmp(const char * s, const char * qs, size_t n);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
/*  Finds the beginning of each key/value pair and stores a pointer in qs_kv.
 | 
			
		||||
 *  Also decodes the value portion of the k/v pair *in-place*.  In a future
 | 
			
		||||
 *  enhancement it will also have a compile-time option of sorting qs_kv
 | 
			
		||||
 *  alphabetically by key.  */
 | 
			
		||||
int qs_parse(char * qs, char * qs_kv[], int qs_kv_size);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
/*  Used by qs_parse to decode the value portion of a k/v pair  */
 | 
			
		||||
int qs_decode(char * qs);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
/*  Looks up the value according to the key on a pre-processed query string
 | 
			
		||||
 *  A future enhancement will be a compile-time option to look up the key
 | 
			
		||||
 *  in a pre-sorted qs_kv array via a binary search.  */
 | 
			
		||||
//char * qs_k2v(const char * key, char * qs_kv[], int qs_kv_size);
 | 
			
		||||
 char * qs_k2v(const char * key, char * const * qs_kv, int qs_kv_size, int nth);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
/*  Non-destructive lookup of value, based on key.  User provides the
 | 
			
		||||
 *  destinaton string and length.  */
 | 
			
		||||
char * qs_scanvalue(const char * key, const char * qs, char * val, size_t val_len);
 | 
			
		||||
 | 
			
		||||
// TODO: implement sorting of the qs_kv array; for now ensure it's not compiled
 | 
			
		||||
#undef _qsSORTING
 | 
			
		||||
 | 
			
		||||
// isxdigit _is_ available in <ctype.h>, but let's avoid another header instead
 | 
			
		||||
#define CROW_QS_ISHEX(x)    ((((x)>='0'&&(x)<='9') || ((x)>='A'&&(x)<='F') || ((x)>='a'&&(x)<='f')) ? 1 : 0)
 | 
			
		||||
#define CROW_QS_HEX2DEC(x)  (((x)>='0'&&(x)<='9') ? (x)-48 : ((x)>='A'&&(x)<='F') ? (x)-55 : ((x)>='a'&&(x)<='f') ? (x)-87 : 0)
 | 
			
		||||
#define CROW_QS_ISQSCHR(x) ((((x)=='=')||((x)=='#')||((x)=='&')||((x)=='\0')) ? 0 : 1)
 | 
			
		||||
 | 
			
		||||
inline int qs_strncmp(const char * s, const char * qs, size_t n)
 | 
			
		||||
{
 | 
			
		||||
    int i=0;
 | 
			
		||||
    unsigned char u1, u2, unyb, lnyb;
 | 
			
		||||
 | 
			
		||||
    while(n-- > 0)
 | 
			
		||||
    {
 | 
			
		||||
        u1 = (unsigned char) *s++;
 | 
			
		||||
        u2 = (unsigned char) *qs++;
 | 
			
		||||
 | 
			
		||||
        if ( ! CROW_QS_ISQSCHR(u1) ) {  u1 = '\0';  }
 | 
			
		||||
        if ( ! CROW_QS_ISQSCHR(u2) ) {  u2 = '\0';  }
 | 
			
		||||
 | 
			
		||||
        if ( u1 == '+' ) {  u1 = ' ';  }
 | 
			
		||||
        if ( u1 == '%' ) // easier/safer than scanf
 | 
			
		||||
        {
 | 
			
		||||
            unyb = (unsigned char) *s++;
 | 
			
		||||
            lnyb = (unsigned char) *s++;
 | 
			
		||||
            if ( CROW_QS_ISHEX(unyb) && CROW_QS_ISHEX(lnyb) )
 | 
			
		||||
                u1 = (CROW_QS_HEX2DEC(unyb) * 16) + CROW_QS_HEX2DEC(lnyb);
 | 
			
		||||
            else
 | 
			
		||||
                u1 = '\0';
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        if ( u2 == '+' ) {  u2 = ' ';  }
 | 
			
		||||
        if ( u2 == '%' ) // easier/safer than scanf
 | 
			
		||||
        {
 | 
			
		||||
            unyb = (unsigned char) *qs++;
 | 
			
		||||
            lnyb = (unsigned char) *qs++;
 | 
			
		||||
            if ( CROW_QS_ISHEX(unyb) && CROW_QS_ISHEX(lnyb) )
 | 
			
		||||
                u2 = (CROW_QS_HEX2DEC(unyb) * 16) + CROW_QS_HEX2DEC(lnyb);
 | 
			
		||||
            else
 | 
			
		||||
                u2 = '\0';
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        if ( u1 != u2 )
 | 
			
		||||
            return u1 - u2;
 | 
			
		||||
        if ( u1 == '\0' )
 | 
			
		||||
            return 0;
 | 
			
		||||
        i++;
 | 
			
		||||
    }
 | 
			
		||||
    if ( CROW_QS_ISQSCHR(*qs) )
 | 
			
		||||
        return -1;
 | 
			
		||||
    else
 | 
			
		||||
        return 0;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
inline int qs_parse(char * qs, char * qs_kv[], int qs_kv_size)
 | 
			
		||||
{
 | 
			
		||||
    int i, j;
 | 
			
		||||
    char * substr_ptr;
 | 
			
		||||
 | 
			
		||||
    for(i=0; i<qs_kv_size; i++)  qs_kv[i] = NULL;
 | 
			
		||||
 | 
			
		||||
    // find the beginning of the k/v substrings or the fragment
 | 
			
		||||
    substr_ptr = qs + strcspn(qs, "?#");
 | 
			
		||||
    if (substr_ptr[0] != '\0')
 | 
			
		||||
        substr_ptr++;
 | 
			
		||||
    else
 | 
			
		||||
        return 0; // no query or fragment
 | 
			
		||||
 | 
			
		||||
    i=0;
 | 
			
		||||
    while(i<qs_kv_size)
 | 
			
		||||
    {
 | 
			
		||||
        qs_kv[i] = substr_ptr;
 | 
			
		||||
        j = strcspn(substr_ptr, "&");
 | 
			
		||||
        if ( substr_ptr[j] == '\0' ) {  break;  }
 | 
			
		||||
        substr_ptr += j + 1;
 | 
			
		||||
        i++;
 | 
			
		||||
    }
 | 
			
		||||
    i++;  // x &'s -> means x iterations of this loop -> means *x+1* k/v pairs
 | 
			
		||||
 | 
			
		||||
    // we only decode the values in place, the keys could have '='s in them
 | 
			
		||||
    // which will hose our ability to distinguish keys from values later
 | 
			
		||||
    for(j=0; j<i; j++)
 | 
			
		||||
    {
 | 
			
		||||
        substr_ptr = qs_kv[j] + strcspn(qs_kv[j], "=&#");
 | 
			
		||||
        if ( substr_ptr[0] == '&' || substr_ptr[0] == '\0')  // blank value: skip decoding
 | 
			
		||||
            substr_ptr[0] = '\0';
 | 
			
		||||
        else
 | 
			
		||||
            qs_decode(++substr_ptr);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
#ifdef _qsSORTING
 | 
			
		||||
// TODO: qsort qs_kv, using qs_strncmp() for the comparison
 | 
			
		||||
#endif
 | 
			
		||||
 | 
			
		||||
    return i;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
inline int qs_decode(char * qs)
 | 
			
		||||
{
 | 
			
		||||
    int i=0, j=0;
 | 
			
		||||
 | 
			
		||||
    while( CROW_QS_ISQSCHR(qs[j]) )
 | 
			
		||||
    {
 | 
			
		||||
        if ( qs[j] == '+' ) {  qs[i] = ' ';  }
 | 
			
		||||
        else if ( qs[j] == '%' ) // easier/safer than scanf
 | 
			
		||||
        {
 | 
			
		||||
            if ( ! CROW_QS_ISHEX(qs[j+1]) || ! CROW_QS_ISHEX(qs[j+2]) )
 | 
			
		||||
            {
 | 
			
		||||
                qs[i] = '\0';
 | 
			
		||||
                return i;
 | 
			
		||||
            }
 | 
			
		||||
            qs[i] = (CROW_QS_HEX2DEC(qs[j+1]) * 16) + CROW_QS_HEX2DEC(qs[j+2]);
 | 
			
		||||
            j+=2;
 | 
			
		||||
        }
 | 
			
		||||
        else
 | 
			
		||||
        {
 | 
			
		||||
            qs[i] = qs[j];
 | 
			
		||||
        }
 | 
			
		||||
        i++;  j++;
 | 
			
		||||
    }
 | 
			
		||||
    qs[i] = '\0';
 | 
			
		||||
 | 
			
		||||
    return i;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
inline char * qs_k2v(const char * key, char * const * qs_kv, int qs_kv_size, int nth = 0)
 | 
			
		||||
{
 | 
			
		||||
    int i;
 | 
			
		||||
    size_t key_len, skip;
 | 
			
		||||
 | 
			
		||||
    key_len = strlen(key);
 | 
			
		||||
 | 
			
		||||
#ifdef _qsSORTING
 | 
			
		||||
// TODO: binary search for key in the sorted qs_kv
 | 
			
		||||
#else  // _qsSORTING
 | 
			
		||||
    for(i=0; i<qs_kv_size; i++)
 | 
			
		||||
    {
 | 
			
		||||
        // we rely on the unambiguous '=' to find the value in our k/v pair
 | 
			
		||||
        if ( qs_strncmp(key, qs_kv[i], key_len) == 0 )
 | 
			
		||||
        {
 | 
			
		||||
            skip = strcspn(qs_kv[i], "=");
 | 
			
		||||
            if ( qs_kv[i][skip] == '=' )
 | 
			
		||||
                skip++;
 | 
			
		||||
            // return (zero-char value) ? ptr to trailing '\0' : ptr to value
 | 
			
		||||
            if(nth == 0)
 | 
			
		||||
                return qs_kv[i] + skip;
 | 
			
		||||
            else 
 | 
			
		||||
                --nth;
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
#endif  // _qsSORTING
 | 
			
		||||
 | 
			
		||||
    return NULL;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
inline char * qs_scanvalue(const char * key, const char * qs, char * val, size_t val_len)
 | 
			
		||||
{
 | 
			
		||||
    size_t i, key_len;
 | 
			
		||||
    const char * tmp;
 | 
			
		||||
 | 
			
		||||
    // find the beginning of the k/v substrings
 | 
			
		||||
    if ( (tmp = strchr(qs, '?')) != NULL )
 | 
			
		||||
        qs = tmp + 1;
 | 
			
		||||
 | 
			
		||||
    key_len = strlen(key);
 | 
			
		||||
    while(qs[0] != '#' && qs[0] != '\0')
 | 
			
		||||
    {
 | 
			
		||||
        if ( qs_strncmp(key, qs, key_len) == 0 )
 | 
			
		||||
            break;
 | 
			
		||||
        qs += strcspn(qs, "&") + 1;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    if ( qs[0] == '\0' ) return NULL;
 | 
			
		||||
 | 
			
		||||
    qs += strcspn(qs, "=&#");
 | 
			
		||||
    if ( qs[0] == '=' )
 | 
			
		||||
    {
 | 
			
		||||
        qs++;
 | 
			
		||||
        i = strcspn(qs, "&=#");
 | 
			
		||||
        strncpy(val, qs, (val_len-1)<(i+1) ? (val_len-1) : (i+1));
 | 
			
		||||
        qs_decode(val);
 | 
			
		||||
    }
 | 
			
		||||
    else
 | 
			
		||||
    {
 | 
			
		||||
        if ( val_len > 0 )
 | 
			
		||||
            val[0] = '\0';
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    return val;
 | 
			
		||||
}
 | 
			
		||||
// ----------------------------------------------------------------------------
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
namespace crow 
 | 
			
		||||
{
 | 
			
		||||
    class query_string
 | 
			
		||||
    {
 | 
			
		||||
    public:
 | 
			
		||||
        static const int MAX_KEY_VALUE_PAIRS_COUNT = 256;
 | 
			
		||||
 | 
			
		||||
        query_string()
 | 
			
		||||
        {
 | 
			
		||||
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        query_string(const query_string& qs)
 | 
			
		||||
            : url_(qs.url_)
 | 
			
		||||
        {
 | 
			
		||||
            for(auto p:qs.key_value_pairs_)
 | 
			
		||||
            {
 | 
			
		||||
                key_value_pairs_.push_back((char*)(p-qs.url_.c_str()+url_.c_str()));
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        query_string& operator = (const query_string& qs)
 | 
			
		||||
        {
 | 
			
		||||
            url_ = qs.url_;
 | 
			
		||||
            key_value_pairs_.clear();
 | 
			
		||||
            for(auto p:qs.key_value_pairs_)
 | 
			
		||||
            {
 | 
			
		||||
                key_value_pairs_.push_back((char*)(p-qs.url_.c_str()+url_.c_str()));
 | 
			
		||||
            }
 | 
			
		||||
            return *this;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        query_string& operator = (query_string&& qs)
 | 
			
		||||
        {
 | 
			
		||||
            key_value_pairs_ = std::move(qs.key_value_pairs_);
 | 
			
		||||
            char* old_data = (char*)qs.url_.c_str();
 | 
			
		||||
            url_ = std::move(qs.url_);
 | 
			
		||||
            for(auto& p:key_value_pairs_)
 | 
			
		||||
            {
 | 
			
		||||
                p += (char*)url_.c_str() - old_data;
 | 
			
		||||
            }
 | 
			
		||||
            return *this;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        query_string(std::string url)
 | 
			
		||||
            : url_(std::move(url))
 | 
			
		||||
        {
 | 
			
		||||
            if (url_.empty())
 | 
			
		||||
                return;
 | 
			
		||||
 | 
			
		||||
            key_value_pairs_.resize(MAX_KEY_VALUE_PAIRS_COUNT);
 | 
			
		||||
 | 
			
		||||
            int count = qs_parse(&url_[0], &key_value_pairs_[0], MAX_KEY_VALUE_PAIRS_COUNT);
 | 
			
		||||
            key_value_pairs_.resize(count);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void clear() 
 | 
			
		||||
        {
 | 
			
		||||
            key_value_pairs_.clear();
 | 
			
		||||
            url_.clear();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        friend std::ostream& operator<<(std::ostream& os, const query_string& qs)
 | 
			
		||||
        {
 | 
			
		||||
            os << "[ ";
 | 
			
		||||
            for(size_t i = 0; i < qs.key_value_pairs_.size(); ++i) {
 | 
			
		||||
                if (i)
 | 
			
		||||
                    os << ", ";
 | 
			
		||||
                os << qs.key_value_pairs_[i];
 | 
			
		||||
            }
 | 
			
		||||
            os << " ]";
 | 
			
		||||
            return os;
 | 
			
		||||
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        char* get (const std::string& name) const
 | 
			
		||||
        {
 | 
			
		||||
            char* ret = qs_k2v(name.c_str(), key_value_pairs_.data(), key_value_pairs_.size());
 | 
			
		||||
            return ret;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        std::vector<char*> get_list (const std::string& name) const
 | 
			
		||||
        {
 | 
			
		||||
            std::vector<char*> ret;
 | 
			
		||||
            std::string plus = name + "[]";            
 | 
			
		||||
            char* element = nullptr;
 | 
			
		||||
 | 
			
		||||
            int count = 0;
 | 
			
		||||
            while(1)
 | 
			
		||||
            {
 | 
			
		||||
                element = qs_k2v(plus.c_str(), key_value_pairs_.data(), key_value_pairs_.size(), count++);
 | 
			
		||||
                if (!element)
 | 
			
		||||
                    break;
 | 
			
		||||
                ret.push_back(element);
 | 
			
		||||
            }
 | 
			
		||||
            return ret;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    private:
 | 
			
		||||
        std::string url_;
 | 
			
		||||
        std::vector<char*> key_value_pairs_;
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
} // end namespace
 | 
			
		||||
							
								
								
									
										892
									
								
								ext/crow/routing.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										892
									
								
								ext/crow/routing.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,892 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <cstdint>
 | 
			
		||||
#include <utility>
 | 
			
		||||
#include <tuple>
 | 
			
		||||
#include <unordered_map>
 | 
			
		||||
#include <memory>
 | 
			
		||||
#include <boost/lexical_cast.hpp>
 | 
			
		||||
#include <vector>
 | 
			
		||||
 | 
			
		||||
#include "common.h"
 | 
			
		||||
#include "http_response.h"
 | 
			
		||||
#include "http_request.h"
 | 
			
		||||
#include "utility.h"
 | 
			
		||||
#include "logging.h"
 | 
			
		||||
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
    class BaseRule
 | 
			
		||||
    {
 | 
			
		||||
    public:
 | 
			
		||||
        BaseRule(std::string rule)
 | 
			
		||||
            : rule_(std::move(rule))
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        virtual ~BaseRule()
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
        
 | 
			
		||||
        virtual void validate() = 0;
 | 
			
		||||
 | 
			
		||||
        virtual void handle(const request&, response&, const routing_params&) = 0;
 | 
			
		||||
 | 
			
		||||
        uint32_t get_methods()
 | 
			
		||||
        {
 | 
			
		||||
            return methods_;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
    protected:
 | 
			
		||||
        uint32_t methods_{1<<(int)HTTPMethod::GET};
 | 
			
		||||
 | 
			
		||||
        std::string rule_;
 | 
			
		||||
        std::string name_;
 | 
			
		||||
        friend class Router;
 | 
			
		||||
        template <typename T>
 | 
			
		||||
        friend struct RuleParameterTraits;
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    namespace detail
 | 
			
		||||
    {
 | 
			
		||||
        namespace routing_handler_call_helper
 | 
			
		||||
        {
 | 
			
		||||
            template <typename T, int Pos>
 | 
			
		||||
            struct call_pair
 | 
			
		||||
            {
 | 
			
		||||
                using type = T;
 | 
			
		||||
                static const int pos = Pos;
 | 
			
		||||
            };
 | 
			
		||||
 | 
			
		||||
            template <typename H1>
 | 
			
		||||
            struct call_params
 | 
			
		||||
            {
 | 
			
		||||
                H1& handler;
 | 
			
		||||
                const routing_params& params;
 | 
			
		||||
                const request& req;
 | 
			
		||||
                response& res;
 | 
			
		||||
            };
 | 
			
		||||
 | 
			
		||||
            template <typename F, int NInt, int NUint, int NDouble, int NString, typename S1, typename S2> 
 | 
			
		||||
            struct call
 | 
			
		||||
            {
 | 
			
		||||
            };
 | 
			
		||||
 | 
			
		||||
            template <typename F, int NInt, int NUint, int NDouble, int NString, typename ... Args1, typename ... Args2> 
 | 
			
		||||
            struct call<F, NInt, NUint, NDouble, NString, black_magic::S<int64_t, Args1...>, black_magic::S<Args2...>>
 | 
			
		||||
            {
 | 
			
		||||
                void operator()(F cparams)
 | 
			
		||||
                {
 | 
			
		||||
                    using pushed = typename black_magic::S<Args2...>::template push_back<call_pair<int64_t, NInt>>;
 | 
			
		||||
                    call<F, NInt+1, NUint, NDouble, NString,
 | 
			
		||||
                        black_magic::S<Args1...>, pushed>()(cparams);
 | 
			
		||||
                }
 | 
			
		||||
            };
 | 
			
		||||
 | 
			
		||||
            template <typename F, int NInt, int NUint, int NDouble, int NString, typename ... Args1, typename ... Args2> 
 | 
			
		||||
            struct call<F, NInt, NUint, NDouble, NString, black_magic::S<uint64_t, Args1...>, black_magic::S<Args2...>>
 | 
			
		||||
            {
 | 
			
		||||
                void operator()(F cparams)
 | 
			
		||||
                {
 | 
			
		||||
                    using pushed = typename black_magic::S<Args2...>::template push_back<call_pair<uint64_t, NUint>>;
 | 
			
		||||
                    call<F, NInt, NUint+1, NDouble, NString,
 | 
			
		||||
                        black_magic::S<Args1...>, pushed>()(cparams);
 | 
			
		||||
                }
 | 
			
		||||
            };
 | 
			
		||||
 | 
			
		||||
            template <typename F, int NInt, int NUint, int NDouble, int NString, typename ... Args1, typename ... Args2> 
 | 
			
		||||
            struct call<F, NInt, NUint, NDouble, NString, black_magic::S<double, Args1...>, black_magic::S<Args2...>>
 | 
			
		||||
            {
 | 
			
		||||
                void operator()(F cparams)
 | 
			
		||||
                {
 | 
			
		||||
                    using pushed = typename black_magic::S<Args2...>::template push_back<call_pair<double, NDouble>>;
 | 
			
		||||
                    call<F, NInt, NUint, NDouble+1, NString,
 | 
			
		||||
                        black_magic::S<Args1...>, pushed>()(cparams);
 | 
			
		||||
                }
 | 
			
		||||
            };
 | 
			
		||||
 | 
			
		||||
            template <typename F, int NInt, int NUint, int NDouble, int NString, typename ... Args1, typename ... Args2> 
 | 
			
		||||
            struct call<F, NInt, NUint, NDouble, NString, black_magic::S<std::string, Args1...>, black_magic::S<Args2...>>
 | 
			
		||||
            {
 | 
			
		||||
                void operator()(F cparams)
 | 
			
		||||
                {
 | 
			
		||||
                    using pushed = typename black_magic::S<Args2...>::template push_back<call_pair<std::string, NString>>;
 | 
			
		||||
                    call<F, NInt, NUint, NDouble, NString+1,
 | 
			
		||||
                        black_magic::S<Args1...>, pushed>()(cparams);
 | 
			
		||||
                }
 | 
			
		||||
            };
 | 
			
		||||
 | 
			
		||||
            template <typename F, int NInt, int NUint, int NDouble, int NString, typename ... Args1> 
 | 
			
		||||
            struct call<F, NInt, NUint, NDouble, NString, black_magic::S<>, black_magic::S<Args1...>>
 | 
			
		||||
            {
 | 
			
		||||
                void operator()(F cparams)
 | 
			
		||||
                {
 | 
			
		||||
                    cparams.handler(
 | 
			
		||||
                        cparams.req,
 | 
			
		||||
                        cparams.res,
 | 
			
		||||
                        cparams.params.template get<typename Args1::type>(Args1::pos)... 
 | 
			
		||||
                    );
 | 
			
		||||
                }
 | 
			
		||||
            };
 | 
			
		||||
 | 
			
		||||
            template <typename Func, typename ... ArgsWrapped>
 | 
			
		||||
            struct Wrapped
 | 
			
		||||
            {
 | 
			
		||||
                template <typename ... Args>
 | 
			
		||||
                void set(Func f, typename std::enable_if<
 | 
			
		||||
                    !std::is_same<typename std::tuple_element<0, std::tuple<Args..., void>>::type, const request&>::value
 | 
			
		||||
                , int>::type = 0)
 | 
			
		||||
                {
 | 
			
		||||
                    handler_ = (
 | 
			
		||||
#ifdef CROW_CAN_USE_CPP14
 | 
			
		||||
                        [f = std::move(f)]
 | 
			
		||||
#else
 | 
			
		||||
                        [f]
 | 
			
		||||
#endif
 | 
			
		||||
                        (const request&, response& res, Args... args){
 | 
			
		||||
                            res = response(f(args...));
 | 
			
		||||
                            res.end();
 | 
			
		||||
                        });
 | 
			
		||||
                }
 | 
			
		||||
 | 
			
		||||
                template <typename Req, typename ... Args>
 | 
			
		||||
                struct req_handler_wrapper
 | 
			
		||||
                {
 | 
			
		||||
                    req_handler_wrapper(Func f)
 | 
			
		||||
                        : f(std::move(f))
 | 
			
		||||
                    {
 | 
			
		||||
                    }
 | 
			
		||||
 | 
			
		||||
                    void operator()(const request& req, response& res, Args... args)
 | 
			
		||||
                    {
 | 
			
		||||
                        res = response(f(req, args...));
 | 
			
		||||
                        res.end();
 | 
			
		||||
                    }
 | 
			
		||||
 | 
			
		||||
                    Func f;
 | 
			
		||||
                };
 | 
			
		||||
 | 
			
		||||
                template <typename ... Args>
 | 
			
		||||
                void set(Func f, typename std::enable_if<
 | 
			
		||||
                        std::is_same<typename std::tuple_element<0, std::tuple<Args..., void>>::type, const request&>::value &&
 | 
			
		||||
                        !std::is_same<typename std::tuple_element<1, std::tuple<Args..., void, void>>::type, response&>::value
 | 
			
		||||
                        , int>::type = 0)
 | 
			
		||||
                {
 | 
			
		||||
                    handler_ = req_handler_wrapper<Args...>(std::move(f));
 | 
			
		||||
                    /*handler_ = (
 | 
			
		||||
                        [f = std::move(f)]
 | 
			
		||||
                        (const request& req, response& res, Args... args){
 | 
			
		||||
                             res = response(f(req, args...));
 | 
			
		||||
                             res.end();
 | 
			
		||||
                        });*/
 | 
			
		||||
                }
 | 
			
		||||
 | 
			
		||||
                template <typename ... Args>
 | 
			
		||||
                void set(Func f, typename std::enable_if<
 | 
			
		||||
                        std::is_same<typename std::tuple_element<0, std::tuple<Args..., void>>::type, const request&>::value &&
 | 
			
		||||
                        std::is_same<typename std::tuple_element<1, std::tuple<Args..., void, void>>::type, response&>::value
 | 
			
		||||
                        , int>::type = 0)
 | 
			
		||||
                {
 | 
			
		||||
                    handler_ = std::move(f);
 | 
			
		||||
                }
 | 
			
		||||
 | 
			
		||||
                template <typename ... Args>
 | 
			
		||||
                struct handler_type_helper
 | 
			
		||||
                {
 | 
			
		||||
                    using type = std::function<void(const crow::request&, crow::response&, Args...)>;
 | 
			
		||||
                    using args_type = black_magic::S<typename black_magic::promote_t<Args>...>; 
 | 
			
		||||
                };
 | 
			
		||||
 | 
			
		||||
                template <typename ... Args>
 | 
			
		||||
                struct handler_type_helper<const request&, Args...>
 | 
			
		||||
                {
 | 
			
		||||
                    using type = std::function<void(const crow::request&, crow::response&, Args...)>;
 | 
			
		||||
                    using args_type = black_magic::S<typename black_magic::promote_t<Args>...>; 
 | 
			
		||||
                };
 | 
			
		||||
 | 
			
		||||
                template <typename ... Args>
 | 
			
		||||
                struct handler_type_helper<const request&, response&, Args...>
 | 
			
		||||
                {
 | 
			
		||||
                    using type = std::function<void(const crow::request&, crow::response&, Args...)>;
 | 
			
		||||
                    using args_type = black_magic::S<typename black_magic::promote_t<Args>...>; 
 | 
			
		||||
                };
 | 
			
		||||
 | 
			
		||||
                typename handler_type_helper<ArgsWrapped...>::type handler_;
 | 
			
		||||
 | 
			
		||||
                void operator()(const request& req, response& res, const routing_params& params)
 | 
			
		||||
                {
 | 
			
		||||
                    detail::routing_handler_call_helper::call<
 | 
			
		||||
                        detail::routing_handler_call_helper::call_params<
 | 
			
		||||
                            decltype(handler_)>,
 | 
			
		||||
                        0, 0, 0, 0, 
 | 
			
		||||
                        typename handler_type_helper<ArgsWrapped...>::args_type,
 | 
			
		||||
                        black_magic::S<>
 | 
			
		||||
                    >()(
 | 
			
		||||
                        detail::routing_handler_call_helper::call_params<
 | 
			
		||||
                            decltype(handler_)>
 | 
			
		||||
                        {handler_, params, req, res}
 | 
			
		||||
                   );
 | 
			
		||||
                }
 | 
			
		||||
            };
 | 
			
		||||
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    template <typename T>
 | 
			
		||||
    struct RuleParameterTraits
 | 
			
		||||
    {
 | 
			
		||||
        using self_t = T;
 | 
			
		||||
        self_t& name(std::string name) noexcept
 | 
			
		||||
        {
 | 
			
		||||
            ((self_t*)this)->name_ = std::move(name);
 | 
			
		||||
            return (self_t&)*this;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        self_t& methods(HTTPMethod method)
 | 
			
		||||
        {
 | 
			
		||||
            ((self_t*)this)->methods_ = 1 << (int)method;
 | 
			
		||||
            return (self_t&)*this;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <typename ... MethodArgs>
 | 
			
		||||
        self_t& methods(HTTPMethod method, MethodArgs ... args_method)
 | 
			
		||||
        {
 | 
			
		||||
            methods(args_method...);
 | 
			
		||||
            ((self_t*)this)->methods_ |= 1 << (int)method;
 | 
			
		||||
            return (self_t&)*this;
 | 
			
		||||
        }
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    class DynamicRule : public BaseRule, public RuleParameterTraits<DynamicRule>
 | 
			
		||||
    {
 | 
			
		||||
    public:
 | 
			
		||||
 | 
			
		||||
        DynamicRule(std::string rule)
 | 
			
		||||
            : BaseRule(std::move(rule))
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void validate() override
 | 
			
		||||
        {
 | 
			
		||||
            if (!erased_handler_)
 | 
			
		||||
            {
 | 
			
		||||
                throw std::runtime_error(name_ + (!name_.empty() ? ": " : "") + "no handler for url " + rule_);
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void handle(const request& req, response& res, const routing_params& params) override
 | 
			
		||||
        {
 | 
			
		||||
            erased_handler_(req, res, params);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <typename Func>
 | 
			
		||||
        void operator()(Func f)
 | 
			
		||||
        {
 | 
			
		||||
#ifdef CROW_MSVC_WORKAROUND
 | 
			
		||||
            using function_t = utility::function_traits<decltype(&Func::operator())>;
 | 
			
		||||
#else
 | 
			
		||||
            using function_t = utility::function_traits<Func>;
 | 
			
		||||
#endif
 | 
			
		||||
            erased_handler_ = wrap(std::move(f), black_magic::gen_seq<function_t::arity>());
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        // enable_if Arg1 == request && Arg2 == response
 | 
			
		||||
        // enable_if Arg1 == request && Arg2 != resposne
 | 
			
		||||
        // enable_if Arg1 != request
 | 
			
		||||
#ifdef CROW_MSVC_WORKAROUND
 | 
			
		||||
        template <typename Func, size_t ... Indices>
 | 
			
		||||
#else
 | 
			
		||||
        template <typename Func, unsigned ... Indices>
 | 
			
		||||
#endif
 | 
			
		||||
        std::function<void(const request&, response&, const routing_params&)> 
 | 
			
		||||
        wrap(Func f, black_magic::seq<Indices...>)
 | 
			
		||||
        {
 | 
			
		||||
#ifdef CROW_MSVC_WORKAROUND
 | 
			
		||||
            using function_t = utility::function_traits<decltype(&Func::operator())>;
 | 
			
		||||
#else
 | 
			
		||||
            using function_t = utility::function_traits<Func>;
 | 
			
		||||
#endif
 | 
			
		||||
            if (!black_magic::is_parameter_tag_compatible(
 | 
			
		||||
                black_magic::get_parameter_tag_runtime(rule_.c_str()), 
 | 
			
		||||
                black_magic::compute_parameter_tag_from_args_list<
 | 
			
		||||
                    typename function_t::template arg<Indices>...>::value))
 | 
			
		||||
            {
 | 
			
		||||
                throw std::runtime_error("route_dynamic: Handler type is mismatched with URL parameters: " + rule_);
 | 
			
		||||
            }
 | 
			
		||||
            auto ret = detail::routing_handler_call_helper::Wrapped<Func, typename function_t::template arg<Indices>...>();
 | 
			
		||||
            ret.template set<
 | 
			
		||||
                typename function_t::template arg<Indices>...
 | 
			
		||||
            >(std::move(f));
 | 
			
		||||
            return ret;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <typename Func>
 | 
			
		||||
        void operator()(std::string name, Func&& f)
 | 
			
		||||
        {
 | 
			
		||||
            name_ = std::move(name);
 | 
			
		||||
            (*this).template operator()<Func>(std::forward(f));
 | 
			
		||||
        }
 | 
			
		||||
    private:
 | 
			
		||||
        std::function<void(const request&, response&, const routing_params&)> erased_handler_;
 | 
			
		||||
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    template <typename ... Args>
 | 
			
		||||
    class TaggedRule : public BaseRule, public RuleParameterTraits<TaggedRule<Args...>>
 | 
			
		||||
    {
 | 
			
		||||
    public:
 | 
			
		||||
        using self_t = TaggedRule<Args...>;
 | 
			
		||||
 | 
			
		||||
        TaggedRule(std::string rule)
 | 
			
		||||
            : BaseRule(std::move(rule))
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void validate()
 | 
			
		||||
        {
 | 
			
		||||
            if (!handler_)
 | 
			
		||||
            {
 | 
			
		||||
                throw std::runtime_error(name_ + (!name_.empty() ? ": " : "") + "no handler for url " + rule_);
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <typename Func>
 | 
			
		||||
        typename std::enable_if<black_magic::CallHelper<Func, black_magic::S<Args...>>::value, void>::type
 | 
			
		||||
        operator()(Func&& f)
 | 
			
		||||
        {
 | 
			
		||||
            static_assert(black_magic::CallHelper<Func, black_magic::S<Args...>>::value ||
 | 
			
		||||
                black_magic::CallHelper<Func, black_magic::S<crow::request, Args...>>::value , 
 | 
			
		||||
                "Handler type is mismatched with URL parameters");
 | 
			
		||||
            static_assert(!std::is_same<void, decltype(f(std::declval<Args>()...))>::value, 
 | 
			
		||||
                "Handler function cannot have void return type; valid return types: string, int, crow::resposne, crow::json::wvalue");
 | 
			
		||||
 | 
			
		||||
                handler_ = [f = std::move(f)](const request&, response& res, Args ... args){
 | 
			
		||||
                    res = response(f(args...));
 | 
			
		||||
                    res.end();
 | 
			
		||||
                };
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <typename Func>
 | 
			
		||||
        typename std::enable_if<
 | 
			
		||||
            !black_magic::CallHelper<Func, black_magic::S<Args...>>::value &&
 | 
			
		||||
            black_magic::CallHelper<Func, black_magic::S<crow::request, Args...>>::value, 
 | 
			
		||||
            void>::type
 | 
			
		||||
        operator()(Func&& f)
 | 
			
		||||
        {
 | 
			
		||||
            static_assert(black_magic::CallHelper<Func, black_magic::S<Args...>>::value ||
 | 
			
		||||
                black_magic::CallHelper<Func, black_magic::S<crow::request, Args...>>::value, 
 | 
			
		||||
                "Handler type is mismatched with URL parameters");
 | 
			
		||||
            static_assert(!std::is_same<void, decltype(f(std::declval<crow::request>(), std::declval<Args>()...))>::value, 
 | 
			
		||||
                "Handler function cannot have void return type; valid return types: string, int, crow::resposne, crow::json::wvalue");
 | 
			
		||||
 | 
			
		||||
                handler_ = [f = std::move(f)](const crow::request& req, crow::response& res, Args ... args){
 | 
			
		||||
                    res = response(f(req, args...));
 | 
			
		||||
                    res.end();
 | 
			
		||||
                };
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <typename Func>
 | 
			
		||||
        typename std::enable_if<
 | 
			
		||||
            !black_magic::CallHelper<Func, black_magic::S<Args...>>::value &&
 | 
			
		||||
            !black_magic::CallHelper<Func, black_magic::S<crow::request, Args...>>::value, 
 | 
			
		||||
            void>::type
 | 
			
		||||
        operator()(Func&& f)
 | 
			
		||||
        {
 | 
			
		||||
            static_assert(black_magic::CallHelper<Func, black_magic::S<Args...>>::value ||
 | 
			
		||||
                black_magic::CallHelper<Func, black_magic::S<crow::request, Args...>>::value ||
 | 
			
		||||
                black_magic::CallHelper<Func, black_magic::S<crow::request, crow::response&, Args...>>::value
 | 
			
		||||
                , 
 | 
			
		||||
                "Handler type is mismatched with URL parameters");
 | 
			
		||||
            static_assert(std::is_same<void, decltype(f(std::declval<crow::request>(), std::declval<crow::response&>(), std::declval<Args>()...))>::value, 
 | 
			
		||||
                "Handler function with response argument should have void return type");
 | 
			
		||||
 | 
			
		||||
                handler_ = std::move(f);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <typename Func>
 | 
			
		||||
        void operator()(std::string name, Func&& f)
 | 
			
		||||
        {
 | 
			
		||||
            name_ = std::move(name);
 | 
			
		||||
            (*this).template operator()<Func>(std::forward(f));
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void handle(const request& req, response& res, const routing_params& params) override
 | 
			
		||||
        {
 | 
			
		||||
            detail::routing_handler_call_helper::call<
 | 
			
		||||
                detail::routing_handler_call_helper::call_params<
 | 
			
		||||
                    decltype(handler_)>, 
 | 
			
		||||
                0, 0, 0, 0, 
 | 
			
		||||
                black_magic::S<Args...>, 
 | 
			
		||||
                black_magic::S<>
 | 
			
		||||
            >()(
 | 
			
		||||
                detail::routing_handler_call_helper::call_params<
 | 
			
		||||
                    decltype(handler_)>
 | 
			
		||||
                {handler_, params, req, res}
 | 
			
		||||
            );
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
    private:
 | 
			
		||||
        std::function<void(const crow::request&, crow::response&, Args...)> handler_;
 | 
			
		||||
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    const int RULE_SPECIAL_REDIRECT_SLASH = 1;
 | 
			
		||||
 | 
			
		||||
    class Trie
 | 
			
		||||
    {
 | 
			
		||||
    public:
 | 
			
		||||
        struct Node
 | 
			
		||||
        {
 | 
			
		||||
            unsigned rule_index{};
 | 
			
		||||
            std::array<unsigned, (int)ParamType::MAX> param_childrens{};
 | 
			
		||||
            std::unordered_map<std::string, unsigned> children;
 | 
			
		||||
 | 
			
		||||
            bool IsSimpleNode() const
 | 
			
		||||
            {
 | 
			
		||||
                return 
 | 
			
		||||
                    !rule_index &&
 | 
			
		||||
                    std::all_of(
 | 
			
		||||
                        std::begin(param_childrens), 
 | 
			
		||||
                        std::end(param_childrens), 
 | 
			
		||||
                        [](unsigned x){ return !x; });
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        Trie() : nodes_(1)
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
private:
 | 
			
		||||
        void optimizeNode(Node* node)
 | 
			
		||||
        {
 | 
			
		||||
            for(auto x : node->param_childrens)
 | 
			
		||||
            {
 | 
			
		||||
                if (!x)
 | 
			
		||||
                    continue;
 | 
			
		||||
                Node* child = &nodes_[x];
 | 
			
		||||
                optimizeNode(child);
 | 
			
		||||
            }
 | 
			
		||||
            if (node->children.empty())
 | 
			
		||||
                return;
 | 
			
		||||
            bool mergeWithChild = true;
 | 
			
		||||
            for(auto& kv : node->children)
 | 
			
		||||
            {
 | 
			
		||||
                Node* child = &nodes_[kv.second];
 | 
			
		||||
                if (!child->IsSimpleNode())
 | 
			
		||||
                {
 | 
			
		||||
                    mergeWithChild = false;
 | 
			
		||||
                    break;
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            if (mergeWithChild)
 | 
			
		||||
            {
 | 
			
		||||
                decltype(node->children) merged;
 | 
			
		||||
                for(auto& kv : node->children)
 | 
			
		||||
                {
 | 
			
		||||
                    Node* child = &nodes_[kv.second];
 | 
			
		||||
                    for(auto& child_kv : child->children)
 | 
			
		||||
                    {
 | 
			
		||||
                        merged[kv.first + child_kv.first] = child_kv.second;
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
                node->children = std::move(merged);
 | 
			
		||||
                optimizeNode(node);
 | 
			
		||||
            }
 | 
			
		||||
            else
 | 
			
		||||
            {
 | 
			
		||||
                for(auto& kv : node->children)
 | 
			
		||||
                {
 | 
			
		||||
                    Node* child = &nodes_[kv.second];
 | 
			
		||||
                    optimizeNode(child);
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void optimize()
 | 
			
		||||
        {
 | 
			
		||||
            optimizeNode(head());
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
public:
 | 
			
		||||
        void validate()
 | 
			
		||||
        {
 | 
			
		||||
            if (!head()->IsSimpleNode())
 | 
			
		||||
                throw std::runtime_error("Internal error: Trie header should be simple!");
 | 
			
		||||
            optimize();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        std::pair<unsigned, routing_params> find(const std::string& req_url, const Node* node = nullptr, unsigned pos = 0, routing_params* params = nullptr) const
 | 
			
		||||
        {
 | 
			
		||||
            routing_params empty;
 | 
			
		||||
            if (params == nullptr)
 | 
			
		||||
                params = ∅
 | 
			
		||||
 | 
			
		||||
            unsigned found{};
 | 
			
		||||
            routing_params match_params;
 | 
			
		||||
 | 
			
		||||
            if (node == nullptr)
 | 
			
		||||
                node = head();
 | 
			
		||||
            if (pos == req_url.size())
 | 
			
		||||
                return {node->rule_index, *params};
 | 
			
		||||
 | 
			
		||||
            auto update_found = [&found, &match_params](std::pair<unsigned, routing_params>& ret)
 | 
			
		||||
            {
 | 
			
		||||
                if (ret.first && (!found || found > ret.first))
 | 
			
		||||
                {
 | 
			
		||||
                    found = ret.first;
 | 
			
		||||
                    match_params = std::move(ret.second);
 | 
			
		||||
                }
 | 
			
		||||
            };
 | 
			
		||||
 | 
			
		||||
            if (node->param_childrens[(int)ParamType::INT])
 | 
			
		||||
            {
 | 
			
		||||
                char c = req_url[pos];
 | 
			
		||||
                if ((c >= '0' && c <= '9') || c == '+' || c == '-')
 | 
			
		||||
                {
 | 
			
		||||
                    char* eptr;
 | 
			
		||||
                    errno = 0;
 | 
			
		||||
                    long long int value = strtoll(req_url.data()+pos, &eptr, 10);
 | 
			
		||||
                    if (errno != ERANGE && eptr != req_url.data()+pos)
 | 
			
		||||
                    {
 | 
			
		||||
                        params->int_params.push_back(value);
 | 
			
		||||
                        auto ret = find(req_url, &nodes_[node->param_childrens[(int)ParamType::INT]], eptr - req_url.data(), params);
 | 
			
		||||
                        update_found(ret);
 | 
			
		||||
                        params->int_params.pop_back();
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            if (node->param_childrens[(int)ParamType::UINT])
 | 
			
		||||
            {
 | 
			
		||||
                char c = req_url[pos];
 | 
			
		||||
                if ((c >= '0' && c <= '9') || c == '+')
 | 
			
		||||
                {
 | 
			
		||||
                    char* eptr;
 | 
			
		||||
                    errno = 0;
 | 
			
		||||
                    unsigned long long int value = strtoull(req_url.data()+pos, &eptr, 10);
 | 
			
		||||
                    if (errno != ERANGE && eptr != req_url.data()+pos)
 | 
			
		||||
                    {
 | 
			
		||||
                        params->uint_params.push_back(value);
 | 
			
		||||
                        auto ret = find(req_url, &nodes_[node->param_childrens[(int)ParamType::UINT]], eptr - req_url.data(), params);
 | 
			
		||||
                        update_found(ret);
 | 
			
		||||
                        params->uint_params.pop_back();
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            if (node->param_childrens[(int)ParamType::DOUBLE])
 | 
			
		||||
            {
 | 
			
		||||
                char c = req_url[pos];
 | 
			
		||||
                if ((c >= '0' && c <= '9') || c == '+' || c == '-' || c == '.')
 | 
			
		||||
                {
 | 
			
		||||
                    char* eptr;
 | 
			
		||||
                    errno = 0;
 | 
			
		||||
                    double value = strtod(req_url.data()+pos, &eptr);
 | 
			
		||||
                    if (errno != ERANGE && eptr != req_url.data()+pos)
 | 
			
		||||
                    {
 | 
			
		||||
                        params->double_params.push_back(value);
 | 
			
		||||
                        auto ret = find(req_url, &nodes_[node->param_childrens[(int)ParamType::DOUBLE]], eptr - req_url.data(), params);
 | 
			
		||||
                        update_found(ret);
 | 
			
		||||
                        params->double_params.pop_back();
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            if (node->param_childrens[(int)ParamType::STRING])
 | 
			
		||||
            {
 | 
			
		||||
                size_t epos = pos;
 | 
			
		||||
                for(; epos < req_url.size(); epos ++)
 | 
			
		||||
                {
 | 
			
		||||
                    if (req_url[epos] == '/')
 | 
			
		||||
                        break;
 | 
			
		||||
                }
 | 
			
		||||
 | 
			
		||||
                if (epos != pos)
 | 
			
		||||
                {
 | 
			
		||||
                    params->string_params.push_back(req_url.substr(pos, epos-pos));
 | 
			
		||||
                    auto ret = find(req_url, &nodes_[node->param_childrens[(int)ParamType::STRING]], epos, params);
 | 
			
		||||
                    update_found(ret);
 | 
			
		||||
                    params->string_params.pop_back();
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            if (node->param_childrens[(int)ParamType::PATH])
 | 
			
		||||
            {
 | 
			
		||||
                size_t epos = req_url.size();
 | 
			
		||||
 | 
			
		||||
                if (epos != pos)
 | 
			
		||||
                {
 | 
			
		||||
                    params->string_params.push_back(req_url.substr(pos, epos-pos));
 | 
			
		||||
                    auto ret = find(req_url, &nodes_[node->param_childrens[(int)ParamType::PATH]], epos, params);
 | 
			
		||||
                    update_found(ret);
 | 
			
		||||
                    params->string_params.pop_back();
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            for(auto& kv : node->children)
 | 
			
		||||
            {
 | 
			
		||||
                const std::string& fragment = kv.first;
 | 
			
		||||
                const Node* child = &nodes_[kv.second];
 | 
			
		||||
 | 
			
		||||
                if (req_url.compare(pos, fragment.size(), fragment) == 0)
 | 
			
		||||
                {
 | 
			
		||||
                    auto ret = find(req_url, child, pos + fragment.size(), params);
 | 
			
		||||
                    update_found(ret);
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            return {found, match_params};
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void add(const std::string& url, unsigned rule_index)
 | 
			
		||||
        {
 | 
			
		||||
            unsigned idx{0};
 | 
			
		||||
 | 
			
		||||
            for(unsigned i = 0; i < url.size(); i ++)
 | 
			
		||||
            {
 | 
			
		||||
                char c = url[i];
 | 
			
		||||
                if (c == '<')
 | 
			
		||||
                {
 | 
			
		||||
                    static struct ParamTraits
 | 
			
		||||
                    {
 | 
			
		||||
                        ParamType type;
 | 
			
		||||
                        std::string name;
 | 
			
		||||
                    } paramTraits[] =
 | 
			
		||||
                    {
 | 
			
		||||
                        { ParamType::INT, "<int>" },
 | 
			
		||||
                        { ParamType::UINT, "<uint>" },
 | 
			
		||||
                        { ParamType::DOUBLE, "<float>" },
 | 
			
		||||
                        { ParamType::DOUBLE, "<double>" },
 | 
			
		||||
                        { ParamType::STRING, "<str>" },
 | 
			
		||||
                        { ParamType::STRING, "<string>" },
 | 
			
		||||
                        { ParamType::PATH, "<path>" },
 | 
			
		||||
                    };
 | 
			
		||||
 | 
			
		||||
                    for(auto& x:paramTraits)
 | 
			
		||||
                    {
 | 
			
		||||
                        if (url.compare(i, x.name.size(), x.name) == 0)
 | 
			
		||||
                        {
 | 
			
		||||
                            if (!nodes_[idx].param_childrens[(int)x.type])
 | 
			
		||||
                            {
 | 
			
		||||
                                auto new_node_idx = new_node();
 | 
			
		||||
                                nodes_[idx].param_childrens[(int)x.type] = new_node_idx;
 | 
			
		||||
                            }
 | 
			
		||||
                            idx = nodes_[idx].param_childrens[(int)x.type];
 | 
			
		||||
                            i += x.name.size();
 | 
			
		||||
                            break;
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
 | 
			
		||||
                    i --;
 | 
			
		||||
                }
 | 
			
		||||
                else
 | 
			
		||||
                {
 | 
			
		||||
                    std::string piece(&c, 1);
 | 
			
		||||
                    if (!nodes_[idx].children.count(piece))
 | 
			
		||||
                    {
 | 
			
		||||
                        auto new_node_idx = new_node();
 | 
			
		||||
                        nodes_[idx].children.emplace(piece, new_node_idx);
 | 
			
		||||
                    }
 | 
			
		||||
                    idx = nodes_[idx].children[piece];
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            if (nodes_[idx].rule_index)
 | 
			
		||||
                throw std::runtime_error("handler already exists for " + url);
 | 
			
		||||
            nodes_[idx].rule_index = rule_index;
 | 
			
		||||
        }
 | 
			
		||||
    private:
 | 
			
		||||
        void debug_node_print(Node* n, int level)
 | 
			
		||||
        {
 | 
			
		||||
            for(int i = 0; i < (int)ParamType::MAX; i ++)
 | 
			
		||||
            {
 | 
			
		||||
                if (n->param_childrens[i])
 | 
			
		||||
                {
 | 
			
		||||
                    CROW_LOG_DEBUG << std::string(2*level, ' ') /*<< "("<<n->param_childrens[i]<<") "*/;
 | 
			
		||||
                    switch((ParamType)i)
 | 
			
		||||
                    {
 | 
			
		||||
                        case ParamType::INT:
 | 
			
		||||
                            CROW_LOG_DEBUG << "<int>";
 | 
			
		||||
                            break;
 | 
			
		||||
                        case ParamType::UINT:
 | 
			
		||||
                            CROW_LOG_DEBUG << "<uint>";
 | 
			
		||||
                            break;
 | 
			
		||||
                        case ParamType::DOUBLE:
 | 
			
		||||
                            CROW_LOG_DEBUG << "<float>";
 | 
			
		||||
                            break;
 | 
			
		||||
                        case ParamType::STRING:
 | 
			
		||||
                            CROW_LOG_DEBUG << "<str>";
 | 
			
		||||
                            break;
 | 
			
		||||
                        case ParamType::PATH:
 | 
			
		||||
                            CROW_LOG_DEBUG << "<path>";
 | 
			
		||||
                            break;
 | 
			
		||||
                        default:
 | 
			
		||||
                            CROW_LOG_DEBUG << "<ERROR>";
 | 
			
		||||
                            break;
 | 
			
		||||
                    }
 | 
			
		||||
 | 
			
		||||
                    debug_node_print(&nodes_[n->param_childrens[i]], level+1);
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            for(auto& kv : n->children)
 | 
			
		||||
            {
 | 
			
		||||
                CROW_LOG_DEBUG << std::string(2*level, ' ') /*<< "(" << kv.second << ") "*/ << kv.first;
 | 
			
		||||
                debug_node_print(&nodes_[kv.second], level+1);
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
    public:
 | 
			
		||||
        void debug_print()
 | 
			
		||||
        {
 | 
			
		||||
            debug_node_print(head(), 0);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
    private:
 | 
			
		||||
        const Node* head() const
 | 
			
		||||
        {
 | 
			
		||||
            return &nodes_.front();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        Node* head()
 | 
			
		||||
        {
 | 
			
		||||
            return &nodes_.front();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        unsigned new_node()
 | 
			
		||||
        {
 | 
			
		||||
            nodes_.resize(nodes_.size()+1);
 | 
			
		||||
            return nodes_.size() - 1;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        std::vector<Node> nodes_;
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    class Router
 | 
			
		||||
    {
 | 
			
		||||
    public:
 | 
			
		||||
        Router() : rules_(2) 
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        DynamicRule& new_rule_dynamic(const std::string& rule)
 | 
			
		||||
        {
 | 
			
		||||
            auto ruleObject = new DynamicRule(rule);
 | 
			
		||||
 | 
			
		||||
            internal_add_rule_object(rule, ruleObject);
 | 
			
		||||
 | 
			
		||||
            return *ruleObject;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <uint64_t N>
 | 
			
		||||
        typename black_magic::arguments<N>::type::template rebind<TaggedRule>& new_rule_tagged(const std::string& rule)
 | 
			
		||||
        {
 | 
			
		||||
            using RuleT = typename black_magic::arguments<N>::type::template rebind<TaggedRule>;
 | 
			
		||||
            auto ruleObject = new RuleT(rule);
 | 
			
		||||
 | 
			
		||||
            internal_add_rule_object(rule, ruleObject);
 | 
			
		||||
 | 
			
		||||
            return *ruleObject;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void internal_add_rule_object(const std::string& rule, BaseRule* ruleObject)
 | 
			
		||||
        {
 | 
			
		||||
            rules_.emplace_back(ruleObject);
 | 
			
		||||
            trie_.add(rule, rules_.size() - 1);
 | 
			
		||||
 | 
			
		||||
            // directory case: 
 | 
			
		||||
            //   request to `/about' url matches `/about/' rule 
 | 
			
		||||
            if (rule.size() > 1 && rule.back() == '/')
 | 
			
		||||
            {
 | 
			
		||||
                std::string rule_without_trailing_slash = rule;
 | 
			
		||||
                rule_without_trailing_slash.pop_back();
 | 
			
		||||
                trie_.add(rule_without_trailing_slash, RULE_SPECIAL_REDIRECT_SLASH);
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void validate()
 | 
			
		||||
        {
 | 
			
		||||
            trie_.validate();
 | 
			
		||||
            for(auto& rule:rules_)
 | 
			
		||||
            {
 | 
			
		||||
                if (rule)
 | 
			
		||||
                    rule->validate();
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void handle(const request& req, response& res)
 | 
			
		||||
        {
 | 
			
		||||
            auto found = trie_.find(req.url);
 | 
			
		||||
 | 
			
		||||
            unsigned rule_index = found.first;
 | 
			
		||||
            CROW_LOG_DEBUG << "???" << rule_index;
 | 
			
		||||
 | 
			
		||||
            if (!rule_index)
 | 
			
		||||
            {
 | 
			
		||||
                CROW_LOG_DEBUG << "Cannot match rules " << req.url;
 | 
			
		||||
                res = response(404);
 | 
			
		||||
                res.end();
 | 
			
		||||
                return;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            if (rule_index >= rules_.size())
 | 
			
		||||
                throw std::runtime_error("Trie internal structure corrupted!");
 | 
			
		||||
 | 
			
		||||
            if (rule_index == RULE_SPECIAL_REDIRECT_SLASH)
 | 
			
		||||
            {
 | 
			
		||||
                CROW_LOG_INFO << "Redirecting to a url with trailing slash: " << req.url;
 | 
			
		||||
                res = response(301);
 | 
			
		||||
 | 
			
		||||
                // TODO absolute url building
 | 
			
		||||
                if (req.get_header_value("Host").empty())
 | 
			
		||||
                {
 | 
			
		||||
                    res.add_header("Location", req.url + "/");
 | 
			
		||||
                }
 | 
			
		||||
                else
 | 
			
		||||
                {
 | 
			
		||||
                    res.add_header("Location", "http://" + req.get_header_value("Host") + req.url + "/");
 | 
			
		||||
                }
 | 
			
		||||
                res.end();
 | 
			
		||||
                return;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            if ((rules_[rule_index]->get_methods() & (1<<(uint32_t)req.method)) == 0)
 | 
			
		||||
            {
 | 
			
		||||
                CROW_LOG_DEBUG << "Rule found but method mismatch: " << req.url << " with " << method_name(req.method) << "(" << (uint32_t)req.method << ") / " << rules_[rule_index]->get_methods();
 | 
			
		||||
                res = response(404);
 | 
			
		||||
                res.end();
 | 
			
		||||
                return;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            CROW_LOG_DEBUG << "Matched rule '" << rules_[rule_index]->rule_ << "' " << (uint32_t)req.method << " / " << rules_[rule_index]->get_methods();
 | 
			
		||||
 | 
			
		||||
            // any uncaught exceptions become 500s
 | 
			
		||||
            try
 | 
			
		||||
            {
 | 
			
		||||
                rules_[rule_index]->handle(req, res, found.second);
 | 
			
		||||
            }
 | 
			
		||||
            catch(std::exception& e)
 | 
			
		||||
            {
 | 
			
		||||
                CROW_LOG_ERROR << "An uncaught exception occurred: " << e.what();
 | 
			
		||||
                res = response(500);
 | 
			
		||||
                res.end();
 | 
			
		||||
                return;   
 | 
			
		||||
            }
 | 
			
		||||
            catch(...)
 | 
			
		||||
            {
 | 
			
		||||
                CROW_LOG_ERROR << "An uncaught exception occurred. The type was unknown so no information was available.";
 | 
			
		||||
                res = response(500);
 | 
			
		||||
                res.end();
 | 
			
		||||
                return;   
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void debug_print()
 | 
			
		||||
        {
 | 
			
		||||
            trie_.debug_print();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
    private:
 | 
			
		||||
        std::vector<std::unique_ptr<BaseRule>> rules_;
 | 
			
		||||
        Trie trie_;
 | 
			
		||||
    };
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										38
									
								
								ext/crow/settings.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										38
									
								
								ext/crow/settings.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,38 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
// settings for crow
 | 
			
		||||
// TODO - replace with runtime config. libucl?
 | 
			
		||||
 | 
			
		||||
/* #ifdef - enables debug mode */
 | 
			
		||||
#define CROW_ENABLE_DEBUG
 | 
			
		||||
 | 
			
		||||
/* #ifdef - enables logging */
 | 
			
		||||
#define CROW_ENABLE_LOGGING
 | 
			
		||||
 | 
			
		||||
/* #ifdef - enables SSL */
 | 
			
		||||
//#define CROW_ENABLE_SSL
 | 
			
		||||
 | 
			
		||||
/* #define - specifies log level */
 | 
			
		||||
/*
 | 
			
		||||
    DEBUG       = 0
 | 
			
		||||
    INFO        = 1
 | 
			
		||||
    WARNING     = 2
 | 
			
		||||
    ERROR       = 3
 | 
			
		||||
    CRITICAL    = 4
 | 
			
		||||
 | 
			
		||||
    default to INFO
 | 
			
		||||
*/
 | 
			
		||||
#define CROW_LOG_LEVEL 1
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
// compiler flags
 | 
			
		||||
#if __cplusplus >= 201402L
 | 
			
		||||
#define CROW_CAN_USE_CPP14
 | 
			
		||||
#endif
 | 
			
		||||
 | 
			
		||||
#if defined(_MSC_VER)
 | 
			
		||||
#if _MSC_VER < 1900
 | 
			
		||||
#define CROW_MSVC_WORKAROUND
 | 
			
		||||
#define constexpr const
 | 
			
		||||
#define noexcept throw()
 | 
			
		||||
#endif
 | 
			
		||||
#endif
 | 
			
		||||
							
								
								
									
										98
									
								
								ext/crow/socket_adaptors.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										98
									
								
								ext/crow/socket_adaptors.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,98 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
#include <boost/asio.hpp>
 | 
			
		||||
#include "settings.h"
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
    using namespace boost;
 | 
			
		||||
    using tcp = asio::ip::tcp;
 | 
			
		||||
 | 
			
		||||
    struct SocketAdaptor
 | 
			
		||||
    {
 | 
			
		||||
        using context = void;
 | 
			
		||||
        SocketAdaptor(boost::asio::io_service& io_service, context*)
 | 
			
		||||
            : socket_(io_service)
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        tcp::socket& raw_socket()
 | 
			
		||||
        {
 | 
			
		||||
            return socket_;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        tcp::socket& socket()
 | 
			
		||||
        {
 | 
			
		||||
            return socket_;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        tcp::endpoint remote_endpoint()
 | 
			
		||||
        {
 | 
			
		||||
            return socket_.remote_endpoint();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        bool is_open()
 | 
			
		||||
        {
 | 
			
		||||
            return socket_.is_open();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void close()
 | 
			
		||||
        {
 | 
			
		||||
            socket_.close();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <typename F> 
 | 
			
		||||
        void start(F f)
 | 
			
		||||
        {
 | 
			
		||||
            f(boost::system::error_code());
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        tcp::socket socket_;
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
#ifdef CROW_ENABLE_SSL
 | 
			
		||||
    struct SSLAdaptor
 | 
			
		||||
    {
 | 
			
		||||
        using context = boost::asio::ssl::context;
 | 
			
		||||
        SSLAdaptor(boost::asio::io_service& io_service, context* ctx)
 | 
			
		||||
            : ssl_socket_(io_service, *ctx)
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        boost::asio::ssl::stream<tcp::socket>& socket()
 | 
			
		||||
        {
 | 
			
		||||
            return ssl_socket_;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        tcp::socket::lowest_layer_type&
 | 
			
		||||
        raw_socket()
 | 
			
		||||
        {
 | 
			
		||||
            return ssl_socket_.lowest_layer();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        tcp::endpoint remote_endpoint()
 | 
			
		||||
        {
 | 
			
		||||
            return raw_socket().remote_endpoint();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        bool is_open()
 | 
			
		||||
        {
 | 
			
		||||
            return raw_socket().is_open();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        void close()
 | 
			
		||||
        {
 | 
			
		||||
            raw_socket().close();
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template <typename F> 
 | 
			
		||||
        void start(F f)
 | 
			
		||||
        {
 | 
			
		||||
            ssl_socket_.async_handshake(boost::asio::ssl::stream_base::server,
 | 
			
		||||
                    [f](const boost::system::error_code& ec) {
 | 
			
		||||
                        f(ec);
 | 
			
		||||
                    });
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        boost::asio::ssl::stream<tcp::socket> ssl_socket_;
 | 
			
		||||
    };
 | 
			
		||||
#endif
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										503
									
								
								ext/crow/utility.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										503
									
								
								ext/crow/utility.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,503 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <cstdint>
 | 
			
		||||
#include <stdexcept>
 | 
			
		||||
#include <tuple>
 | 
			
		||||
#include <type_traits>
 | 
			
		||||
#include <cstring>
 | 
			
		||||
#include <functional>
 | 
			
		||||
 | 
			
		||||
namespace crow
 | 
			
		||||
{
 | 
			
		||||
    namespace black_magic
 | 
			
		||||
    {
 | 
			
		||||
#ifndef CROW_MSVC_WORKAROUND
 | 
			
		||||
        struct OutOfRange
 | 
			
		||||
        {
 | 
			
		||||
            OutOfRange(unsigned pos, unsigned length) {}
 | 
			
		||||
        };
 | 
			
		||||
        constexpr unsigned requires_in_range( unsigned i, unsigned len )
 | 
			
		||||
        {
 | 
			
		||||
            return i >= len ? throw OutOfRange(i, len) : i;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        class const_str
 | 
			
		||||
        {
 | 
			
		||||
            const char * const begin_;
 | 
			
		||||
            unsigned size_;
 | 
			
		||||
 | 
			
		||||
            public:
 | 
			
		||||
            template< unsigned N >
 | 
			
		||||
                constexpr const_str( const char(&arr)[N] ) : begin_(arr), size_(N - 1) {
 | 
			
		||||
                    static_assert( N >= 1, "not a string literal");
 | 
			
		||||
                }
 | 
			
		||||
            constexpr char operator[]( unsigned i ) const { 
 | 
			
		||||
                return requires_in_range(i, size_), begin_[i]; 
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            constexpr operator const char *() const { 
 | 
			
		||||
                return begin_; 
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            constexpr const char* begin() const { return begin_; }
 | 
			
		||||
            constexpr const char* end() const { return begin_ + size_; }
 | 
			
		||||
 | 
			
		||||
            constexpr unsigned size() const { 
 | 
			
		||||
                return size_; 
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        constexpr unsigned find_closing_tag(const_str s, unsigned p)
 | 
			
		||||
        {
 | 
			
		||||
            return s[p] == '>' ? p : find_closing_tag(s, p+1);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        constexpr bool is_valid(const_str s, unsigned i = 0, int f = 0)
 | 
			
		||||
        {
 | 
			
		||||
            return 
 | 
			
		||||
                i == s.size()
 | 
			
		||||
                    ? f == 0 :
 | 
			
		||||
                f < 0 || f >= 2
 | 
			
		||||
                    ? false :
 | 
			
		||||
                s[i] == '<'
 | 
			
		||||
                    ? is_valid(s, i+1, f+1) :
 | 
			
		||||
                s[i] == '>'
 | 
			
		||||
                    ? is_valid(s, i+1, f-1) :
 | 
			
		||||
                is_valid(s, i+1, f);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        constexpr bool is_equ_p(const char* a, const char* b, unsigned n)
 | 
			
		||||
        {
 | 
			
		||||
            return
 | 
			
		||||
                *a == 0 && *b == 0 && n == 0 
 | 
			
		||||
                    ? true :
 | 
			
		||||
                (*a == 0 || *b == 0)
 | 
			
		||||
                    ? false :
 | 
			
		||||
                n == 0
 | 
			
		||||
                    ? true :
 | 
			
		||||
                *a != *b
 | 
			
		||||
                    ? false :
 | 
			
		||||
                is_equ_p(a+1, b+1, n-1);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        constexpr bool is_equ_n(const_str a, unsigned ai, const_str b, unsigned bi, unsigned n)
 | 
			
		||||
        {
 | 
			
		||||
            return 
 | 
			
		||||
                ai + n > a.size() || bi + n > b.size() 
 | 
			
		||||
                    ? false :
 | 
			
		||||
                n == 0 
 | 
			
		||||
                    ? true : 
 | 
			
		||||
                a[ai] != b[bi] 
 | 
			
		||||
                    ? false : 
 | 
			
		||||
                is_equ_n(a,ai+1,b,bi+1,n-1);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        constexpr bool is_int(const_str s, unsigned i)
 | 
			
		||||
        {
 | 
			
		||||
            return is_equ_n(s, i, "<int>", 0, 5);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        constexpr bool is_uint(const_str s, unsigned i)
 | 
			
		||||
        {
 | 
			
		||||
            return is_equ_n(s, i, "<uint>", 0, 6);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        constexpr bool is_float(const_str s, unsigned i)
 | 
			
		||||
        {
 | 
			
		||||
            return is_equ_n(s, i, "<float>", 0, 7) ||
 | 
			
		||||
                is_equ_n(s, i, "<double>", 0, 8);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        constexpr bool is_str(const_str s, unsigned i)
 | 
			
		||||
        {
 | 
			
		||||
            return is_equ_n(s, i, "<str>", 0, 5) ||
 | 
			
		||||
                is_equ_n(s, i, "<string>", 0, 8);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        constexpr bool is_path(const_str s, unsigned i)
 | 
			
		||||
        {
 | 
			
		||||
            return is_equ_n(s, i, "<path>", 0, 6);
 | 
			
		||||
        }
 | 
			
		||||
#endif
 | 
			
		||||
        template <typename T> 
 | 
			
		||||
        struct parameter_tag
 | 
			
		||||
        {
 | 
			
		||||
            static const int value = 0;
 | 
			
		||||
        };
 | 
			
		||||
#define CROW_INTERNAL_PARAMETER_TAG(t, i) \
 | 
			
		||||
template <> \
 | 
			
		||||
struct parameter_tag<t> \
 | 
			
		||||
{ \
 | 
			
		||||
    static const int value = i; \
 | 
			
		||||
};
 | 
			
		||||
        CROW_INTERNAL_PARAMETER_TAG(int, 1);
 | 
			
		||||
        CROW_INTERNAL_PARAMETER_TAG(char, 1);
 | 
			
		||||
        CROW_INTERNAL_PARAMETER_TAG(short, 1);
 | 
			
		||||
        CROW_INTERNAL_PARAMETER_TAG(long, 1);
 | 
			
		||||
        CROW_INTERNAL_PARAMETER_TAG(long long, 1);
 | 
			
		||||
        CROW_INTERNAL_PARAMETER_TAG(unsigned int, 2);
 | 
			
		||||
        CROW_INTERNAL_PARAMETER_TAG(unsigned char, 2);
 | 
			
		||||
        CROW_INTERNAL_PARAMETER_TAG(unsigned short, 2);
 | 
			
		||||
        CROW_INTERNAL_PARAMETER_TAG(unsigned long, 2);
 | 
			
		||||
        CROW_INTERNAL_PARAMETER_TAG(unsigned long long, 2);
 | 
			
		||||
        CROW_INTERNAL_PARAMETER_TAG(double, 3);
 | 
			
		||||
        CROW_INTERNAL_PARAMETER_TAG(std::string, 4);
 | 
			
		||||
#undef CROW_INTERNAL_PARAMETER_TAG
 | 
			
		||||
        template <typename ... Args>
 | 
			
		||||
        struct compute_parameter_tag_from_args_list;
 | 
			
		||||
 | 
			
		||||
        template <>
 | 
			
		||||
        struct compute_parameter_tag_from_args_list<>
 | 
			
		||||
        {
 | 
			
		||||
            static const int value = 0;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <typename Arg, typename ... Args>
 | 
			
		||||
        struct compute_parameter_tag_from_args_list<Arg, Args...>
 | 
			
		||||
        {
 | 
			
		||||
            static const int sub_value = 
 | 
			
		||||
                compute_parameter_tag_from_args_list<Args...>::value;
 | 
			
		||||
            static const int value = 
 | 
			
		||||
                parameter_tag<typename std::decay<Arg>::type>::value
 | 
			
		||||
                ? sub_value* 6 + parameter_tag<typename std::decay<Arg>::type>::value
 | 
			
		||||
                : sub_value;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        static inline bool is_parameter_tag_compatible(uint64_t a, uint64_t b)
 | 
			
		||||
        {
 | 
			
		||||
            if (a == 0)
 | 
			
		||||
                return b == 0;
 | 
			
		||||
            if (b == 0)
 | 
			
		||||
                return a == 0;
 | 
			
		||||
            int sa = a%6;
 | 
			
		||||
            int sb = a%6;
 | 
			
		||||
            if (sa == 5) sa = 4;
 | 
			
		||||
            if (sb == 5) sb = 4;
 | 
			
		||||
            if (sa != sb)
 | 
			
		||||
                return false;
 | 
			
		||||
            return is_parameter_tag_compatible(a/6, b/6);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        static inline unsigned find_closing_tag_runtime(const char* s, unsigned p)
 | 
			
		||||
        {
 | 
			
		||||
            return
 | 
			
		||||
                s[p] == 0
 | 
			
		||||
                ? throw std::runtime_error("unmatched tag <") :
 | 
			
		||||
                s[p] == '>'
 | 
			
		||||
                ? p : find_closing_tag_runtime(s, p + 1);
 | 
			
		||||
        }
 | 
			
		||||
        
 | 
			
		||||
        static inline uint64_t get_parameter_tag_runtime(const char* s, unsigned p = 0)
 | 
			
		||||
        {
 | 
			
		||||
            return
 | 
			
		||||
                s[p] == 0
 | 
			
		||||
                    ?  0 :
 | 
			
		||||
                s[p] == '<' ? (
 | 
			
		||||
                    std::strncmp(s+p, "<int>", 5) == 0
 | 
			
		||||
                        ? get_parameter_tag_runtime(s, find_closing_tag_runtime(s, p)) * 6 + 1 :
 | 
			
		||||
                    std::strncmp(s+p, "<uint>", 6) == 0
 | 
			
		||||
                        ? get_parameter_tag_runtime(s, find_closing_tag_runtime(s, p)) * 6 + 2 :
 | 
			
		||||
                    (std::strncmp(s+p, "<float>", 7) == 0 ||
 | 
			
		||||
                    std::strncmp(s+p, "<double>", 8) == 0)
 | 
			
		||||
                        ? get_parameter_tag_runtime(s, find_closing_tag_runtime(s, p)) * 6 + 3 :
 | 
			
		||||
                    (std::strncmp(s+p, "<str>", 5) == 0 ||
 | 
			
		||||
                    std::strncmp(s+p, "<string>", 8) == 0)
 | 
			
		||||
                        ? get_parameter_tag_runtime(s, find_closing_tag_runtime(s, p)) * 6 + 4 :
 | 
			
		||||
                    std::strncmp(s+p, "<path>", 6) == 0
 | 
			
		||||
                        ? get_parameter_tag_runtime(s, find_closing_tag_runtime(s, p)) * 6 + 5 :
 | 
			
		||||
                    throw std::runtime_error("invalid parameter type")
 | 
			
		||||
                    ) :
 | 
			
		||||
                get_parameter_tag_runtime(s, p+1);
 | 
			
		||||
        }
 | 
			
		||||
#ifndef CROW_MSVC_WORKAROUND
 | 
			
		||||
        constexpr uint64_t get_parameter_tag(const_str s, unsigned p = 0)
 | 
			
		||||
        {
 | 
			
		||||
            return
 | 
			
		||||
                p == s.size() 
 | 
			
		||||
                    ?  0 :
 | 
			
		||||
                s[p] == '<' ? ( 
 | 
			
		||||
                    is_int(s, p)
 | 
			
		||||
                        ? get_parameter_tag(s, find_closing_tag(s, p)) * 6 + 1 :
 | 
			
		||||
                    is_uint(s, p)
 | 
			
		||||
                        ? get_parameter_tag(s, find_closing_tag(s, p)) * 6 + 2 :
 | 
			
		||||
                    is_float(s, p)
 | 
			
		||||
                        ? get_parameter_tag(s, find_closing_tag(s, p)) * 6 + 3 :
 | 
			
		||||
                    is_str(s, p)
 | 
			
		||||
                        ? get_parameter_tag(s, find_closing_tag(s, p)) * 6 + 4 :
 | 
			
		||||
                    is_path(s, p)
 | 
			
		||||
                        ? get_parameter_tag(s, find_closing_tag(s, p)) * 6 + 5 :
 | 
			
		||||
                    throw std::runtime_error("invalid parameter type")
 | 
			
		||||
                    ) : 
 | 
			
		||||
                get_parameter_tag(s, p+1);
 | 
			
		||||
        }
 | 
			
		||||
#endif
 | 
			
		||||
 | 
			
		||||
        template <typename ... T>
 | 
			
		||||
        struct S
 | 
			
		||||
        {
 | 
			
		||||
            template <typename U>
 | 
			
		||||
            using push = S<U, T...>;
 | 
			
		||||
            template <typename U>
 | 
			
		||||
            using push_back = S<T..., U>;
 | 
			
		||||
            template <template<typename ... Args> class U>
 | 
			
		||||
            using rebind = U<T...>;
 | 
			
		||||
        };
 | 
			
		||||
template <typename F, typename Set>
 | 
			
		||||
        struct CallHelper;
 | 
			
		||||
        template <typename F, typename ...Args>
 | 
			
		||||
        struct CallHelper<F, S<Args...>>
 | 
			
		||||
        {
 | 
			
		||||
            template <typename F1, typename ...Args1, typename = 
 | 
			
		||||
                decltype(std::declval<F1>()(std::declval<Args1>()...))
 | 
			
		||||
                >
 | 
			
		||||
            static char __test(int);
 | 
			
		||||
 | 
			
		||||
            template <typename ...>
 | 
			
		||||
            static int __test(...);
 | 
			
		||||
 | 
			
		||||
            static constexpr bool value = sizeof(__test<F, Args...>(0)) == sizeof(char);
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        template <int N>
 | 
			
		||||
        struct single_tag_to_type
 | 
			
		||||
        {
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <>
 | 
			
		||||
        struct single_tag_to_type<1>
 | 
			
		||||
        {
 | 
			
		||||
            using type = int64_t;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <>
 | 
			
		||||
        struct single_tag_to_type<2>
 | 
			
		||||
        {
 | 
			
		||||
            using type = uint64_t;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <>
 | 
			
		||||
        struct single_tag_to_type<3>
 | 
			
		||||
        {
 | 
			
		||||
            using type = double;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <>
 | 
			
		||||
        struct single_tag_to_type<4>
 | 
			
		||||
        {
 | 
			
		||||
            using type = std::string;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <>
 | 
			
		||||
        struct single_tag_to_type<5>
 | 
			
		||||
        {
 | 
			
		||||
            using type = std::string;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        template <uint64_t Tag> 
 | 
			
		||||
        struct arguments
 | 
			
		||||
        {
 | 
			
		||||
            using subarguments = typename arguments<Tag/6>::type;
 | 
			
		||||
            using type = 
 | 
			
		||||
                typename subarguments::template push<typename single_tag_to_type<Tag%6>::type>;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <> 
 | 
			
		||||
        struct arguments<0>
 | 
			
		||||
        {
 | 
			
		||||
            using type = S<>;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <typename ... T>
 | 
			
		||||
        struct last_element_type
 | 
			
		||||
        {
 | 
			
		||||
            using type = typename std::tuple_element<sizeof...(T)-1, std::tuple<T...>>::type;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        template <>
 | 
			
		||||
        struct last_element_type<>
 | 
			
		||||
        {
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        // from http://stackoverflow.com/questions/13072359/c11-compile-time-array-with-logarithmic-evaluation-depth
 | 
			
		||||
        template<class T> using Invoke = typename T::type;
 | 
			
		||||
 | 
			
		||||
        template<unsigned...> struct seq{ using type = seq; };
 | 
			
		||||
 | 
			
		||||
        template<class S1, class S2> struct concat;
 | 
			
		||||
 | 
			
		||||
        template<unsigned... I1, unsigned... I2>
 | 
			
		||||
        struct concat<seq<I1...>, seq<I2...>>
 | 
			
		||||
          : seq<I1..., (sizeof...(I1)+I2)...>{};
 | 
			
		||||
 | 
			
		||||
        template<class S1, class S2>
 | 
			
		||||
        using Concat = Invoke<concat<S1, S2>>;
 | 
			
		||||
 | 
			
		||||
        template<unsigned N> struct gen_seq;
 | 
			
		||||
        template<unsigned N> using GenSeq = Invoke<gen_seq<N>>;
 | 
			
		||||
 | 
			
		||||
        template<unsigned N>
 | 
			
		||||
        struct gen_seq : Concat<GenSeq<N/2>, GenSeq<N - N/2>>{};
 | 
			
		||||
 | 
			
		||||
        template<> struct gen_seq<0> : seq<>{};
 | 
			
		||||
        template<> struct gen_seq<1> : seq<0>{};
 | 
			
		||||
 | 
			
		||||
        template <typename Seq, typename Tuple> 
 | 
			
		||||
        struct pop_back_helper;
 | 
			
		||||
 | 
			
		||||
        template <unsigned ... N, typename Tuple>
 | 
			
		||||
        struct pop_back_helper<seq<N...>, Tuple>
 | 
			
		||||
        {
 | 
			
		||||
            template <template <typename ... Args> class U>
 | 
			
		||||
            using rebind = U<typename std::tuple_element<N, Tuple>::type...>;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <typename ... T>
 | 
			
		||||
        struct pop_back //: public pop_back_helper<typename gen_seq<sizeof...(T)-1>::type, std::tuple<T...>>
 | 
			
		||||
        {
 | 
			
		||||
            template <template <typename ... Args> class U>
 | 
			
		||||
            using rebind = typename pop_back_helper<typename gen_seq<sizeof...(T)-1>::type, std::tuple<T...>>::template rebind<U>;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <>
 | 
			
		||||
        struct pop_back<>
 | 
			
		||||
        {
 | 
			
		||||
            template <template <typename ... Args> class U>
 | 
			
		||||
            using rebind = U<>;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        // from http://stackoverflow.com/questions/2118541/check-if-c0x-parameter-pack-contains-a-type
 | 
			
		||||
        template < typename Tp, typename... List >
 | 
			
		||||
        struct contains : std::true_type {};
 | 
			
		||||
 | 
			
		||||
        template < typename Tp, typename Head, typename... Rest >
 | 
			
		||||
        struct contains<Tp, Head, Rest...>
 | 
			
		||||
        : std::conditional< std::is_same<Tp, Head>::value,
 | 
			
		||||
            std::true_type,
 | 
			
		||||
            contains<Tp, Rest...>
 | 
			
		||||
        >::type {};
 | 
			
		||||
 | 
			
		||||
        template < typename Tp >
 | 
			
		||||
        struct contains<Tp> : std::false_type {};
 | 
			
		||||
 | 
			
		||||
        template <typename T>
 | 
			
		||||
        struct empty_context
 | 
			
		||||
        {
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <typename T>
 | 
			
		||||
        struct promote
 | 
			
		||||
        {
 | 
			
		||||
            using type = T;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
#define CROW_INTERNAL_PROMOTE_TYPE(t1, t2) \
 | 
			
		||||
        template<> \
 | 
			
		||||
        struct promote<t1> \
 | 
			
		||||
        {  \
 | 
			
		||||
            using type = t2; \
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        CROW_INTERNAL_PROMOTE_TYPE(char, int64_t);
 | 
			
		||||
        CROW_INTERNAL_PROMOTE_TYPE(short, int64_t);
 | 
			
		||||
        CROW_INTERNAL_PROMOTE_TYPE(int, int64_t);
 | 
			
		||||
        CROW_INTERNAL_PROMOTE_TYPE(long, int64_t);
 | 
			
		||||
        CROW_INTERNAL_PROMOTE_TYPE(long long, int64_t);
 | 
			
		||||
        CROW_INTERNAL_PROMOTE_TYPE(unsigned char, uint64_t);
 | 
			
		||||
        CROW_INTERNAL_PROMOTE_TYPE(unsigned short, uint64_t);
 | 
			
		||||
        CROW_INTERNAL_PROMOTE_TYPE(unsigned int, uint64_t);
 | 
			
		||||
        CROW_INTERNAL_PROMOTE_TYPE(unsigned long, uint64_t);
 | 
			
		||||
        CROW_INTERNAL_PROMOTE_TYPE(unsigned long long, uint64_t);
 | 
			
		||||
        CROW_INTERNAL_PROMOTE_TYPE(float, double);
 | 
			
		||||
#undef CROW_INTERNAL_PROMOTE_TYPE
 | 
			
		||||
 | 
			
		||||
        template <typename T>
 | 
			
		||||
        using promote_t = typename promote<T>::type;
 | 
			
		||||
 | 
			
		||||
    } // namespace black_magic
 | 
			
		||||
 | 
			
		||||
    namespace detail
 | 
			
		||||
    {
 | 
			
		||||
 | 
			
		||||
        template <class T, std::size_t N, class... Args>
 | 
			
		||||
        struct get_index_of_element_from_tuple_by_type_impl
 | 
			
		||||
        {
 | 
			
		||||
            static constexpr auto value = N;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <class T, std::size_t N, class... Args>
 | 
			
		||||
        struct get_index_of_element_from_tuple_by_type_impl<T, N, T, Args...>
 | 
			
		||||
        {
 | 
			
		||||
            static constexpr auto value = N;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template <class T, std::size_t N, class U, class... Args>
 | 
			
		||||
        struct get_index_of_element_from_tuple_by_type_impl<T, N, U, Args...>
 | 
			
		||||
        {
 | 
			
		||||
            static constexpr auto value = get_index_of_element_from_tuple_by_type_impl<T, N + 1, Args...>::value;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
    } // namespace detail
 | 
			
		||||
 | 
			
		||||
    namespace utility
 | 
			
		||||
    {
 | 
			
		||||
        template <class T, class... Args>
 | 
			
		||||
        T& get_element_by_type(std::tuple<Args...>& t)
 | 
			
		||||
        {
 | 
			
		||||
            return std::get<detail::get_index_of_element_from_tuple_by_type_impl<T, 0, Args...>::value>(t);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        template<typename T> 
 | 
			
		||||
        struct function_traits;  
 | 
			
		||||
 | 
			
		||||
#ifndef CROW_MSVC_WORKAROUND
 | 
			
		||||
        template<typename T> 
 | 
			
		||||
        struct function_traits : public function_traits<decltype(&T::operator())>
 | 
			
		||||
        {
 | 
			
		||||
            using parent_t = function_traits<decltype(&T::operator())>;
 | 
			
		||||
            static const size_t arity = parent_t::arity;
 | 
			
		||||
            using result_type = typename parent_t::result_type;
 | 
			
		||||
            template <size_t i>
 | 
			
		||||
            using arg = typename parent_t::template arg<i>;
 | 
			
		||||
        
 | 
			
		||||
        };  
 | 
			
		||||
#endif
 | 
			
		||||
 | 
			
		||||
        template<typename ClassType, typename R, typename ...Args> 
 | 
			
		||||
        struct function_traits<R(ClassType::*)(Args...) const>
 | 
			
		||||
        {
 | 
			
		||||
            static const size_t arity = sizeof...(Args);
 | 
			
		||||
 | 
			
		||||
            typedef R result_type;
 | 
			
		||||
 | 
			
		||||
            template <size_t i>
 | 
			
		||||
            using arg = typename std::tuple_element<i, std::tuple<Args...>>::type;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template<typename ClassType, typename R, typename ...Args> 
 | 
			
		||||
        struct function_traits<R(ClassType::*)(Args...)>
 | 
			
		||||
        {
 | 
			
		||||
            static const size_t arity = sizeof...(Args);
 | 
			
		||||
 | 
			
		||||
            typedef R result_type;
 | 
			
		||||
 | 
			
		||||
            template <size_t i>
 | 
			
		||||
            using arg = typename std::tuple_element<i, std::tuple<Args...>>::type;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        template<typename R, typename ...Args> 
 | 
			
		||||
        struct function_traits<std::function<R(Args...)>>
 | 
			
		||||
        {
 | 
			
		||||
            static const size_t arity = sizeof...(Args);
 | 
			
		||||
 | 
			
		||||
            typedef R result_type;
 | 
			
		||||
 | 
			
		||||
            template <size_t i>
 | 
			
		||||
            using arg = typename std::tuple_element<i, std::tuple<Args...>>::type;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
    } // namespace utility
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										35
									
								
								ext/dateparser.cpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										35
									
								
								ext/dateparser.cpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,35 @@
 | 
			
		|||
//
 | 
			
		||||
// Created by marcin on 22/11/15.
 | 
			
		||||
//
 | 
			
		||||
 | 
			
		||||
#include "dateparser.h"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
dateparser::dateparser(std::string fmt)
 | 
			
		||||
{
 | 
			
		||||
    // set format
 | 
			
		||||
    using namespace boost::local_time;
 | 
			
		||||
    local_time_input_facet* input_facet = new local_time_input_facet();
 | 
			
		||||
    input_facet->format(fmt.c_str());
 | 
			
		||||
    ss.imbue(std::locale(ss.getloc(), input_facet));
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
bool
 | 
			
		||||
dateparser::operator()(std::string const& text)
 | 
			
		||||
{
 | 
			
		||||
    ss.clear();
 | 
			
		||||
    ss.str(text);
 | 
			
		||||
 | 
			
		||||
    bool ok = bool(ss >> pt);
 | 
			
		||||
 | 
			
		||||
    if (ok)
 | 
			
		||||
    {
 | 
			
		||||
        auto tm = to_tm(pt);
 | 
			
		||||
        year    = tm.tm_year;
 | 
			
		||||
        month   = tm.tm_mon + 1; // for 1-based (1:jan, .. 12:dec)
 | 
			
		||||
        day     = tm.tm_mday;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    return ok;
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										27
									
								
								ext/dateparser.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										27
									
								
								ext/dateparser.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,27 @@
 | 
			
		|||
//
 | 
			
		||||
// Created by marcin on 22/11/15.
 | 
			
		||||
//
 | 
			
		||||
 | 
			
		||||
#ifndef XMR2CSV_DATEPARSER_H
 | 
			
		||||
#define XMR2CSV_DATEPARSER_H
 | 
			
		||||
 | 
			
		||||
#include <iostream>
 | 
			
		||||
#include <boost/date_time/local_time/local_time.hpp>
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
// taken from: http://stackoverflow.com/a/19482908/248823
 | 
			
		||||
struct dateparser
 | 
			
		||||
{
 | 
			
		||||
    boost::posix_time::ptime pt;
 | 
			
		||||
    unsigned year, month, day;
 | 
			
		||||
 | 
			
		||||
    dateparser(std::string fmt);
 | 
			
		||||
 | 
			
		||||
    bool
 | 
			
		||||
    operator()(std::string const& text);
 | 
			
		||||
 | 
			
		||||
private:
 | 
			
		||||
    std::stringstream ss;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
#endif //XMR2CSV_DATEPARSER_H
 | 
			
		||||
							
								
								
									
										1378
									
								
								ext/format.cc
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										1378
									
								
								ext/format.cc
									
										
									
									
									
										Normal file
									
								
							
										
											
												File diff suppressed because it is too large
												Load diff
											
										
									
								
							
							
								
								
									
										3157
									
								
								ext/format.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										3157
									
								
								ext/format.h
									
										
									
									
									
										Normal file
									
								
							
										
											
												File diff suppressed because it is too large
												Load diff
											
										
									
								
							
							
								
								
									
										762
									
								
								ext/minicsv.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										762
									
								
								ext/minicsv.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,762 @@
 | 
			
		|||
// The MIT License (MIT)
 | 
			
		||||
// Minimalistic CSV Streams 1.6
 | 
			
		||||
// Copyright (C) 2014, by Wong Shao Voon (shaovoon@yahoo.com)
 | 
			
		||||
//
 | 
			
		||||
// http://opensource.org/licenses/MIT
 | 
			
		||||
//
 | 
			
		||||
// version 1.2   : make use of make_shared
 | 
			
		||||
// version 1.3   : fixed: to when reading the last line and it does not have linefeed
 | 
			
		||||
//                 added: skip_1st_line and skip_line functions to ifstream class
 | 
			
		||||
// version 1.4   : Removed the use of smart ptr.
 | 
			
		||||
// version 1.5   : Performance increase on writing without flushing every line.
 | 
			
		||||
// version 1.6   : Add string streams
 | 
			
		||||
// version 1.7   : You MUST specify the escape/unescape string when calling set_delimiter. Option to surround/trim string with quotes
 | 
			
		||||
// version 1.7.1 : Add stream operator overload usage in example.cpp
 | 
			
		||||
//                 Disable the surround/trim quote on text by default
 | 
			
		||||
// version 1.7.2 : Stream operator overload for const char*
 | 
			
		||||
// version 1.7.3 : Add num_of_delimiter method to ifstream and istringstream
 | 
			
		||||
//                 Fix g++ compilation errors
 | 
			
		||||
// version 1.7.4 : Add get_rest_of_line
 | 
			
		||||
// version 1.7.5 : Add terminate_on_blank_line variable. Set to false if your file format has blank lines in between.
 | 
			
		||||
// version 1.7.6 : Ignore delimiters within quotes during reading when enable_trim_quote_on_str is true;
 | 
			
		||||
// version 1.7.7 : Fixed multiple symbol linkage errors
 | 
			
		||||
 | 
			
		||||
//#define USE_BOOST_LEXICAL_CAST
 | 
			
		||||
 | 
			
		||||
#ifndef MiniCSV_H
 | 
			
		||||
	#define MiniCSV_H
 | 
			
		||||
 | 
			
		||||
#include <string>
 | 
			
		||||
#include <sstream>
 | 
			
		||||
#include <fstream>
 | 
			
		||||
 | 
			
		||||
#ifdef USE_BOOST_LEXICAL_CAST
 | 
			
		||||
#	include <boost/lexical_cast.hpp>
 | 
			
		||||
#endif
 | 
			
		||||
 | 
			
		||||
#define NEWLINE '\n'
 | 
			
		||||
 | 
			
		||||
namespace csv
 | 
			
		||||
{
 | 
			
		||||
	inline std::string const & replace(std::string & src, std::string const & to_find, std::string const & to_replace)
 | 
			
		||||
	{
 | 
			
		||||
		size_t pos = 0;
 | 
			
		||||
		while (std::string::npos != pos)
 | 
			
		||||
		{
 | 
			
		||||
			pos = src.find(to_find, pos);
 | 
			
		||||
 | 
			
		||||
			if (std::string::npos != pos)
 | 
			
		||||
			{
 | 
			
		||||
				src.erase(pos, to_find.size());
 | 
			
		||||
				src.insert(pos, to_replace);
 | 
			
		||||
				pos += to_replace.size();
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		return src;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	inline std::string trim_right(const std::string& str, const std::string& trimChars)
 | 
			
		||||
	{
 | 
			
		||||
		std::string result = "";
 | 
			
		||||
		size_t endpos = str.find_last_not_of(trimChars);
 | 
			
		||||
		if (std::string::npos != endpos)
 | 
			
		||||
		{
 | 
			
		||||
			result = str.substr(0, endpos + 1);
 | 
			
		||||
		}
 | 
			
		||||
		else
 | 
			
		||||
			result = str;
 | 
			
		||||
 | 
			
		||||
		return result;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	inline std::string trim_left(const std::string& str, const std::string& trimChars)
 | 
			
		||||
	{
 | 
			
		||||
		std::string result = "";
 | 
			
		||||
 | 
			
		||||
		size_t startpos = str.find_first_not_of(trimChars);
 | 
			
		||||
		if (std::string::npos != startpos)
 | 
			
		||||
		{
 | 
			
		||||
			result = str.substr(startpos);
 | 
			
		||||
		}
 | 
			
		||||
		else
 | 
			
		||||
			result = str;
 | 
			
		||||
 | 
			
		||||
		return result;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	inline std::string trim(const std::string& str, const std::string& trimChars)
 | 
			
		||||
	{
 | 
			
		||||
		return trim_left(trim_right(str, trimChars), trimChars);
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	class ifstream
 | 
			
		||||
	{
 | 
			
		||||
	public:
 | 
			
		||||
		ifstream() : str(""), pos(0), delimiter(","), unescape_str("##"), trim_quote_on_str(false), trim_quote('\"'), terminate_on_blank_line(true)
 | 
			
		||||
		{
 | 
			
		||||
		}
 | 
			
		||||
		ifstream(const char * file)
 | 
			
		||||
		{
 | 
			
		||||
			open(file);
 | 
			
		||||
		}
 | 
			
		||||
		void open(const char * file)
 | 
			
		||||
		{
 | 
			
		||||
			init();
 | 
			
		||||
			istm.open(file, std::ios_base::in);
 | 
			
		||||
		}
 | 
			
		||||
		void init()
 | 
			
		||||
		{
 | 
			
		||||
			str = "";
 | 
			
		||||
			pos = 0;
 | 
			
		||||
			delimiter = ',';
 | 
			
		||||
			unescape_str = "##";
 | 
			
		||||
			trim_quote_on_str = false;
 | 
			
		||||
			trim_quote = '\"';
 | 
			
		||||
			terminate_on_blank_line = true;
 | 
			
		||||
		}
 | 
			
		||||
		void close()
 | 
			
		||||
		{
 | 
			
		||||
			istm.close();
 | 
			
		||||
		}
 | 
			
		||||
		bool is_open()
 | 
			
		||||
		{
 | 
			
		||||
			return istm.is_open();
 | 
			
		||||
		}
 | 
			
		||||
		void enable_trim_quote_on_str(bool enable, char quote)
 | 
			
		||||
		{
 | 
			
		||||
			trim_quote_on_str = enable;
 | 
			
		||||
			trim_quote = quote;
 | 
			
		||||
		}
 | 
			
		||||
		// eof is replaced by read_line
 | 
			
		||||
		//bool eof() const
 | 
			
		||||
		void set_delimiter(char delimiter_, std::string const & unescape_str_)
 | 
			
		||||
		{
 | 
			
		||||
			delimiter = delimiter_;
 | 
			
		||||
			unescape_str = unescape_str_;
 | 
			
		||||
		}
 | 
			
		||||
		std::string const &  get_delimiter() const
 | 
			
		||||
		{
 | 
			
		||||
			return delimiter;
 | 
			
		||||
		}
 | 
			
		||||
		std::string const &  get_unescape_str() const
 | 
			
		||||
		{
 | 
			
		||||
			return unescape_str;
 | 
			
		||||
		}
 | 
			
		||||
		void skip_line()
 | 
			
		||||
		{
 | 
			
		||||
			if(!istm.eof())
 | 
			
		||||
			{
 | 
			
		||||
				std::getline(istm, str);
 | 
			
		||||
				pos = 0;
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		bool read_line()
 | 
			
		||||
		{
 | 
			
		||||
			this->str = "";
 | 
			
		||||
			while(!istm.eof())
 | 
			
		||||
			{
 | 
			
		||||
				std::getline(istm, this->str);
 | 
			
		||||
				pos = 0;
 | 
			
		||||
 | 
			
		||||
				if (this->str.empty())
 | 
			
		||||
				{
 | 
			
		||||
					if (terminate_on_blank_line)
 | 
			
		||||
						break;
 | 
			
		||||
					else
 | 
			
		||||
						continue;
 | 
			
		||||
				}
 | 
			
		||||
 | 
			
		||||
				return true;
 | 
			
		||||
			}
 | 
			
		||||
			return false;
 | 
			
		||||
		}
 | 
			
		||||
		std::string get_delimited_str()
 | 
			
		||||
		{
 | 
			
		||||
			std::string str = "";
 | 
			
		||||
			char ch = '\0';
 | 
			
		||||
			bool within_quote = false;
 | 
			
		||||
			do
 | 
			
		||||
			{
 | 
			
		||||
				if(pos>=this->str.size())
 | 
			
		||||
				{
 | 
			
		||||
					this->str = "";
 | 
			
		||||
 | 
			
		||||
					return unescape(str);
 | 
			
		||||
				}
 | 
			
		||||
 | 
			
		||||
				ch = this->str[pos];
 | 
			
		||||
				if (trim_quote_on_str)
 | 
			
		||||
				{
 | 
			
		||||
					if (within_quote == false && ch == trim_quote && ((pos > 0 && this->str[pos - 1] == delimiter[0]) || pos == 0))
 | 
			
		||||
						within_quote = true;
 | 
			
		||||
					else if (within_quote && ch == trim_quote)
 | 
			
		||||
						within_quote = false;
 | 
			
		||||
				}
 | 
			
		||||
 | 
			
		||||
				++(pos);
 | 
			
		||||
 | 
			
		||||
				if (ch == delimiter[0] && within_quote == false)
 | 
			
		||||
					break;
 | 
			
		||||
				if (ch == '\r' || ch == '\n')
 | 
			
		||||
					break;
 | 
			
		||||
 | 
			
		||||
				str += ch;
 | 
			
		||||
			}
 | 
			
		||||
			while(true);
 | 
			
		||||
 | 
			
		||||
			return unescape(str);
 | 
			
		||||
		}
 | 
			
		||||
		std::string unescape(std::string & src)
 | 
			
		||||
		{
 | 
			
		||||
			src = unescape_str.empty() ? src : replace(src, unescape_str, delimiter);
 | 
			
		||||
			return trim_quote_on_str ? trim(src, std::string(1, trim_quote)) : src;
 | 
			
		||||
		}
 | 
			
		||||
		size_t num_of_delimiter() const
 | 
			
		||||
		{
 | 
			
		||||
			if (delimiter.size() == 0)
 | 
			
		||||
				return 0;
 | 
			
		||||
 | 
			
		||||
			size_t cnt = 0;
 | 
			
		||||
			for (size_t i = 0; i < str.size(); ++i)
 | 
			
		||||
			{
 | 
			
		||||
				if (str[i] == delimiter[0])
 | 
			
		||||
					++cnt;
 | 
			
		||||
			}
 | 
			
		||||
			return cnt;
 | 
			
		||||
		}
 | 
			
		||||
		std::string get_rest_of_line() const
 | 
			
		||||
		{
 | 
			
		||||
			return str.substr(pos);
 | 
			
		||||
		}
 | 
			
		||||
		const std::string& get_line() const
 | 
			
		||||
		{
 | 
			
		||||
			return str;
 | 
			
		||||
		}
 | 
			
		||||
		void enable_terminate_on_blank_line(bool enable)
 | 
			
		||||
		{
 | 
			
		||||
			terminate_on_blank_line = enable;
 | 
			
		||||
		}
 | 
			
		||||
		bool is_terminate_on_blank_line() const
 | 
			
		||||
		{
 | 
			
		||||
			return terminate_on_blank_line;
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
	private:
 | 
			
		||||
		std::ifstream istm;
 | 
			
		||||
		std::string str;
 | 
			
		||||
		size_t pos;
 | 
			
		||||
		std::string delimiter;
 | 
			
		||||
		std::string unescape_str;
 | 
			
		||||
		bool trim_quote_on_str;
 | 
			
		||||
		char trim_quote;
 | 
			
		||||
		bool terminate_on_blank_line;
 | 
			
		||||
	};
 | 
			
		||||
 | 
			
		||||
	class ofstream
 | 
			
		||||
	{
 | 
			
		||||
	public:
 | 
			
		||||
 | 
			
		||||
		ofstream() : after_newline(true), delimiter(","), escape_str("##"), surround_quote_on_str(false), surround_quote('\"')
 | 
			
		||||
		{
 | 
			
		||||
		}
 | 
			
		||||
		ofstream(const char * file)
 | 
			
		||||
		{
 | 
			
		||||
			open(file);
 | 
			
		||||
		}
 | 
			
		||||
		void open(const char * file)
 | 
			
		||||
		{
 | 
			
		||||
			init();
 | 
			
		||||
			ostm.open(file, std::ios_base::out);
 | 
			
		||||
		}
 | 
			
		||||
		void init()
 | 
			
		||||
		{
 | 
			
		||||
			after_newline = true; 
 | 
			
		||||
			delimiter = ',';
 | 
			
		||||
			escape_str = "##";
 | 
			
		||||
			surround_quote_on_str = false;
 | 
			
		||||
			surround_quote = '\"';
 | 
			
		||||
		}
 | 
			
		||||
		void flush()
 | 
			
		||||
		{
 | 
			
		||||
			ostm.flush();
 | 
			
		||||
		}
 | 
			
		||||
		void close()
 | 
			
		||||
		{
 | 
			
		||||
			ostm.close();
 | 
			
		||||
		}
 | 
			
		||||
		bool is_open()
 | 
			
		||||
		{
 | 
			
		||||
			return ostm.is_open();
 | 
			
		||||
		}
 | 
			
		||||
		void enable_surround_quote_on_str(bool enable, char quote)
 | 
			
		||||
		{
 | 
			
		||||
			surround_quote_on_str = enable;
 | 
			
		||||
			surround_quote = quote;
 | 
			
		||||
		}
 | 
			
		||||
		void set_delimiter(char delimiter_, std::string const & escape_str_)
 | 
			
		||||
		{
 | 
			
		||||
			delimiter = delimiter_;
 | 
			
		||||
			escape_str = escape_str_;
 | 
			
		||||
		}
 | 
			
		||||
		std::string const &  get_delimiter() const
 | 
			
		||||
		{
 | 
			
		||||
			return delimiter;
 | 
			
		||||
		}
 | 
			
		||||
		std::string const &  get_escape_str() const
 | 
			
		||||
		{
 | 
			
		||||
			return escape_str;
 | 
			
		||||
		}
 | 
			
		||||
		void set_after_newline(bool after_newline_)
 | 
			
		||||
		{
 | 
			
		||||
			after_newline = after_newline_;
 | 
			
		||||
		}
 | 
			
		||||
		bool get_after_newline() const
 | 
			
		||||
		{
 | 
			
		||||
			return after_newline;
 | 
			
		||||
		}
 | 
			
		||||
		std::ofstream& get_ofstream()
 | 
			
		||||
		{
 | 
			
		||||
			return ostm;
 | 
			
		||||
		}
 | 
			
		||||
		void escape_and_output(std::string src)
 | 
			
		||||
		{
 | 
			
		||||
			ostm << ((escape_str.empty()) ? src : replace(src, delimiter, escape_str));
 | 
			
		||||
		}
 | 
			
		||||
		void escape_str_and_output(std::string src)
 | 
			
		||||
		{
 | 
			
		||||
			src = ((escape_str.empty()) ? src : replace(src, delimiter, escape_str));
 | 
			
		||||
			if (surround_quote_on_str)
 | 
			
		||||
			{
 | 
			
		||||
				ostm << surround_quote << src << surround_quote;
 | 
			
		||||
			}
 | 
			
		||||
			else
 | 
			
		||||
			{
 | 
			
		||||
				ostm << src;
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	private:
 | 
			
		||||
		std::ofstream ostm;
 | 
			
		||||
		bool after_newline;
 | 
			
		||||
		std::string delimiter;
 | 
			
		||||
		std::string escape_str;
 | 
			
		||||
		bool surround_quote_on_str;
 | 
			
		||||
		char surround_quote;
 | 
			
		||||
	};
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
} // ns csv
 | 
			
		||||
 | 
			
		||||
template<typename T>
 | 
			
		||||
csv::ifstream& operator >> (csv::ifstream& istm, T& val)
 | 
			
		||||
{
 | 
			
		||||
	std::string str = istm.get_delimited_str();
 | 
			
		||||
	
 | 
			
		||||
#ifdef USE_BOOST_LEXICAL_CAST
 | 
			
		||||
	val = boost::lexical_cast<T>(str);
 | 
			
		||||
#else
 | 
			
		||||
	std::istringstream is(str);
 | 
			
		||||
	is >> val;
 | 
			
		||||
#endif
 | 
			
		||||
 | 
			
		||||
	return istm;
 | 
			
		||||
}
 | 
			
		||||
template<>
 | 
			
		||||
inline csv::ifstream& operator >> (csv::ifstream& istm, std::string& val)
 | 
			
		||||
{
 | 
			
		||||
	val = istm.get_delimited_str();
 | 
			
		||||
 | 
			
		||||
	return istm;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
template<typename T>
 | 
			
		||||
csv::ofstream& operator << (csv::ofstream& ostm, const T& val)
 | 
			
		||||
{
 | 
			
		||||
	if(!ostm.get_after_newline())
 | 
			
		||||
		ostm.get_ofstream() << ostm.get_delimiter();
 | 
			
		||||
 | 
			
		||||
	std::ostringstream os_temp;
 | 
			
		||||
 | 
			
		||||
	os_temp << val;
 | 
			
		||||
 | 
			
		||||
	ostm.escape_and_output(os_temp.str());
 | 
			
		||||
 | 
			
		||||
	ostm.set_after_newline(false);
 | 
			
		||||
 | 
			
		||||
	return ostm;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
template<typename T>
 | 
			
		||||
csv::ofstream& operator << (csv::ofstream& ostm, const T* val)
 | 
			
		||||
{
 | 
			
		||||
	if (!ostm.get_after_newline())
 | 
			
		||||
		ostm.get_ofstream() << ostm.get_delimiter();
 | 
			
		||||
 | 
			
		||||
	std::ostringstream os_temp;
 | 
			
		||||
 | 
			
		||||
	os_temp << *val;
 | 
			
		||||
 | 
			
		||||
	ostm.escape_and_output(os_temp.str());
 | 
			
		||||
 | 
			
		||||
	ostm.set_after_newline(false);
 | 
			
		||||
 | 
			
		||||
	return ostm;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
template<>
 | 
			
		||||
inline csv::ofstream& operator << (csv::ofstream& ostm, const std::string& val)
 | 
			
		||||
{
 | 
			
		||||
	if (!ostm.get_after_newline())
 | 
			
		||||
		ostm.get_ofstream() << ostm.get_delimiter();
 | 
			
		||||
 | 
			
		||||
	std::string temp = val;
 | 
			
		||||
	ostm.escape_str_and_output(temp);
 | 
			
		||||
 | 
			
		||||
	ostm.set_after_newline(false);
 | 
			
		||||
 | 
			
		||||
	return ostm;
 | 
			
		||||
}
 | 
			
		||||
template<>
 | 
			
		||||
inline csv::ofstream& operator << (csv::ofstream& ostm, const char& val)
 | 
			
		||||
{
 | 
			
		||||
	if(val==NEWLINE)
 | 
			
		||||
	{
 | 
			
		||||
		ostm.get_ofstream() << NEWLINE;
 | 
			
		||||
 | 
			
		||||
		ostm.set_after_newline(true);
 | 
			
		||||
	}
 | 
			
		||||
	else
 | 
			
		||||
	{
 | 
			
		||||
		std::ostringstream os_temp;
 | 
			
		||||
 | 
			
		||||
		os_temp << val;
 | 
			
		||||
 | 
			
		||||
		ostm.escape_and_output(os_temp.str());
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return ostm;
 | 
			
		||||
}
 | 
			
		||||
template<>
 | 
			
		||||
inline csv::ofstream& operator << (csv::ofstream& ostm, const char* val)
 | 
			
		||||
{
 | 
			
		||||
	const std::string temp = val;
 | 
			
		||||
 | 
			
		||||
	ostm << temp;
 | 
			
		||||
 | 
			
		||||
	return ostm;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
namespace csv
 | 
			
		||||
{
 | 
			
		||||
 | 
			
		||||
class istringstream
 | 
			
		||||
{
 | 
			
		||||
public:
 | 
			
		||||
	istringstream(const char * text)
 | 
			
		||||
		: str("")
 | 
			
		||||
		, pos(0)
 | 
			
		||||
		, delimiter(",")
 | 
			
		||||
		, unescape_str("##")
 | 
			
		||||
		, trim_quote_on_str(false)
 | 
			
		||||
		, trim_quote('\"')
 | 
			
		||||
		, terminate_on_blank_line(true)
 | 
			
		||||
	{
 | 
			
		||||
		istm.str(text);
 | 
			
		||||
	}
 | 
			
		||||
	void enable_trim_quote_on_str(bool enable, char quote)
 | 
			
		||||
	{
 | 
			
		||||
		trim_quote_on_str = enable;
 | 
			
		||||
		trim_quote = quote;
 | 
			
		||||
	}
 | 
			
		||||
	void set_delimiter(char delimiter_, std::string const & unescape_str_)
 | 
			
		||||
	{
 | 
			
		||||
		delimiter = delimiter_;
 | 
			
		||||
		unescape_str = unescape_str_;
 | 
			
		||||
	}
 | 
			
		||||
	std::string const & get_delimiter() const
 | 
			
		||||
	{
 | 
			
		||||
		return delimiter;
 | 
			
		||||
	}
 | 
			
		||||
	std::string const &  get_unescape_str() const
 | 
			
		||||
	{
 | 
			
		||||
		return unescape_str;
 | 
			
		||||
	}
 | 
			
		||||
	void skip_line()
 | 
			
		||||
	{
 | 
			
		||||
		std::getline(istm, str);
 | 
			
		||||
		pos = 0;
 | 
			
		||||
	}
 | 
			
		||||
	bool read_line()
 | 
			
		||||
	{
 | 
			
		||||
		this->str = "";
 | 
			
		||||
		while (!istm.eof())
 | 
			
		||||
		{
 | 
			
		||||
			std::getline(istm, this->str);
 | 
			
		||||
			pos = 0;
 | 
			
		||||
 | 
			
		||||
			if (this->str.empty())
 | 
			
		||||
			{
 | 
			
		||||
				if (terminate_on_blank_line)
 | 
			
		||||
					break;
 | 
			
		||||
				else
 | 
			
		||||
					continue;
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			return true;
 | 
			
		||||
		}
 | 
			
		||||
		return false;
 | 
			
		||||
		return false;
 | 
			
		||||
	}
 | 
			
		||||
	std::string get_delimited_str()
 | 
			
		||||
	{
 | 
			
		||||
		std::string str = "";
 | 
			
		||||
		char ch = '\0';
 | 
			
		||||
		bool within_quote = false;
 | 
			
		||||
		do
 | 
			
		||||
		{
 | 
			
		||||
			if (pos >= this->str.size())
 | 
			
		||||
			{
 | 
			
		||||
				this->str = "";
 | 
			
		||||
 | 
			
		||||
				return unescape(str);
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			ch = this->str[pos];
 | 
			
		||||
			if (trim_quote_on_str)
 | 
			
		||||
			{
 | 
			
		||||
				if (within_quote == false && ch == trim_quote && ((pos > 0 && this->str[pos - 1] == delimiter[0]) || pos == 0))
 | 
			
		||||
					within_quote = true;
 | 
			
		||||
				else if (within_quote && ch == trim_quote)
 | 
			
		||||
					within_quote = false;
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			++(pos);
 | 
			
		||||
 | 
			
		||||
			if (ch == delimiter[0] && within_quote == false)
 | 
			
		||||
				break;
 | 
			
		||||
			if (ch == '\r' || ch == '\n')
 | 
			
		||||
				break;
 | 
			
		||||
 | 
			
		||||
			str += ch;
 | 
			
		||||
		} while (true);
 | 
			
		||||
 | 
			
		||||
		return unescape(str);
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	std::string unescape(std::string & src)
 | 
			
		||||
	{
 | 
			
		||||
		src = unescape_str.empty() ? src : replace(src, unescape_str, delimiter);
 | 
			
		||||
		return trim_quote_on_str ? trim(src, std::string(1, trim_quote)) : src;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	size_t num_of_delimiter() const
 | 
			
		||||
	{
 | 
			
		||||
		if (delimiter.size() == 0)
 | 
			
		||||
			return 0;
 | 
			
		||||
 | 
			
		||||
		size_t cnt = 0;
 | 
			
		||||
		for (size_t i = 0; i < str.size(); ++i)
 | 
			
		||||
		{
 | 
			
		||||
			if (str[i] == delimiter[0])
 | 
			
		||||
				++cnt;
 | 
			
		||||
		}
 | 
			
		||||
		return cnt;
 | 
			
		||||
	}
 | 
			
		||||
	std::string get_rest_of_line() const
 | 
			
		||||
	{
 | 
			
		||||
		return str.substr(pos);
 | 
			
		||||
	}
 | 
			
		||||
	const std::string& get_line() const
 | 
			
		||||
	{
 | 
			
		||||
		return str;
 | 
			
		||||
	}
 | 
			
		||||
	void enable_terminate_on_blank_line(bool enable)
 | 
			
		||||
	{
 | 
			
		||||
		terminate_on_blank_line = enable;
 | 
			
		||||
	}
 | 
			
		||||
	bool is_terminate_on_blank_line() const
 | 
			
		||||
	{
 | 
			
		||||
		return terminate_on_blank_line;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
private:
 | 
			
		||||
	std::istringstream istm;
 | 
			
		||||
	std::string str;
 | 
			
		||||
	size_t pos;
 | 
			
		||||
	std::string delimiter;
 | 
			
		||||
	std::string unescape_str;
 | 
			
		||||
	bool trim_quote_on_str;
 | 
			
		||||
	char trim_quote;
 | 
			
		||||
	bool terminate_on_blank_line;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
class ostringstream
 | 
			
		||||
{
 | 
			
		||||
public:
 | 
			
		||||
 | 
			
		||||
	ostringstream() : after_newline(true), delimiter(","), escape_str("##"), surround_quote_on_str(false), surround_quote('\"')
 | 
			
		||||
	{
 | 
			
		||||
	}
 | 
			
		||||
	void enable_surround_quote_on_str(bool enable, char quote)
 | 
			
		||||
	{
 | 
			
		||||
		surround_quote_on_str = enable;
 | 
			
		||||
		surround_quote = quote;
 | 
			
		||||
	}
 | 
			
		||||
	void set_delimiter(char delimiter_, std::string const & escape_str_)
 | 
			
		||||
	{
 | 
			
		||||
		delimiter = delimiter_;
 | 
			
		||||
		escape_str = escape_str_;
 | 
			
		||||
	}
 | 
			
		||||
	std::string const & get_delimiter() const
 | 
			
		||||
	{
 | 
			
		||||
		return delimiter;
 | 
			
		||||
	}
 | 
			
		||||
	std::string const &  get_escape_str() const
 | 
			
		||||
	{
 | 
			
		||||
		return escape_str;
 | 
			
		||||
	}
 | 
			
		||||
	void set_after_newline(bool after_newline_)
 | 
			
		||||
	{
 | 
			
		||||
		after_newline = after_newline_;
 | 
			
		||||
	}
 | 
			
		||||
	bool get_after_newline() const
 | 
			
		||||
	{
 | 
			
		||||
		return after_newline;
 | 
			
		||||
	}
 | 
			
		||||
	std::ostringstream& get_ostringstream()
 | 
			
		||||
	{
 | 
			
		||||
		return ostm;
 | 
			
		||||
	}
 | 
			
		||||
	std::string get_text()
 | 
			
		||||
	{
 | 
			
		||||
		return ostm.str();
 | 
			
		||||
	}
 | 
			
		||||
	void escape_and_output(std::string src)
 | 
			
		||||
	{
 | 
			
		||||
		ostm << ((escape_str.empty()) ? src : replace(src, delimiter, escape_str));
 | 
			
		||||
	}
 | 
			
		||||
	void escape_str_and_output(std::string src)
 | 
			
		||||
	{
 | 
			
		||||
		src = ((escape_str.empty()) ? src : replace(src, delimiter, escape_str));
 | 
			
		||||
		if (surround_quote_on_str)
 | 
			
		||||
		{
 | 
			
		||||
			ostm << surround_quote << src << surround_quote;
 | 
			
		||||
		}
 | 
			
		||||
		else
 | 
			
		||||
		{
 | 
			
		||||
			ostm << src;
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
private:
 | 
			
		||||
	std::ostringstream ostm;
 | 
			
		||||
	bool after_newline;
 | 
			
		||||
	std::string delimiter;
 | 
			
		||||
	std::string escape_str;
 | 
			
		||||
	bool surround_quote_on_str;
 | 
			
		||||
	char surround_quote;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
} // ns csv
 | 
			
		||||
 | 
			
		||||
template<typename T>
 | 
			
		||||
csv::istringstream& operator >> (csv::istringstream& istm, T& val)
 | 
			
		||||
{
 | 
			
		||||
	std::string str = istm.get_delimited_str();
 | 
			
		||||
 | 
			
		||||
#ifdef USE_BOOST_LEXICAL_CAST
 | 
			
		||||
	val = boost::lexical_cast<T>(str);
 | 
			
		||||
#else
 | 
			
		||||
	std::istringstream is(str);
 | 
			
		||||
	is >> val;
 | 
			
		||||
#endif
 | 
			
		||||
 | 
			
		||||
	return istm;
 | 
			
		||||
}
 | 
			
		||||
template<>
 | 
			
		||||
inline csv::istringstream& operator >> (csv::istringstream& istm, std::string& val)
 | 
			
		||||
{
 | 
			
		||||
	val = istm.get_delimited_str();
 | 
			
		||||
 | 
			
		||||
	return istm;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
template<typename T>
 | 
			
		||||
csv::ostringstream& operator << (csv::ostringstream& ostm, const T& val)
 | 
			
		||||
{
 | 
			
		||||
	if (!ostm.get_after_newline())
 | 
			
		||||
		ostm.get_ostringstream() << ostm.get_delimiter();
 | 
			
		||||
 | 
			
		||||
	std::ostringstream os_temp;
 | 
			
		||||
 | 
			
		||||
	os_temp << val;
 | 
			
		||||
 | 
			
		||||
	ostm.escape_and_output(os_temp.str());
 | 
			
		||||
 | 
			
		||||
	ostm.set_after_newline(false);
 | 
			
		||||
 | 
			
		||||
	return ostm;
 | 
			
		||||
}
 | 
			
		||||
template<typename T>
 | 
			
		||||
csv::ostringstream& operator << (csv::ostringstream& ostm, const T* val)
 | 
			
		||||
{
 | 
			
		||||
	if (!ostm.get_after_newline())
 | 
			
		||||
		ostm.get_ostringstream() << ostm.get_delimiter();
 | 
			
		||||
 | 
			
		||||
	std::ostringstream os_temp;
 | 
			
		||||
 | 
			
		||||
	os_temp << val;
 | 
			
		||||
 | 
			
		||||
	ostm.escape_and_output(os_temp.str());
 | 
			
		||||
 | 
			
		||||
	ostm.set_after_newline(false);
 | 
			
		||||
 | 
			
		||||
	return ostm;
 | 
			
		||||
}
 | 
			
		||||
template<>
 | 
			
		||||
inline csv::ostringstream& operator << (csv::ostringstream& ostm, const std::string& val)
 | 
			
		||||
{
 | 
			
		||||
	if (!ostm.get_after_newline())
 | 
			
		||||
		ostm.get_ostringstream() << ostm.get_delimiter();
 | 
			
		||||
 | 
			
		||||
	std::string temp = val;
 | 
			
		||||
	ostm.escape_str_and_output(temp);
 | 
			
		||||
 | 
			
		||||
	ostm.set_after_newline(false);
 | 
			
		||||
 | 
			
		||||
	return ostm;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
template<>
 | 
			
		||||
inline csv::ostringstream& operator << (csv::ostringstream& ostm, const char& val)
 | 
			
		||||
{
 | 
			
		||||
	if (val == NEWLINE)
 | 
			
		||||
	{
 | 
			
		||||
		ostm.get_ostringstream() << NEWLINE;
 | 
			
		||||
 | 
			
		||||
		ostm.set_after_newline(true);
 | 
			
		||||
	}
 | 
			
		||||
	else
 | 
			
		||||
	{
 | 
			
		||||
		std::ostringstream os_temp;
 | 
			
		||||
 | 
			
		||||
		os_temp << val;
 | 
			
		||||
 | 
			
		||||
		ostm.escape_and_output(os_temp.str());
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return ostm;
 | 
			
		||||
}
 | 
			
		||||
template<>
 | 
			
		||||
inline csv::ostringstream& operator << (csv::ostringstream& ostm, const char* val)
 | 
			
		||||
{
 | 
			
		||||
	const std::string temp = val;
 | 
			
		||||
 | 
			
		||||
	ostm << temp;
 | 
			
		||||
 | 
			
		||||
	return ostm;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
#endif // MiniCSV_H
 | 
			
		||||
							
								
								
									
										17
									
								
								ext/mstch/CMakeLists.txt
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								ext/mstch/CMakeLists.txt
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,17 @@
 | 
			
		|||
cmake_minimum_required(VERSION 3.0.2)
 | 
			
		||||
project(mstch)
 | 
			
		||||
 | 
			
		||||
option(WITH_UNIT_TESTS "enable building unit test executable" OFF)
 | 
			
		||||
option(WITH_BENCHMARK "enable building benchmark executable" OFF)
 | 
			
		||||
 | 
			
		||||
set(CMAKE_INCLUDE_CURRENT_DIR ON)
 | 
			
		||||
set(CMAKE_INCLUDE_CURRENT_DIR_IN_INTERFACE ON)
 | 
			
		||||
set(CMAKE_BUILD_TYPE Debug)
 | 
			
		||||
 | 
			
		||||
set(mstch_VERSION 1.0.1)
 | 
			
		||||
 | 
			
		||||
if(NOT MSVC)
 | 
			
		||||
  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -Wall -Wextra -O3")
 | 
			
		||||
endif()
 | 
			
		||||
 | 
			
		||||
add_subdirectory(src)
 | 
			
		||||
							
								
								
									
										22
									
								
								ext/mstch/LICENSE
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								ext/mstch/LICENSE
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,22 @@
 | 
			
		|||
The MIT License (MIT)
 | 
			
		||||
 | 
			
		||||
Copyright (c) 2015 Daniel Sipka
 | 
			
		||||
 | 
			
		||||
Permission is hereby granted, free of charge, to any person obtaining a copy
 | 
			
		||||
of this software and associated documentation files (the "Software"), to deal
 | 
			
		||||
in the Software without restriction, including without limitation the rights
 | 
			
		||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 | 
			
		||||
copies of the Software, and to permit persons to whom the Software is
 | 
			
		||||
furnished to do so, subject to the following conditions:
 | 
			
		||||
 | 
			
		||||
The above copyright notice and this permission notice shall be included in all
 | 
			
		||||
copies or substantial portions of the Software.
 | 
			
		||||
 | 
			
		||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 | 
			
		||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 | 
			
		||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 | 
			
		||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 | 
			
		||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 | 
			
		||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 | 
			
		||||
SOFTWARE.
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										281
									
								
								ext/mstch/README.md
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										281
									
								
								ext/mstch/README.md
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,281 @@
 | 
			
		|||
# mstch - {{mustache}} templates in C++11 
 | 
			
		||||
 | 
			
		||||

 | 
			
		||||
 | 
			
		||||
mstch is a complete implementation of [{{mustache}}](http://mustache.github.io/) 
 | 
			
		||||
templates using modern C++. It's compliant with [specifications](https://github.com/mustache/spec)
 | 
			
		||||
v1.1.3, including the lambda module.
 | 
			
		||||
 | 
			
		||||
[](http://melpon.org/wandbox/permlink/EqyOe7IBRYPGVk5f)
 | 
			
		||||
[](http://badge.fury.io/gh/no1msd%2Fmstch)
 | 
			
		||||
[](https://travis-ci.org/no1msd/mstch)
 | 
			
		||||
[](https://ci.appveyor.com/project/no1msd/mstch)
 | 
			
		||||
 | 
			
		||||
## Supported features
 | 
			
		||||
 | 
			
		||||
mstch supports the complete feature set described in the `mustache(5)` [manpage](http://mustache.github.com/mustache.5.html):
 | 
			
		||||
 | 
			
		||||
 - JSON-like data structure using [Boost.Variant](http://www.boost.org/libs/variant)
 | 
			
		||||
 - variables, sections, inverted sections
 | 
			
		||||
 - partials
 | 
			
		||||
 - changing the delimiter
 | 
			
		||||
 - C++11 lambdas
 | 
			
		||||
 - C++ objects as view models
 | 
			
		||||
 | 
			
		||||
## Basic usage
 | 
			
		||||
 | 
			
		||||
```c++
 | 
			
		||||
#include <iostream>
 | 
			
		||||
#include <mstch/mstch.hpp>
 | 
			
		||||
 | 
			
		||||
int main() {
 | 
			
		||||
  std::string view{"{{#names}}Hi {{name}}!\n{{/names}}"};
 | 
			
		||||
  mstch::map context{
 | 
			
		||||
    {"names", mstch::array{
 | 
			
		||||
      mstch::map{{"name", std::string{"Chris"}}},
 | 
			
		||||
      mstch::map{{"name", std::string{"Mark"}}},
 | 
			
		||||
      mstch::map{{"name", std::string{"Scott"}}},
 | 
			
		||||
    }}
 | 
			
		||||
  };
 | 
			
		||||
  
 | 
			
		||||
  std::cout << mstch::render(view, context) << std::endl;
 | 
			
		||||
  
 | 
			
		||||
  return 0;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
The output of this example will be:
 | 
			
		||||
 | 
			
		||||
```html
 | 
			
		||||
Hi Chris!
 | 
			
		||||
Hi Mark!
 | 
			
		||||
Hi Scott!
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
### Data structure
 | 
			
		||||
 | 
			
		||||
The types in the example above, `mstch::array` and `mstch::map` are  actually 
 | 
			
		||||
aliases for standard types:
 | 
			
		||||
 | 
			
		||||
```c++
 | 
			
		||||
using map = std::map<const std::string, node>;
 | 
			
		||||
using array = std::vector<node>;
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
`mstch::node` is a `boost::variant` that can hold a `std::string`, `int`, 
 | 
			
		||||
`double`, `bool`, `mstch::lambda` or a `std::shared_ptr<mstch::object>` 
 | 
			
		||||
(see below), also a map or an array recursively. Essentially it works just like 
 | 
			
		||||
a JSON object.
 | 
			
		||||
 | 
			
		||||
Note that when using a `std::string` as value you must explicitly specify the 
 | 
			
		||||
type, since a `const char*` literal like `"foobar"` would be implicitly 
 | 
			
		||||
converted to `bool`. Alternatively you can use [C++14 string_literals](http://en.cppreference.com/w/cpp/string/basic_string/operator%22%22s)
 | 
			
		||||
if your compiler supports it.
 | 
			
		||||
 | 
			
		||||
## Advanced usage
 | 
			
		||||
 | 
			
		||||
### Partials
 | 
			
		||||
 | 
			
		||||
Partials can be passed in a `std::map` as the third parameter of the 
 | 
			
		||||
`mstch::render` function:
 | 
			
		||||
 | 
			
		||||
```c++
 | 
			
		||||
std::string view{"{{#names}}{{> user}}{{/names}}"};
 | 
			
		||||
std::string user_view{"<strong>{{name}}\n</strong>"};
 | 
			
		||||
mstch::map context{
 | 
			
		||||
  {"names", mstch::array{
 | 
			
		||||
    mstch::map{{"name", std::string{"Chris"}}},
 | 
			
		||||
    mstch::map{{"name", std::string{"Mark"}}},
 | 
			
		||||
    mstch::map{{"name", std::string{"Scott"}}},
 | 
			
		||||
  }}
 | 
			
		||||
};
 | 
			
		||||
  
 | 
			
		||||
std::cout << mstch::render(view, context, {{"user", user_view}}) << std::endl;
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
Output:
 | 
			
		||||
 | 
			
		||||
```html
 | 
			
		||||
<strong>Chris</strong>
 | 
			
		||||
<strong>Mark</strong>
 | 
			
		||||
<strong>Scott</strong>
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
### Lambdas
 | 
			
		||||
 | 
			
		||||
C++11 lambda expressions can be used to add logic to your templates. Like a
 | 
			
		||||
`const char*` literal, lambdas can be implicitly converted to `bool`, so they
 | 
			
		||||
must be wrapped in a `mstch::lambda` object when used in a `mstch::node`. The 
 | 
			
		||||
lambda expression passed to `mstch::lambda` must itself return a `mstch::node`.
 | 
			
		||||
The returned node will be rendered to a string, then it will be parsed as a
 | 
			
		||||
template.
 | 
			
		||||
 | 
			
		||||
The lambda expression accepts either no parameters:
 | 
			
		||||
 | 
			
		||||
```c++
 | 
			
		||||
std::string view{"Hello {{lambda}}!"};
 | 
			
		||||
mstch::map context{
 | 
			
		||||
  {"lambda", mstch::lambda{[]() -> mstch::node {
 | 
			
		||||
    return std::string{"World"};
 | 
			
		||||
  }}}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
std::cout << mstch::render(view, context) << std::endl;
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
Output:
 | 
			
		||||
 | 
			
		||||
```html
 | 
			
		||||
Hello World!
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
Or it accepts a `const std::string&` that gets the unrendered literal block:
 | 
			
		||||
 | 
			
		||||
```c++
 | 
			
		||||
std::string view{"{{#bold}}{{yay}} :){{/bold}}"};
 | 
			
		||||
mstch::map context{
 | 
			
		||||
  {"yay", std::string{"Yay!"}},
 | 
			
		||||
  {"bold", mstch::lambda{[](const std::string& text) -> mstch::node {
 | 
			
		||||
    return "<b>" + text + "</b>";
 | 
			
		||||
  }}}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
std::cout << mstch::render(view, context) << std::endl;
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
Output:
 | 
			
		||||
 | 
			
		||||
```html
 | 
			
		||||
<b>Yay! :)</b>
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
### Objects
 | 
			
		||||
 | 
			
		||||
Custom objects can also be used as context for rendering templates. The class 
 | 
			
		||||
must inherit from `mstch::object`, and register it's exported methods with
 | 
			
		||||
`register_methods`. Exported methods must have the return type of `mstch::node`.
 | 
			
		||||
Objects must be created as a `std::shared_ptr`.
 | 
			
		||||
 | 
			
		||||
```c++
 | 
			
		||||
class example: public mstch::object {
 | 
			
		||||
 public:
 | 
			
		||||
  example(): m_value(1) {
 | 
			
		||||
    register_methods(this, {
 | 
			
		||||
      {"count", &example::count},
 | 
			
		||||
      {"names", &example::names}  
 | 
			
		||||
    });
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  mstch::node count() {
 | 
			
		||||
    return m_value++;
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  mstch::node names() {
 | 
			
		||||
    return mstch::array{
 | 
			
		||||
      std::string{"Chris"}, std::string{"Mark"}, std::string{"Scott"}};
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
 private:
 | 
			
		||||
  int m_value;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
std::string view{"{{#names}}<b>{{count}}</b>: {{.}}\n{{/names}}"};
 | 
			
		||||
const auto context = std::make_shared<example>();
 | 
			
		||||
 | 
			
		||||
std::cout << mstch::render(view, context) << std::endl;
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
Output:
 | 
			
		||||
 | 
			
		||||
```html
 | 
			
		||||
<b>1</b>: Chris
 | 
			
		||||
<b>2</b>: Mark
 | 
			
		||||
<b>3</b>: Scott
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
### Custom escape function
 | 
			
		||||
 | 
			
		||||
By default, mstch uses HTML escaping on the output, as per specification. This
 | 
			
		||||
is not useful if your output is not HTML, so mstch provides a way to supply
 | 
			
		||||
your own escape implementation. Just assign any callable object to the static
 | 
			
		||||
`mstch::config::escape`, which is an initially empty
 | 
			
		||||
`std::function<std::string(const std::string&)>`. 
 | 
			
		||||
 | 
			
		||||
For example you can turn off escaping entirely with a lambda:
 | 
			
		||||
 | 
			
		||||
```c++
 | 
			
		||||
mstch::config::escape = [](const std::string& str) -> std::string {
 | 
			
		||||
  return str;
 | 
			
		||||
};
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
## Requirements
 | 
			
		||||
 | 
			
		||||
 - A C++ compiler with decent C++11 support. Currently tested with:
 | 
			
		||||
   - GCC 4.7, 4.8, 4.9, 5.1
 | 
			
		||||
   - clang 3.5, 3.6, 3.7 (both libstdc++ and libc++ are supported)
 | 
			
		||||
   - MSVC 2013, 2015
 | 
			
		||||
 - Boost 1.54+ for [Boost.Variant](http://www.boost.org/libs/variant)
 | 
			
		||||
 - CMake 3.0+ for building
 | 
			
		||||
 | 
			
		||||
## Using mstch in your project
 | 
			
		||||
 | 
			
		||||
If you are using CMake, the easiest way to include mstch in your project is to 
 | 
			
		||||
copy the whole directory to your source tree, and use `add_subdirectory` in your 
 | 
			
		||||
CMakeLists.txt. This will set a variable named `mstch_INCLUDE_DIR` that contains 
 | 
			
		||||
its include path, and add a static library target named `mstch`. For example:
 | 
			
		||||
 | 
			
		||||
```cmake
 | 
			
		||||
add_subdirectory(external/mstch)
 | 
			
		||||
include_directories(${mstch_INCLUDE_DIR})
 | 
			
		||||
target_link_libraries(your_project mstch)
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
If you prefer to install the library globally, you can simply do the following 
 | 
			
		||||
from the root of the source tree:
 | 
			
		||||
 | 
			
		||||
```bash
 | 
			
		||||
 $ mkdir build
 | 
			
		||||
 $ cd build
 | 
			
		||||
 $ cmake ..
 | 
			
		||||
 $ make
 | 
			
		||||
 $ make install
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
The install command may require root privileges. This will also install CMake 
 | 
			
		||||
config files, so you can use use `find_package` in your CMakeLists.txt:
 | 
			
		||||
 | 
			
		||||
```cmake
 | 
			
		||||
find_package(mstch)
 | 
			
		||||
target_link_libraries(your_project mstch::mstch)
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
## Running the unit tests
 | 
			
		||||
 | 
			
		||||
Unit tests are using the [Catch](https://github.com/philsquared/Catch) framework
 | 
			
		||||
and [rapidjson](http://rapidjson.org/) to parse the
 | 
			
		||||
[Mustache specifications](https://github.com/mustache/spec), all of which are 
 | 
			
		||||
included in the repository as git submodules. Various 
 | 
			
		||||
[Boost](http://www.boost.org/) libraries are also required to build them.
 | 
			
		||||
 | 
			
		||||
Don't forget to initialize submodules:
 | 
			
		||||
 | 
			
		||||
```bash
 | 
			
		||||
 $ git submodule init
 | 
			
		||||
 $ git submodule update
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
To build and run the unit tests:
 | 
			
		||||
 | 
			
		||||
```bash
 | 
			
		||||
 $ mkdir build
 | 
			
		||||
 $ cd build
 | 
			
		||||
 $ cmake -DWITH_UNIT_TESTS=ON ..
 | 
			
		||||
 $ make
 | 
			
		||||
 $ make test
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
## License
 | 
			
		||||
 | 
			
		||||
mstch is licensed under the [MIT license](https://github.com/no1msd/mstch/blob/master/LICENSE).
 | 
			
		||||
							
								
								
									
										1
									
								
								ext/mstch/cmake/mstch-config.cmake
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								ext/mstch/cmake/mstch-config.cmake
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1 @@
 | 
			
		|||
include("${CMAKE_CURRENT_LIST_DIR}/mstch-targets.cmake")
 | 
			
		||||
							
								
								
									
										113
									
								
								ext/mstch/include/mstch/mstch.hpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										113
									
								
								ext/mstch/include/mstch/mstch.hpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,113 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <vector>
 | 
			
		||||
#include <map>
 | 
			
		||||
#include <string>
 | 
			
		||||
#include <memory>
 | 
			
		||||
#include <functional>
 | 
			
		||||
 | 
			
		||||
#include <boost/variant.hpp>
 | 
			
		||||
 | 
			
		||||
namespace mstch {
 | 
			
		||||
 | 
			
		||||
struct config {
 | 
			
		||||
  static std::function<std::string(const std::string&)> escape;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
namespace internal {
 | 
			
		||||
 | 
			
		||||
template<class N>
 | 
			
		||||
class object_t {
 | 
			
		||||
 public:
 | 
			
		||||
  const N& at(const std::string& name) const {
 | 
			
		||||
    cache[name] = (methods.at(name))();
 | 
			
		||||
    return cache[name];
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  bool has(const std::string name) const {
 | 
			
		||||
    return methods.count(name) != 0;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
 protected:
 | 
			
		||||
  template<class S>
 | 
			
		||||
  void register_methods(S* s, std::map<std::string,N(S::*)()> methods) {
 | 
			
		||||
    for(auto& item: methods)
 | 
			
		||||
      this->methods.insert({item.first, std::bind(item.second, s)});
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
 private:
 | 
			
		||||
  std::map<std::string, std::function<N()>> methods;
 | 
			
		||||
  mutable std::map<std::string, N> cache;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
template<class T, class N>
 | 
			
		||||
class is_fun {
 | 
			
		||||
 private:
 | 
			
		||||
  using not_fun = char;
 | 
			
		||||
  using fun_without_args = char[2];
 | 
			
		||||
  using fun_with_args = char[3];
 | 
			
		||||
  template <typename U, U> struct really_has;
 | 
			
		||||
  template <typename C> static fun_without_args& test(
 | 
			
		||||
      really_has<N(C::*)() const, &C::operator()>*);
 | 
			
		||||
  template <typename C> static fun_with_args& test(
 | 
			
		||||
      really_has<N(C::*)(const std::string&) const,
 | 
			
		||||
      &C::operator()>*);
 | 
			
		||||
  template <typename> static not_fun& test(...);
 | 
			
		||||
 | 
			
		||||
 public:
 | 
			
		||||
  static bool const no_args = sizeof(test<T>(0)) == sizeof(fun_without_args);
 | 
			
		||||
  static bool const has_args = sizeof(test<T>(0)) == sizeof(fun_with_args);
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
template<class N>
 | 
			
		||||
using node_renderer = std::function<std::string(const N& n)>;
 | 
			
		||||
 | 
			
		||||
template<class N>
 | 
			
		||||
class lambda_t {
 | 
			
		||||
 public:
 | 
			
		||||
  template<class F>
 | 
			
		||||
  lambda_t(F f, typename std::enable_if<is_fun<F, N>::no_args>::type* = 0):
 | 
			
		||||
      fun([f](node_renderer<N> renderer, const std::string&) {
 | 
			
		||||
        return renderer(f());
 | 
			
		||||
      })
 | 
			
		||||
  {
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  template<class F>
 | 
			
		||||
  lambda_t(F f, typename std::enable_if<is_fun<F, N>::has_args>::type* = 0):
 | 
			
		||||
      fun([f](node_renderer<N> renderer, const std::string& text) {
 | 
			
		||||
        return renderer(f(text));
 | 
			
		||||
      })
 | 
			
		||||
  {
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  std::string operator()(node_renderer<N> renderer,
 | 
			
		||||
      const std::string& text = "") const
 | 
			
		||||
  {
 | 
			
		||||
    return fun(renderer, text);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
 private:
 | 
			
		||||
  std::function<std::string(node_renderer<N> renderer, const std::string&)> fun;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
using node = boost::make_recursive_variant<
 | 
			
		||||
    std::nullptr_t, std::string, int, double, bool,
 | 
			
		||||
    internal::lambda_t<boost::recursive_variant_>,
 | 
			
		||||
    std::shared_ptr<internal::object_t<boost::recursive_variant_>>,
 | 
			
		||||
    std::map<const std::string, boost::recursive_variant_>,
 | 
			
		||||
    std::vector<boost::recursive_variant_>>::type;
 | 
			
		||||
using object = internal::object_t<node>;
 | 
			
		||||
using lambda = internal::lambda_t<node>;
 | 
			
		||||
using map = std::map<const std::string, node>;
 | 
			
		||||
using array = std::vector<node>;
 | 
			
		||||
 | 
			
		||||
std::string render(
 | 
			
		||||
    const std::string& tmplt,
 | 
			
		||||
    const node& root,
 | 
			
		||||
    const std::map<std::string,std::string>& partials =
 | 
			
		||||
        std::map<std::string,std::string>());
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										66
									
								
								ext/mstch/src/CMakeLists.txt
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										66
									
								
								ext/mstch/src/CMakeLists.txt
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,66 @@
 | 
			
		|||
find_package(Boost 1.54 REQUIRED)
 | 
			
		||||
 | 
			
		||||
set(mstch_INCLUDE_DIR
 | 
			
		||||
      ${PROJECT_SOURCE_DIR}/include CACHE STRING "mstch include directory")
 | 
			
		||||
 | 
			
		||||
# /home/mwo/crow-monero-test/ext/mstch
 | 
			
		||||
message(${PROJECT_SOURCE_DIR})
 | 
			
		||||
#
 | 
			
		||||
include_directories(
 | 
			
		||||
    ${Boost_INCLUDE_DIR})
 | 
			
		||||
 | 
			
		||||
set(SRC
 | 
			
		||||
    state/in_section.cpp
 | 
			
		||||
    state/outside_section.cpp
 | 
			
		||||
    state/render_state.hpp
 | 
			
		||||
    visitor/get_token.hpp
 | 
			
		||||
    visitor/has_token.hpp
 | 
			
		||||
    visitor/is_node_empty.hpp
 | 
			
		||||
    visitor/render_node.hpp
 | 
			
		||||
    visitor/render_section.hpp
 | 
			
		||||
    mstch.cpp
 | 
			
		||||
    render_context.cpp
 | 
			
		||||
    template_type.cpp
 | 
			
		||||
    token.cpp
 | 
			
		||||
    utils.cpp)
 | 
			
		||||
 | 
			
		||||
add_library(mstch STATIC ${SRC})
 | 
			
		||||
#
 | 
			
		||||
set_property(TARGET mstch PROPERTY VERSION ${mstch_VERSION})
 | 
			
		||||
#
 | 
			
		||||
#install(
 | 
			
		||||
#    TARGETS mstch EXPORT mstchTargets
 | 
			
		||||
#    LIBRARY DESTINATION lib
 | 
			
		||||
#    ARCHIVE DESTINATION lib)
 | 
			
		||||
#
 | 
			
		||||
#install(
 | 
			
		||||
#    FILES "${PROJECT_SOURCE_DIR}/include/mstch/mstch.hpp"
 | 
			
		||||
#    DESTINATION include/mstch
 | 
			
		||||
#    COMPONENT Devel)
 | 
			
		||||
#
 | 
			
		||||
#include(CMakePackageConfigHelpers)
 | 
			
		||||
#write_basic_package_version_file(
 | 
			
		||||
#    "${CMAKE_CURRENT_BINARY_DIR}/mstch/mstch-config-version.cmake"
 | 
			
		||||
#    VERSION ${mstch_VERSION}
 | 
			
		||||
#    COMPATIBILITY AnyNewerVersion)
 | 
			
		||||
#
 | 
			
		||||
#export(
 | 
			
		||||
#    EXPORT mstchTargets
 | 
			
		||||
#    FILE "${CMAKE_CURRENT_BINARY_DIR}/mstch/mstch-targets.cmake"
 | 
			
		||||
#    NAMESPACE mstch::)
 | 
			
		||||
#
 | 
			
		||||
#configure_file(
 | 
			
		||||
#    "${PROJECT_SOURCE_DIR}/cmake/mstch-config.cmake"
 | 
			
		||||
#    "${CMAKE_CURRENT_BINARY_DIR}/mstch/mstch-config.cmake")
 | 
			
		||||
#
 | 
			
		||||
#install(
 | 
			
		||||
#    EXPORT mstchTargets
 | 
			
		||||
#    FILE mstch-targets.cmake
 | 
			
		||||
#    NAMESPACE mstch::
 | 
			
		||||
#    DESTINATION lib/cmake/mstch)
 | 
			
		||||
#
 | 
			
		||||
#install(FILES
 | 
			
		||||
#    "${PROJECT_SOURCE_DIR}/cmake/mstch-config.cmake"
 | 
			
		||||
#    "${CMAKE_CURRENT_BINARY_DIR}/mstch/mstch-config-version.cmake"
 | 
			
		||||
#    DESTINATION lib/cmake/mstch
 | 
			
		||||
#    COMPONENT Devel)
 | 
			
		||||
							
								
								
									
										20
									
								
								ext/mstch/src/mstch.cpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										20
									
								
								ext/mstch/src/mstch.cpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,20 @@
 | 
			
		|||
#include <iostream>
 | 
			
		||||
 | 
			
		||||
#include "mstch/mstch.hpp"
 | 
			
		||||
#include "render_context.hpp"
 | 
			
		||||
 | 
			
		||||
using namespace mstch;
 | 
			
		||||
 | 
			
		||||
std::function<std::string(const std::string&)> mstch::config::escape;
 | 
			
		||||
 | 
			
		||||
std::string mstch::render(
 | 
			
		||||
    const std::string& tmplt,
 | 
			
		||||
    const node& root,
 | 
			
		||||
    const std::map<std::string,std::string>& partials)
 | 
			
		||||
{
 | 
			
		||||
  std::map<std::string, template_type> partial_templates;
 | 
			
		||||
  for (auto& partial: partials)
 | 
			
		||||
    partial_templates.insert({partial.first, {partial.second}});
 | 
			
		||||
 | 
			
		||||
  return render_context(root, partial_templates).render(tmplt);
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										72
									
								
								ext/mstch/src/render_context.cpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										72
									
								
								ext/mstch/src/render_context.cpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,72 @@
 | 
			
		|||
#include "render_context.hpp"
 | 
			
		||||
#include "state/outside_section.hpp"
 | 
			
		||||
#include "visitor/get_token.hpp"
 | 
			
		||||
 | 
			
		||||
using namespace mstch;
 | 
			
		||||
 | 
			
		||||
const mstch::node render_context::null_node;
 | 
			
		||||
 | 
			
		||||
render_context::push::push(render_context& context, const mstch::node& node):
 | 
			
		||||
    m_context(context)
 | 
			
		||||
{
 | 
			
		||||
  context.m_nodes.emplace_front(node);
 | 
			
		||||
  context.m_node_ptrs.emplace_front(&node);
 | 
			
		||||
  context.m_state.push(std::unique_ptr<render_state>(new outside_section));
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
render_context::push::~push() {
 | 
			
		||||
  m_context.m_nodes.pop_front();
 | 
			
		||||
  m_context.m_node_ptrs.pop_front();
 | 
			
		||||
  m_context.m_state.pop();
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
std::string render_context::push::render(const template_type& templt) {
 | 
			
		||||
  return m_context.render(templt);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
render_context::render_context(
 | 
			
		||||
    const mstch::node& node,
 | 
			
		||||
    const std::map<std::string, template_type>& partials):
 | 
			
		||||
    m_partials(partials), m_nodes(1, node), m_node_ptrs(1, &node)
 | 
			
		||||
{
 | 
			
		||||
  m_state.push(std::unique_ptr<render_state>(new outside_section));
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
const mstch::node& render_context::find_node(
 | 
			
		||||
    const std::string& token,
 | 
			
		||||
    std::list<node const*> current_nodes)
 | 
			
		||||
{
 | 
			
		||||
  if (token != "." && token.find('.') != std::string::npos)
 | 
			
		||||
    return find_node(token.substr(token.rfind('.') + 1),
 | 
			
		||||
        {&find_node(token.substr(0, token.rfind('.')), current_nodes)});
 | 
			
		||||
  else
 | 
			
		||||
    for (auto& node: current_nodes)
 | 
			
		||||
      if (visit(has_token(token), *node))
 | 
			
		||||
        return visit(get_token(token, *node), *node);
 | 
			
		||||
  return null_node;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
const mstch::node& render_context::get_node(const std::string& token) {
 | 
			
		||||
    return find_node(token, m_node_ptrs);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
std::string render_context::render(
 | 
			
		||||
    const template_type& templt, const std::string& prefix)
 | 
			
		||||
{
 | 
			
		||||
  std::string output;
 | 
			
		||||
  bool prev_eol = true;
 | 
			
		||||
  for (auto& token: templt) {
 | 
			
		||||
    if (prev_eol && prefix.length() != 0)
 | 
			
		||||
      output += m_state.top()->render(*this, {prefix});
 | 
			
		||||
    output += m_state.top()->render(*this, token);
 | 
			
		||||
    prev_eol = token.eol();
 | 
			
		||||
  }
 | 
			
		||||
  return output;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
std::string render_context::render_partial(
 | 
			
		||||
    const std::string& partial_name, const std::string& prefix)
 | 
			
		||||
{
 | 
			
		||||
  return m_partials.count(partial_name) ?
 | 
			
		||||
      render(m_partials.at(partial_name), prefix) : "";
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										51
									
								
								ext/mstch/src/render_context.hpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										51
									
								
								ext/mstch/src/render_context.hpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,51 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <deque>
 | 
			
		||||
#include <list>
 | 
			
		||||
#include <sstream>
 | 
			
		||||
#include <string>
 | 
			
		||||
#include <stack>
 | 
			
		||||
 | 
			
		||||
#include "mstch/mstch.hpp"
 | 
			
		||||
#include "state/render_state.hpp"
 | 
			
		||||
#include "template_type.hpp"
 | 
			
		||||
 | 
			
		||||
namespace mstch {
 | 
			
		||||
 | 
			
		||||
class render_context {
 | 
			
		||||
 public:
 | 
			
		||||
  class push {
 | 
			
		||||
   public:
 | 
			
		||||
    push(render_context& context, const mstch::node& node = {});
 | 
			
		||||
    ~push();
 | 
			
		||||
    std::string render(const template_type& templt);
 | 
			
		||||
   private:
 | 
			
		||||
    render_context& m_context;
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  render_context(
 | 
			
		||||
      const mstch::node& node,
 | 
			
		||||
      const std::map<std::string, template_type>& partials);
 | 
			
		||||
  const mstch::node& get_node(const std::string& token);
 | 
			
		||||
  std::string render(
 | 
			
		||||
      const template_type& templt, const std::string& prefix = "");
 | 
			
		||||
  std::string render_partial(
 | 
			
		||||
      const std::string& partial_name, const std::string& prefix);
 | 
			
		||||
  template<class T, class... Args>
 | 
			
		||||
  void set_state(Args&& ... args) {
 | 
			
		||||
    m_state.top() = std::unique_ptr<render_state>(
 | 
			
		||||
        new T(std::forward<Args>(args)...));
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
 private:
 | 
			
		||||
  static const mstch::node null_node;
 | 
			
		||||
  const mstch::node& find_node(
 | 
			
		||||
      const std::string& token,
 | 
			
		||||
      std::list<node const*> current_nodes);
 | 
			
		||||
  std::map<std::string, template_type> m_partials;
 | 
			
		||||
  std::deque<mstch::node> m_nodes;
 | 
			
		||||
  std::list<const mstch::node*> m_node_ptrs;
 | 
			
		||||
  std::stack<std::unique_ptr<render_state>> m_state;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										34
									
								
								ext/mstch/src/state/in_section.cpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										34
									
								
								ext/mstch/src/state/in_section.cpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,34 @@
 | 
			
		|||
#include "in_section.hpp"
 | 
			
		||||
#include "outside_section.hpp"
 | 
			
		||||
#include "visitor/is_node_empty.hpp"
 | 
			
		||||
#include "visitor/render_section.hpp"
 | 
			
		||||
 | 
			
		||||
using namespace mstch;
 | 
			
		||||
 | 
			
		||||
in_section::in_section(type type, const token& start_token):
 | 
			
		||||
    m_type(type), m_start_token(start_token), m_skipped_openings(0)
 | 
			
		||||
{
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
std::string in_section::render(render_context& ctx, const token& token) {
 | 
			
		||||
  if (token.token_type() == token::type::section_close)
 | 
			
		||||
    if (token.name() == m_start_token.name() && m_skipped_openings == 0) {
 | 
			
		||||
      auto& node = ctx.get_node(m_start_token.name());
 | 
			
		||||
      std::string out;
 | 
			
		||||
 | 
			
		||||
      if (m_type == type::normal && !visit(is_node_empty(), node))
 | 
			
		||||
        out = visit(render_section(ctx, m_section, m_start_token.delims()), node);
 | 
			
		||||
      else if (m_type == type::inverted && visit(is_node_empty(), node))
 | 
			
		||||
        out = render_context::push(ctx).render(m_section);
 | 
			
		||||
 | 
			
		||||
      ctx.set_state<outside_section>();
 | 
			
		||||
      return out;
 | 
			
		||||
    } else
 | 
			
		||||
      m_skipped_openings--;
 | 
			
		||||
  else if (token.token_type() == token::type::inverted_section_open ||
 | 
			
		||||
      token.token_type() == token::type::section_open)
 | 
			
		||||
    m_skipped_openings++;
 | 
			
		||||
 | 
			
		||||
  m_section << token;
 | 
			
		||||
  return "";
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										24
									
								
								ext/mstch/src/state/in_section.hpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										24
									
								
								ext/mstch/src/state/in_section.hpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,24 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <sstream>
 | 
			
		||||
#include <vector>
 | 
			
		||||
 | 
			
		||||
#include "render_state.hpp"
 | 
			
		||||
#include "template_type.hpp"
 | 
			
		||||
 | 
			
		||||
namespace mstch {
 | 
			
		||||
 | 
			
		||||
class in_section: public render_state {
 | 
			
		||||
 public:
 | 
			
		||||
  enum class type { inverted, normal };
 | 
			
		||||
  in_section(type type, const token& start_token);
 | 
			
		||||
  std::string render(render_context& context, const token& token) override;
 | 
			
		||||
 | 
			
		||||
 private:
 | 
			
		||||
  const type m_type;
 | 
			
		||||
  const token& m_start_token;
 | 
			
		||||
  template_type m_section;
 | 
			
		||||
  int m_skipped_openings;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										32
									
								
								ext/mstch/src/state/outside_section.cpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										32
									
								
								ext/mstch/src/state/outside_section.cpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,32 @@
 | 
			
		|||
#include "outside_section.hpp"
 | 
			
		||||
 | 
			
		||||
#include "visitor/render_node.hpp"
 | 
			
		||||
#include "in_section.hpp"
 | 
			
		||||
#include "render_context.hpp"
 | 
			
		||||
 | 
			
		||||
using namespace mstch;
 | 
			
		||||
 | 
			
		||||
std::string outside_section::render(
 | 
			
		||||
    render_context& ctx, const token& token)
 | 
			
		||||
{
 | 
			
		||||
  using flag = render_node::flag;
 | 
			
		||||
  switch (token.token_type()) {
 | 
			
		||||
    case token::type::section_open:
 | 
			
		||||
      ctx.set_state<in_section>(in_section::type::normal, token);
 | 
			
		||||
      break;
 | 
			
		||||
    case token::type::inverted_section_open:
 | 
			
		||||
      ctx.set_state<in_section>(in_section::type::inverted, token);
 | 
			
		||||
      break;
 | 
			
		||||
    case token::type::variable:
 | 
			
		||||
      return visit(render_node(ctx, flag::escape_html), ctx.get_node(token.name()));
 | 
			
		||||
    case token::type::unescaped_variable:
 | 
			
		||||
      return visit(render_node(ctx, flag::none), ctx.get_node(token.name()));
 | 
			
		||||
    case token::type::text:
 | 
			
		||||
      return token.raw();
 | 
			
		||||
    case token::type::partial:
 | 
			
		||||
      return ctx.render_partial(token.name(), token.partial_prefix());
 | 
			
		||||
    default:
 | 
			
		||||
      break;
 | 
			
		||||
  }
 | 
			
		||||
  return "";
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										12
									
								
								ext/mstch/src/state/outside_section.hpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								ext/mstch/src/state/outside_section.hpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,12 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include "render_state.hpp"
 | 
			
		||||
 | 
			
		||||
namespace mstch {
 | 
			
		||||
 | 
			
		||||
class outside_section: public render_state {
 | 
			
		||||
 public:
 | 
			
		||||
  std::string render(render_context& context, const token& token) override;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										17
									
								
								ext/mstch/src/state/render_state.hpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								ext/mstch/src/state/render_state.hpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,17 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <memory>
 | 
			
		||||
 | 
			
		||||
#include "token.hpp"
 | 
			
		||||
 | 
			
		||||
namespace mstch {
 | 
			
		||||
 | 
			
		||||
class render_context;
 | 
			
		||||
 | 
			
		||||
class render_state {
 | 
			
		||||
 public:
 | 
			
		||||
  virtual ~render_state() {}
 | 
			
		||||
  virtual std::string render(render_context& context, const token& token) = 0;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										104
									
								
								ext/mstch/src/template_type.cpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										104
									
								
								ext/mstch/src/template_type.cpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,104 @@
 | 
			
		|||
#include "template_type.hpp"
 | 
			
		||||
 | 
			
		||||
using namespace mstch;
 | 
			
		||||
 | 
			
		||||
template_type::template_type(const std::string& str, const delim_type& delims):
 | 
			
		||||
    m_open(delims.first), m_close(delims.second)
 | 
			
		||||
{
 | 
			
		||||
  tokenize(str);
 | 
			
		||||
  strip_whitespace();
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
template_type::template_type(const std::string& str):
 | 
			
		||||
    m_open("{{"), m_close("}}")
 | 
			
		||||
{
 | 
			
		||||
  tokenize(str);
 | 
			
		||||
  strip_whitespace();
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void template_type::process_text(citer begin, citer end) {
 | 
			
		||||
  if (begin == end)
 | 
			
		||||
    return;
 | 
			
		||||
  auto start = begin;
 | 
			
		||||
  for (auto it = begin; it != end; ++it)
 | 
			
		||||
    if (*it == '\n' || it == end - 1) {
 | 
			
		||||
      m_tokens.push_back({{start, it + 1}});
 | 
			
		||||
      start = it + 1;
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void template_type::tokenize(const std::string& tmp) {
 | 
			
		||||
  citer beg = tmp.begin();
 | 
			
		||||
  auto npos = std::string::npos;
 | 
			
		||||
 | 
			
		||||
  for (std::size_t cur_pos = 0; cur_pos < tmp.size();) {
 | 
			
		||||
    auto open_pos = tmp.find(m_open, cur_pos);
 | 
			
		||||
    auto close_pos = tmp.find(
 | 
			
		||||
        m_close, open_pos == npos ? open_pos : open_pos + 1);
 | 
			
		||||
 | 
			
		||||
    if (close_pos != npos && open_pos != npos) {
 | 
			
		||||
      if (*(beg + open_pos + m_open.size()) == '{' &&
 | 
			
		||||
          *(beg + close_pos + m_close.size()) == '}')
 | 
			
		||||
        ++close_pos;
 | 
			
		||||
 | 
			
		||||
      process_text(beg + cur_pos, beg + open_pos);
 | 
			
		||||
      cur_pos = close_pos + m_close.size();
 | 
			
		||||
      m_tokens.push_back({{beg + open_pos, beg + close_pos + m_close.size()},
 | 
			
		||||
          m_open.size(), m_close.size()});
 | 
			
		||||
 | 
			
		||||
      if (cur_pos == tmp.size()) {
 | 
			
		||||
        m_tokens.push_back({{""}});
 | 
			
		||||
        m_tokens.back().eol(true);
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      if (*(beg + open_pos + m_open.size()) == '=' &&
 | 
			
		||||
          *(beg + close_pos - 1) == '=')
 | 
			
		||||
      {
 | 
			
		||||
        auto tok_beg = beg + open_pos + m_open.size() + 1;
 | 
			
		||||
        auto tok_end = beg + close_pos - 1;
 | 
			
		||||
        auto front_skip = first_not_ws(tok_beg, tok_end);
 | 
			
		||||
        auto back_skip = first_not_ws(reverse(tok_end), reverse(tok_beg));
 | 
			
		||||
        m_open = {front_skip, beg + tmp.find(' ', front_skip - beg)};
 | 
			
		||||
        m_close = {beg + tmp.rfind(' ', back_skip - beg) + 1, back_skip + 1};
 | 
			
		||||
      }
 | 
			
		||||
    } else {
 | 
			
		||||
      process_text(beg + cur_pos, tmp.end());
 | 
			
		||||
      cur_pos = close_pos;
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void template_type::strip_whitespace() {
 | 
			
		||||
  auto line_begin = m_tokens.begin();
 | 
			
		||||
  bool has_tag = false, non_space = false;
 | 
			
		||||
 | 
			
		||||
  for (auto it = m_tokens.begin(); it != m_tokens.end(); ++it) {
 | 
			
		||||
    auto type = (*it).token_type();
 | 
			
		||||
    if (type != token::type::text && type != token::type::variable &&
 | 
			
		||||
        type != token::type::unescaped_variable)
 | 
			
		||||
      has_tag = true;
 | 
			
		||||
    else if (!(*it).ws_only())
 | 
			
		||||
      non_space = true;
 | 
			
		||||
 | 
			
		||||
    if ((*it).eol()) {
 | 
			
		||||
      if (has_tag && !non_space) {
 | 
			
		||||
        store_prefixes(line_begin);
 | 
			
		||||
 | 
			
		||||
        auto c = line_begin;
 | 
			
		||||
        for (bool end = false; !end; (*c).ws_only() ? c = m_tokens.erase(c) : ++c)
 | 
			
		||||
          if ((end = (*c).eol()))
 | 
			
		||||
            it = c - 1;
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      non_space = has_tag = false;
 | 
			
		||||
      line_begin = it + 1;
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void template_type::store_prefixes(std::vector<token>::iterator beg) {
 | 
			
		||||
  for (auto cur = beg; !(*cur).eol(); ++cur)
 | 
			
		||||
    if ((*cur).token_type() == token::type::partial &&
 | 
			
		||||
        cur != beg && (*(cur - 1)).ws_only())
 | 
			
		||||
      (*cur).partial_prefix((*(cur - 1)).raw());
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										30
									
								
								ext/mstch/src/template_type.hpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										30
									
								
								ext/mstch/src/template_type.hpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,30 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <string>
 | 
			
		||||
#include <vector>
 | 
			
		||||
 | 
			
		||||
#include "token.hpp"
 | 
			
		||||
#include "utils.hpp"
 | 
			
		||||
 | 
			
		||||
namespace mstch {
 | 
			
		||||
 | 
			
		||||
class template_type {
 | 
			
		||||
 public:
 | 
			
		||||
  template_type() = default;
 | 
			
		||||
  template_type(const std::string& str);
 | 
			
		||||
  template_type(const std::string& str, const delim_type& delims);
 | 
			
		||||
  std::vector<token>::const_iterator begin() const { return m_tokens.begin(); }
 | 
			
		||||
  std::vector<token>::const_iterator end() const { return m_tokens.end(); }
 | 
			
		||||
  void operator<<(const token& token) { m_tokens.push_back(token); }
 | 
			
		||||
 | 
			
		||||
 private:
 | 
			
		||||
  std::vector<token> m_tokens;
 | 
			
		||||
  std::string m_open;
 | 
			
		||||
  std::string m_close;
 | 
			
		||||
  void strip_whitespace();
 | 
			
		||||
  void process_text(citer beg, citer end);
 | 
			
		||||
  void tokenize(const std::string& tmp);
 | 
			
		||||
  void store_prefixes(std::vector<token>::iterator beg);
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										42
									
								
								ext/mstch/src/token.cpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										42
									
								
								ext/mstch/src/token.cpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,42 @@
 | 
			
		|||
#include "token.hpp"
 | 
			
		||||
#include "utils.hpp"
 | 
			
		||||
 | 
			
		||||
using namespace mstch;
 | 
			
		||||
 | 
			
		||||
token::type token::token_info(char c) {
 | 
			
		||||
  switch (c) {
 | 
			
		||||
    case '>': return type::partial;
 | 
			
		||||
    case '^': return type::inverted_section_open;
 | 
			
		||||
    case '/': return type::section_close;
 | 
			
		||||
    case '&': return type::unescaped_variable;
 | 
			
		||||
    case '#': return type::section_open;
 | 
			
		||||
    case '!': return type::comment;
 | 
			
		||||
    default: return type::variable;
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
token::token(const std::string& str, std::size_t left, std::size_t right):
 | 
			
		||||
    m_raw(str), m_eol(false), m_ws_only(false)
 | 
			
		||||
{
 | 
			
		||||
  if (left != 0 && right != 0) {
 | 
			
		||||
    if (str[left] == '=' && str[str.size() - right - 1] == '=') {
 | 
			
		||||
      m_type = type::delimiter_change;
 | 
			
		||||
    } else if (str[left] == '{' && str[str.size() - right - 1] == '}') {
 | 
			
		||||
      m_type = type::unescaped_variable;
 | 
			
		||||
      m_name = {first_not_ws(str.begin() + left + 1, str.end() - right),
 | 
			
		||||
          first_not_ws(str.rbegin() + 1 + right, str.rend() - left) + 1};
 | 
			
		||||
    } else {
 | 
			
		||||
      auto c = first_not_ws(str.begin() + left, str.end() - right);
 | 
			
		||||
      m_type = token_info(*c);
 | 
			
		||||
      if (m_type != type::variable)
 | 
			
		||||
        c = first_not_ws(c + 1, str.end() - right);
 | 
			
		||||
      m_name = {c, first_not_ws(str.rbegin() + right, str.rend() - left) + 1};
 | 
			
		||||
      m_delims = {{str.begin(), str.begin() + left},
 | 
			
		||||
          {str.end() - right, str.end()}};
 | 
			
		||||
    }
 | 
			
		||||
  } else {
 | 
			
		||||
    m_type = type::text;
 | 
			
		||||
    m_eol = (str.size() > 0 && str[str.size() - 1] == '\n');
 | 
			
		||||
    m_ws_only = (str.find_first_not_of(" \r\n\t") == std::string::npos);
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										39
									
								
								ext/mstch/src/token.hpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										39
									
								
								ext/mstch/src/token.hpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,39 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <string>
 | 
			
		||||
 | 
			
		||||
namespace mstch {
 | 
			
		||||
 | 
			
		||||
using delim_type = std::pair<std::string, std::string>;
 | 
			
		||||
 | 
			
		||||
class token {
 | 
			
		||||
 public:
 | 
			
		||||
  enum class type {
 | 
			
		||||
    text, variable, section_open, section_close, inverted_section_open,
 | 
			
		||||
    unescaped_variable, comment, partial, delimiter_change
 | 
			
		||||
  };
 | 
			
		||||
  token(const std::string& str, std::size_t left = 0, std::size_t right = 0);
 | 
			
		||||
  type token_type() const { return m_type; };
 | 
			
		||||
  const std::string& raw() const { return m_raw; };
 | 
			
		||||
  const std::string& name() const { return m_name; };
 | 
			
		||||
  const std::string& partial_prefix() const { return m_partial_prefix; };
 | 
			
		||||
  const delim_type& delims() const { return m_delims; };
 | 
			
		||||
  void partial_prefix(const std::string& p_partial_prefix) {
 | 
			
		||||
    m_partial_prefix = p_partial_prefix;
 | 
			
		||||
  };
 | 
			
		||||
  bool eol() const { return m_eol; }
 | 
			
		||||
  void eol(bool eol) { m_eol = eol; }
 | 
			
		||||
  bool ws_only() const { return m_ws_only; }
 | 
			
		||||
 | 
			
		||||
 private:
 | 
			
		||||
  type m_type;
 | 
			
		||||
  std::string m_name;
 | 
			
		||||
  std::string m_raw;
 | 
			
		||||
  std::string m_partial_prefix;
 | 
			
		||||
  delim_type m_delims;
 | 
			
		||||
  bool m_eol;
 | 
			
		||||
  bool m_ws_only;
 | 
			
		||||
  type token_info(char c);
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										44
									
								
								ext/mstch/src/utils.cpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										44
									
								
								ext/mstch/src/utils.cpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,44 @@
 | 
			
		|||
#include "utils.hpp"
 | 
			
		||||
#include "mstch/mstch.hpp"
 | 
			
		||||
 | 
			
		||||
mstch::citer mstch::first_not_ws(mstch::citer begin, mstch::citer end) {
 | 
			
		||||
  for (auto it = begin; it != end; ++it)
 | 
			
		||||
    if (*it != ' ') return it;
 | 
			
		||||
  return end;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
mstch::citer mstch::first_not_ws(mstch::criter begin, mstch::criter end) {
 | 
			
		||||
  for (auto rit = begin; rit != end; ++rit)
 | 
			
		||||
    if (*rit != ' ') return --(rit.base());
 | 
			
		||||
  return --(end.base());
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
mstch::criter mstch::reverse(mstch::citer it) {
 | 
			
		||||
  return std::reverse_iterator<mstch::citer>(it);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
std::string mstch::html_escape(const std::string& str) {
 | 
			
		||||
  if (mstch::config::escape)
 | 
			
		||||
    return mstch::config::escape(str);
 | 
			
		||||
  
 | 
			
		||||
  std::string out;
 | 
			
		||||
  citer start = str.begin();
 | 
			
		||||
 | 
			
		||||
  auto add_escape = [&out, &start](const std::string& escaped, citer& it) {
 | 
			
		||||
    out += std::string{start, it} + escaped;
 | 
			
		||||
    start = it + 1;
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  for (auto it = str.begin(); it != str.end(); ++it)
 | 
			
		||||
    switch (*it) {
 | 
			
		||||
      case '&': add_escape("&", it); break;
 | 
			
		||||
      case '\'': add_escape("'", it); break;
 | 
			
		||||
      case '"': add_escape(""", it); break;
 | 
			
		||||
      case '<': add_escape("<", it); break;
 | 
			
		||||
      case '>': add_escape(">", it); break;
 | 
			
		||||
      case '/': add_escape("/", it); break;
 | 
			
		||||
      default: break;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
  return out + std::string{start, str.end()};
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										23
									
								
								ext/mstch/src/utils.hpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										23
									
								
								ext/mstch/src/utils.hpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,23 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <string>
 | 
			
		||||
#include <boost/variant/apply_visitor.hpp>
 | 
			
		||||
 | 
			
		||||
namespace mstch {
 | 
			
		||||
 | 
			
		||||
using citer = std::string::const_iterator;
 | 
			
		||||
using criter = std::string::const_reverse_iterator;
 | 
			
		||||
 | 
			
		||||
citer first_not_ws(citer begin, citer end);
 | 
			
		||||
citer first_not_ws(criter begin, criter end);
 | 
			
		||||
std::string html_escape(const std::string& str);
 | 
			
		||||
criter reverse(citer it);
 | 
			
		||||
 | 
			
		||||
template<class... Args>
 | 
			
		||||
auto visit(Args&&... args) -> decltype(boost::apply_visitor(
 | 
			
		||||
    std::forward<Args>(args)...))
 | 
			
		||||
{
 | 
			
		||||
  return boost::apply_visitor(std::forward<Args>(args)...);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										35
									
								
								ext/mstch/src/visitor/get_token.hpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										35
									
								
								ext/mstch/src/visitor/get_token.hpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,35 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <boost/variant/static_visitor.hpp>
 | 
			
		||||
 | 
			
		||||
#include "mstch/mstch.hpp"
 | 
			
		||||
#include "has_token.hpp"
 | 
			
		||||
 | 
			
		||||
namespace mstch {
 | 
			
		||||
 | 
			
		||||
class get_token: public boost::static_visitor<const mstch::node&> {
 | 
			
		||||
 public:
 | 
			
		||||
  get_token(const std::string& token, const mstch::node& node):
 | 
			
		||||
      m_token(token), m_node(node)
 | 
			
		||||
  {
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  template<class T>
 | 
			
		||||
  const mstch::node& operator()(const T&) const {
 | 
			
		||||
    return m_node;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  const mstch::node& operator()(const map& map) const {
 | 
			
		||||
    return map.at(m_token);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  const mstch::node& operator()(const std::shared_ptr<object>& object) const {
 | 
			
		||||
    return object->at(m_token);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
 private:
 | 
			
		||||
  const std::string& m_token;
 | 
			
		||||
  const mstch::node& m_node;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										31
									
								
								ext/mstch/src/visitor/has_token.hpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								ext/mstch/src/visitor/has_token.hpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,31 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <boost/variant/static_visitor.hpp>
 | 
			
		||||
 | 
			
		||||
#include "mstch/mstch.hpp"
 | 
			
		||||
 | 
			
		||||
namespace mstch {
 | 
			
		||||
 | 
			
		||||
class has_token: public boost::static_visitor<bool> {
 | 
			
		||||
 public:
 | 
			
		||||
  has_token(const std::string& token): m_token(token) {
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  template<class T>
 | 
			
		||||
  bool operator()(const T&) const {
 | 
			
		||||
    return m_token == ".";
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  bool operator()(const map& map) const {
 | 
			
		||||
    return map.count(m_token) == 1;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  bool operator()(const std::shared_ptr<object>& object) const {
 | 
			
		||||
    return object->has(m_token);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
 private:
 | 
			
		||||
  const std::string& m_token;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										41
									
								
								ext/mstch/src/visitor/is_node_empty.hpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										41
									
								
								ext/mstch/src/visitor/is_node_empty.hpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,41 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <boost/variant/static_visitor.hpp>
 | 
			
		||||
 | 
			
		||||
#include "mstch/mstch.hpp"
 | 
			
		||||
 | 
			
		||||
namespace mstch {
 | 
			
		||||
 | 
			
		||||
class is_node_empty: public boost::static_visitor<bool> {
 | 
			
		||||
 public:
 | 
			
		||||
  template<class T>
 | 
			
		||||
  bool operator()(const T&) const {
 | 
			
		||||
    return false;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  bool operator()(const std::nullptr_t&) const {
 | 
			
		||||
    return true;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  bool operator()(const int& value) const {
 | 
			
		||||
    return value == 0;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  bool operator()(const double& value) const {
 | 
			
		||||
    return value == 0;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  bool operator()(const bool& value) const {
 | 
			
		||||
    return !value;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  bool operator()(const std::string& value) const {
 | 
			
		||||
    return value == "";
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  bool operator()(const array& array) const {
 | 
			
		||||
    return array.size() == 0;
 | 
			
		||||
  }
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										56
									
								
								ext/mstch/src/visitor/render_node.hpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										56
									
								
								ext/mstch/src/visitor/render_node.hpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,56 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <sstream>
 | 
			
		||||
#include <boost/variant/static_visitor.hpp>
 | 
			
		||||
 | 
			
		||||
#include "render_context.hpp"
 | 
			
		||||
#include "mstch/mstch.hpp"
 | 
			
		||||
#include "utils.hpp"
 | 
			
		||||
 | 
			
		||||
namespace mstch {
 | 
			
		||||
 | 
			
		||||
class render_node: public boost::static_visitor<std::string> {
 | 
			
		||||
 public:
 | 
			
		||||
  enum class flag { none, escape_html };
 | 
			
		||||
  render_node(render_context& ctx, flag p_flag = flag::none):
 | 
			
		||||
      m_ctx(ctx), m_flag(p_flag)
 | 
			
		||||
  {
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  template<class T>
 | 
			
		||||
  std::string operator()(const T&) const {
 | 
			
		||||
    return "";
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  std::string operator()(const int& value) const {
 | 
			
		||||
    return std::to_string(value);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  std::string operator()(const double& value) const {
 | 
			
		||||
    std::stringstream ss;
 | 
			
		||||
    ss << value;
 | 
			
		||||
    return ss.str();
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  std::string operator()(const bool& value) const {
 | 
			
		||||
    return value ? "true" : "false";
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  std::string operator()(const lambda& value) const {
 | 
			
		||||
    template_type interpreted{value([this](const mstch::node& n) {
 | 
			
		||||
      return visit(render_node(m_ctx), n);
 | 
			
		||||
    })};
 | 
			
		||||
    auto rendered = render_context::push(m_ctx).render(interpreted);
 | 
			
		||||
    return (m_flag == flag::escape_html) ? html_escape(rendered) : rendered;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  std::string operator()(const std::string& value) const {
 | 
			
		||||
    return (m_flag == flag::escape_html) ? html_escape(value) : value;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
 private:
 | 
			
		||||
  render_context& m_ctx;
 | 
			
		||||
  flag m_flag;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										57
									
								
								ext/mstch/src/visitor/render_section.hpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										57
									
								
								ext/mstch/src/visitor/render_section.hpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,57 @@
 | 
			
		|||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <boost/variant/static_visitor.hpp>
 | 
			
		||||
 | 
			
		||||
#include "render_context.hpp"
 | 
			
		||||
#include "mstch/mstch.hpp"
 | 
			
		||||
#include "utils.hpp"
 | 
			
		||||
#include "render_node.hpp"
 | 
			
		||||
 | 
			
		||||
namespace mstch {
 | 
			
		||||
 | 
			
		||||
class render_section: public boost::static_visitor<std::string> {
 | 
			
		||||
 public:
 | 
			
		||||
  enum class flag { none, keep_array };
 | 
			
		||||
  render_section(
 | 
			
		||||
      render_context& ctx,
 | 
			
		||||
      const template_type& section,
 | 
			
		||||
      const delim_type& delims,
 | 
			
		||||
      flag p_flag = flag::none):
 | 
			
		||||
      m_ctx(ctx), m_section(section), m_delims(delims), m_flag(p_flag)
 | 
			
		||||
  {
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  template<class T>
 | 
			
		||||
  std::string operator()(const T& t) const {
 | 
			
		||||
    return render_context::push(m_ctx, t).render(m_section);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  std::string operator()(const lambda& fun) const {
 | 
			
		||||
    std::string section_str;
 | 
			
		||||
    for (auto& token: m_section)
 | 
			
		||||
      section_str += token.raw();
 | 
			
		||||
    template_type interpreted{fun([this](const mstch::node& n) {
 | 
			
		||||
      return visit(render_node(m_ctx), n);
 | 
			
		||||
    }, section_str), m_delims};
 | 
			
		||||
    return render_context::push(m_ctx).render(interpreted);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  std::string operator()(const array& array) const {
 | 
			
		||||
    std::string out;
 | 
			
		||||
    if (m_flag == flag::keep_array)
 | 
			
		||||
      return render_context::push(m_ctx, array).render(m_section);
 | 
			
		||||
    else
 | 
			
		||||
      for (auto& item: array)
 | 
			
		||||
        out += visit(render_section(
 | 
			
		||||
            m_ctx, m_section, m_delims, flag::keep_array), item);
 | 
			
		||||
    return out;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
 private:
 | 
			
		||||
  render_context& m_ctx;
 | 
			
		||||
  const template_type& m_section;
 | 
			
		||||
  const delim_type& m_delims;
 | 
			
		||||
  flag m_flag;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										42
									
								
								main.cpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										42
									
								
								main.cpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,42 @@
 | 
			
		|||
 | 
			
		||||
#include "src/MicroCore.h"
 | 
			
		||||
 | 
			
		||||
#include "ext/crow/crow.h"
 | 
			
		||||
#include "mstch/mstch.hpp"
 | 
			
		||||
#include "ext/format.h"
 | 
			
		||||
 | 
			
		||||
#include <iostream>
 | 
			
		||||
 | 
			
		||||
using boost::filesystem::path;
 | 
			
		||||
 | 
			
		||||
using namespace std;
 | 
			
		||||
 | 
			
		||||
int main() {
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    path blockchain_path {"/home/mwo/.bitmonero/lmdb"};
 | 
			
		||||
 | 
			
		||||
    fmt::print("Blockchain path      : {}\n", blockchain_path);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    std::string view{"{{#names}}Hi {{name}}!\n{{/names}}"};
 | 
			
		||||
 | 
			
		||||
    mstch::map context{
 | 
			
		||||
            {"names", mstch::array{
 | 
			
		||||
                    mstch::map{{"name", std::string{"Chris"}}},
 | 
			
		||||
                    mstch::map{{"name", std::string{"Mark"}}},
 | 
			
		||||
                    mstch::map{{"name", std::string{"Scott"}}},
 | 
			
		||||
            }}
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    crow::SimpleApp app;
 | 
			
		||||
 | 
			
		||||
    CROW_ROUTE(app, "/")
 | 
			
		||||
            ([&]() {
 | 
			
		||||
                return mstch::render(view, context);
 | 
			
		||||
            });
 | 
			
		||||
 | 
			
		||||
    app.port(8080).multithreaded().run();
 | 
			
		||||
 | 
			
		||||
    return 0;
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										22
									
								
								src/CMakeLists.txt
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								src/CMakeLists.txt
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,22 @@
 | 
			
		|||
cmake_minimum_required(VERSION 3.5)
 | 
			
		||||
 | 
			
		||||
project(myxrm)
 | 
			
		||||
 | 
			
		||||
set(SOURCE_HEADERS
 | 
			
		||||
        MicroCore.h
 | 
			
		||||
		tools.h
 | 
			
		||||
		monero_headers.h
 | 
			
		||||
		tx_details.h)
 | 
			
		||||
 | 
			
		||||
set(SOURCE_FILES
 | 
			
		||||
		MicroCore.cpp
 | 
			
		||||
		tools.cpp
 | 
			
		||||
		CmdLineOptions.cpp
 | 
			
		||||
		tx_details.cpp)
 | 
			
		||||
 | 
			
		||||
# make static library called libmyxrm
 | 
			
		||||
# that we are going to link to
 | 
			
		||||
# in the root CMakeLists.txt file
 | 
			
		||||
add_library(myxrm
 | 
			
		||||
		STATIC
 | 
			
		||||
		${SOURCE_FILES})
 | 
			
		||||
							
								
								
									
										80
									
								
								src/CmdLineOptions.cpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										80
									
								
								src/CmdLineOptions.cpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,80 @@
 | 
			
		|||
//
 | 
			
		||||
// Created by mwo on 6/11/15.
 | 
			
		||||
//
 | 
			
		||||
 | 
			
		||||
#include "CmdLineOptions.h"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
namespace xmreg
 | 
			
		||||
{
 | 
			
		||||
    /**
 | 
			
		||||
     * Take the acc and *avv[] from the main() and check and parse
 | 
			
		||||
     * all the options given
 | 
			
		||||
     */
 | 
			
		||||
    CmdLineOptions::CmdLineOptions(int acc, const char *avv[]) {
 | 
			
		||||
 | 
			
		||||
        positional_options_description p;
 | 
			
		||||
 | 
			
		||||
        p.add("txhash", -1);
 | 
			
		||||
 | 
			
		||||
        options_description desc(
 | 
			
		||||
                "showmixins, shows mixin outputs used for each input in a given transaction");
 | 
			
		||||
 | 
			
		||||
        desc.add_options()
 | 
			
		||||
                ("help,h", value<bool>()->default_value(false)->implicit_value(true),
 | 
			
		||||
                 "produce help message")
 | 
			
		||||
                ("txhash,t", value<string>(),
 | 
			
		||||
                 "transaction hash")
 | 
			
		||||
                ("viewkey,v", value<string>(),
 | 
			
		||||
                 "private view key string")
 | 
			
		||||
                ("address,a", value<string>(),
 | 
			
		||||
                 "monero address string")
 | 
			
		||||
                ("bc-path,b", value<string>(),
 | 
			
		||||
                 "path to lmdb blockchain")
 | 
			
		||||
                ("testnet",  value<bool>()->default_value(false)->implicit_value(true),
 | 
			
		||||
                 "is the address from testnet network");
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        store(command_line_parser(acc, avv)
 | 
			
		||||
                          .options(desc)
 | 
			
		||||
                          .positional(p)
 | 
			
		||||
                          .run(), vm);
 | 
			
		||||
 | 
			
		||||
        notify(vm);
 | 
			
		||||
 | 
			
		||||
        if (vm.count("help"))
 | 
			
		||||
        {
 | 
			
		||||
            if (vm["help"].as<bool>())
 | 
			
		||||
                cout << desc << "\n";
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Return the value of the argument passed to the program
 | 
			
		||||
     * in wrapped around boost::optional
 | 
			
		||||
     */
 | 
			
		||||
    template<typename T>
 | 
			
		||||
    boost::optional<T>
 | 
			
		||||
    CmdLineOptions::get_option(const string & opt_name) const
 | 
			
		||||
    {
 | 
			
		||||
 | 
			
		||||
        if (!vm.count(opt_name))
 | 
			
		||||
        {
 | 
			
		||||
            return boost::none;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        return vm[opt_name].as<T>();
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    // explicit instantiations of get_option template function
 | 
			
		||||
    template  boost::optional<string>
 | 
			
		||||
    CmdLineOptions::get_option<string>(const string & opt_name) const;
 | 
			
		||||
 | 
			
		||||
    template  boost::optional<bool>
 | 
			
		||||
            CmdLineOptions::get_option<bool>(const string & opt_name) const;
 | 
			
		||||
 | 
			
		||||
    template  boost::optional<size_t>
 | 
			
		||||
            CmdLineOptions::get_option<size_t>(const string & opt_name) const;
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										37
									
								
								src/CmdLineOptions.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										37
									
								
								src/CmdLineOptions.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,37 @@
 | 
			
		|||
//
 | 
			
		||||
// Created by mwo on 6/11/15.
 | 
			
		||||
//
 | 
			
		||||
 | 
			
		||||
#ifndef XMREG01_CMDLINEOPTIONS_H
 | 
			
		||||
#define XMREG01_CMDLINEOPTIONS_H
 | 
			
		||||
 | 
			
		||||
#include <iostream>
 | 
			
		||||
#include <string>
 | 
			
		||||
 | 
			
		||||
#include <boost/program_options.hpp>
 | 
			
		||||
#include <boost/optional.hpp>
 | 
			
		||||
 | 
			
		||||
namespace xmreg
 | 
			
		||||
{
 | 
			
		||||
 | 
			
		||||
    using namespace std;
 | 
			
		||||
    using namespace boost::program_options;
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Manages program options of this example program.
 | 
			
		||||
     *
 | 
			
		||||
     * Basically a wrapper for boost::program_options
 | 
			
		||||
     */
 | 
			
		||||
    class CmdLineOptions {
 | 
			
		||||
        variables_map vm;
 | 
			
		||||
 | 
			
		||||
    public:
 | 
			
		||||
        CmdLineOptions(int acc, const char *avv[]);
 | 
			
		||||
 | 
			
		||||
    template<typename T>
 | 
			
		||||
    boost::optional<T> get_option(const string & opt_name) const;
 | 
			
		||||
    };
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
#endif //XMREG01_CMDLINEOPTIONS_H
 | 
			
		||||
							
								
								
									
										276
									
								
								src/MicroCore.cpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										276
									
								
								src/MicroCore.cpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,276 @@
 | 
			
		|||
//
 | 
			
		||||
// Created by mwo on 5/11/15.
 | 
			
		||||
//
 | 
			
		||||
 | 
			
		||||
#include "MicroCore.h"
 | 
			
		||||
 | 
			
		||||
namespace xmreg
 | 
			
		||||
{
 | 
			
		||||
    /**
 | 
			
		||||
     * The constructor is interesting, as
 | 
			
		||||
     * m_mempool and m_blockchain_storage depend
 | 
			
		||||
     * on each other.
 | 
			
		||||
     *
 | 
			
		||||
     * So basically m_mempool is initialized with
 | 
			
		||||
     * reference to Blockchain (i.e., Blockchain&)
 | 
			
		||||
     * and m_blockchain_storage is initialized with
 | 
			
		||||
     * reference to m_mempool (i.e., tx_memory_pool&)
 | 
			
		||||
     *
 | 
			
		||||
     * The same is done in cryptonode::core.
 | 
			
		||||
     */
 | 
			
		||||
    MicroCore::MicroCore():
 | 
			
		||||
            m_mempool(m_blockchain_storage),
 | 
			
		||||
            m_blockchain_storage(m_mempool)
 | 
			
		||||
    {}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Initialized the MicroCore object.
 | 
			
		||||
     *
 | 
			
		||||
     * Create BlockchainLMDB on the heap.
 | 
			
		||||
     * Open database files located in blockchain_path.
 | 
			
		||||
     * Initialize m_blockchain_storage with the BlockchainLMDB object.
 | 
			
		||||
     */
 | 
			
		||||
    bool
 | 
			
		||||
    MicroCore::init(const string& blockchain_path)
 | 
			
		||||
    {
 | 
			
		||||
        int db_flags = 0;
 | 
			
		||||
 | 
			
		||||
        //db_flags |= MDB_RDONLY;
 | 
			
		||||
        //db_flags |= MDB_NOLOCK;
 | 
			
		||||
        db_flags |= MDB_NOSYNC;
 | 
			
		||||
 | 
			
		||||
        BlockchainDB* db = nullptr;
 | 
			
		||||
        db = new BlockchainLMDB();
 | 
			
		||||
 | 
			
		||||
        try
 | 
			
		||||
        {
 | 
			
		||||
            // try opening lmdb database files
 | 
			
		||||
            db->open(blockchain_path, db_flags);
 | 
			
		||||
        }
 | 
			
		||||
        catch (const std::exception& e)
 | 
			
		||||
        {
 | 
			
		||||
            cerr << "Error opening database: " << e.what();
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        // check if the blockchain database
 | 
			
		||||
        // is successful opened
 | 
			
		||||
        if(!db->is_open())
 | 
			
		||||
        {
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        // initialize Blockchain object to manage
 | 
			
		||||
        // the database.
 | 
			
		||||
        return m_blockchain_storage.init(db, false);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
    * Get m_blockchain_storage.
 | 
			
		||||
    * Initialize m_blockchain_storage with the BlockchainLMDB object.
 | 
			
		||||
    */
 | 
			
		||||
    Blockchain&
 | 
			
		||||
    MicroCore::get_core()
 | 
			
		||||
    {
 | 
			
		||||
        return m_blockchain_storage;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Get block by its height
 | 
			
		||||
     *
 | 
			
		||||
     * returns true if success
 | 
			
		||||
     */
 | 
			
		||||
    bool
 | 
			
		||||
    MicroCore::get_block_by_height(const uint64_t& height, block& blk)
 | 
			
		||||
    {
 | 
			
		||||
 | 
			
		||||
        crypto::hash block_id;
 | 
			
		||||
 | 
			
		||||
        try
 | 
			
		||||
        {
 | 
			
		||||
            block_id = m_blockchain_storage.get_block_id_by_height(height);
 | 
			
		||||
        }
 | 
			
		||||
        catch (const exception& e)
 | 
			
		||||
        {
 | 
			
		||||
            cerr << e.what() << endl;
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        if (!m_blockchain_storage.get_block_by_hash(block_id, blk))
 | 
			
		||||
        {
 | 
			
		||||
            cerr << "Block with hash " << block_id
 | 
			
		||||
                 << "and height " << height << " not found!"
 | 
			
		||||
                 << endl;
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        return true;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Get transaction tx from the blockchain using it hash
 | 
			
		||||
     */
 | 
			
		||||
    bool
 | 
			
		||||
    MicroCore::get_tx(const crypto::hash& tx_hash, transaction& tx)
 | 
			
		||||
    {
 | 
			
		||||
        try
 | 
			
		||||
        {
 | 
			
		||||
            // get transaction with given hash
 | 
			
		||||
            tx = m_blockchain_storage.get_db().get_tx(tx_hash);
 | 
			
		||||
        }
 | 
			
		||||
        catch (const exception& e)
 | 
			
		||||
        {
 | 
			
		||||
            cerr << e.what() << endl;
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        return true;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Find output with given public key in a given transaction
 | 
			
		||||
     */
 | 
			
		||||
    bool
 | 
			
		||||
    MicroCore::find_output_in_tx(const transaction& tx,
 | 
			
		||||
                                 const public_key& output_pubkey,
 | 
			
		||||
                                 tx_out& out,
 | 
			
		||||
                                 size_t& output_index)
 | 
			
		||||
    {
 | 
			
		||||
 | 
			
		||||
        size_t idx {0};
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        // search in the ouputs for an output which
 | 
			
		||||
        // public key matches to what we want
 | 
			
		||||
        auto it = std::find_if(tx.vout.begin(), tx.vout.end(),
 | 
			
		||||
                               [&](const tx_out& o)
 | 
			
		||||
                               {
 | 
			
		||||
                                   const txout_to_key& tx_in_to_key
 | 
			
		||||
                                           = boost::get<txout_to_key>(o.target);
 | 
			
		||||
 | 
			
		||||
                                   ++idx;
 | 
			
		||||
 | 
			
		||||
                                   return tx_in_to_key.key == output_pubkey;
 | 
			
		||||
                               });
 | 
			
		||||
 | 
			
		||||
        if (it != tx.vout.end())
 | 
			
		||||
        {
 | 
			
		||||
            // we found the desired public key
 | 
			
		||||
            out = *it;
 | 
			
		||||
            output_index = idx > 0 ? idx - 1 : idx;
 | 
			
		||||
 | 
			
		||||
            //cout << idx << " " << output_index << endl;
 | 
			
		||||
 | 
			
		||||
            return true;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        return false;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Returns tx hash in a given block which
 | 
			
		||||
     * contains given output's public key
 | 
			
		||||
     */
 | 
			
		||||
    bool
 | 
			
		||||
    MicroCore::get_tx_hash_from_output_pubkey(const public_key& output_pubkey,
 | 
			
		||||
                                              const uint64_t& block_height,
 | 
			
		||||
                                              crypto::hash& tx_hash,
 | 
			
		||||
                                              cryptonote::transaction& tx_found)
 | 
			
		||||
    {
 | 
			
		||||
 | 
			
		||||
        tx_hash = null_hash;
 | 
			
		||||
 | 
			
		||||
        // get block of given height
 | 
			
		||||
        block blk;
 | 
			
		||||
        if (!get_block_by_height(block_height, blk))
 | 
			
		||||
        {
 | 
			
		||||
            cerr << "Cant get block of height: " << block_height << endl;
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        // get all transactions in the block found
 | 
			
		||||
        // initialize the first list with transaction for solving
 | 
			
		||||
        // the block i.e. coinbase.
 | 
			
		||||
        list<transaction> txs {blk.miner_tx};
 | 
			
		||||
        list<crypto::hash> missed_txs;
 | 
			
		||||
 | 
			
		||||
        if (!m_blockchain_storage.get_transactions(blk.tx_hashes, txs, missed_txs))
 | 
			
		||||
        {
 | 
			
		||||
            cerr << "Cant find transcations in block: " << block_height << endl;
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        if (!missed_txs.empty())
 | 
			
		||||
        {
 | 
			
		||||
            cerr << "Transactions not found in blk: " << block_height << endl;
 | 
			
		||||
 | 
			
		||||
            for (const crypto::hash& h : missed_txs)
 | 
			
		||||
            {
 | 
			
		||||
                cerr << " - tx hash: " << h << endl;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        // search outputs in each transactions
 | 
			
		||||
        // until output with pubkey of interest is found
 | 
			
		||||
        for (const transaction& tx : txs)
 | 
			
		||||
        {
 | 
			
		||||
 | 
			
		||||
            tx_out found_out;
 | 
			
		||||
 | 
			
		||||
            // we dont need here output_index
 | 
			
		||||
            size_t output_index;
 | 
			
		||||
 | 
			
		||||
            if (find_output_in_tx(tx, output_pubkey, found_out, output_index))
 | 
			
		||||
            {
 | 
			
		||||
                // we found the desired public key
 | 
			
		||||
                tx_hash = get_transaction_hash(tx);
 | 
			
		||||
                tx_found = tx;
 | 
			
		||||
 | 
			
		||||
                return true;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        return false;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    uint64_t
 | 
			
		||||
    MicroCore::get_blk_timestamp(uint64_t blk_height)
 | 
			
		||||
    {
 | 
			
		||||
        cryptonote::block blk;
 | 
			
		||||
 | 
			
		||||
        if (!get_block_by_height(blk_height, blk))
 | 
			
		||||
        {
 | 
			
		||||
            cerr << "Cant get block by height: " << blk_height << endl;
 | 
			
		||||
            return 0;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        return blk.timestamp;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * De-initialized Blockchain.
 | 
			
		||||
     *
 | 
			
		||||
     * since blockchain is opened as MDB_RDONLY
 | 
			
		||||
     * need to manually free memory taken on heap
 | 
			
		||||
     * by BlockchainLMDB
 | 
			
		||||
     */
 | 
			
		||||
    MicroCore::~MicroCore()
 | 
			
		||||
    {
 | 
			
		||||
        delete &m_blockchain_storage.get_db();
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										72
									
								
								src/MicroCore.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										72
									
								
								src/MicroCore.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,72 @@
 | 
			
		|||
//
 | 
			
		||||
// Created by mwo on 5/11/15.
 | 
			
		||||
//
 | 
			
		||||
 | 
			
		||||
#ifndef XMREG01_MICROCORE_H
 | 
			
		||||
#define XMREG01_MICROCORE_H
 | 
			
		||||
 | 
			
		||||
#include <iostream>
 | 
			
		||||
 | 
			
		||||
#include "monero_headers.h"
 | 
			
		||||
#include "tx_details.h"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
namespace xmreg
 | 
			
		||||
{
 | 
			
		||||
    using namespace cryptonote;
 | 
			
		||||
    using namespace crypto;
 | 
			
		||||
    using namespace std;
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Micro version of cryptonode::core class
 | 
			
		||||
     * Micro version of constructor,
 | 
			
		||||
     * init and destructor are implemted.
 | 
			
		||||
     *
 | 
			
		||||
     * Just enough to read the blockchain
 | 
			
		||||
     * database for use in the example.
 | 
			
		||||
     */
 | 
			
		||||
    class MicroCore {
 | 
			
		||||
 | 
			
		||||
        tx_memory_pool m_mempool;
 | 
			
		||||
        Blockchain m_blockchain_storage;
 | 
			
		||||
 | 
			
		||||
    public:
 | 
			
		||||
        MicroCore();
 | 
			
		||||
 | 
			
		||||
        bool
 | 
			
		||||
        init(const string& blockchain_path);
 | 
			
		||||
 | 
			
		||||
        Blockchain&
 | 
			
		||||
        get_core();
 | 
			
		||||
 | 
			
		||||
        bool
 | 
			
		||||
        get_block_by_height(const uint64_t& height, block& blk);
 | 
			
		||||
 | 
			
		||||
        bool
 | 
			
		||||
        get_tx(const crypto::hash& tx_hash, transaction& tx);
 | 
			
		||||
 | 
			
		||||
        bool
 | 
			
		||||
        find_output_in_tx(const transaction& tx,
 | 
			
		||||
                          const public_key& output_pubkey,
 | 
			
		||||
                          tx_out& out,
 | 
			
		||||
                          size_t& output_index);
 | 
			
		||||
 | 
			
		||||
        bool
 | 
			
		||||
        get_tx_hash_from_output_pubkey(const public_key& output_pubkey,
 | 
			
		||||
                                       const uint64_t& block_height,
 | 
			
		||||
                                       crypto::hash& tx_hash,
 | 
			
		||||
                                       transaction& tx_found);
 | 
			
		||||
 | 
			
		||||
        uint64_t
 | 
			
		||||
        get_blk_timestamp(uint64_t blk_height);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        virtual ~MicroCore();
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
#endif //XMREG01_MICROCORE_H
 | 
			
		||||
							
								
								
									
										22
									
								
								src/monero_headers.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								src/monero_headers.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,22 @@
 | 
			
		|||
//
 | 
			
		||||
// Created by mwo on 5/11/15.
 | 
			
		||||
//
 | 
			
		||||
 | 
			
		||||
#ifndef XMREG01_MONERO_HEADERS_H_H
 | 
			
		||||
#define XMREG01_MONERO_HEADERS_H_H
 | 
			
		||||
 | 
			
		||||
#define DB_LMDB   2
 | 
			
		||||
#define BLOCKCHAIN_DB DB_LMDB
 | 
			
		||||
 | 
			
		||||
#include "cryptonote_core/cryptonote_basic.h"
 | 
			
		||||
#include "cryptonote_core/blockchain_storage.h"
 | 
			
		||||
#include "cryptonote_core/blockchain.h"
 | 
			
		||||
 | 
			
		||||
#include "blockchain_db/lmdb/db_lmdb.h"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
#include "common/base58.h"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
#endif //XMREG01_MONERO_HEADERS_H_H
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										337
									
								
								src/tools.cpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										337
									
								
								src/tools.cpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,337 @@
 | 
			
		|||
//
 | 
			
		||||
// Created by marcin on 5/11/15.
 | 
			
		||||
//
 | 
			
		||||
 | 
			
		||||
#include "tools.h"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
namespace xmreg
 | 
			
		||||
{
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Parse key string, e.g., a viewkey in a string
 | 
			
		||||
     * into crypto::secret_key or crypto::public_key
 | 
			
		||||
     * depending on the template argument.
 | 
			
		||||
     */
 | 
			
		||||
    template <typename T>
 | 
			
		||||
    bool
 | 
			
		||||
    parse_str_secret_key(const string& key_str, T& secret_key)
 | 
			
		||||
    {
 | 
			
		||||
 | 
			
		||||
        // hash and keys have same structure, so to parse string of
 | 
			
		||||
        // a key, e.g., a view key, we can first parse it into the hash
 | 
			
		||||
        // object using parse_hash256 function, and then copy the reslting
 | 
			
		||||
        // hash data into secret key.
 | 
			
		||||
        crypto::hash hash_;
 | 
			
		||||
 | 
			
		||||
        if(!parse_hash256(key_str, hash_))
 | 
			
		||||
        {
 | 
			
		||||
            cerr << "Cant parse a key (e.g. viewkey): " << key_str << endl;
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        // crypto::hash and crypto::secret_key have basicly same
 | 
			
		||||
        // structure. They both keep they key/hash as c-style char array
 | 
			
		||||
        // of fixed size. Thus we can just copy data from hash
 | 
			
		||||
        // to key
 | 
			
		||||
        copy(begin(hash_.data), end(hash_.data), secret_key.data);
 | 
			
		||||
 | 
			
		||||
        return true;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // explicit instantiations of get template function
 | 
			
		||||
    template bool parse_str_secret_key<crypto::secret_key>(const string& key_str, crypto::secret_key& secret_key);
 | 
			
		||||
    template bool parse_str_secret_key<crypto::public_key>(const string& key_str, crypto::public_key& secret_key);
 | 
			
		||||
    template bool parse_str_secret_key<crypto::hash>(const string& key_str, crypto::hash& secret_key);
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Get transaction tx using given tx hash. Hash is represent as string here,
 | 
			
		||||
     * so before we can tap into the blockchain, we need to pare it into
 | 
			
		||||
     * crypto::hash object.
 | 
			
		||||
     */
 | 
			
		||||
    bool
 | 
			
		||||
    get_tx_pub_key_from_str_hash(Blockchain& core_storage, const string& hash_str, transaction& tx)
 | 
			
		||||
    {
 | 
			
		||||
        crypto::hash tx_hash;
 | 
			
		||||
        parse_hash256(hash_str, tx_hash);
 | 
			
		||||
 | 
			
		||||
        try
 | 
			
		||||
        {
 | 
			
		||||
            // get transaction with given hash
 | 
			
		||||
            tx = core_storage.get_db().get_tx(tx_hash);
 | 
			
		||||
        }
 | 
			
		||||
        catch (const TX_DNE& e)
 | 
			
		||||
        {
 | 
			
		||||
            cerr << e.what() << endl;
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        return true;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Parse monero address in a string form into
 | 
			
		||||
     * cryptonote::account_public_address object
 | 
			
		||||
     */
 | 
			
		||||
    bool
 | 
			
		||||
    parse_str_address(const string& address_str,
 | 
			
		||||
                      account_public_address& address,
 | 
			
		||||
                      bool testnet)
 | 
			
		||||
    {
 | 
			
		||||
 | 
			
		||||
        if (!get_account_address_from_str(address, testnet, address_str))
 | 
			
		||||
        {
 | 
			
		||||
            cerr << "Error getting address: " << address_str << endl;
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        return true;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Return string representation of monero address
 | 
			
		||||
     */
 | 
			
		||||
    string
 | 
			
		||||
    print_address(const account_public_address& address, bool testnet)
 | 
			
		||||
    {
 | 
			
		||||
        return "<" + get_account_address_as_str(testnet, address) + ">";
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    string
 | 
			
		||||
    print_sig (const signature& sig)
 | 
			
		||||
    {
 | 
			
		||||
        stringstream ss;
 | 
			
		||||
 | 
			
		||||
        ss << "c: <" << epee::string_tools::pod_to_hex(sig.c) << "> "
 | 
			
		||||
           << "r: <" << epee::string_tools::pod_to_hex(sig.r) << ">";
 | 
			
		||||
 | 
			
		||||
        return ss.str();
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Check if a character is a path seprator
 | 
			
		||||
     */
 | 
			
		||||
    inline bool
 | 
			
		||||
    is_separator(char c)
 | 
			
		||||
    {
 | 
			
		||||
        // default linux path separator
 | 
			
		||||
        const char separator = PATH_SEPARARTOR;
 | 
			
		||||
 | 
			
		||||
        return c == separator;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Remove trailinig path separator.
 | 
			
		||||
     */
 | 
			
		||||
    string
 | 
			
		||||
    remove_trailing_path_separator(const string& in_path)
 | 
			
		||||
    {
 | 
			
		||||
        string new_string = in_path;
 | 
			
		||||
        if (!new_string.empty() && is_separator(new_string[new_string.size() - 1]))
 | 
			
		||||
            new_string.erase(new_string.size() - 1);
 | 
			
		||||
        return new_string;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    bf::path
 | 
			
		||||
    remove_trailing_path_separator(const bf::path& in_path)
 | 
			
		||||
    {
 | 
			
		||||
        string path_str = in_path.native();
 | 
			
		||||
        return bf::path(remove_trailing_path_separator(path_str));
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    string
 | 
			
		||||
    timestamp_to_str(time_t timestamp, const char* format)
 | 
			
		||||
    {
 | 
			
		||||
 | 
			
		||||
        const int TIME_LENGTH = 60;
 | 
			
		||||
 | 
			
		||||
        char str_buff[TIME_LENGTH];
 | 
			
		||||
 | 
			
		||||
        tm *tm_ptr;
 | 
			
		||||
        tm_ptr = localtime(×tamp);
 | 
			
		||||
 | 
			
		||||
        size_t len;
 | 
			
		||||
 | 
			
		||||
        len = std::strftime(str_buff, TIME_LENGTH, format, tm_ptr);
 | 
			
		||||
 | 
			
		||||
        return string(str_buff, len);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    ostream&
 | 
			
		||||
    operator<< (ostream& os, const account_public_address& addr)
 | 
			
		||||
    {
 | 
			
		||||
        os << get_account_address_as_str(false, addr);
 | 
			
		||||
        return os;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    /*
 | 
			
		||||
     * Generate key_image of foran ith output
 | 
			
		||||
     */
 | 
			
		||||
    bool
 | 
			
		||||
    generate_key_image(const crypto::key_derivation& derivation,
 | 
			
		||||
                       const std::size_t i,
 | 
			
		||||
                       const crypto::secret_key& sec_key,
 | 
			
		||||
                       const crypto::public_key& pub_key,
 | 
			
		||||
                       crypto::key_image& key_img)
 | 
			
		||||
    {
 | 
			
		||||
 | 
			
		||||
        cryptonote::keypair in_ephemeral;
 | 
			
		||||
 | 
			
		||||
        if (!crypto::derive_public_key(derivation, i,
 | 
			
		||||
                                       pub_key,
 | 
			
		||||
                                       in_ephemeral.pub))
 | 
			
		||||
        {
 | 
			
		||||
            cerr << "Error generating publick key " << pub_key << endl;
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        try
 | 
			
		||||
        {
 | 
			
		||||
 | 
			
		||||
            crypto::derive_secret_key(derivation, i,
 | 
			
		||||
                                      sec_key,
 | 
			
		||||
                                      in_ephemeral.sec);
 | 
			
		||||
        }
 | 
			
		||||
        catch(const std::exception& e)
 | 
			
		||||
        {
 | 
			
		||||
            cerr << "Error generate secret image: " << e.what() << endl;
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        try
 | 
			
		||||
        {
 | 
			
		||||
            crypto::generate_key_image(in_ephemeral.pub,
 | 
			
		||||
                                       in_ephemeral.sec,
 | 
			
		||||
                                       key_img);
 | 
			
		||||
        }
 | 
			
		||||
        catch(const std::exception& e)
 | 
			
		||||
        {
 | 
			
		||||
            cerr << "Error generate key image: " << e.what() << endl;
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        return true;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    string
 | 
			
		||||
    get_default_lmdb_folder()
 | 
			
		||||
    {
 | 
			
		||||
        // default path to monero folder
 | 
			
		||||
        // on linux this is /home/<username>/.bitmonero
 | 
			
		||||
        string default_monero_dir = tools::get_default_data_dir();
 | 
			
		||||
 | 
			
		||||
        // the default folder of the lmdb blockchain database
 | 
			
		||||
        // is therefore as follows
 | 
			
		||||
        return default_monero_dir + string("/lmdb");
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    /*
 | 
			
		||||
     * Ge blockchain exception from command line option
 | 
			
		||||
     *
 | 
			
		||||
     * If not given, provide default path
 | 
			
		||||
     */
 | 
			
		||||
    bool
 | 
			
		||||
    get_blockchain_path(const boost::optional<string>& bc_path, bf::path& blockchain_path )
 | 
			
		||||
    {
 | 
			
		||||
        // the default folder of the lmdb blockchain database
 | 
			
		||||
        string default_lmdb_dir   = xmreg::get_default_lmdb_folder();
 | 
			
		||||
 | 
			
		||||
        blockchain_path = bc_path
 | 
			
		||||
                          ? bf::path(*bc_path)
 | 
			
		||||
                          : bf::path(default_lmdb_dir);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        if (!bf::is_directory(blockchain_path))
 | 
			
		||||
        {
 | 
			
		||||
            cerr << "Given path \"" << blockchain_path   << "\" "
 | 
			
		||||
                 << "is not a folder or does not exist" << " "
 | 
			
		||||
                 << endl;
 | 
			
		||||
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        blockchain_path = xmreg::remove_trailing_path_separator(blockchain_path);
 | 
			
		||||
 | 
			
		||||
        return true;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Rough estimate of block height from the time provided
 | 
			
		||||
     *
 | 
			
		||||
     */
 | 
			
		||||
    uint64_t
 | 
			
		||||
    estimate_bc_height(const string& date, const char* format)
 | 
			
		||||
    {
 | 
			
		||||
        const pt::ptime MONERO_START {gt::date(2014,04,18)};
 | 
			
		||||
        const uint64_t MONERO_BLOCK_TIME {60}; // seconds
 | 
			
		||||
 | 
			
		||||
        dateparser parser {format};
 | 
			
		||||
 | 
			
		||||
        if (!parser(date))
 | 
			
		||||
        {
 | 
			
		||||
           throw runtime_error(string("Date format is incorrect: ") + date);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        pt::ptime requested_date = parser.pt;
 | 
			
		||||
 | 
			
		||||
        if (requested_date < MONERO_START)
 | 
			
		||||
        {
 | 
			
		||||
            return 0;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        pt::time_duration td = requested_date - MONERO_START;
 | 
			
		||||
 | 
			
		||||
        return static_cast<uint64_t>(td.total_seconds()) / MONERO_BLOCK_TIME;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    array<size_t, 5>
 | 
			
		||||
    timestamp_difference(uint64_t t1, uint64_t t2)
 | 
			
		||||
    {
 | 
			
		||||
 | 
			
		||||
        uint64_t timestamp_diff = t1 - t2;
 | 
			
		||||
 | 
			
		||||
        // calculate difference of timestamps from current block to the mixin one
 | 
			
		||||
        if (t2 > t1)
 | 
			
		||||
        {
 | 
			
		||||
            timestamp_diff = t2 - t1;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        uint64_t time_diff_years = timestamp_diff / 31536000;
 | 
			
		||||
 | 
			
		||||
        timestamp_diff -=  time_diff_years * 31536000;
 | 
			
		||||
 | 
			
		||||
        uint64_t time_diff_days = timestamp_diff / 86400;
 | 
			
		||||
 | 
			
		||||
        timestamp_diff -=  time_diff_days * 86400;
 | 
			
		||||
 | 
			
		||||
        uint64_t time_diff_hours = timestamp_diff / 3600;
 | 
			
		||||
 | 
			
		||||
        timestamp_diff -=  time_diff_hours * 3600;
 | 
			
		||||
 | 
			
		||||
        uint64_t time_diff_minutes = timestamp_diff / 60;
 | 
			
		||||
 | 
			
		||||
        timestamp_diff -=  time_diff_minutes * 60;
 | 
			
		||||
 | 
			
		||||
        uint64_t time_diff_seconds = timestamp_diff ;
 | 
			
		||||
 | 
			
		||||
        return array<size_t, 5> {time_diff_years, time_diff_days,
 | 
			
		||||
                                 time_diff_hours, time_diff_minutes,
 | 
			
		||||
                                 time_diff_seconds};
 | 
			
		||||
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										151
									
								
								src/tools.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										151
									
								
								src/tools.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,151 @@
 | 
			
		|||
//
 | 
			
		||||
// Created by marcin on 5/11/15.
 | 
			
		||||
//
 | 
			
		||||
 | 
			
		||||
#ifndef XMREG01_TOOLS_H
 | 
			
		||||
#define XMREG01_TOOLS_H
 | 
			
		||||
 | 
			
		||||
#define PATH_SEPARARTOR '/'
 | 
			
		||||
 | 
			
		||||
#include "monero_headers.h"
 | 
			
		||||
#include "tx_details.h"
 | 
			
		||||
 | 
			
		||||
#include "../ext/dateparser.h"
 | 
			
		||||
 | 
			
		||||
#include <boost/filesystem.hpp>
 | 
			
		||||
#include <boost/optional.hpp>
 | 
			
		||||
#include "boost/date_time/posix_time/posix_time.hpp"
 | 
			
		||||
 | 
			
		||||
#include <string>
 | 
			
		||||
#include <vector>
 | 
			
		||||
#include <array>
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Some helper functions used in the example.
 | 
			
		||||
 * Names are rather self-explanatory, so I think
 | 
			
		||||
 * there is no reason for any detailed explanations here
 | 
			
		||||
 */
 | 
			
		||||
namespace xmreg
 | 
			
		||||
{
 | 
			
		||||
    using namespace cryptonote;
 | 
			
		||||
    using namespace crypto;
 | 
			
		||||
    using namespace std;
 | 
			
		||||
 | 
			
		||||
    namespace bf = boost::filesystem;
 | 
			
		||||
    namespace pt = boost::posix_time;
 | 
			
		||||
    namespace gt = boost::gregorian;
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    struct outputs_visitor
 | 
			
		||||
    {
 | 
			
		||||
        std::vector<crypto::public_key >& m_output_keys;
 | 
			
		||||
 | 
			
		||||
        const Blockchain& m_bch;
 | 
			
		||||
 | 
			
		||||
        outputs_visitor(std::vector<crypto::public_key>& output_keys, const Blockchain& bch) :
 | 
			
		||||
                m_output_keys(output_keys), m_bch(bch)
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        bool handle_output(uint64_t unlock_time, const crypto::public_key &pubkey)
 | 
			
		||||
        {
 | 
			
		||||
            //check tx unlock time
 | 
			
		||||
//            if (!m_bch.is_tx_spendtime_unlocked(unlock_time))
 | 
			
		||||
//            {
 | 
			
		||||
//                LOG_PRINT_L1("One of outputs for one of inputs has wrong tx.unlock_time = " << unlock_time);
 | 
			
		||||
//                return false;
 | 
			
		||||
//            }
 | 
			
		||||
 | 
			
		||||
            m_output_keys.push_back(pubkey);
 | 
			
		||||
 | 
			
		||||
            return true;
 | 
			
		||||
        }
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    template <typename T>
 | 
			
		||||
    bool
 | 
			
		||||
    parse_str_secret_key(const string& key_str, T& secret_key);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    bool
 | 
			
		||||
    get_tx_pub_key_from_str_hash(Blockchain& core_storage,
 | 
			
		||||
                             const string& hash_str,
 | 
			
		||||
                             transaction& tx);
 | 
			
		||||
 | 
			
		||||
    bool
 | 
			
		||||
    parse_str_address(const string& address_str,
 | 
			
		||||
                      account_public_address& address,
 | 
			
		||||
                      bool testnet = false);
 | 
			
		||||
 | 
			
		||||
    inline bool
 | 
			
		||||
    is_separator(char c);
 | 
			
		||||
 | 
			
		||||
    string
 | 
			
		||||
    print_address(const account_public_address& address,
 | 
			
		||||
                  bool testnet = false);
 | 
			
		||||
 | 
			
		||||
    string
 | 
			
		||||
    print_sig (const signature& sig);
 | 
			
		||||
 | 
			
		||||
    string
 | 
			
		||||
    remove_trailing_path_separator(const string& in_path);
 | 
			
		||||
 | 
			
		||||
    bf::path
 | 
			
		||||
    remove_trailing_path_separator(const bf::path& in_path);
 | 
			
		||||
 | 
			
		||||
    string
 | 
			
		||||
    timestamp_to_str(time_t timestamp, const char* format = "%F %T");
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    ostream&
 | 
			
		||||
    operator<< (ostream& os, const account_public_address& addr);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    string
 | 
			
		||||
    get_default_lmdb_folder();
 | 
			
		||||
 | 
			
		||||
    bool
 | 
			
		||||
    generate_key_image(const crypto::key_derivation& derivation,
 | 
			
		||||
                       const std::size_t output_index,
 | 
			
		||||
                       const crypto::secret_key& sec_key,
 | 
			
		||||
                       const crypto::public_key& pub_key,
 | 
			
		||||
                       crypto::key_image& key_img);
 | 
			
		||||
 | 
			
		||||
    bool
 | 
			
		||||
    get_blockchain_path(const boost::optional<string>& bc_path,
 | 
			
		||||
                        bf::path& blockchain_path);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    inline void
 | 
			
		||||
    enable_monero_log() {
 | 
			
		||||
        uint32_t log_level = 0;
 | 
			
		||||
        epee::log_space::get_set_log_detalisation_level(true, log_level);
 | 
			
		||||
        epee::log_space::log_singletone::add_logger(LOGGER_CONSOLE, NULL, NULL);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    uint64_t
 | 
			
		||||
    estimate_bc_height(const string& date, const char* format = "%Y-%m-%d");
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    inline double
 | 
			
		||||
    get_xmr(uint64_t core_amount)
 | 
			
		||||
    {
 | 
			
		||||
        return  static_cast<double>(core_amount) / 1e12;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    array<size_t, 5>
 | 
			
		||||
    timestamp_difference(uint64_t t1, uint64_t t2);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#endif //XMREG01_TOOLS_H
 | 
			
		||||
							
								
								
									
										292
									
								
								src/tx_details.cpp
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										292
									
								
								src/tx_details.cpp
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,292 @@
 | 
			
		|||
//
 | 
			
		||||
// Created by mwo on 14/11/15.
 | 
			
		||||
//
 | 
			
		||||
 | 
			
		||||
#include "tx_details.h"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
namespace xmreg
 | 
			
		||||
{
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    crypto::hash
 | 
			
		||||
    transfer_details::tx_hash() const
 | 
			
		||||
    {
 | 
			
		||||
        return get_transaction_hash(m_tx);
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    uint64_t
 | 
			
		||||
    transfer_details::amount() const
 | 
			
		||||
    {
 | 
			
		||||
        return m_tx.vout[m_internal_output_index].amount;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    ostream&
 | 
			
		||||
    operator<<(ostream& os, const transfer_details& td)
 | 
			
		||||
    {
 | 
			
		||||
        os << "Block: "    << td.m_block_height
 | 
			
		||||
           << " time: "    << timestamp_to_str(td.m_block_timestamp)
 | 
			
		||||
           << " tx hash: " << td.tx_hash()
 | 
			
		||||
           << " out idx: " << td.m_internal_output_index
 | 
			
		||||
           << " amount: "  << print_money(td.amount());
 | 
			
		||||
 | 
			
		||||
        return os;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Get tx outputs associated with the given private view and public spend keys
 | 
			
		||||
     *
 | 
			
		||||
     *
 | 
			
		||||
     */
 | 
			
		||||
    vector<xmreg::transfer_details>
 | 
			
		||||
    get_belonging_outputs(const block& blk,
 | 
			
		||||
                          const transaction& tx,
 | 
			
		||||
                          const secret_key& private_view_key,
 | 
			
		||||
                          const public_key& public_spend_key,
 | 
			
		||||
                          uint64_t block_height)
 | 
			
		||||
    {
 | 
			
		||||
        // vector to be returned
 | 
			
		||||
        vector<xmreg::transfer_details> our_outputs;
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        // get transaction's public key
 | 
			
		||||
        public_key pub_tx_key = get_tx_pub_key_from_extra(tx);
 | 
			
		||||
 | 
			
		||||
        // check if transaction has valid public key
 | 
			
		||||
        // if no, then skip
 | 
			
		||||
        if (pub_tx_key == null_pkey)
 | 
			
		||||
        {
 | 
			
		||||
            return our_outputs;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        // get the total number of outputs in a transaction.
 | 
			
		||||
        size_t output_no = tx.vout.size();
 | 
			
		||||
 | 
			
		||||
        // check if the given transaction has any outputs
 | 
			
		||||
        // if no, then finish
 | 
			
		||||
        if (output_no == 0)
 | 
			
		||||
        {
 | 
			
		||||
            return our_outputs;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        // public transaction key is combined with our viewkey
 | 
			
		||||
        // to create, so called, derived key.
 | 
			
		||||
        key_derivation derivation;
 | 
			
		||||
 | 
			
		||||
        if (!generate_key_derivation(pub_tx_key, private_view_key, derivation))
 | 
			
		||||
        {
 | 
			
		||||
            cerr << "Cant get dervied key for: "  << "\n"
 | 
			
		||||
                 << "pub_tx_key: " << private_view_key << " and "
 | 
			
		||||
                 << "prv_view_key" << private_view_key << endl;
 | 
			
		||||
            return our_outputs;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        // each tx that we (or the address we are checking) received
 | 
			
		||||
        // contains a number of outputs.
 | 
			
		||||
        // some of them are ours, some not. so we need to go through
 | 
			
		||||
        // all of them in a given tx block, to check which outputs are ours.
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        // sum amount of xmr sent to us
 | 
			
		||||
        // in the given transaction
 | 
			
		||||
        uint64_t money_transfered {0};
 | 
			
		||||
 | 
			
		||||
        // loop through outputs in the given tx
 | 
			
		||||
        // to check which outputs our ours. we compare outputs'
 | 
			
		||||
        // public keys with the public key that would had been
 | 
			
		||||
        // generated for us if we had gotten the outputs.
 | 
			
		||||
        // not sure this is the case though, but that's my understanding.
 | 
			
		||||
        for (size_t i = 0; i < output_no; ++i)
 | 
			
		||||
        {
 | 
			
		||||
            // get the tx output public key
 | 
			
		||||
            // that normally would be generated for us,
 | 
			
		||||
            // if someone had sent us some xmr.
 | 
			
		||||
            public_key pubkey;
 | 
			
		||||
 | 
			
		||||
            derive_public_key(derivation,
 | 
			
		||||
                              i,
 | 
			
		||||
                              public_spend_key,
 | 
			
		||||
                              pubkey);
 | 
			
		||||
 | 
			
		||||
            // get tx output public key
 | 
			
		||||
            const txout_to_key tx_out_to_key
 | 
			
		||||
                    = boost::get<txout_to_key>(tx.vout[i].target);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
            //cout << "Output no: " << i << ", " << tx_out_to_key.key;
 | 
			
		||||
 | 
			
		||||
            // check if the output's public key is ours
 | 
			
		||||
            if (tx_out_to_key.key == pubkey)
 | 
			
		||||
            {
 | 
			
		||||
                // if so, then add this output to the
 | 
			
		||||
                // returned vector
 | 
			
		||||
                //our_outputs.push_back(tx.vout[i]);
 | 
			
		||||
                our_outputs.push_back(
 | 
			
		||||
                        xmreg::transfer_details {block_height,
 | 
			
		||||
                                                 blk.timestamp,
 | 
			
		||||
                                                 tx, i, false}
 | 
			
		||||
                );
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        return our_outputs;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    /**
 | 
			
		||||
     * Check if given output (specified by output_index)
 | 
			
		||||
     * belongs is ours based
 | 
			
		||||
     * on our private view key and public spend key
 | 
			
		||||
     */
 | 
			
		||||
    bool
 | 
			
		||||
    is_output_ours(const size_t& output_index,
 | 
			
		||||
                   const transaction& tx,
 | 
			
		||||
                   const secret_key& private_view_key,
 | 
			
		||||
                   const public_key& public_spend_key)
 | 
			
		||||
    {
 | 
			
		||||
        // get transaction's public key
 | 
			
		||||
        public_key pub_tx_key = get_tx_pub_key_from_extra(tx);
 | 
			
		||||
 | 
			
		||||
        // check if transaction has valid public key
 | 
			
		||||
        // if no, then skip
 | 
			
		||||
        if (pub_tx_key == null_pkey)
 | 
			
		||||
        {
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        // public transaction key is combined with our viewkey
 | 
			
		||||
        // to create, so called, derived key.
 | 
			
		||||
        key_derivation derivation;
 | 
			
		||||
 | 
			
		||||
        if (!generate_key_derivation(pub_tx_key, private_view_key, derivation))
 | 
			
		||||
        {
 | 
			
		||||
            cerr << "Cant get dervied key for: "  << "\n"
 | 
			
		||||
                 << "pub_tx_key: " << pub_tx_key  << " and "
 | 
			
		||||
                 << "prv_view_key" << private_view_key << endl;
 | 
			
		||||
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        // get the tx output public key
 | 
			
		||||
        // that normally would be generated for us,
 | 
			
		||||
        // if someone had sent us some xmr.
 | 
			
		||||
        public_key pubkey;
 | 
			
		||||
 | 
			
		||||
        derive_public_key(derivation,
 | 
			
		||||
                          output_index,
 | 
			
		||||
                          public_spend_key,
 | 
			
		||||
                          pubkey);
 | 
			
		||||
 | 
			
		||||
        //cout << "\n" << tx.vout.size() << " " << output_index << endl;
 | 
			
		||||
 | 
			
		||||
        // get tx output public key
 | 
			
		||||
        const txout_to_key tx_out_to_key
 | 
			
		||||
                = boost::get<txout_to_key>(tx.vout[output_index].target);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        if (tx_out_to_key.key == pubkey)
 | 
			
		||||
        {
 | 
			
		||||
            return true;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        return false;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    bool
 | 
			
		||||
    get_payment_id(const transaction& tx,
 | 
			
		||||
                   crypto::hash& payment_id)
 | 
			
		||||
    {
 | 
			
		||||
 | 
			
		||||
        payment_id = null_hash;
 | 
			
		||||
 | 
			
		||||
        std::vector<tx_extra_field> tx_extra_fields;
 | 
			
		||||
 | 
			
		||||
        if(!parse_tx_extra(tx.extra, tx_extra_fields))
 | 
			
		||||
        {
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        tx_extra_nonce extra_nonce;
 | 
			
		||||
 | 
			
		||||
        if (find_tx_extra_field_by_type(tx_extra_fields, extra_nonce))
 | 
			
		||||
        {
 | 
			
		||||
            if (!get_payment_id_from_tx_extra_nonce(extra_nonce.nonce, payment_id))
 | 
			
		||||
            {
 | 
			
		||||
                return false;
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        else
 | 
			
		||||
        {
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        return true;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    bool
 | 
			
		||||
    get_encrypted_payment_id(const transaction& tx,
 | 
			
		||||
                             crypto::hash8& payment_id8)
 | 
			
		||||
    {
 | 
			
		||||
        payment_id8 = null_hash8;
 | 
			
		||||
 | 
			
		||||
        std::vector<tx_extra_field> tx_extra_fields;
 | 
			
		||||
 | 
			
		||||
        if(!parse_tx_extra(tx.extra, tx_extra_fields))
 | 
			
		||||
        {
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        tx_extra_nonce extra_nonce;
 | 
			
		||||
 | 
			
		||||
        if (find_tx_extra_field_by_type(tx_extra_fields, extra_nonce))
 | 
			
		||||
        {
 | 
			
		||||
 | 
			
		||||
            if (!get_encrypted_payment_id_from_tx_extra_nonce(extra_nonce.nonce, payment_id8))
 | 
			
		||||
            {
 | 
			
		||||
    //            if (decrypt_payment_id(payment_id8, addr.m_view_public_key, ptx.tx_key))
 | 
			
		||||
    //            {
 | 
			
		||||
    //                memcpy(payment_id8.data, payment_id8.data, 8);
 | 
			
		||||
    //            }
 | 
			
		||||
                return false;
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        else
 | 
			
		||||
        {
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        return true;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
template<>
 | 
			
		||||
csv::ofstream&
 | 
			
		||||
operator<<(csv::ofstream& ostm, const xmreg::transfer_details& td)
 | 
			
		||||
{
 | 
			
		||||
 | 
			
		||||
    ostm << xmreg::timestamp_to_str(td.m_block_timestamp, "%F");
 | 
			
		||||
    ostm << xmreg::timestamp_to_str(td.m_block_timestamp, "%T");
 | 
			
		||||
    ostm << td.m_block_height;
 | 
			
		||||
    ostm << td.tx_hash();
 | 
			
		||||
    ostm << td.m_internal_output_index;
 | 
			
		||||
    ostm << cryptonote::print_money(td.amount());
 | 
			
		||||
 | 
			
		||||
    return ostm;
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										71
									
								
								src/tx_details.h
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										71
									
								
								src/tx_details.h
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,71 @@
 | 
			
		|||
//
 | 
			
		||||
// Created by mwo on 14/11/15.
 | 
			
		||||
//
 | 
			
		||||
 | 
			
		||||
#ifndef XMR2CSV_TXDATA_H
 | 
			
		||||
#define XMR2CSV_TXDATA_H
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
#include "../ext/minicsv.h"
 | 
			
		||||
 | 
			
		||||
#include "monero_headers.h"
 | 
			
		||||
#include "tools.h"
 | 
			
		||||
 | 
			
		||||
namespace xmreg
 | 
			
		||||
{
 | 
			
		||||
 | 
			
		||||
    using namespace cryptonote;
 | 
			
		||||
    using namespace crypto;
 | 
			
		||||
    using namespace std;
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    struct transfer_details
 | 
			
		||||
    {
 | 
			
		||||
        uint64_t m_block_height;
 | 
			
		||||
        uint64_t m_block_timestamp;
 | 
			
		||||
        transaction m_tx;
 | 
			
		||||
        size_t m_internal_output_index;
 | 
			
		||||
        bool m_spent;
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        crypto::hash tx_hash() const;
 | 
			
		||||
 | 
			
		||||
        uint64_t amount() const;
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    ostream&
 | 
			
		||||
    operator<<(ostream& os, const transfer_details& dt);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    vector<xmreg::transfer_details>
 | 
			
		||||
    get_belonging_outputs(const block& blk,
 | 
			
		||||
                          const transaction& tx,
 | 
			
		||||
                          const secret_key& private_view_key,
 | 
			
		||||
                          const public_key& public_spend_key,
 | 
			
		||||
                          uint64_t block_height = 0);
 | 
			
		||||
 | 
			
		||||
    bool
 | 
			
		||||
    is_output_ours(const size_t& output_index,
 | 
			
		||||
                   const transaction& tx,
 | 
			
		||||
                   const secret_key& private_view_key,
 | 
			
		||||
                   const public_key& public_spend_key);
 | 
			
		||||
 | 
			
		||||
    bool
 | 
			
		||||
    get_payment_id(const transaction& tx,
 | 
			
		||||
                   crypto::hash& payment_id);
 | 
			
		||||
 | 
			
		||||
    bool
 | 
			
		||||
    get_encrypted_payment_id(const transaction& tx,
 | 
			
		||||
                             crypto::hash8& payment_id);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
template<>
 | 
			
		||||
csv::ostringstream&
 | 
			
		||||
operator<<(csv::ostringstream& ostm, const xmreg::transfer_details& td);
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
#endif //XMR2CSV_TXDATA_H
 | 
			
		||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue