Refactoring how conditions without URL attributes are handled.

chrome/browser/extensions/api/declarative_webrequest/webrequest_condition.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_CONDITION_H_
 #define CHROME_BROWSER_EXTENSIONS_API_DECLARATIVE_WEBREQUEST_WEBREQUEST_CONDITION_H_
 
+#include <map>
+#include <set>
 #include <string>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/memory/linked_ptr.h"
 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_condition_attribute.h"
 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_rule.h"
 #include "chrome/common/extensions/matcher/url_matcher.h"
 
 namespace extensions {
@@ skipping 22 matching lines @@
       const WebRequestConditionAttributes& condition_attributes);
   ~WebRequestCondition();
 
   // Factory method that instantiates a WebRequestCondition according to
   // the description |condition| passed by the extension API.
   static scoped_ptr<WebRequestCondition> Create(
       URLMatcherConditionFactory* url_matcher_condition_factory,
       const base::Value& condition,
       std::string* error);
 
-  // Returns whether the request is a match, given that the URLMatcher found
-  // a match for |url_matcher_conditions_|.
-  bool IsFulfilled(const WebRequestRule::RequestData& request_data) const;
+  // Returns whether the request matches this condition.  |url_matches| lists
+  // the IDs that match the request's URL. |url_match_trigger| is expected to
+  // be |this->url_matcher_conditions_->id()| if the latter is defined and also
+  // presend in |url_matches|, or -1 otherwise.
+  bool IsFulfilled(URLMatcherConditionSet::ID url_match_trigger,
+                   const std::set<URLMatcherConditionSet::ID> &url_matches,
+                   const WebRequestRule::RequestData &request_data) const;
 
   // Returns a URLMatcherConditionSet::ID which is the canonical representation
   // for all URL patterns that need to be matched by this WebRequestCondition.
   // This ID is registered in a URLMatcher that can inform us in case of a
   // match.
   URLMatcherConditionSet::ID url_matcher_condition_set_id() const {
+    DCHECK(url_matcher_conditions_.get());
     return url_matcher_conditions_->id();
   }
 
-  // Returns the set of conditions that are checked on the URL. This is the
-  // primary trigger for WebRequestCondition and therefore never empty.
-  // (If it was empty, the URLMatcher would never notify us about network
-  // requests which might fulfill the entire WebRequestCondition).
+  // Returns the set of conditions that are checked on the URL. May be NULL.
   scoped_refptr<URLMatcherConditionSet> url_matcher_condition_set() const {
     return url_matcher_conditions_;
   }
 
   // Returns the condition attributes checked by this condition.
   const WebRequestConditionAttributes condition_attributes() const {
     return condition_attributes_;
   }
 
   // Returns a bit vector representing extensions::RequestStage. The bit vector
   // contains a 1 for each request stage during which the condition can be
   // tested.
   int stages() const { return applicable_request_stages_; }
 
  private:
+  // Represents the 'url' attribute of this condition. If NULL, then there was
+  // no 'url' attribute in this condition.
   scoped_refptr<URLMatcherConditionSet> url_matcher_conditions_;
+
+  // All non-UrlFilter attributes of this condition.
   WebRequestConditionAttributes condition_attributes_;
 
   // Bit vector indicating all RequestStage during which all
   // |condition_attributes_| can be evaluated.
   int applicable_request_stages_;
 
   DISALLOW_COPY_AND_ASSIGN(WebRequestCondition);
 };
 
 // This class stores a set of conditions that may be part of a WebRequestRule.
 // If any condition is fulfilled, the WebRequestActions of the WebRequestRule
 // can be triggered.
 class WebRequestConditionSet {
  public:
   typedef std::vector<linked_ptr<json_schema_compiler::any::Any> > AnyVector;
   typedef std::vector<linked_ptr<WebRequestCondition> > Conditions;
 
-  explicit WebRequestConditionSet(const Conditions& conditions);
   ~WebRequestConditionSet();
 
   // Factory method that creates an WebRequestConditionSet according to the JSON
   // array |conditions| passed by the extension API.
   // Sets |error| and returns NULL in case of an error.
   static scoped_ptr<WebRequestConditionSet> Create(
       URLMatcherConditionFactory* url_matcher_condition_factory,
       const AnyVector& conditions,
       std::string* error);
 
   const Conditions& conditions() const {
     return conditions_;
   }
 
-  // Returns whether any condition in the condition set is fulfilled
-  // based on a match |url_match| and the value of |request_data.request|.
-  // This function should be called for each URLMatcherConditionSet::ID
-  // that was found by the URLMatcher to ensure that the each trigger in
-  // |match_triggers_| is found.
+  // If |url_match_trigger| is a member of |url_matches|, then this returns
+  // whether the corresponding condition is fulfilled wrt. |request_data|. Else
+  // it returns whether any of the conditions without URL attributes is
+  // satisfied.
   bool IsFulfilled(
-      URLMatcherConditionSet::ID url_match,
+      URLMatcherConditionSet::ID url_match_trigger,
+      const std::set<URLMatcherConditionSet::ID>& url_matches,
       const WebRequestRule::RequestData& request_data) const;
 
   // Appends the URLMatcherConditionSet from all conditions to |condition_sets|.
   void GetURLMatcherConditionSets(
       URLMatcherConditionSet::Vector* condition_sets) const;
 
+  // Returns whether there are some conditions without UrlFilter attributes.
+  bool HasConditionsWithoutUrls() const;
+
  private:
-  Conditions conditions_;
+  typedef std::map<URLMatcherConditionSet::ID, const WebRequestCondition*>
+      URLMatcherIdToCondition;
 
-  typedef std::map<URLMatcherConditionSet::ID, WebRequestCondition*>
-      MatchTriggers;
-  MatchTriggers match_triggers_;
+  WebRequestConditionSet(
+      const Conditions& conditions,
+      const URLMatcherIdToCondition& match_id_to_condition,
+      const std::vector<const WebRequestCondition*>& conditions_without_urls);
+
+  const URLMatcherIdToCondition match_id_to_condition_;
+  const Conditions conditions_;
+  const std::vector<const WebRequestCondition*> conditions_without_urls_;
 
   DISALLOW_COPY_AND_ASSIGN(WebRequestConditionSet);
 };
 
 }  // namespace extensions
 
 #endif  // CHROME_BROWSER_EXTENSIONS_API_DECLARATIVE_WEBREQUEST_WEBREQUEST_CONDITION_H_