From 067d6fadd90ecd6efc07669561bbefd72a5d9f14 Mon Sep 17 00:00:00 2001
From: Vladislav Kaznacheev <kaznacheev@google.com>
Date: Tue, 3 May 2016 15:37:21 -0700
Subject: [PATCH] Collect mime type data correctly for all URIs added to
 ClipData

Bug: 28452937
Change-Id: I883ae2d22e09000137fa484ef66d5070855cd0ed
---
 .../android/documentsui/DocumentClipper.java  | 25 ++++++++++++++++---
 1 file changed, 22 insertions(+), 3 deletions(-)

diff --git a/packages/DocumentsUI/src/com/android/documentsui/DocumentClipper.java b/packages/DocumentsUI/src/com/android/documentsui/DocumentClipper.java
index 059b5e09cc908..72387de496823 100644
--- a/packages/DocumentsUI/src/com/android/documentsui/DocumentClipper.java
+++ b/packages/DocumentsUI/src/com/android/documentsui/DocumentClipper.java
@@ -32,7 +32,9 @@ import com.android.documentsui.model.DocumentInfo;
 import libcore.io.IoUtils;
 
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.Collections;
+import java.util.HashSet;
 import java.util.List;
 
 /**
@@ -125,22 +127,39 @@ public final class DocumentClipper {
      */
     public @Nullable ClipData getClipDataForDocuments(List<DocumentInfo> docs) {
         final ContentResolver resolver = mContext.getContentResolver();
+        final String[] mimeTypes = getMimeTypes(resolver, docs);
         ClipData clipData = null;
         for (DocumentInfo doc : docs) {
-            final Uri uri = DocumentsContract.buildDocumentUri(doc.authority, doc.documentId);
             if (clipData == null) {
                 // TODO: figure out what this string should be.
                 // Currently it is not displayed anywhere in the UI, but this might change.
                 final String label = "";
-                clipData = ClipData.newUri(resolver, label, uri);
+                clipData = new ClipData(label, mimeTypes, new ClipData.Item(doc.derivedUri));
             } else {
                 // TODO: update list of mime types in ClipData.
-                clipData.addItem(new ClipData.Item(uri));
+                clipData.addItem(new ClipData.Item(doc.derivedUri));
             }
         }
         return clipData;
     }
 
+    private static String[] getMimeTypes(ContentResolver resolver, List<DocumentInfo> docs) {
+        final HashSet<String> mimeTypes = new HashSet<>();
+        for (DocumentInfo doc : docs) {
+            assert(doc != null);
+            assert(doc.derivedUri != null);
+            final Uri uri = doc.derivedUri;
+            if ("content".equals(uri.getScheme())) {
+                mimeTypes.add(resolver.getType(uri));
+                final String[] streamTypes = resolver.getStreamTypes(uri, "*/*");
+                if (streamTypes != null) {
+                    mimeTypes.addAll(Arrays.asList(streamTypes));
+                }
+            }
+        }
+        return mimeTypes.toArray(new String[0]);
+    }
+
     public void clipDocuments(List<DocumentInfo> docs) {
         ClipData data = getClipDataForDocuments(docs);
         mClipboard.setPrimaryClip(data);