headersTest.h

/* Copyright (C) 2012-2013 Justin Berger 
   The full license is available in the LICENSE file at the root of this project and is also available at http://opensource.org/licenses/MIT. */

#include <rikitiki/rikitiki>
#include <rikitiki/mongoose/server.h>
#include <rikitiki/configuration/configuration>


using namespace rikitiki;
using namespace rikitiki::mongoose;

namespace rikitiki {
  namespace examples {

    struct HeadersTestModule {
      void Register(Server& server){
    typedef HeadersTestModule T;
    server.AddHandler( CreateRoute<>::With(this, "/headers", &T::headers) );
    server.AddHandler( CreateRoute<std::string, std::string>::With(this, "/cookies/{name}/{value}", &T::set_cookie) );
    server.AddHandler( CreateRoute<>::With(this, "/cookies", &T::cookies) );
      }

      void headers(ConnContext& ctx){
    ctx << ContentType::text_plain
        << "Headers: \n";
    for(auto header : ctx.Headers())
      ctx << header.first << ": " << header.second << "\n";
      }

      void cookies(ConnContext& ctx){
    if(ctx.QueryString()["name"].size())
      ctx << Cookie(ctx.QueryString()["name"], ctx.QueryString()["value"]);
    
    ctx << ContentType::text_plain
        << "Cookies: \n";

    for(auto cook : ctx.Cookies())
      ctx << cook.first << ": " << cook.second << "\n";
    ctx << "Headers: \n";
    for(auto header : ctx.Headers())
      ctx << header.first << ": " << header.second << "\n";
      }

      void set_cookie(ConnContext& ctx, const std::string& name, const std::string& value){
    ctx << Cookie(name, value)
        << "<a href='/cookies/test/test'>test</a>\n";
      }
    };
  }
}