#include "config.h"
#include "worker_thread.h"
#include "global_config.h"
#include "base_view.h"
#ifndef CPPCMS_EMBEDDED
#include <boost/iostreams/filtering_stream.hpp>
#include <boost/iostreams/filter/gzip.hpp>
#endif
#include "manager.h"
using namespace cgicc;
namespace cppcms {
worker_thread::worker_thread(manager const &s) :
url(this),
app(s),
cache(this),
cout(&(this->out_buf)),
on_start(),
on_end(),
session(*this)
{
caching_module=app.cache->get();
static const transtext::trans tr;
gt=&tr;
} ;
worker_thread::~worker_thread()
{
app.cache->del(caching_module);
}
void worker_thread::main()
{
on_start();
cout<<"<h1>Hello World</h2>\n";
on_end();
}
void worker_thread::set_header(HTTPHeader *h)
{
response_header=auto_ptr<HTTPHeader>(h);
};
void worker_thread::add_header(string s) {
other_headers.push_back(s);
};
void worker_thread::set_cookie(cgicc::HTTPCookie const &c)
{
response_header->setCookie(c);
}
void worker_thread::set_user_io()
{
ostream &cout=cgi_conn->cout();
for(list<string>::iterator h=other_headers.begin();h!=other_headers.end();h++) {
cout<<*h<<"\n";
}
session.save();
cout<<header();
user_io=true;
}
void worker_thread::set_lang(string const &s)
{
lang=s;
gt=&app.gettext->get(s,"");
}
transtext::trans const *worker_thread::domain_gettext(string const &domain)
{
return &app.gettext->get(lang,domain);
}
HTTPHeader &worker_thread::header()
{
return *response_header;
}
void worker_thread::run(cgicc_connection &cgi_conn)
{
cgi=&cgi_conn.cgi();
env=&(cgi->getEnvironment());
ostream &cgi_out=cgi_conn.cout();
other_headers.clear();
cache.reset();
set_lang("");
out_buf.str("");
this->cgi_conn=&cgi_conn;
set_header(new HTTPHTMLHeader);
gzip=gzip_done=false;
user_io=false;
#ifndef CPPCMS_EMBEDDED
string encoding;
if((encoding=cgi_conn.env("HTTP_ACCEPT_ENCODING"))!="") {
if(strstr(encoding.c_str(),"gzip")!=NULL) {
gzip=app.config.lval("gzip.enable",1);
}
}
#endif
if(app.config.lval("server.disable_xpowered_by",0)==0) {
add_header("X-Powered-By: " PACKAGE_NAME "/" PACKAGE_VERSION);
}
try {
/**********/
session.on_start();
main();
session.on_end();
/**********/
if(response_header.get() == NULL) {
throw cppcms_error("Looks like a bug");
}
}
catch(std::exception const &e) {
string msg=e.what();
cgi_out<<"Content-Type: text/html\n";
cgi_out<<HTTPStatusHeader(500,msg);
cgi_out<<"<html><body><p>"+msg+"</p><body></html>";
gzip=gzip_done=false;
other_headers.clear();
out_buf.str("");
return;
}
if(user_io) {
// user controls it's IO
return;
}
for(list<string>::iterator h=other_headers.begin();h!=other_headers.end();h++) {
cgi_out<<*h<<"\n";
}
string out=out_buf.str();
out_buf.str("");
#ifndef CPPCMS_EMBEDDED
if(gzip) {
if(out.size()>0) {
if(gzip_done){
cgi_out<<"Content-Length: "<<out.size()<<"\n";
}
cgi_out<<"Content-Encoding: gzip\n";
cgi_out<<*response_header;
if(gzip_done) {
cgi_out<<out;
}
else{
int level=app.config.ival("gzip.level",-1);
int length=app.config.ival("gzip.buffer",-1);
deflate(out,cgi_out,level,length);
}
}
else {
cgi_out<<*response_header;
}
}
else
#endif
{
cgi_out<<"Content-Length: "<<out.size()<<"\n";
cgi_out<<*response_header;
cgi_out<<out;
}
}
void worker_thread::no_gzip()
{
gzip=false;
}
void worker_thread::render(string tmpl,string name,base_content &content,ostream &out )
{
using cppcms::details::views_storage;
base_view::settings s(this,&out);
auto_ptr<base_view> p(views_storage::instance().fetch_view(tmpl,name,s,&content));
if(!p.get()) throw cppcms_error("Template `"+name+"' not found in template set `" + tmpl +"'");
p->render();
};
void worker_thread::render(string tmpl,string name,base_content &content)
{
render(tmpl,name,content,cout);
};
void worker_thread::render(string name,base_content &content,ostream &o)
{
render(current_template,name,content,o);
};
void worker_thread::render(string name,base_content &content)
{
render(current_template,name,content,cout);
};
}
