Enables specifying mojo url mapping on command line

mojo/shell/desktop/mojo_main.cc

 // Copyright 2013 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 <algorithm>
 #include <iostream>
 
 #include "base/at_exit.h"
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/logging.h"
 #include "base/message_loop/message_loop.h"
 #include "base/strings/string_split.h"
 #include "base/strings/utf_string_conversions.h"
 #include "mojo/shell/child_process.h"
 #include "mojo/shell/context.h"
 #include "mojo/shell/init.h"
+#include "mojo/shell/mojo_url_resolver.h"
 #include "mojo/shell/switches.h"
 
 #if defined(COMPONENT_BUILD)
 #include "ui/gl/gl_surface.h"
 #endif
 
 namespace {
 
 #if defined(OS_LINUX)
 // Copied from ui/gfx/switches.cc to avoid a dependency on //ui/gfx
@@ skipping 35 matching lines @@
 
 void RunApps(mojo::shell::Context* context) {
   const auto& command_line = *base::CommandLine::ForCurrentProcess();
   for (const auto& arg : command_line.GetArgs())
     context->Run(GetAppURLAndSetArgs(arg, context));
 }
 
 void Usage() {
   std::cerr << "Launch Mojo applications.\n";
   std::cerr
-    << "Usage: mojo_shell"
-    << " [--" << switches::kArgsFor << "=<mojo-app>]"
-    << " [--" << switches::kContentHandlers << "=<handlers>]"
-    << " [--" << switches::kEnableExternalApplications << "]"
-    << " [--" << switches::kDisableCache << "]"
-    << " [--" << switches::kEnableMultiprocess << "]"
-    << " [--" << switches::kOrigin << "=<url-lib-path>]"
-    << " <mojo-app> ...\n\n"
-    << "A <mojo-app> is a Mojo URL or a Mojo URL and arguments within quotes.\n"
-    << "Example: mojo_shell \"mojo://mojo_js_standalone test.js\".\n"
-    << "<url-lib-path> is searched for shared libraries named by mojo URLs.\n"
-    << "The value of <handlers> is a comma separated list like:\n"
-    << "text/html,mojo://mojo_html_viewer,"
-    << "application/javascript,mojo://mojo_js_content_handler\n";
+      << "Usage: mojo_shell"
+      << " [--" << switches::kArgsFor << "=<mojo-app>]"
+      << " [--" << switches::kContentHandlers << "=<handlers>]"
+      << " [--" << switches::kEnableExternalApplications << "]"
+      << " [--" << switches::kDisableCache << "]"
+      << " [--" << switches::kEnableMultiprocess << "]"
+      << " [--" << switches::kOrigin << "=<url-lib-path>]"
+      << " [--" << switches::kURLMappings << "=from1=to1,from2=to2]"
+      << " <mojo-app> ...\n\n"
+      << "A <mojo-app> is a Mojo URL or a Mojo URL and arguments within "
+      << "quotes.\n"
+      << "Example: mojo_shell \"mojo://mojo_js_standalone test.js\".\n"
+      << "<url-lib-path> is searched for shared libraries named by mojo URLs.\n"
+      << "The value of <handlers> is a comma separated list like:\n"
+      << "text/html,mojo://mojo_html_viewer,"
+      << "application/javascript,mojo://mojo_js_content_handler\n";
+}
+
+bool ConfigureURLMappings(const std::string& mappings,
+                          mojo::shell::MojoURLResolver* resolver) {
+  base::StringPairs pairs;
+  if (!base::SplitStringIntoKeyValuePairs(mappings, '=', ',', &pairs))
+    return false;
+  using StringPair = std::pair<std::string, std::string>;
+  for (const StringPair& pair : pairs) {
+    const GURL from(pair.first);
+    const GURL to(pair.second);
+    if (!from.is_valid() || !to.is_valid())
+      return false;
+    resolver->AddCustomMapping(from, to);
+  }
+  return true;
 }
 
 }  // namespace
 
 int main(int argc, char** argv) {
   base::AtExitManager at_exit;
   base::CommandLine::Init(argc, argv);
 
   const base::CommandLine& command_line =
     *base::CommandLine::ForCurrentProcess();
@@ skipping 26 matching lines @@
     mojo::shell::Context shell_context;
     {
       base::MessageLoop message_loop;
       shell_context.Init();
 
       if (command_line.HasSwitch(switches::kOrigin)) {
         shell_context.mojo_url_resolver()->SetBaseURL(
             GURL(command_line.GetSwitchValueASCII(switches::kOrigin)));
       }
 
+      if (command_line.HasSwitch(switches::kURLMappings) &&
+          !ConfigureURLMappings(
+              command_line.GetSwitchValueASCII(switches::kURLMappings),
+              shell_context.mojo_url_resolver())) {
+        Usage();
+        return 0;
+      }
+
       for (const auto& kv : command_line.GetSwitches()) {
         if (kv.first == switches::kArgsFor)
           GetAppURLAndSetArgs(kv.second, &shell_context);
       }
 
       message_loop.PostTask(FROM_HERE, base::Bind(RunApps, &shell_context));
       message_loop.Run();
     }
   }
   return 0;
 }