1 files changed, 79 insertions, 0 deletions
diff --git a/chrome/browser/extensions/app_file_handler_util.cc b/chrome/browser/extensions/app_file_handler_util.cc
new file mode 100644
index 0000000..df31c28
--- /dev/null
+++ b/chrome/browser/extensions/app_file_handler_util.cc
@@ -0,0 +1,79 @@
+// Copyright (c) 2012 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 "chrome/browser/extensions/app_file_handler_util.h"
+
+#include "net/base/mime_util.h"
+
+using extensions::Extension;
+
+namespace app_file_handler_util {
+
+typedef std::vector<Extension::FileHandlerInfo> FileHandlerList;
+
+const Extension::FileHandlerInfo* FileHandlerForId(
+    const Extension& app,
+    const std::string& handler_id) {
+  const FileHandlerList& file_handlers = app.file_handlers();
+  for (FileHandlerList::const_iterator i = file_handlers.begin();
+       i != file_handlers.end(); i++) {
+    if (i->id == handler_id)
+      return &*i;
+  }
+  return NULL;
+}
+
+const Extension::FileHandlerInfo* FirstFileHandlerForMimeType(
+    const Extension& app,
+    const std::string& mime_type) {
+  const FileHandlerList& file_handlers = app.file_handlers();
+  for (FileHandlerList::const_iterator i = file_handlers.begin();
+       i != file_handlers.end(); i++) {
+    for (std::set<std::string>::const_iterator t = i->types.begin();
+         t != i->types.end(); t++) {
+      if (net::MatchesMimeType(*t, mime_type))
+        return &*i;
+    }
+  }
+  return NULL;
+}
+
+std::vector<const Extension::FileHandlerInfo*> FindFileHandlersForMimeTypes(
+    const Extension& app,
+    const std::set<std::string>& mime_types) {
+  std::vector<const Extension::FileHandlerInfo*> handlers;
+  if (mime_types.empty())
+    return handlers;
+
+  // Look for file handlers which can handle all the MIME types specified.
+  const FileHandlerList& file_handlers = app.file_handlers();
+  for (FileHandlerList::const_iterator data = file_handlers.begin();
+       data != file_handlers.end(); ++data) {
+    bool handles_all_types = true;
+    for (std::set<std::string>::const_iterator type_iter = mime_types.begin();
+         type_iter != mime_types.end(); ++type_iter) {
+      if (!FileHandlerCanHandleFileWithMimeType(*data, *type_iter)) {
+        handles_all_types = false;
+        break;
+      }
+    }
+    if (handles_all_types)
+      handlers.push_back(&*data);
+  }
+  return handlers;
+}
+
+bool FileHandlerCanHandleFileWithMimeType(
+    const extensions::Extension::FileHandlerInfo& handler,
+    const std::string& mime_type) {
+  // TODO(benwells): this should check the file's extension as well.
+  for (std::set<std::string>::const_iterator type = handler.types.begin();
+       type != handler.types.end(); ++type) {
+    if (net::MatchesMimeType(*type, mime_type))
+      return true;
+  }
+  return false;
+}
+
+}  // namespace