content-disposition.hpp

Go to the documentation of this file.

/*
 * RESTinio
 */

 
#pragma once
 
#include <restinio/helpers/http_field_parsers/basics.hpp>
 
#include <restinio/helpers/http_field_parsers/details/pct_encoded_symbols.hpp>
 
namespace restinio
{
 
namespace http_field_parsers
{
 
namespace content_disposition_details
{
 
namespace ep_impl = restinio::easy_parser::impl;
namespace hfp_impl = restinio::http_field_parsers::impl;
namespace hfp_details = restinio::http_field_parsers::details;
 
//
// regular_token_producer_t
//
class regular_token_producer_t
    :   public hfp_impl::token_producer_t
{
public:
    [[nodiscard]]
    expected_t< result_type, parse_error_t >
    try_parse( ep_impl::source_t & from ) const
    {
        ep_impl::source_t::content_consumer_t consumer{ from };
        const auto result = hfp_impl::token_producer_t::try_parse( from );
        if( result )
        {
            if( '*' == *(result->rbegin()) )
            {
                // Regular token can't have the trailing '*'.
                return make_unexpected( parse_error_t{
                        consumer.started_at() + result->size() - 1,
                        error_reason_t::unexpected_character
                    } );
            }
 
            consumer.commit();
        }
 
        return result;
    }
};
 
//
// ext_token_producer_t
//
class ext_token_producer_t
    :   public hfp_impl::token_producer_t
{
public:
    [[nodiscard]]
    expected_t< result_type, parse_error_t >
    try_parse( ep_impl::source_t & from ) const
    {
        ep_impl::source_t::content_consumer_t consumer{ from };
        const auto result = hfp_impl::token_producer_t::try_parse( from );
        if( result )
        {
            if( '*' != *(result->rbegin()) )
            {
                // Extended token should have the trailing '*'.
                return make_unexpected( parse_error_t{
                        consumer.started_at(),
                        error_reason_t::pattern_not_found
                    } );
            }
 
            consumer.commit();
        }
 
        return result;
    }
};
 
//
// mime_charsetc_predicate_t
//
struct mime_charsetc_predicate_t
{
    [[nodiscard]]
    bool
    operator()( const char actual ) const noexcept
    {
        return hfp_impl::is_alpha(actual)
                || hfp_impl::is_digit(actual)
                || '!' == actual
                || '#' == actual
                || '$' == actual
                || '%' == actual
                || '&' == actual
                || '+' == actual
                || '-' == actual
                || '^' == actual
                || '_' == actual
                || '`' == actual
                || '{' == actual
                || '}' == actual
                || '~' == actual
                ;
    }
};
 
//
// mime_charsetc_symbol_producer
//
[[nodiscard]]
inline auto
mime_charsetc_symbol_p()
{
    return ep_impl::symbol_producer_template_t< mime_charsetc_predicate_t >{};
}
 
//
// language_predicate_t
//
struct language_predicate_t
{
    [[nodiscard]]
    bool
    operator()( const char actual ) const noexcept
    {
        return hfp_impl::is_alpha(actual)
                || hfp_impl::is_digit(actual)
                || '-' == actual
                ;
    }
};
 
//
// language_symbol_producer
//
[[nodiscard]]
inline auto
language_symbol_p()
{
    return ep_impl::symbol_producer_template_t< language_predicate_t >{};
}
 
//
// attr_char_predicate_t
//
struct attr_char_predicate_t
{
    [[nodiscard]]
    bool
    operator()( const char actual ) const noexcept
    {
        return hfp_impl::is_alpha(actual)
                || hfp_impl::is_digit(actual)
                || '!' == actual
                || '#' == actual
                || '$' == actual
                || '&' == actual
                || '+' == actual
                || '-' == actual
                || '.' == actual
                || '^' == actual
                || '_' == actual
                || '`' == actual
                || '|' == actual
                || '~' == actual
                ;
    }
};
 
//
// attr_char_symbol_producer
//
[[nodiscard]]
inline auto
attr_char_symbol_p()
{
    return ep_impl::symbol_producer_template_t< attr_char_predicate_t >{};
}
 
//
// ext_parameter_value_producer
//
[[nodiscard]]
inline auto
ext_parameter_value_p()
{
    return produce< std::string >(
            repeat( 1, N, mime_charsetc_symbol_p() >> to_container() ),
            symbol_p( '\'' ) >> to_container(),
            repeat( 0, N, language_symbol_p() >> to_container() ),
            symbol_p( '\'' ) >> to_container(),
            repeat( 0, N,
                alternatives(
                    attr_char_symbol_p() >> to_container(),
                    hfp_details::pct_encoded_symbols_p() >>
                            hfp_details::pct_encoded_symbols_consumer_t{} )
            )
        );
}
 
} /* namespace content_disposition_details */
 
//
// content_disposition_value_t
//
struct content_disposition_value_t
{
    using parameter_t = parameter_with_mandatory_value_t;
 
    using parameter_container_t = parameter_with_mandatory_value_container_t;
 
    std::string value;
    parameter_container_t parameters;

    [[nodiscard]]
    static auto
    make_parser()
    {
        using namespace content_disposition_details;
 
        return produce< content_disposition_value_t >(
            token_p() >> to_lower()
                    >> &content_disposition_value_t::value,
            produce< parameter_container_t >(
                repeat( 0, N,
                    produce< parameter_t >(
                        ows(),
                        symbol(';'),
                        ows(),
                        alternatives(
                            sequence(
                                regular_token_producer_t{}
                                        >> to_lower() >> &parameter_t::first,
                                symbol('='),
                                alternatives(
                                    token_p() >> &parameter_t::second,
                                    quoted_string_p() >> &parameter_t::second
                                )
                            ),
                            sequence(
                                ext_token_producer_t{}
                                        >> to_lower() >> &parameter_t::first,
                                symbol('='),
                                ext_parameter_value_p() >> &parameter_t::second
                            )
                        )
                    ) >> to_container()
                )
            ) >> &content_disposition_value_t::parameters
        );
    }

    [[nodiscard]]
    static expected_t< content_disposition_value_t, restinio::easy_parser::parse_error_t >
    try_parse( string_view_t what )
    {
        return restinio::easy_parser::try_parse( what, make_parser() );
    }
};
 
} /* namespace http_field_parsers */
 
} /* namespace restinio */