Add GetMimeTypeFromFile to Mojo's network service.

mojo/shell/local_fetcher.cc

 // Copyright 2015 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/shell/local_fetcher.h"
 
 #include "base/bind.h"
 #include "base/files/file_util.h"
 #include "base/format_macros.h"
 #include "base/message_loop/message_loop.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "base/trace_event/trace_event.h"
 #include "mojo/common/common_type_converters.h"
 #include "mojo/common/data_pipe_utils.h"
 #include "mojo/common/url_type_converters.h"
+#include "mojo/services/network/public/interfaces/network_service.mojom.h"
 #include "mojo/util/filename_util.h"
 #include "url/url_util.h"
 
 namespace mojo {
 namespace shell {
 
 namespace {
 
 void IgnoreResult(bool result) {
 }
 
 }  // namespace
 
 // A loader for local files.
-LocalFetcher::LocalFetcher(const GURL& url,
+LocalFetcher::LocalFetcher(NetworkService* network_service,
+                           const GURL& url,
                            const GURL& url_without_query,
                            const FetchCallback& loader_callback)
     : Fetcher(loader_callback),
       url_(url),
       path_(util::UrlToFilePath(url_without_query)) {
   TRACE_EVENT1("mojo_shell", "LocalFetcher::LocalFetcher", "url", url.spec());
+  const std::string ext(base::FilePath(path_.Extension()).AsUTF8Unsafe());
+  if (network_service && !base::EqualsCaseInsensitiveASCII(ext, ".mojo")) {
+    network_service->GetMimeTypeFromFile(
+      path_.AsUTF8Unsafe(),
+      base::Bind(&LocalFetcher::GetMimeTypeFromFileCallback,
+                 base::Unretained(this)));
+  } else {
+    loader_callback_.Run(make_scoped_ptr(this));
+  }
+}
+
+void LocalFetcher::GetMimeTypeFromFileCallback(const mojo::String& mime_type) {
+  mime_type_ = mime_type.To<std::string>();
   loader_callback_.Run(make_scoped_ptr(this));
 }
 
 const GURL& LocalFetcher::GetURL() const {
   return url_;
 }
 
 GURL LocalFetcher::GetRedirectURL() const {
   return GURL::EmptyGURL();
 }
@@ skipping 24 matching lines @@
 
 void LocalFetcher::AsPath(
     base::TaskRunner* task_runner,
     base::Callback<void(const base::FilePath&, bool)> callback) {
   // Async for consistency with network case.
   base::MessageLoop::current()->PostTask(
       FROM_HERE, base::Bind(callback, path_, base::PathExists(path_)));
 }
 
 std::string LocalFetcher::MimeType() {
-  // TODO(msw): Expand support; use net::GetMimeTypeFromExtension or similar.
-  std::string extension(base::FilePath(path_.Extension()).AsUTF8Unsafe());
-  if (base::EqualsCaseInsensitiveASCII(extension, ".html") ||
-      base::EqualsCaseInsensitiveASCII(extension, ".htm") ||
-      base::EqualsCaseInsensitiveASCII(extension, ".shtml") ||
-      base::EqualsCaseInsensitiveASCII(extension, ".shtm") ||
-      base::EqualsCaseInsensitiveASCII(extension, ".ehtml")) {
-    return "text/html";
-  }
-  return std::string();
+  return mime_type_;
 }
 
 bool LocalFetcher::HasMojoMagic() {
   std::string magic;
   ReadFileToString(path_, &magic, strlen(kMojoMagic));
   return magic == kMojoMagic;
 }
 
 bool LocalFetcher::PeekFirstLine(std::string* line) {
   std::string start_of_file;
   ReadFileToString(path_, &start_of_file, kMaxShebangLength);
   size_t return_position = start_of_file.find('\n');
   if (return_position == std::string::npos)
     return false;
   *line = start_of_file.substr(0, return_position + 1);
   return true;
 }
 
 }  // namespace shell
 }  // namespace mojo