MD Downloads: gzip vulcanized resources via compress="gzip"

content/browser/webui/web_ui_data_source_impl.cc

 // 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.
 
 #include "content/browser/webui/web_ui_data_source_impl.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <string>
@@ skipping 70 matching lines @@
   }
   std::string GetContentSecurityPolicyChildSrc() const override {
     if (parent_->frame_src_set_)
       return parent_->frame_src_;
     return URLDataSource::GetContentSecurityPolicyChildSrc();
   }
   bool ShouldDenyXFrameOptions() const override {
     return parent_->deny_xframe_options_;
   }
   bool IsGzipped(const std::string& path) const override {
-    return parent_->use_gzip_for_all_paths_ &&
+    return parent_->use_gzip_ &&
         parent_->excluded_paths_.find(path) == parent_->excluded_paths_.end();
   }
 
  private:
   WebUIDataSourceImpl* parent_;
 };
 
 WebUIDataSourceImpl::WebUIDataSourceImpl(const std::string& source_name)
     : URLDataSourceImpl(source_name, new InternalDataSource(this)),
       source_name_(source_name),
       default_resource_(-1),
       add_csp_(true),
       script_src_set_(false),
       object_src_set_(false),
       frame_src_set_(false),
       deny_xframe_options_(true),
       add_load_time_data_defaults_(true),
       replace_existing_source_(true),
-      use_gzip_for_all_paths_(false) {}
+      use_gzip_(false) {}
 
 WebUIDataSourceImpl::~WebUIDataSourceImpl() {
 }
 
 void WebUIDataSourceImpl::AddString(const std::string& name,
                                     const base::string16& value) {
   // TODO(dschuyler): Share only one copy of these strings.
   localized_strings_.SetString(name, value);
   replacements_[name] = base::UTF16ToUTF8(value);
 }
@@ skipping 37 matching lines @@
 
 void WebUIDataSourceImpl::AddInteger(const std::string& name, int32_t value) {
   localized_strings_.SetInteger(name, value);
 }
 
 void WebUIDataSourceImpl::SetJsonPath(const std::string& path) {
   DCHECK(json_path_.empty());
   DCHECK(!path.empty());
 
   json_path_ = path;
-  ExcludePathFromGzip(json_path_);
+  excluded_paths_.insert(json_path_);
 }
 
 void WebUIDataSourceImpl::AddResourcePath(const std::string &path,
                                           int resource_id) {
   path_to_idr_map_[path] = resource_id;
 }
 
 void WebUIDataSourceImpl::SetDefaultResource(int resource_id) {
   default_resource_ = resource_id;
 }
 
 void WebUIDataSourceImpl::SetRequestFilter(
     const WebUIDataSource::HandleRequestCallback& callback) {
   filter_callback_ = callback;
 }
 
 void WebUIDataSourceImpl::DisableReplaceExistingSource() {
   replace_existing_source_ = false;
 }
 
-void WebUIDataSourceImpl::ExcludePathFromGzip(const std::string& path) {
-  excluded_paths_.insert(path);
-}
-
 bool WebUIDataSourceImpl::IsWebUIDataSourceImpl() const {
   return true;
 }
 
 void WebUIDataSourceImpl::DisableContentSecurityPolicy() {
   add_csp_ = false;
 }
 
 void WebUIDataSourceImpl::OverrideContentSecurityPolicyScriptSrc(
     const std::string& data) {
@@ skipping 10 matching lines @@
 void WebUIDataSourceImpl::OverrideContentSecurityPolicyChildSrc(
     const std::string& data) {
   frame_src_set_ = true;
   frame_src_ = data;
 }
 
 void WebUIDataSourceImpl::DisableDenyXFrameOptions() {
   deny_xframe_options_ = false;
 }
 
-void WebUIDataSourceImpl::DisableI18nAndUseGzipForAllPaths() {
-  use_gzip_for_all_paths_ = true;
+void WebUIDataSourceImpl::UseGzip(
+    const std::unordered_set<std::string>& excluded_paths) {
+  use_gzip_ = true;
+  for (const auto& path : excluded_paths)
+    excluded_paths_.insert(path);
 }
 
 const ui::TemplateReplacements* WebUIDataSourceImpl::GetReplacements() const {
   return &replacements_;
 }
 
 void WebUIDataSourceImpl::EnsureLoadTimeDataDefaultsAdded() {
   if (!add_load_time_data_defaults_)
     return;
 
@@ skipping 60 matching lines @@
 }
 
 void WebUIDataSourceImpl::SendLocalizedStringsAsJSON(
     const URLDataSource::GotDataCallback& callback) {
   std::string template_data;
   webui::AppendJsonJS(&localized_strings_, &template_data);
   callback.Run(base::RefCountedString::TakeString(&template_data));
 }
 
 }  // namespace content