instaweb_handler.h

Go to the documentation of this file.

// Copyright 2010 Google Inc.

#ifndef PAGESPEED_APACHE_INSTAWEB_HANDLER_H_
#define PAGESPEED_APACHE_INSTAWEB_HANDLER_H_

#include "pagespeed/apache/apache_fetch.h"
#include "net/instaweb/http/public/request_context.h"
#include "net/instaweb/rewriter/public/rewrite_driver.h"
#include "net/instaweb/rewriter/public/rewrite_options.h"
#include "net/instaweb/rewriter/public/rewrite_query.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_util.h"
#include "pagespeed/kernel/http/content_type.h"
#include "pagespeed/kernel/http/google_url.h"
#include "pagespeed/kernel/http/query_params.h"
#include "pagespeed/kernel/http/request_headers.h"
#include "pagespeed/kernel/http/response_headers.h"

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

namespace net_instaweb {

class ApacheConfig;
class ApacheRequestContext;
class ApacheRewriteDriverFactory;
class ApacheServerContext;
class InPlaceResourceRecorder;

class InstawebHandler {
 public:
  explicit InstawebHandler(request_rec* request);
  ~InstawebHandler();

  const GoogleUrl& stripped_gurl() const { return stripped_gurl_; }
  const RequestContextPtr request_context() const { return request_context_; }
  bool use_custom_options() const { return custom_options_.get() != NULL; }
  const QueryParams& query_params() { return rewrite_query_.query_params(); }
  const QueryParams& pagespeed_query_params() {
    return rewrite_query_.pagespeed_query_params();
  }
  const QueryParams& pagespeed_option_cookies() {
    return rewrite_query_.pagespeed_option_cookies();
  }

  void RemoveStrippedResponseHeadersFromApacheRequest();

  RewriteDriver* MakeDriver();

  void DisownDriver();

  ApacheFetch* MakeFetch(
      const GoogleString& url, bool buffered, StringPiece debug_info);

  ApacheFetch* MakeFetch(bool buffered, StringPiece debug_info) {
    return MakeFetch(original_url_, buffered, debug_info);
  }

  bool HandleAsProxy();

  void HandleAsProxyForAll();

  bool HandleAsInPlace();

  void HandleAsPagespeedResource();

  void WaitForFetch();

  bool ProxyUrl();

  bool AuthenticateProxy();

  RequestHeaders* ReleaseRequestHeaders() { return request_headers_.release(); }

  const ApacheConfig* options() { return options_; }

  static bool is_pagespeed_subrequest(request_rec* request);

  static apr_status_t instaweb_handler(request_rec* request);

  static apr_status_t save_url_hook(request_rec *request);

  static apr_status_t save_url_in_note(request_rec *request,
                                       ApacheServerContext* server_context);

  static apr_status_t instaweb_map_to_storage(request_rec* request);

  static void AboutToBeDoneWithRecorder(request_rec* request,
                                        InPlaceResourceRecorder* recorder);

 private:
  void ComputeCustomOptions();

  static bool IsCompressibleContentType(const char* content_type);

  static void send_out_headers_and_body(
      request_rec* request,
      const ResponseHeaders& response_headers,
      const GoogleString& output);

  static bool handle_as_resource(ApacheServerContext* server_context,
                                 request_rec* request,
                                 GoogleUrl* gurl);

  static void write_handler_response(const StringPiece& output,
                                     request_rec* request,
                                     ContentType content_type,
                                     const StringPiece& cache_control);
  static void write_handler_response(const StringPiece& output,
                                     request_rec* request);

  static const char* get_instaweb_resource_url(
      request_rec* request, ApacheServerContext* server_context);

  static int log_request_headers(void* logging_data, const char* key,
                                 const char* value);

  static void instaweb_static_handler(request_rec* request,
                                      ApacheServerContext* server_context);

  static apr_status_t instaweb_statistics_handler(
      request_rec* request, ApacheServerContext* server_context,
      ApacheRewriteDriverFactory* factory);

  static bool parse_query_params(const request_rec* request, GoogleString* data,
                                 apr_status_t* ret);

  static bool parse_body_from_post(const request_rec* request,
                                   GoogleString* data, apr_status_t* ret);

  static apr_status_t instaweb_beacon_handler(
      request_rec* request, ApacheServerContext* server_context);

  static bool IsBeaconUrl(const RewriteOptions::BeaconUrl& beacons,
                          const GoogleUrl& gurl);

  request_rec* request_;
  RequestContextPtr request_context_;
  ApacheRequestContext* apache_request_context_; 
  ApacheServerContext* server_context_;
  scoped_ptr<RequestHeaders> request_headers_;
  scoped_ptr<ResponseHeaders> response_headers_;
  GoogleString original_url_;
  GoogleUrl stripped_gurl_; 
  scoped_ptr<ApacheConfig> custom_options_;

  const ApacheConfig* options_;
  RewriteDriver* rewrite_driver_;
  bool driver_owned_;
  int num_response_attributes_;
  RewriteQuery rewrite_query_;
  ApacheFetch* fetch_;

  
};

}  

#endif