| // Copyright 2014 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 "chrome/browser/dom_distiller/profile_utils.h" |
| |
| #include <memory> |
| #include <utility> |
| |
| #include "base/command_line.h" |
| #include "chrome/browser/dom_distiller/dom_distiller_service_factory.h" |
| #include "chrome/browser/dom_distiller/lazy_dom_distiller_service.h" |
| #include "chrome/browser/profiles/profile.h" |
| #include "chrome/common/chrome_isolated_world_ids.h" |
| #include "chrome/common/chrome_switches.h" |
| #include "components/dom_distiller/content/browser/distiller_javascript_utils.h" |
| #include "components/dom_distiller/content/browser/distiller_ui_handle.h" |
| #include "components/dom_distiller/content/browser/dom_distiller_viewer_source.h" |
| #include "components/dom_distiller/core/dom_distiller_switches.h" |
| #include "components/dom_distiller/core/url_constants.h" |
| |
| #if defined(OS_ANDROID) |
| #include "chrome/browser/android/dom_distiller/distiller_ui_handle_android.h" |
| #endif // defined(OS_ANDROID) |
| |
| namespace dom_distiller { |
| |
| void RegisterViewerSource(Profile* profile) { |
| bool enabled_distiller = base::CommandLine::ForCurrentProcess()->HasSwitch( |
| switches::kEnableDomDistiller); |
| if (!enabled_distiller) |
| return; |
| |
| DomDistillerServiceFactory* dom_distiller_service_factory = |
| DomDistillerServiceFactory::GetInstance(); |
| // The LazyDomDistillerService deletes itself when the profile is destroyed. |
| LazyDomDistillerService* lazy_service = |
| new LazyDomDistillerService(profile, dom_distiller_service_factory); |
| std::unique_ptr<DistillerUIHandle> ui_handle; |
| |
| #if defined(OS_ANDROID) |
| ui_handle = |
| std::make_unique<dom_distiller::android::DistillerUIHandleAndroid>(); |
| #endif // defined(OS_ANDROID) |
| |
| // Set the JavaScript world ID. |
| if (!DistillerJavaScriptWorldIdIsSet()) |
| SetDistillerJavaScriptWorldId(ISOLATED_WORLD_ID_CHROME_INTERNAL); |
| |
| content::URLDataSource::Add( |
| profile, std::make_unique<DomDistillerViewerSource>( |
| lazy_service, kDomDistillerScheme, std::move(ui_handle))); |
| } |
| |
| } // namespace dom_distiller |