Index: mojo/runner/url_resolver.cc |
diff --git a/mojo/runner/url_resolver.cc b/mojo/runner/url_resolver.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..4222ec7cec3989a215c5bd4ca5d3e22d6c9f55e5 |
--- /dev/null |
+++ b/mojo/runner/url_resolver.cc |
@@ -0,0 +1,117 @@ |
+// 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 "mojo/runner/url_resolver.h" |
+ |
+#include "base/base_paths.h" |
+#include "base/logging.h" |
+#include "base/strings/utf_string_conversions.h" |
+#include "mojo/runner/switches.h" |
+#include "mojo/shell/query_util.h" |
+#include "mojo/util/filename_util.h" |
+#include "url/url_util.h" |
+ |
+namespace mojo { |
+namespace runner { |
+ |
+URLResolver::URLResolver() { |
+ // Needed to treat first component of mojo URLs as host, not path. |
+ url::AddStandardScheme("mojo", url::SCHEME_WITHOUT_AUTHORITY); |
+} |
+ |
+URLResolver::~URLResolver() { |
+} |
+ |
+// static |
+std::vector<URLResolver::OriginMapping> URLResolver::GetOriginMappings( |
+ const base::CommandLine::StringVector& args) { |
+ std::vector<OriginMapping> origin_mappings; |
+ const std::string kArgsForSwitches[] = { |
+ "-" + std::string(switches::kMapOrigin) + "=", |
+ "--" + std::string(switches::kMapOrigin) + "=", |
+ }; |
+ for (auto& arg : args) { |
+ for (size_t i = 0; i < arraysize(kArgsForSwitches); i++) { |
+ const std::string& argsfor_switch = kArgsForSwitches[i]; |
+ std::string arg_string; |
+#if defined(OS_WIN) |
+ arg_string = base::UTF16ToUTF8(arg); |
+#else |
+ arg_string = arg; |
+#endif |
+ if (arg_string.compare(0, argsfor_switch.size(), argsfor_switch) == 0) { |
+ std::string value = |
+ arg_string.substr(argsfor_switch.size(), std::string::npos); |
+ size_t delim = value.find('='); |
+ if (delim <= 0 || delim >= value.size()) |
+ continue; |
+ origin_mappings.push_back( |
+ OriginMapping(value.substr(0, delim), |
+ value.substr(delim + 1, std::string::npos))); |
+ } |
+ } |
+ } |
+ return origin_mappings; |
+} |
+ |
+void URLResolver::AddURLMapping(const GURL& url, const GURL& mapped_url) { |
+ url_map_[url] = mapped_url; |
+} |
+ |
+void URLResolver::AddOriginMapping(const GURL& origin, const GURL& base_url) { |
+ if (!origin.is_valid() || !base_url.is_valid() || |
+ origin != origin.GetOrigin()) { |
+ // Disallow invalid mappings. |
+ LOG(ERROR) << "Invalid origin for mapping: " << origin; |
+ return; |
+ } |
+ // Force both origin and base_url to have trailing slashes. |
+ origin_map_[origin] = util::AddTrailingSlashIfNeeded(base_url); |
+} |
+ |
+GURL URLResolver::ApplyMappings(const GURL& url) const { |
+ std::string query; |
+ GURL mapped_url = shell::GetBaseURLAndQuery(url, &query); |
+ for (;;) { |
+ const auto& url_it = url_map_.find(mapped_url); |
+ if (url_it != url_map_.end()) { |
+ mapped_url = url_it->second; |
+ continue; |
+ } |
+ |
+ GURL origin = mapped_url.GetOrigin(); |
+ const auto& origin_it = origin_map_.find(origin); |
+ if (origin_it == origin_map_.end()) |
+ break; |
+ mapped_url = GURL(origin_it->second.spec() + |
+ mapped_url.spec().substr(origin.spec().length())); |
+ } |
+ |
+ if (query.length()) |
+ mapped_url = GURL(mapped_url.spec() + query); |
+ return mapped_url; |
+} |
+ |
+void URLResolver::SetMojoBaseURL(const GURL& mojo_base_url) { |
+ DCHECK(mojo_base_url.is_valid()); |
+ // Force a trailing slash on the base_url to simplify resolving |
+ // relative files and URLs below. |
+ mojo_base_url_ = util::AddTrailingSlashIfNeeded(mojo_base_url); |
+} |
+ |
+GURL URLResolver::ResolveMojoURL(const GURL& mojo_url) const { |
+ if (mojo_url.scheme() != "mojo") { |
+ // The mapping has produced some sort of non-mojo: URL - file:, http:, etc. |
+ return mojo_url; |
+ } |
+ |
+ // It's still a mojo: URL, use the default mapping scheme. |
+ std::string query; |
+ GURL base_url = shell::GetBaseURLAndQuery(mojo_url, &query); |
+ const std::string host = base_url.host(); |
+ return mojo_base_url_.Resolve(host + "/" + host + ".mojo" + query); |
+} |
+ |
+} // namespace runner |
+} // namespace mojo |