140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski/* 240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski * Copyright (C) 2016 The Android Open Source Project 340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski * 440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski * Licensed under the Apache License, Version 2.0 (the "License"); 540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski * you may not use this file except in compliance with the License. 640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski * You may obtain a copy of the License at 740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski * 840cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski * http://www.apache.org/licenses/LICENSE-2.0 940cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski * 1040cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski * Unless required by applicable law or agreed to in writing, software 1140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski * distributed under the License is distributed on an "AS IS" BASIS, 1240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski * See the License for the specific language governing permissions and 1440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski * limitations under the License. 1540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski */ 1640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 1740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskipackage com.android.documentsui.archives; 1840cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 1940cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport com.android.documentsui.tests.R; 2040cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 2140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport android.content.Context; 2240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport android.content.res.AssetFileDescriptor; 2340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport android.database.Cursor; 2440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport android.database.MatrixCursor; 2540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport android.database.MatrixCursor.RowBuilder; 2640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport android.net.Uri; 2740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport android.os.CancellationSignal; 2840cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport android.os.ParcelFileDescriptor; 2940cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport android.provider.DocumentsContract; 3040cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport android.provider.DocumentsContract.Document; 3140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport android.provider.DocumentsContract.Root; 3240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport android.provider.DocumentsProvider; 3340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport android.webkit.MimeTypeMap; 3440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 3540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport libcore.io.IoUtils; 3640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 3740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport java.io.File; 3840cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport java.io.FileNotFoundException; 3940cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport java.util.HashMap; 4040cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport java.util.Map; 4140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport java.util.concurrent.ExecutorService; 4240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskiimport java.util.concurrent.Executors; 4340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 4440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 4540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewskipublic class ResourcesProvider extends DocumentsProvider { 4640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski public static final String AUTHORITY = "com.android.documentsui.archives.resourcesprovider"; 4740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 4840cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski private static final String[] DEFAULT_ROOT_PROJECTION = new String[] { 4940cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski Root.COLUMN_ROOT_ID, Root.COLUMN_FLAGS, Root.COLUMN_TITLE, Root.COLUMN_DOCUMENT_ID 5040cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski }; 5140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski private static final String[] DEFAULT_DOCUMENT_PROJECTION = new String[] { 5240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski Document.COLUMN_DOCUMENT_ID, Document.COLUMN_MIME_TYPE, Document.COLUMN_DISPLAY_NAME, 5340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski Document.COLUMN_LAST_MODIFIED, Document.COLUMN_FLAGS, Document.COLUMN_SIZE, 5440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski }; 5540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 5640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski private static final Map<String, Integer> RESOURCES = new HashMap<>(); 5740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski static { 5840cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski RESOURCES.put("archive.zip", R.raw.archive); 5940cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski RESOURCES.put("empty_dirs.zip", R.raw.empty_dirs); 6040cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski RESOURCES.put("no_dirs.zip", R.raw.no_dirs); 6140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski RESOURCES.put("broken.zip", R.raw.broken); 6240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski } 6340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 6440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski private ExecutorService mExecutor = null; 6540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski private TestUtils mTestUtils = null; 6640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 6740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski @Override 6840cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski public boolean onCreate() { 6940cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski mExecutor = Executors.newSingleThreadExecutor(); 7040cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski mTestUtils = new TestUtils(getContext(), getContext(), mExecutor); 7140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski return true; 7240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski } 7340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 7440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski @Override 7540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski public Cursor queryRoots(String[] projection) throws FileNotFoundException { 7640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski final MatrixCursor result = new MatrixCursor(projection != null ? projection 7740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski : DEFAULT_ROOT_PROJECTION); 7840cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski final RowBuilder row = result.newRow(); 7940cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski row.add(Root.COLUMN_ROOT_ID, "root-id"); 8040cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski row.add(Root.COLUMN_FLAGS, 0); 8140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski row.add(Root.COLUMN_TITLE, "ResourcesProvider"); 8240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski row.add(Root.COLUMN_DOCUMENT_ID, "root-document-id"); 8340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski return result; 8440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski } 8540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 8640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski @Override 8740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski public Cursor queryDocument(String documentId, String[] projection) 8840cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski throws FileNotFoundException { 8940cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski final MatrixCursor result = new MatrixCursor(projection != null ? projection 9040cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski : DEFAULT_DOCUMENT_PROJECTION); 9140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski if ("root-document-id".equals(documentId)) { 9240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski final RowBuilder row = result.newRow(); 9340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski row.add(Document.COLUMN_DOCUMENT_ID, "root-document-id"); 9440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski row.add(Document.COLUMN_FLAGS, 0); 9540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski row.add(Document.COLUMN_DISPLAY_NAME, "ResourcesProvider"); 9640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski row.add(Document.COLUMN_SIZE, 0); 9740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski row.add(Document.COLUMN_MIME_TYPE, Document.MIME_TYPE_DIR); 9840cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski return result; 9940cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski } 10040cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 10140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski includeDocument(result, documentId); 10240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski return result; 10340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski } 10440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 10540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski @Override 10640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski public Cursor queryChildDocuments( 10740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski String parentDocumentId, String[] projection, String sortOrder) 10840cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski throws FileNotFoundException { 10940cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski if (!"root-document-id".equals(parentDocumentId)) { 11040cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski throw new FileNotFoundException(); 11140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski } 11240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 11340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski final MatrixCursor result = new MatrixCursor(projection != null ? projection 11440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski : DEFAULT_DOCUMENT_PROJECTION); 11540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski for (String documentId : RESOURCES.keySet()) { 11640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski includeDocument(result, documentId); 11740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski } 11840cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski return result; 11940cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski } 12040cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 12140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski @Override 12240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski public ParcelFileDescriptor openDocument(String docId, String mode, CancellationSignal signal) 12340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski throws FileNotFoundException { 12440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski final Integer resourceId = RESOURCES.get(docId); 12540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski if (resourceId == null) { 12640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski throw new FileNotFoundException(); 12740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski } 12840cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski return mTestUtils.getSeekableDescriptor(resourceId); 12940cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski } 13040cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 13140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski void includeDocument(MatrixCursor result, String documentId) throws FileNotFoundException { 13240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski final Integer resourceId = RESOURCES.get(documentId); 13340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski if (resourceId == null) { 13440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski throw new FileNotFoundException(); 13540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski } 13640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 13740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski AssetFileDescriptor fd = null; 13840cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski try { 13940cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski fd = getContext().getResources().openRawResourceFd(resourceId); 14040cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski final RowBuilder row = result.newRow(); 14140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski row.add(Document.COLUMN_DOCUMENT_ID, documentId); 14240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski row.add(Document.COLUMN_FLAGS, 0); 14340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski row.add(Document.COLUMN_DISPLAY_NAME, documentId); 14440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 14540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski final int lastDot = documentId.lastIndexOf('.'); 14640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski assert(lastDot > 0); 14740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski final String extension = documentId.substring(lastDot + 1).toLowerCase(); 14840cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski final String mimeType = MimeTypeMap.getSingleton().getMimeTypeFromExtension(extension); 14940cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski 15040cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski row.add(Document.COLUMN_MIME_TYPE, mimeType); 15140cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski row.add(Document.COLUMN_SIZE, fd.getLength()); 15240cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski } 15340cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski finally { 15440cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski IoUtils.closeQuietly(fd); 15540cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski } 15640cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski } 15740cec2ae53d637e186d0f6de105df4859bef2201Tomasz Mikolajewski} 158