1 files changed, 107 insertions, 0 deletions
diff --git a/mojo/shell/local_fetcher.cc b/mojo/shell/local_fetcher.cc
new file mode 100644
index 0000000..a945dc2
--- /dev/null
+++ b/mojo/shell/local_fetcher.cc
@@ -0,0 +1,107 @@
+// 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/stringprintf.h"
+#include "base/strings/utf_string_conversions.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 "url/url_util.h"
+
+namespace mojo {
+namespace shell {
+
+namespace {
+
+void IgnoreResult(bool result) {
+}
+
+}  // namespace
+
+// A loader for local files.
+LocalFetcher::LocalFetcher(const GURL& url,
+                           const GURL& url_without_query,
+                           const FetchCallback& loader_callback)
+    : Fetcher(loader_callback), url_(url), path_(UrlToFile(url_without_query)) {
+  TRACE_EVENT1("mojo_shell", "LocalFetcher::LocalFetcher", "url", url.spec());
+  loader_callback_.Run(make_scoped_ptr(this));
+}
+
+base::FilePath LocalFetcher::UrlToFile(const GURL& url) {
+  DCHECK(url.SchemeIsFile());
+  url::RawCanonOutputW<1024> output;
+  url::DecodeURLEscapeSequences(url.path().data(),
+                                static_cast<int>(url.path().length()), &output);
+  base::string16 decoded_path = base::string16(output.data(), output.length());
+#if defined(OS_WIN)
+  base::TrimString(decoded_path, L"/", &decoded_path);
+  base::FilePath path(decoded_path);
+#else
+  base::FilePath path(base::UTF16ToUTF8(decoded_path));
+#endif
+  return path;
+}
+
+const GURL& LocalFetcher::GetURL() const {
+  return url_;
+}
+
+GURL LocalFetcher::GetRedirectURL() const {
+  return GURL::EmptyGURL();
+}
+
+URLResponsePtr LocalFetcher::AsURLResponse(base::TaskRunner* task_runner,
+                                           uint32_t skip) {
+  URLResponsePtr response(URLResponse::New());
+  response->url = String::From(url_);
+  DataPipe data_pipe;
+  response->body = data_pipe.consumer_handle.Pass();
+  int64 file_size;
+  if (base::GetFileSize(path_, &file_size)) {
+    response->headers = Array<String>(1);
+    response->headers[0] =
+        base::StringPrintf("Content-Length: %" PRId64, file_size);
+  }
+  common::CopyFromFile(path_, data_pipe.producer_handle.Pass(), skip,
+                       task_runner, base::Bind(&IgnoreResult));
+  return response.Pass();
+}
+
+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() {
+  return "";
+}
+
+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