Change Lo-Fi bool to bitmask to support multiple Previews types

components/data_reduction_proxy/content/browser/content_lofi_decider.cc

 // Copyright 2015 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.
 
 #include "components/data_reduction_proxy/content/browser/content_lofi_decider.h"
 
 #include <string>
 
 #include "base/strings/string_split.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "components/data_reduction_proxy/core/common/data_reduction_proxy_headers.h"
 #include "components/data_reduction_proxy/core/common/data_reduction_proxy_params.h"
 #include "content/public/browser/resource_request_info.h"
+#include "content/public/common/previews_state.h"
 #include "content/public/common/resource_type.h"
 #include "net/base/load_flags.h"
 #include "net/http/http_request_headers.h"
 #include "net/url_request/url_request.h"
 
 namespace data_reduction_proxy {
 
 ContentLoFiDecider::ContentLoFiDecider() {}
 
 ContentLoFiDecider::~ContentLoFiDecider() {}
 
 bool ContentLoFiDecider::IsUsingLoFiMode(const net::URLRequest& request) const {
   const content::ResourceRequestInfo* request_info =
       content::ResourceRequestInfo::ForRequest(&request);
   // The Lo-Fi directive should not be added for users in the Lo-Fi field
   // trial "Control" group. Check that the user is in a group that can get
   // "q=low".
   bool lofi_enabled_via_flag_or_field_trial =
       params::IsLoFiOnViaFlags() || params::IsIncludedInLoFiEnabledFieldTrial();
 
   // Return if the user is using Lo-Fi and not part of the "Control" group.
-  if (request_info)
-    return request_info->IsUsingLoFi() && lofi_enabled_via_flag_or_field_trial;
+  if (request_info) {
+    return (request_info->GetPreviewsState() & content::SERVER_LOFI_ON) &&
+           lofi_enabled_via_flag_or_field_trial;
+  }
   return false;
 }
 
 void ContentLoFiDecider::MaybeSetAcceptTransformHeader(
     const net::URLRequest& request,
     bool is_previews_disabled,
     net::HttpRequestHeaders* headers) const {
   const content::ResourceRequestInfo* request_info =
       content::ResourceRequestInfo::ForRequest(&request);
 
@@ skipping 54 matching lines @@
   if (lite_page_enabled_via_flags_or_field_trial) {
     if (resource_type == content::RESOURCE_TYPE_MAIN_FRAME)
       accept_transform_value = lite_page_directive();
   } else if (lofi_enabled_via_flags_or_field_trial) {
     if (resource_type_supports_empty_image)
       accept_transform_value = empty_image_directive();
   }
   if (accept_transform_value.empty())
     return;
 
-  if (!request_info->IsUsingLoFi())
+  if (!(request_info->GetPreviewsState() & content::SERVER_LOFI_ON))
     accept_transform_value += base::StringPrintf(";%s", if_heavy_qualifier());
 
   headers->SetHeader(chrome_proxy_accept_transform_header(),
                      accept_transform_value);
 }
 
 bool ContentLoFiDecider::IsSlowPagePreviewRequested(
     const net::HttpRequestHeaders& headers) const {
   std::string accept_transform_header_value;
   if (!headers.GetHeader(chrome_proxy_accept_transform_header(),
@@ skipping 50 matching lines @@
       chrome_proxy_lite_page_ignore_blacklist_directive();
   headers->SetHeader(chrome_proxy_header(), chrome_proxy_header_value);
 }
 
 bool ContentLoFiDecider::ShouldRecordLoFiUMA(
     const net::URLRequest& request) const {
   const content::ResourceRequestInfo* request_info =
       content::ResourceRequestInfo::ForRequest(&request);
 
   // User is not using Lo-Fi.
-  if (!request_info || !request_info->IsUsingLoFi())
+  if (!request_info ||
+      !(request_info->GetPreviewsState() & content::SERVER_LOFI_ON)) {
     return false;
+  }
 
   return params::IsIncludedInLoFiEnabledFieldTrial() ||
          params::IsIncludedInLoFiControlFieldTrial();
 }
 
 }  // namespace data_reduction_proxy