Index: chrome/android/java/src/org/chromium/chrome/browser/photo_picker/MimeTypeFileFilter.java |
diff --git a/chrome/android/java/src/org/chromium/chrome/browser/photo_picker/MimeTypeFileFilter.java b/chrome/android/java/src/org/chromium/chrome/browser/photo_picker/MimeTypeFileFilter.java |
new file mode 100644 |
index 0000000000000000000000000000000000000000..058d05b9915ef1020982c1af3ba80d65f3b605d1 |
--- /dev/null |
+++ b/chrome/android/java/src/org/chromium/chrome/browser/photo_picker/MimeTypeFileFilter.java |
@@ -0,0 +1,89 @@ |
+// Copyright 2017 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. |
+ |
+package org.chromium.chrome.browser.photo_picker; |
+ |
+import android.support.annotation.NonNull; |
+import android.webkit.MimeTypeMap; |
+ |
+import java.io.File; |
+import java.io.FileFilter; |
+import java.util.HashSet; |
+import java.util.Locale; |
+ |
+/** |
+ * A file filter for handling extensions and MIME types (images/jpeg) and |
Theresa
2017/04/11 17:01:28
nit: This description should be updated since it c
Finnur
2017/04/12 11:04:43
Correct. Done.
|
+ * supertypes (images/*). |
+ */ |
+class MimeTypeFileFilter implements FileFilter { |
+ private HashSet<String> mExtensions = new HashSet<>(); |
+ private HashSet<String> mMimeTypes = new HashSet<>(); |
+ private HashSet<String> mMimeSupertypes = new HashSet<>(); |
+ private MimeTypeMap mMimeTypeMap; |
+ |
+ public MimeTypeFileFilter(@NonNull String acceptAttr) { |
Theresa
2017/04/11 17:01:28
JavaDoc
Finnur
2017/04/12 11:04:42
Done.
|
+ for (String field : acceptAttr.toLowerCase(Locale.US).split(",")) { |
+ field = field.trim(); |
+ if (field.startsWith(".")) { |
+ mExtensions.add(field.substring(1)); |
+ } else if (field.endsWith("/*")) { |
+ mMimeSupertypes.add(field.substring(0, field.length() - 2)); |
+ } else if (field.contains("/")) { |
+ mMimeTypes.add(field); |
+ } else { |
+ // Throw exception? |
+ } |
+ } |
+ |
+ mMimeTypeMap = MimeTypeMap.getSingleton(); |
+ } |
+ |
+ @Override |
+ public boolean accept(@NonNull File file) { |
+ if (file.isDirectory()) { |
+ return true; |
+ } |
+ |
+ String uri = file.toURI().toString(); |
+ String ext = MimeTypeMap.getFileExtensionFromUrl(uri).toLowerCase(Locale.US); |
+ if (mExtensions.contains(ext)) { |
+ return true; |
+ } |
+ |
+ String mimeType = getMimeTypeFromExtension(ext); |
+ if (mimeType != null) { |
+ if (mMimeTypes.contains(mimeType) |
+ || mMimeSupertypes.contains(getMimeSupertype(mimeType))) { |
+ return true; |
+ } |
+ } |
+ |
+ return false; |
+ } |
+ |
+ private HashSet<String> getAcceptedSupertypes() { |
+ HashSet<String> supertypes = new HashSet<>(); |
+ supertypes.addAll(mMimeSupertypes); |
+ for (String mimeType : mMimeTypes) { |
+ supertypes.add(getMimeSupertype(mimeType)); |
+ } |
+ for (String ext : mExtensions) { |
+ String mimeType = getMimeTypeFromExtension(ext); |
+ if (mimeType != null) { |
+ supertypes.add(getMimeSupertype(mimeType)); |
+ } |
+ } |
+ return supertypes; |
+ } |
+ |
+ private String getMimeTypeFromExtension(@NonNull String ext) { |
+ String mimeType = mMimeTypeMap.getMimeTypeFromExtension(ext); |
+ return (mimeType != null) ? mimeType.toLowerCase(Locale.US) : null; |
+ } |
+ |
+ @NonNull |
+ private String getMimeSupertype(@NonNull String mimeType) { |
+ return mimeType.split("/", 2)[0]; |
+ } |
+} |