// Copyright 2012 The Chromium Authors // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. #ifndef CONTENT_BROWSER_BROWSER_URL_HANDLER_IMPL_H_ #define CONTENT_BROWSER_BROWSER_URL_HANDLER_IMPL_H_ #include <utility> #include <vector> #include "base/gtest_prod_util.h" #include "base/memory/singleton.h" #include "content/common/content_export.h" #include "content/public/browser/browser_url_handler.h" class GURL; namespace content { class BrowserContext; class CONTENT_EXPORT BrowserURLHandlerImpl : public BrowserURLHandler { public: // Returns the singleton instance. static BrowserURLHandlerImpl* GetInstance(); BrowserURLHandlerImpl(const BrowserURLHandlerImpl&) = delete; BrowserURLHandlerImpl& operator=(const BrowserURLHandlerImpl&) = delete; // BrowserURLHandler implementation: void RewriteURLIfNecessary(GURL* url, BrowserContext* browser_context) override; std::vector<GURL> GetPossibleRewrites( const GURL& url, BrowserContext* browser_context) override; void AddHandlerPair(URLHandler handler, URLHandler reverse_handler) override; // Like the //content-public RewriteURLIfNecessary overload (overridden // above), but if the original URL needs to be adjusted if the modified URL is // redirected, this method sets |*reverse_on_redirect| to true. void RewriteURLIfNecessary(GURL* url, BrowserContext* browser_context, bool* reverse_on_redirect); // Reverses the rewriting that was done for |original| using the new |url|. bool ReverseURLRewrite(GURL* url, const GURL& original, BrowserContext* browser_context); // Reverses |AddHandlerPair| for the given |handler|. void RemoveHandlerForTesting(URLHandler handler); private: // This object is a singleton: BrowserURLHandlerImpl(); ~BrowserURLHandlerImpl() override; friend struct base::DefaultSingletonTraits<BrowserURLHandlerImpl>; // The list of known URLHandlers, optionally with reverse-rewriters. typedef std::pair<URLHandler, URLHandler> HandlerPair; std::vector<HandlerPair> url_handlers_; FRIEND_TEST_ALL_PREFIXES(BrowserURLHandlerImplTest, BasicRewriteAndReverse); FRIEND_TEST_ALL_PREFIXES(BrowserURLHandlerImplTest, NullHandlerReverse); FRIEND_TEST_ALL_PREFIXES(BrowserURLHandlerImplTest, ViewSourceReverse); FRIEND_TEST_ALL_PREFIXES(BrowserURLHandlerImplTest, GetPossibleRewrites); }; } // namespace content #endif // CONTENT_BROWSER_BROWSER_URL_HANDLER_IMPL_H_