servlet.h

Go to the documentation of this file.

/*
Copyright (c) 2016 Alexei Novakov
https://github.com/novalexei

Distributed under the Boost Software License, Version 1.0.
http://boost.org/LICENSE_1_0.txt
*/
#ifndef SERVLET_SERVLET_H
#define SERVLET_SERVLET_H

#include <string>

#include <servlet/request.h>
#include <servlet/response.h>
#include <servlet/context.h>

#define SERVLET_EXPORT(factoryName, className) extern "C" servlet::http_servlet* factoryName() { return new className{}; }

namespace servlet
{

class http_servlet
{
public:
    http_servlet() = default;
    virtual ~http_servlet() noexcept = default;

    virtual void init(servlet_config& config);

    virtual void init() {}

    virtual void service(http_request& req, http_response& resp);

    const servlet_config &get_servlet_config() const { return *_cfg; }

    const std::string &get_servlet_name() const { return _cfg->get_servlet_name(); }

    const std::map<std::string, std::string, std::less<>>& get_init_parameters() const
    { return _cfg->get_init_parameters(); };

    template <typename KeyType>
    optional_ref<const std::string> get_init_parameter(const KeyType& name) const
    { return _cfg->get_init_parameter(name); }

protected:
    virtual void do_get(http_request& req, http_response& resp);

    virtual void do_post(http_request& req, http_response& resp);

    virtual void do_put(http_request& req, http_response& resp);

    virtual void do_delete(http_request& req, http_response& resp);
    virtual void do_head(http_request& req, http_response& resp);

    virtual void do_trace(http_request& req, http_response& resp);

    virtual void do_options(http_request& req, http_response& resp);

    virtual long get_last_modified(http_request& req);

    constexpr static int GET_ALLOWED     = 1;
    constexpr static int POST_ALLOWED    = 1 << 1;
    constexpr static int PUT_ALLOWED     = 1 << 2;
    constexpr static int DELETE_ALLOWED  = 1 << 3;
    constexpr static int HEAD_ALLOWED    = 1 << 4;
    constexpr static int TRACE_ALLOWED   = 1 << 5;
    constexpr static int OPTIONS_ALLOWED = 1 << 6;

    virtual int get_allowed_methods();

private:
    void _maybe_set_last_modified(http_response &resp, long lastModifiedSec);

    static const std::string METHOD_DELETE;
    static const std::string METHOD_HEAD;
    static const std::string METHOD_GET;
    static const std::string METHOD_OPTIONS;
    static const std::string METHOD_POST;
    static const std::string METHOD_PUT;
    static const std::string METHOD_TRACE;

    static const std::string HEADER_IFMODSINCE;
    static const std::string HEADER_LASTMOD;

    servlet_config* _cfg;
};

}

#endif // SERVLET_SERVLET_H