instaweb_context.h

Go to the documentation of this file.

// Copyright 2010 Google Inc.

#ifndef PAGESPEED_APACHE_INSTAWEB_CONTEXT_H_
#define PAGESPEED_APACHE_INSTAWEB_CONTEXT_H_

#include "net/instaweb/http/public/request_context.h"
#include "pagespeed/automatic/html_detector.h"
#include "pagespeed/kernel/base/basictypes.h"
#include "pagespeed/kernel/base/scoped_ptr.h"
#include "pagespeed/kernel/base/string.h"
#include "pagespeed/kernel/base/string_writer.h"
#include "pagespeed/kernel/http/content_type.h"

#include "pagespeed/apache/apache_httpd_includes.h"
#include "apr_pools.h"

namespace net_instaweb {

class ApacheServerContext;
class GzipInflater;
class QueryParams;
class RequestHeaders;
class ResponseHeaders;
class RewriteDriver;
class RewriteOptions;

const char kPagespeedOriginalUrl[] = "mod_pagespeed_original_url";

template <class T>
apr_status_t apache_cleanup(void* object) {
  T* resolved = static_cast<T*>(object);
  delete resolved;
  return APR_SUCCESS;
}

class InstawebContext {
 public:
  enum ContentEncoding { kNone, kGzip, kDeflate, kOther };
  enum ContentDetectionState { kStart, kHtml, kNotHtml };

  InstawebContext(request_rec* request,
                  RequestHeaders* request_headers,
                  const ContentType& content_type,
                  ApacheServerContext* server_context,
                  const GoogleString& base_url,
                  const RequestContextPtr& request_context,
                  const QueryParams& pagespeed_query_params,
                  const QueryParams& pagespeed_option_cookies,
                  bool use_custom_options,
                  const RewriteOptions& options);
  ~InstawebContext();

  void Rewrite(const char* input, int size);
  void Flush();
  void Finish();

  apr_bucket_brigade* bucket_brigade() const { return bucket_brigade_; }
  ContentEncoding content_encoding() const { return  content_encoding_; }
  ApacheServerContext* apache_server_context() { return server_context_; }
  const GoogleString& output() { return output_; }
  bool empty() const { return output_.empty(); }
  void clear() { output_.clear(); }  

  ResponseHeaders* response_headers() {
    return response_headers_.get();
  }

  bool sent_headers() { return sent_headers_; }
  void set_sent_headers(bool sent) { sent_headers_ = sent; }

  void PopulateHeaders(request_rec* request);

  static ApacheServerContext* ServerContextFromServerRec(server_rec* server);

  static const char* MakeRequestUrl(const RewriteOptions& global_options,
                                    request_rec* request);

 private:
  void ComputeContentEncoding(request_rec* request);
  void BlockingPropertyCacheLookup();
  void ProcessBytes(const char* input, int size);

  void SetExperimentStateAndCookie(request_rec* request,
                                   RewriteOptions* options);

  GoogleString output_; 
  apr_bucket_brigade* bucket_brigade_;
  ContentEncoding content_encoding_;
  const ContentType content_type_;

  ApacheServerContext* server_context_;
  RewriteDriver* rewrite_driver_;
  StringWriter string_writer_;
  scoped_ptr<GzipInflater> inflater_;
  HtmlDetector html_detector_;
  GoogleString absolute_url_;
  scoped_ptr<RequestHeaders> request_headers_;
  scoped_ptr<ResponseHeaders> response_headers_;
  bool started_parse_;
  bool sent_headers_;
  bool populated_headers_;

  
};

}  

#endif