Support redirects by regular expression in declarative WebRequest API

chrome/browser/extensions/api/declarative_webrequest/webrequest_action.h

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CHROME_BROWSER_EXTENSIONS_API_DECLARATIVE_WEBREQUEST_WEBREQUEST_ACTION_H_
 #define CHROME_BROWSER_EXTENSIONS_API_DECLARATIVE_WEBREQUEST_WEBREQUEST_ACTION_H_
 #pragma once
 
 #include <list>
 #include <string>
 #include <vector>
 
 #include "base/compiler_specific.h"
 #include "base/memory/linked_ptr.h"
 #include "chrome/browser/extensions/api/declarative_webrequest/request_stages.h"
 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_rule.h"
 #include "chrome/common/extensions/api/events.h"
 #include "googleurl/src/gurl.h"
+#include "unicode/regex.h"
 
 namespace base {
 class DictionaryValue;
 class Time;
 class Value;
 }
 
 namespace extension_web_request_api_helpers {
 struct EventResponseDelta;
 }
@@ skipping 11 matching lines @@
 //
 // TODO(battre): Add method that corresponds to executing the action.
 class WebRequestAction {
  public:
   // Type identifiers for concrete WebRequestActions.
   enum Type {
     ACTION_CANCEL_REQUEST,
     ACTION_REDIRECT_REQUEST,
     ACTION_REDIRECT_TO_TRANSPARENT_IMAGE,
     ACTION_REDIRECT_TO_EMPTY_DOCUMENT,
+    ACTION_REDIRECT_BY_REGEX_DOCUMENT,
     ACTION_SET_REQUEST_HEADER,
     ACTION_REMOVE_REQUEST_HEADER,
     ACTION_ADD_RESPONSE_HEADER,
     ACTION_REMOVE_RESPONSE_HEADER,
     ACTION_IGNORE_RULES,
   };
 
   WebRequestAction();
   virtual ~WebRequestAction();
 
@@ skipping 149 matching lines @@
       net::URLRequest* request,
       RequestStages request_stage,
       const WebRequestRule::OptionalRequestData& optional_request_data,
       const std::string& extension_id,
       const base::Time& extension_install_time) const OVERRIDE;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(WebRequestRedirectToEmptyDocumentAction);
 };
 
+// Action that instructs to redirect a network request.
+class WebRequestRedirectByRegExAction : public WebRequestAction {
+ public:
+  // The |to_pattern| has to be passed in ICU syntax.
+  // TODO(battre): Change this to Perl style when migrated to RE2.
+  explicit WebRequestRedirectByRegExAction(
+      scoped_ptr<icu::RegexPattern> from_pattern,
+      const std::string& to_pattern);
+  virtual ~WebRequestRedirectByRegExAction();
+
+  // Conversion of capture group styles between Perl style ($1, $2, ...) and
+  // RE2 (\1, \2, ...).
+  static std::string PerlToRe2Style(const std::string& perl);
+
+  // Implementation of WebRequestAction:
+  virtual int GetStages() const OVERRIDE;
+  virtual Type GetType() const OVERRIDE;
+  virtual LinkedPtrEventResponseDelta CreateDelta(
+      net::URLRequest* request,
+      RequestStages request_stage,
+      const WebRequestRule::OptionalRequestData& optional_request_data,
+      const std::string& extension_id,
+      const base::Time& extension_install_time) const OVERRIDE;
+
+ private:
+  scoped_ptr<icu::RegexPattern> from_pattern_;
+  icu::UnicodeString to_pattern_;
+
+  DISALLOW_COPY_AND_ASSIGN(WebRequestRedirectByRegExAction);
+};
+
 // Action that instructs to set a request header.
 class WebRequestSetRequestHeaderAction : public WebRequestAction {
  public:
   WebRequestSetRequestHeaderAction(const std::string& name,
                                    const std::string& value);
   virtual ~WebRequestSetRequestHeaderAction();
 
   // Implementation of WebRequestAction:
   virtual int GetStages() const OVERRIDE;
   virtual Type GetType() const OVERRIDE;
@@ skipping 100 matching lines @@
   int minimum_priority_;
   DISALLOW_COPY_AND_ASSIGN(WebRequestIgnoreRulesAction);
 };
 
 // TODO(battre) Implement further actions:
 // Redirect by RegEx, Cookie manipulations, ...
 
 }  // namespace extensions
 
 #endif  // CHROME_BROWSER_EXTENSIONS_API_DECLARATIVE_WEBREQUEST_WEBREQUEST_ACTION_H_