Generalize media input for use with Audio.

app/src/main/java/org/thoughtcrime/securesms/media/DecryptableUriMediaInput.java

@@ -0,0 +1,54 @@
+package org.thoughtcrime.securesms.media;
+
+import android.content.Context;
+import android.media.MediaDataSource;
+import android.net.Uri;
+
+import androidx.annotation.NonNull;
+import androidx.annotation.RequiresApi;
+
+import org.thoughtcrime.securesms.attachments.AttachmentId;
+import org.thoughtcrime.securesms.database.DatabaseFactory;
+import org.thoughtcrime.securesms.mms.PartAuthority;
+import org.thoughtcrime.securesms.mms.PartUriParser;
+import org.thoughtcrime.securesms.providers.BlobProvider;
+import org.thoughtcrime.securesms.media.MediaInput;
+
+import java.io.IOException;
+
+@RequiresApi(api = 23)
+public final class DecryptableUriMediaInput {
+
+  private DecryptableUriMediaInput() {
+  }
+
+  public static @NonNull MediaInput createForUri(@NonNull Context context, @NonNull Uri uri) throws IOException {
+
+    if (BlobProvider.isAuthority(uri)) {
+      return new MediaInput.MediaDataSourceMediaInput(BlobProvider.getInstance().getMediaDataSource(context, uri));
+    }
+
+    if (PartAuthority.isLocalUri(uri)) {
+      return createForAttachmentUri(context, uri);
+    }
+
+    return new MediaInput.UriMediaInput(context, uri);
+  }
+
+  private static @NonNull MediaInput createForAttachmentUri(@NonNull Context context, @NonNull Uri uri) {
+    AttachmentId partId = new PartUriParser(uri).getPartId();
+
+    if (!partId.isValid()) {
+      throw new AssertionError();
+    }
+
+    MediaDataSource mediaDataSource = DatabaseFactory.getAttachmentDatabase(context)
+                                                     .mediaDataSourceFor(partId);
+
+    if (mediaDataSource == null) {
+      throw new AssertionError();
+    }
+
+    return new MediaInput.MediaDataSourceMediaInput(mediaDataSource);
+  }
+}

app/src/main/java/org/thoughtcrime/securesms/media/MediaInput.java

@@ -0,0 +1,83 @@
+package org.thoughtcrime.securesms.media;
+
+import android.content.Context;
+import android.media.MediaDataSource;
+import android.media.MediaExtractor;
+import android.net.Uri;
+
+import androidx.annotation.NonNull;
+import androidx.annotation.RequiresApi;
+
+import java.io.Closeable;
+import java.io.File;
+import java.io.IOException;
+
+public abstract class MediaInput implements Closeable {
+
+  @NonNull
+  public abstract MediaExtractor createExtractor() throws IOException;
+
+  public static class FileMediaInput extends MediaInput {
+
+    private final File file;
+
+    public FileMediaInput(@NonNull File file) {
+      this.file = file;
+    }
+
+    @Override
+    public @NonNull MediaExtractor createExtractor() throws IOException {
+      final MediaExtractor extractor = new MediaExtractor();
+      extractor.setDataSource(file.getAbsolutePath());
+      return extractor;
+    }
+
+    @Override
+    public void close() {
+    }
+  }
+
+  public static class UriMediaInput extends MediaInput {
+
+    private final Uri     uri;
+    private final Context context;
+
+    public UriMediaInput(@NonNull Context context, @NonNull Uri uri) {
+      this.uri     = uri;
+      this.context = context;
+    }
+
+    @Override
+    public @NonNull MediaExtractor createExtractor() throws IOException {
+      final MediaExtractor extractor = new MediaExtractor();
+      extractor.setDataSource(context, uri, null);
+      return extractor;
+    }
+
+    @Override
+    public void close() {
+    }
+  }
+
+  @RequiresApi(23)
+  public static class MediaDataSourceMediaInput extends MediaInput {
+
+    private final MediaDataSource mediaDataSource;
+
+    public MediaDataSourceMediaInput(@NonNull MediaDataSource mediaDataSource) {
+      this.mediaDataSource = mediaDataSource;
+    }
+
+    @Override
+    public @NonNull MediaExtractor createExtractor() throws IOException {
+      final MediaExtractor extractor = new MediaExtractor();
+      extractor.setDataSource(mediaDataSource);
+      return extractor;
+    }
+
+    @Override
+    public void close() throws IOException {
+      mediaDataSource.close();
+    }
+  }
+}