Konverting the MediaSendViewModel (#922)
* Konverting the MediaSendViewModel * PR Feedbackpull/1710/head
ThomasSession
2 months ago
committed by
GitHub
parent
b134b82daa
commit
14948df4b5
@ -1,360 +0,0 @@
|
||||
package org.thoughtcrime.securesms.mediasend;
|
||||
|
||||
import android.app.Application;
|
||||
import android.content.Context;
|
||||
import android.net.Uri;
|
||||
import androidx.annotation.NonNull;
|
||||
import androidx.lifecycle.LiveData;
|
||||
import androidx.lifecycle.MutableLiveData;
|
||||
import androidx.lifecycle.ViewModel;
|
||||
import androidx.lifecycle.ViewModelProvider;
|
||||
import com.annimon.stream.Stream;
|
||||
import java.util.Collections;
|
||||
import java.util.HashMap;
|
||||
import java.util.LinkedList;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import org.session.libsession.utilities.FileUtils;
|
||||
import org.session.libsession.utilities.Util;
|
||||
import org.session.libsignal.utilities.Log;
|
||||
import org.session.libsignal.utilities.guava.Optional;
|
||||
import org.thoughtcrime.securesms.mms.MediaConstraints;
|
||||
import org.thoughtcrime.securesms.providers.BlobProvider;
|
||||
import org.thoughtcrime.securesms.util.MediaUtil;
|
||||
import org.thoughtcrime.securesms.util.SingleLiveEvent;
|
||||
|
||||
/**
|
||||
* Manages the observable datasets available in {@link MediaSendActivity}.
|
||||
*/
|
||||
class MediaSendViewModel extends ViewModel {
|
||||
|
||||
private static final String TAG = MediaSendViewModel.class.getSimpleName();
|
||||
|
||||
private static final int MAX_SELECTION = 32;
|
||||
|
||||
private final Application application;
|
||||
private final MediaRepository repository;
|
||||
private final MutableLiveData<List<Media>> selectedMedia;
|
||||
private final MutableLiveData<List<Media>> bucketMedia;
|
||||
private final MutableLiveData<Integer> position;
|
||||
private final MutableLiveData<String> bucketId;
|
||||
private final MutableLiveData<List<MediaFolder>> folders;
|
||||
private final MutableLiveData<CountButtonState> countButtonState;
|
||||
private final MutableLiveData<Boolean> cameraButtonVisibility;
|
||||
private final SingleLiveEvent<Error> error;
|
||||
private final Map<Uri, Object> savedDrawState;
|
||||
|
||||
private final MediaConstraints mediaConstraints = MediaConstraints.getPushMediaConstraints();
|
||||
|
||||
private CharSequence body;
|
||||
private CountButtonState.Visibility countButtonVisibility;
|
||||
private boolean sentMedia;
|
||||
private Optional<Media> lastImageCapture;
|
||||
|
||||
private MediaSendViewModel(@NonNull Application application, @NonNull MediaRepository repository) {
|
||||
this.application = application;
|
||||
this.repository = repository;
|
||||
this.selectedMedia = new MutableLiveData<>();
|
||||
this.bucketMedia = new MutableLiveData<>();
|
||||
this.position = new MutableLiveData<>();
|
||||
this.bucketId = new MutableLiveData<>();
|
||||
this.folders = new MutableLiveData<>();
|
||||
this.countButtonState = new MutableLiveData<>();
|
||||
this.cameraButtonVisibility = new MutableLiveData<>();
|
||||
this.error = new SingleLiveEvent<>();
|
||||
this.savedDrawState = new HashMap<>();
|
||||
this.countButtonVisibility = CountButtonState.Visibility.FORCED_OFF;
|
||||
this.lastImageCapture = Optional.absent();
|
||||
this.body = "";
|
||||
|
||||
position.setValue(-1);
|
||||
countButtonState.setValue(new CountButtonState(0, countButtonVisibility));
|
||||
cameraButtonVisibility.setValue(false);
|
||||
}
|
||||
|
||||
void onSelectedMediaChanged(@NonNull Context context, @NonNull List<Media> newMedia) {
|
||||
repository.getPopulatedMedia(context, newMedia, populatedMedia -> {
|
||||
Util.runOnMain(() -> {
|
||||
|
||||
List<Media> filteredMedia = getFilteredMedia(context, populatedMedia, mediaConstraints);
|
||||
|
||||
if (filteredMedia.size() != newMedia.size()) {
|
||||
error.setValue(Error.ITEM_TOO_LARGE);
|
||||
} else if (filteredMedia.size() > MAX_SELECTION) {
|
||||
filteredMedia = filteredMedia.subList(0, MAX_SELECTION);
|
||||
error.setValue(Error.TOO_MANY_ITEMS);
|
||||
}
|
||||
|
||||
if (filteredMedia.size() > 0) {
|
||||
String computedId = Stream.of(filteredMedia)
|
||||
.skip(1)
|
||||
.reduce(filteredMedia.get(0).getBucketId().or(Media.ALL_MEDIA_BUCKET_ID), (id, m) -> {
|
||||
if (Util.equals(id, m.getBucketId().or(Media.ALL_MEDIA_BUCKET_ID))) {
|
||||
return id;
|
||||
} else {
|
||||
return Media.ALL_MEDIA_BUCKET_ID;
|
||||
}
|
||||
});
|
||||
bucketId.setValue(computedId);
|
||||
} else {
|
||||
bucketId.setValue(Media.ALL_MEDIA_BUCKET_ID);
|
||||
countButtonVisibility = CountButtonState.Visibility.CONDITIONAL;
|
||||
}
|
||||
|
||||
selectedMedia.setValue(filteredMedia);
|
||||
countButtonState.setValue(new CountButtonState(filteredMedia.size(), countButtonVisibility));
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
void onSingleMediaSelected(@NonNull Context context, @NonNull Media media) {
|
||||
repository.getPopulatedMedia(context, Collections.singletonList(media), populatedMedia -> {
|
||||
Util.runOnMain(() -> {
|
||||
List<Media> filteredMedia = getFilteredMedia(context, populatedMedia, mediaConstraints);
|
||||
|
||||
if (filteredMedia.isEmpty()) {
|
||||
error.setValue(Error.ITEM_TOO_LARGE);
|
||||
bucketId.setValue(Media.ALL_MEDIA_BUCKET_ID);
|
||||
} else {
|
||||
bucketId.setValue(filteredMedia.get(0).getBucketId().or(Media.ALL_MEDIA_BUCKET_ID));
|
||||
}
|
||||
|
||||
countButtonVisibility = CountButtonState.Visibility.FORCED_OFF;
|
||||
|
||||
selectedMedia.setValue(filteredMedia);
|
||||
countButtonState.setValue(new CountButtonState(filteredMedia.size(), countButtonVisibility));
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
void onMultiSelectStarted() {
|
||||
countButtonVisibility = CountButtonState.Visibility.FORCED_ON;
|
||||
countButtonState.setValue(new CountButtonState(getSelectedMediaOrDefault().size(), countButtonVisibility));
|
||||
}
|
||||
|
||||
void onImageEditorStarted() {
|
||||
countButtonVisibility = CountButtonState.Visibility.FORCED_OFF;
|
||||
countButtonState.setValue(new CountButtonState(getSelectedMediaOrDefault().size(), countButtonVisibility));
|
||||
cameraButtonVisibility.setValue(false);
|
||||
}
|
||||
|
||||
void onCameraStarted() {
|
||||
countButtonVisibility = CountButtonState.Visibility.CONDITIONAL;
|
||||
countButtonState.setValue(new CountButtonState(getSelectedMediaOrDefault().size(), countButtonVisibility));
|
||||
cameraButtonVisibility.setValue(false);
|
||||
}
|
||||
|
||||
void onItemPickerStarted() {
|
||||
countButtonVisibility = CountButtonState.Visibility.CONDITIONAL;
|
||||
countButtonState.setValue(new CountButtonState(getSelectedMediaOrDefault().size(), countButtonVisibility));
|
||||
cameraButtonVisibility.setValue(true);
|
||||
}
|
||||
|
||||
void onFolderPickerStarted() {
|
||||
countButtonVisibility = CountButtonState.Visibility.CONDITIONAL;
|
||||
countButtonState.setValue(new CountButtonState(getSelectedMediaOrDefault().size(), countButtonVisibility));
|
||||
cameraButtonVisibility.setValue(true);
|
||||
}
|
||||
|
||||
void onBodyChanged(@NonNull CharSequence body) {
|
||||
this.body = body;
|
||||
}
|
||||
|
||||
void onFolderSelected(@NonNull String bucketId) {
|
||||
this.bucketId.setValue(bucketId);
|
||||
bucketMedia.setValue(Collections.emptyList());
|
||||
}
|
||||
|
||||
void onPageChanged(int position) {
|
||||
if (position < 0 || position >= getSelectedMediaOrDefault().size()) {
|
||||
Log.w(TAG, "Tried to move to an out-of-bounds item. Size: " + getSelectedMediaOrDefault().size() + ", position: " + position);
|
||||
return;
|
||||
}
|
||||
|
||||
this.position.setValue(position);
|
||||
}
|
||||
|
||||
void onMediaItemRemoved(@NonNull Context context, int position) {
|
||||
if (position < 0 || position >= getSelectedMediaOrDefault().size()) {
|
||||
Log.w(TAG, "Tried to remove an out-of-bounds item. Size: " + getSelectedMediaOrDefault().size() + ", position: " + position);
|
||||
return;
|
||||
}
|
||||
|
||||
Media removed = getSelectedMediaOrDefault().remove(position);
|
||||
|
||||
if (removed != null && BlobProvider.isAuthority(removed.getUri())) {
|
||||
BlobProvider.getInstance().delete(context, removed.getUri());
|
||||
}
|
||||
|
||||
selectedMedia.setValue(selectedMedia.getValue());
|
||||
}
|
||||
|
||||
void onImageCaptured(@NonNull Media media) {
|
||||
List<Media> selected = selectedMedia.getValue();
|
||||
|
||||
if (selected == null) {
|
||||
selected = new LinkedList<>();
|
||||
}
|
||||
|
||||
if (selected.size() >= MAX_SELECTION) {
|
||||
error.setValue(Error.TOO_MANY_ITEMS);
|
||||
return;
|
||||
}
|
||||
|
||||
lastImageCapture = Optional.of(media);
|
||||
|
||||
selected.add(media);
|
||||
selectedMedia.setValue(selected);
|
||||
position.setValue(selected.size() - 1);
|
||||
bucketId.setValue(Media.ALL_MEDIA_BUCKET_ID);
|
||||
|
||||
if (selected.size() == 1) {
|
||||
countButtonVisibility = CountButtonState.Visibility.FORCED_OFF;
|
||||
} else {
|
||||
countButtonVisibility = CountButtonState.Visibility.CONDITIONAL;
|
||||
}
|
||||
|
||||
countButtonState.setValue(new CountButtonState(selected.size(), countButtonVisibility));
|
||||
}
|
||||
|
||||
void onImageCaptureUndo(@NonNull Context context) {
|
||||
List<Media> selected = getSelectedMediaOrDefault();
|
||||
|
||||
if (lastImageCapture.isPresent() && selected.contains(lastImageCapture.get()) && selected.size() == 1) {
|
||||
selected.remove(lastImageCapture.get());
|
||||
selectedMedia.setValue(selected);
|
||||
countButtonState.setValue(new CountButtonState(selected.size(), countButtonVisibility));
|
||||
BlobProvider.getInstance().delete(context, lastImageCapture.get().getUri());
|
||||
}
|
||||
}
|
||||
|
||||
void saveDrawState(@NonNull Map<Uri, Object> state) {
|
||||
savedDrawState.clear();
|
||||
savedDrawState.putAll(state);
|
||||
}
|
||||
|
||||
void onSendClicked() {
|
||||
sentMedia = true;
|
||||
}
|
||||
|
||||
@NonNull Map<Uri, Object> getDrawState() {
|
||||
return savedDrawState;
|
||||
}
|
||||
|
||||
@NonNull LiveData<List<Media>> getSelectedMedia() {
|
||||
return selectedMedia;
|
||||
}
|
||||
|
||||
@NonNull LiveData<List<Media>> getMediaInBucket(@NonNull Context context, @NonNull String bucketId) {
|
||||
repository.getMediaInBucket(context, bucketId, bucketMedia::postValue);
|
||||
return bucketMedia;
|
||||
}
|
||||
|
||||
@NonNull LiveData<List<MediaFolder>> getFolders(@NonNull Context context) {
|
||||
repository.getFolders(context, folders::postValue);
|
||||
return folders;
|
||||
}
|
||||
|
||||
@NonNull LiveData<CountButtonState> getCountButtonState() {
|
||||
return countButtonState;
|
||||
}
|
||||
|
||||
@NonNull LiveData<Boolean> getCameraButtonVisibility() {
|
||||
return cameraButtonVisibility;
|
||||
}
|
||||
|
||||
@NonNull CharSequence getBody() {
|
||||
return body;
|
||||
}
|
||||
|
||||
@NonNull LiveData<Integer> getPosition() {
|
||||
return position;
|
||||
}
|
||||
|
||||
@NonNull LiveData<String> getBucketId() {
|
||||
return bucketId;
|
||||
}
|
||||
|
||||
@NonNull LiveData<Error> getError() {
|
||||
return error;
|
||||
}
|
||||
|
||||
int getMaxSelection() {
|
||||
return MAX_SELECTION;
|
||||
}
|
||||
|
||||
private @NonNull List<Media> getSelectedMediaOrDefault() {
|
||||
return selectedMedia.getValue() == null ? Collections.emptyList()
|
||||
: selectedMedia.getValue();
|
||||
}
|
||||
|
||||
private @NonNull List<Media> getFilteredMedia(@NonNull Context context, @NonNull List<Media> media, @NonNull MediaConstraints mediaConstraints) {
|
||||
return Stream.of(media).filter(m -> MediaUtil.isGif(m.getMimeType()) ||
|
||||
MediaUtil.isImageType(m.getMimeType()) ||
|
||||
MediaUtil.isVideoType(m.getMimeType()))
|
||||
.filter(m -> {
|
||||
return (MediaUtil.isImageType(m.getMimeType()) && !MediaUtil.isGif(m.getMimeType())) ||
|
||||
(MediaUtil.isGif(m.getMimeType()) && m.getSize() < mediaConstraints.getGifMaxSize(context)) ||
|
||||
(MediaUtil.isVideoType(m.getMimeType()) && m.getSize() < mediaConstraints.getVideoMaxSize(context));
|
||||
}).toList();
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void onCleared() {
|
||||
if (!sentMedia) {
|
||||
Stream.of(getSelectedMediaOrDefault())
|
||||
.map(Media::getUri)
|
||||
.filter(BlobProvider::isAuthority)
|
||||
.forEach(uri -> BlobProvider.getInstance().delete(application.getApplicationContext(), uri));
|
||||
}
|
||||
}
|
||||
|
||||
enum Error {
|
||||
ITEM_TOO_LARGE, TOO_MANY_ITEMS
|
||||
}
|
||||
|
||||
static class CountButtonState {
|
||||
private final int count;
|
||||
private final Visibility visibility;
|
||||
|
||||
private CountButtonState(int count, @NonNull Visibility visibility) {
|
||||
this.count = count;
|
||||
this.visibility = visibility;
|
||||
}
|
||||
|
||||
int getCount() {
|
||||
return count;
|
||||
}
|
||||
|
||||
boolean isVisible() {
|
||||
switch (visibility) {
|
||||
case FORCED_ON: return true;
|
||||
case FORCED_OFF: return false;
|
||||
case CONDITIONAL: return count > 0;
|
||||
default: return false;
|
||||
}
|
||||
}
|
||||
|
||||
enum Visibility {
|
||||
CONDITIONAL, FORCED_ON, FORCED_OFF
|
||||
}
|
||||
}
|
||||
|
||||
static class Factory extends ViewModelProvider.NewInstanceFactory {
|
||||
|
||||
private final Application application;
|
||||
private final MediaRepository repository;
|
||||
|
||||
Factory(@NonNull Application application, @NonNull MediaRepository repository) {
|
||||
this.application = application;
|
||||
this.repository = repository;
|
||||
}
|
||||
|
||||
@Override
|
||||
public @NonNull <T extends ViewModel> T create(@NonNull Class<T> modelClass) {
|
||||
return modelClass.cast(new MediaSendViewModel(application, repository));
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,367 @@
|
||||
package org.thoughtcrime.securesms.mediasend
|
||||
|
||||
import android.app.Application
|
||||
import android.content.Context
|
||||
import android.net.Uri
|
||||
import androidx.lifecycle.LiveData
|
||||
import androidx.lifecycle.MutableLiveData
|
||||
import androidx.lifecycle.ViewModel
|
||||
import com.annimon.stream.Stream
|
||||
import dagger.hilt.android.lifecycle.HiltViewModel
|
||||
import org.session.libsession.utilities.Util.equals
|
||||
import org.session.libsession.utilities.Util.runOnMain
|
||||
import org.session.libsignal.utilities.Log
|
||||
import org.session.libsignal.utilities.guava.Optional
|
||||
import org.thoughtcrime.securesms.mms.MediaConstraints
|
||||
import org.thoughtcrime.securesms.providers.BlobProvider
|
||||
import org.thoughtcrime.securesms.util.MediaUtil
|
||||
import org.thoughtcrime.securesms.util.SingleLiveEvent
|
||||
import java.util.LinkedList
|
||||
import javax.inject.Inject
|
||||
|
||||
/**
|
||||
* Manages the observable datasets available in [MediaSendActivity].
|
||||
*/
|
||||
|
||||
@HiltViewModel
|
||||
internal class MediaSendViewModel @Inject constructor(
|
||||
private val application: Application
|
||||
) : ViewModel() {
|
||||
private val selectedMedia: MutableLiveData<List<Media>?>
|
||||
private val bucketMedia: MutableLiveData<List<Media>>
|
||||
private val position: MutableLiveData<Int>
|
||||
private val bucketId: MutableLiveData<String>
|
||||
private val folders: MutableLiveData<List<MediaFolder>>
|
||||
private val countButtonState: MutableLiveData<CountButtonState>
|
||||
private val cameraButtonVisibility: MutableLiveData<Boolean>
|
||||
private val error: SingleLiveEvent<Error>
|
||||
private val savedDrawState: MutableMap<Uri, Any>
|
||||
|
||||
private val mediaConstraints: MediaConstraints = MediaConstraints.getPushMediaConstraints()
|
||||
private val repository: MediaRepository = MediaRepository()
|
||||
|
||||
var body: CharSequence
|
||||
private set
|
||||
private var countButtonVisibility: CountButtonState.Visibility
|
||||
private var sentMedia: Boolean = false
|
||||
private var lastImageCapture: Optional<Media>
|
||||
|
||||
init {
|
||||
this.selectedMedia = MutableLiveData()
|
||||
this.bucketMedia = MutableLiveData()
|
||||
this.position = MutableLiveData()
|
||||
this.bucketId = MutableLiveData()
|
||||
this.folders = MutableLiveData()
|
||||
this.countButtonState = MutableLiveData()
|
||||
this.cameraButtonVisibility = MutableLiveData()
|
||||
this.error = SingleLiveEvent()
|
||||
this.savedDrawState = HashMap()
|
||||
this.countButtonVisibility = CountButtonState.Visibility.FORCED_OFF
|
||||
this.lastImageCapture = Optional.absent()
|
||||
this.body = ""
|
||||
|
||||
position.value = -1
|
||||
countButtonState.value = CountButtonState(0, countButtonVisibility)
|
||||
cameraButtonVisibility.value = false
|
||||
}
|
||||
|
||||
fun onSelectedMediaChanged(context: Context, newMedia: List<Media?>) {
|
||||
repository.getPopulatedMedia(context, newMedia,
|
||||
{ populatedMedia: List<Media> ->
|
||||
runOnMain(
|
||||
{
|
||||
var filteredMedia: List<Media> =
|
||||
getFilteredMedia(context, populatedMedia, mediaConstraints)
|
||||
if (filteredMedia.size != newMedia.size) {
|
||||
error.setValue(Error.ITEM_TOO_LARGE)
|
||||
} else if (filteredMedia.size > MAX_SELECTED_FILES) {
|
||||
filteredMedia = filteredMedia.subList(0, MAX_SELECTED_FILES)
|
||||
error.setValue(Error.TOO_MANY_ITEMS)
|
||||
}
|
||||
|
||||
if (filteredMedia.size > 0) {
|
||||
val computedId: String = Stream.of(filteredMedia)
|
||||
.skip(1)
|
||||
.reduce(filteredMedia.get(0).bucketId.or(Media.ALL_MEDIA_BUCKET_ID),
|
||||
{ id: String?, m: Media ->
|
||||
if (equals(id, m.bucketId.or(Media.ALL_MEDIA_BUCKET_ID))) {
|
||||
return@reduce id
|
||||
} else {
|
||||
return@reduce Media.ALL_MEDIA_BUCKET_ID
|
||||
}
|
||||
})
|
||||
bucketId.setValue(computedId)
|
||||
} else {
|
||||
bucketId.setValue(Media.ALL_MEDIA_BUCKET_ID)
|
||||
countButtonVisibility = CountButtonState.Visibility.CONDITIONAL
|
||||
}
|
||||
|
||||
selectedMedia.setValue(filteredMedia)
|
||||
countButtonState.setValue(
|
||||
CountButtonState(
|
||||
filteredMedia.size,
|
||||
countButtonVisibility
|
||||
)
|
||||
)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
fun onSingleMediaSelected(context: Context, media: Media) {
|
||||
repository.getPopulatedMedia(context, listOf(media),
|
||||
{ populatedMedia: List<Media> ->
|
||||
runOnMain(
|
||||
{
|
||||
val filteredMedia: List<Media> =
|
||||
getFilteredMedia(context, populatedMedia, mediaConstraints)
|
||||
if (filteredMedia.isEmpty()) {
|
||||
error.setValue(Error.ITEM_TOO_LARGE)
|
||||
bucketId.setValue(Media.ALL_MEDIA_BUCKET_ID)
|
||||
} else {
|
||||
bucketId.setValue(filteredMedia.get(0).bucketId.or(Media.ALL_MEDIA_BUCKET_ID))
|
||||
}
|
||||
|
||||
countButtonVisibility = CountButtonState.Visibility.FORCED_OFF
|
||||
|
||||
selectedMedia.value = filteredMedia
|
||||
countButtonState.setValue(
|
||||
CountButtonState(
|
||||
filteredMedia.size,
|
||||
countButtonVisibility
|
||||
)
|
||||
)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
fun onMultiSelectStarted() {
|
||||
countButtonVisibility = CountButtonState.Visibility.FORCED_ON
|
||||
countButtonState.value =
|
||||
CountButtonState(selectedMediaOrDefault.size, countButtonVisibility)
|
||||
}
|
||||
|
||||
fun onImageEditorStarted() {
|
||||
countButtonVisibility = CountButtonState.Visibility.FORCED_OFF
|
||||
countButtonState.value =
|
||||
CountButtonState(selectedMediaOrDefault.size, countButtonVisibility)
|
||||
cameraButtonVisibility.value = false
|
||||
}
|
||||
|
||||
fun onCameraStarted() {
|
||||
countButtonVisibility = CountButtonState.Visibility.CONDITIONAL
|
||||
countButtonState.value =
|
||||
CountButtonState(selectedMediaOrDefault.size, countButtonVisibility)
|
||||
cameraButtonVisibility.value = false
|
||||
}
|
||||
|
||||
fun onItemPickerStarted() {
|
||||
countButtonVisibility = CountButtonState.Visibility.CONDITIONAL
|
||||
countButtonState.value =
|
||||
CountButtonState(selectedMediaOrDefault.size, countButtonVisibility)
|
||||
cameraButtonVisibility.value = true
|
||||
}
|
||||
|
||||
fun onFolderPickerStarted() {
|
||||
countButtonVisibility = CountButtonState.Visibility.CONDITIONAL
|
||||
countButtonState.value =
|
||||
CountButtonState(selectedMediaOrDefault.size, countButtonVisibility)
|
||||
cameraButtonVisibility.value = true
|
||||
}
|
||||
|
||||
fun onBodyChanged(body: CharSequence) {
|
||||
this.body = body
|
||||
}
|
||||
|
||||
fun onFolderSelected(bucketId: String) {
|
||||
this.bucketId.value = bucketId
|
||||
bucketMedia.value =
|
||||
emptyList()
|
||||
}
|
||||
|
||||
fun onPageChanged(position: Int) {
|
||||
if (position < 0 || position >= selectedMediaOrDefault.size) {
|
||||
Log.w(TAG,
|
||||
"Tried to move to an out-of-bounds item. Size: " + selectedMediaOrDefault.size + ", position: " + position
|
||||
)
|
||||
return
|
||||
}
|
||||
|
||||
this.position.value = position
|
||||
}
|
||||
|
||||
fun onMediaItemRemoved(context: Context, position: Int) {
|
||||
if (position < 0 || position >= selectedMediaOrDefault.size) {
|
||||
Log.w(
|
||||
TAG,
|
||||
"Tried to remove an out-of-bounds item. Size: " + selectedMediaOrDefault.size + ", position: " + position
|
||||
)
|
||||
return
|
||||
}
|
||||
|
||||
val updatedList = selectedMediaOrDefault.toMutableList()
|
||||
val removed: Media = updatedList.removeAt(position)
|
||||
|
||||
if (BlobProvider.isAuthority(removed.uri)) {
|
||||
BlobProvider.getInstance().delete(context, removed.uri)
|
||||
}
|
||||
|
||||
selectedMedia.setValue(updatedList)
|
||||
}
|
||||
|
||||
fun onImageCaptured(media: Media) {
|
||||
var selected: MutableList<Media>? = selectedMedia.value?.toMutableList()
|
||||
|
||||
if (selected == null) {
|
||||
selected = LinkedList()
|
||||
}
|
||||
|
||||
if (selected.size >= MAX_SELECTED_FILES) {
|
||||
error.setValue(Error.TOO_MANY_ITEMS)
|
||||
return
|
||||
}
|
||||
|
||||
lastImageCapture = Optional.of(media)
|
||||
|
||||
selected.add(media)
|
||||
selectedMedia.setValue(selected)
|
||||
position.setValue(selected.size - 1)
|
||||
bucketId.setValue(Media.ALL_MEDIA_BUCKET_ID)
|
||||
|
||||
if (selected.size == 1) {
|
||||
countButtonVisibility = CountButtonState.Visibility.FORCED_OFF
|
||||
} else {
|
||||
countButtonVisibility = CountButtonState.Visibility.CONDITIONAL
|
||||
}
|
||||
|
||||
countButtonState.setValue(CountButtonState(selected.size, countButtonVisibility))
|
||||
}
|
||||
|
||||
fun onImageCaptureUndo(context: Context) {
|
||||
val selected: MutableList<Media> = selectedMediaOrDefault.toMutableList()
|
||||
|
||||
if (lastImageCapture.isPresent && selected.contains(lastImageCapture.get()) && selected.size == 1) {
|
||||
selected.remove(lastImageCapture.get())
|
||||
selectedMedia.value = selected
|
||||
countButtonState.value = CountButtonState(selected.size, countButtonVisibility)
|
||||
BlobProvider.getInstance().delete(context, lastImageCapture.get().uri)
|
||||
}
|
||||
}
|
||||
|
||||
fun saveDrawState(state: Map<Uri, Any>) {
|
||||
savedDrawState.clear()
|
||||
savedDrawState.putAll(state)
|
||||
}
|
||||
|
||||
fun onSendClicked() {
|
||||
sentMedia = true
|
||||
}
|
||||
|
||||
val drawState: Map<Uri, Any>
|
||||
get() = savedDrawState
|
||||
|
||||
fun getSelectedMedia(): LiveData<List<Media>?> {
|
||||
return selectedMedia
|
||||
}
|
||||
|
||||
fun getMediaInBucket(context: Context, bucketId: String): LiveData<List<Media>> {
|
||||
repository.getMediaInBucket(context, bucketId,
|
||||
{ value: List<Media> -> bucketMedia.postValue(value) })
|
||||
return bucketMedia
|
||||
}
|
||||
|
||||
fun getFolders(context: Context): LiveData<List<MediaFolder>> {
|
||||
repository.getFolders(context,
|
||||
{ value: List<MediaFolder> -> folders.postValue(value) })
|
||||
return folders
|
||||
}
|
||||
|
||||
fun getCountButtonState(): LiveData<CountButtonState> {
|
||||
return countButtonState
|
||||
}
|
||||
|
||||
fun getCameraButtonVisibility(): LiveData<Boolean> {
|
||||
return cameraButtonVisibility
|
||||
}
|
||||
|
||||
fun getPosition(): LiveData<Int> {
|
||||
return position
|
||||
}
|
||||
|
||||
fun getBucketId(): LiveData<String> {
|
||||
return bucketId
|
||||
}
|
||||
|
||||
fun getError(): LiveData<Error> {
|
||||
return error
|
||||
}
|
||||
|
||||
private val selectedMediaOrDefault: List<Media>
|
||||
get() = if (selectedMedia.value == null) emptyList() else
|
||||
selectedMedia.value!!
|
||||
|
||||
private fun getFilteredMedia(
|
||||
context: Context,
|
||||
media: List<Media>,
|
||||
mediaConstraints: MediaConstraints
|
||||
): List<Media> {
|
||||
return Stream.of(media).filter(
|
||||
{ m: Media ->
|
||||
MediaUtil.isGif(m.mimeType) ||
|
||||
MediaUtil.isImageType(m.mimeType) ||
|
||||
MediaUtil.isVideoType(m.mimeType)
|
||||
})
|
||||
.filter({ m: Media ->
|
||||
(MediaUtil.isImageType(m.mimeType) && !MediaUtil.isGif(m.mimeType)) ||
|
||||
(MediaUtil.isGif(m.mimeType) && m.size < mediaConstraints.getGifMaxSize(
|
||||
context
|
||||
)) ||
|
||||
(MediaUtil.isVideoType(m.mimeType) && m.size < mediaConstraints.getVideoMaxSize(
|
||||
context
|
||||
))
|
||||
}).toList()
|
||||
}
|
||||
|
||||
override fun onCleared() {
|
||||
if (!sentMedia) {
|
||||
Stream.of(selectedMediaOrDefault)
|
||||
.map({ obj: Media -> obj.uri })
|
||||
.filter({ uri: Uri? ->
|
||||
BlobProvider.isAuthority(
|
||||
uri!!
|
||||
)
|
||||
})
|
||||
.forEach({ uri: Uri? ->
|
||||
BlobProvider.getInstance().delete(
|
||||
application.applicationContext, uri!!
|
||||
)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
internal enum class Error {
|
||||
ITEM_TOO_LARGE, TOO_MANY_ITEMS
|
||||
}
|
||||
|
||||
internal class CountButtonState(val count: Int, private val visibility: Visibility) {
|
||||
val isVisible: Boolean
|
||||
get() {
|
||||
when (visibility) {
|
||||
Visibility.FORCED_ON -> return true
|
||||
Visibility.FORCED_OFF -> return false
|
||||
Visibility.CONDITIONAL -> return count > 0
|
||||
else -> return false
|
||||
}
|
||||
}
|
||||
|
||||
internal enum class Visibility {
|
||||
CONDITIONAL, FORCED_ON, FORCED_OFF
|
||||
}
|
||||
}
|
||||
|
||||
companion object {
|
||||
private val TAG: String = MediaSendViewModel::class.java.simpleName
|
||||
|
||||
// the maximum amount of files that can be selected to send as attachment
|
||||
const val MAX_SELECTED_FILES: Int = 32
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue