1package eu.siacs.conversations.ui.adapter;
2
3import android.content.res.ColorStateList;
4import android.content.res.Resources;
5import android.graphics.Bitmap;
6import android.graphics.Color;
7import android.graphics.drawable.BitmapDrawable;
8import android.graphics.drawable.Drawable;
9import android.os.AsyncTask;
10import android.view.LayoutInflater;
11import android.view.ViewGroup;
12import android.widget.ImageView;
13import androidx.annotation.DimenRes;
14import androidx.annotation.DrawableRes;
15import androidx.annotation.NonNull;
16import androidx.core.widget.ImageViewCompat;
17import androidx.databinding.DataBindingUtil;
18import androidx.recyclerview.widget.RecyclerView;
19import com.google.android.material.color.MaterialColors;
20import com.google.common.base.Strings;
21import com.google.common.collect.ImmutableList;
22import eu.siacs.conversations.R;
23import eu.siacs.conversations.databinding.ItemMediaBinding;
24import eu.siacs.conversations.ui.XmppActivity;
25import eu.siacs.conversations.ui.util.Attachment;
26import eu.siacs.conversations.ui.util.ViewUtil;
27import eu.siacs.conversations.utils.MimeUtils;
28import eu.siacs.conversations.worker.ExportBackupWorker;
29import java.lang.ref.WeakReference;
30import java.util.ArrayList;
31import java.util.Arrays;
32import java.util.List;
33import java.util.concurrent.RejectedExecutionException;
34
35public class MediaAdapter extends RecyclerView.Adapter<MediaAdapter.MediaViewHolder> {
36
37 public static final List<String> DOCUMENT_MIMES =
38 new ImmutableList.Builder<String>()
39 .add("application/pdf")
40 .add("text/x-tex")
41 .add("text/plain")
42 .addAll(MimeUtils.WORD_DOCUMENT_MIMES)
43 .build();
44 public static final List<String> SPREAD_SHEET_MIMES =
45 Arrays.asList(
46 "text/comma-separated-values",
47 "application/vnd.ms-excel",
48 "application/vnd.stardivision.calc",
49 "application/vnd.oasis.opendocument.spreadsheet",
50 "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
51
52 public static final List<String> SLIDE_SHOW_MIMES =
53 Arrays.asList(
54 "application/vnd.ms-powerpoint",
55 "application/vnd.stardivision.impress",
56 "application/vnd.oasis.opendocument.presentation",
57 "application/vnd.openxmlformats-officedocument.presentationml.presentation",
58 "application/vnd.openxmlformats-officedocument.presentationml.slideshow");
59
60 private static final List<String> ARCHIVE_MIMES =
61 Arrays.asList(
62 "application/x-7z-compressed",
63 "application/zip",
64 "application/rar",
65 "application/x-gtar",
66 "application/x-tar");
67 public static final List<String> CODE_MIMES = Arrays.asList("text/html", "text/xml");
68
69 private final ArrayList<Attachment> attachments = new ArrayList<>();
70
71 private final XmppActivity activity;
72
73 private int mediaSize = 0;
74
75 public MediaAdapter(XmppActivity activity, @DimenRes int mediaSize) {
76 this.activity = activity;
77 this.mediaSize = Math.round(activity.getResources().getDimension(mediaSize));
78 }
79
80 @SuppressWarnings("rawtypes")
81 public static void setMediaSize(final RecyclerView recyclerView, int mediaSize) {
82 final RecyclerView.Adapter adapter = recyclerView.getAdapter();
83 if (adapter instanceof MediaAdapter mediaAdapter) {
84 mediaAdapter.setMediaSize(mediaSize);
85 }
86 }
87
88 public static @DrawableRes int getImageDrawable(final Attachment attachment) {
89 if (attachment.getType() == Attachment.Type.LOCATION) {
90 return R.drawable.ic_location_pin_48dp;
91 } else if (attachment.getType() == Attachment.Type.RECORDING) {
92 return R.drawable.ic_mic_48dp;
93 } else {
94 return getImageDrawable(attachment.getMime());
95 }
96 }
97
98 private static @DrawableRes int getImageDrawable(final String mime) {
99
100 // TODO ideas for more mime types: XML, HTML documents, GPG/PGP files, eml files,
101 // spreadsheets (table symbol)
102
103 // add bz2 and tar.gz to archive detection
104
105 if (Strings.isNullOrEmpty(mime)) {
106 return R.drawable.ic_help_center_48dp;
107 } else if (mime.equals("audio/x-m4b")) {
108 return R.drawable.ic_play_lesson_48dp;
109 } else if (mime.startsWith("audio/")) {
110 return R.drawable.ic_headphones_48dp;
111 } else if (mime.equals("text/calendar") || (mime.equals("text/x-vcalendar"))) {
112 return R.drawable.ic_event_48dp;
113 } else if (mime.equals("text/x-vcard")) {
114 return R.drawable.ic_person_48dp;
115 } else if (mime.equals("application/vnd.android.package-archive")) {
116 return R.drawable.ic_adb_48dp;
117 } else if (ARCHIVE_MIMES.contains(mime)) {
118 return R.drawable.ic_archive_48dp;
119 } else if (mime.equals("application/epub+zip")
120 || mime.equals("application/vnd.amazon.mobi8-ebook")) {
121 return R.drawable.ic_book_48dp;
122 } else if (mime.equals(ExportBackupWorker.MIME_TYPE)) {
123 return R.drawable.ic_backup_48dp;
124 } else if (DOCUMENT_MIMES.contains(mime)) {
125 return R.drawable.ic_description_48dp;
126 } else if (SPREAD_SHEET_MIMES.contains(mime)) {
127 return R.drawable.ic_table_48dp;
128 } else if (SLIDE_SHOW_MIMES.contains(mime)) {
129 return R.drawable.ic_slideshow_48dp;
130 } else if (mime.equals("application/gpx+xml")) {
131 return R.drawable.ic_tour_48dp;
132 } else if (mime.startsWith("image/")) {
133 return R.drawable.ic_image_48dp;
134 } else if (mime.startsWith("video/")) {
135 return R.drawable.ic_movie_48dp;
136 } else if (CODE_MIMES.contains(mime)) {
137 return R.drawable.ic_code_48dp;
138 } else if (mime.equals("message/rfc822")) {
139 return R.drawable.ic_email_48dp;
140 } else {
141 return R.drawable.ic_help_center_48dp;
142 }
143 }
144
145 static void renderPreview(final Attachment attachment, final ImageView imageView) {
146 ImageViewCompat.setImageTintList(
147 imageView,
148 ColorStateList.valueOf(
149 MaterialColors.getColor(
150 imageView, com.google.android.material.R.attr.colorOnSurface)));
151 imageView.setImageResource(getImageDrawable(attachment));
152 imageView.setBackgroundColor(
153 MaterialColors.getColor(
154 imageView,
155 com.google.android.material.R.attr.colorSurfaceContainerHighest));
156 }
157
158 private static boolean cancelPotentialWork(Attachment attachment, ImageView imageView) {
159 final BitmapWorkerTask bitmapWorkerTask = getBitmapWorkerTask(imageView);
160
161 if (bitmapWorkerTask != null) {
162 final Attachment oldAttachment = bitmapWorkerTask.attachment;
163 if (oldAttachment == null || !oldAttachment.equals(attachment)) {
164 bitmapWorkerTask.cancel(true);
165 } else {
166 return false;
167 }
168 }
169 return true;
170 }
171
172 private static BitmapWorkerTask getBitmapWorkerTask(ImageView imageView) {
173 if (imageView != null) {
174 final Drawable drawable = imageView.getDrawable();
175 if (drawable instanceof AsyncDrawable asyncDrawable) {
176 return asyncDrawable.getBitmapWorkerTask();
177 }
178 }
179 return null;
180 }
181
182 @NonNull
183 @Override
184 public MediaViewHolder onCreateViewHolder(@NonNull ViewGroup parent, int viewType) {
185 final LayoutInflater layoutInflater = LayoutInflater.from(parent.getContext());
186 ItemMediaBinding binding =
187 DataBindingUtil.inflate(layoutInflater, R.layout.item_media, parent, false);
188 return new MediaViewHolder(binding);
189 }
190
191 @Override
192 public void onBindViewHolder(@NonNull MediaViewHolder holder, int position) {
193 final Attachment attachment = attachments.get(position);
194 if (attachment.renderThumbnail()) {
195 loadPreview(attachment, holder.binding.media);
196 } else {
197 cancelPotentialWork(attachment, holder.binding.media);
198 renderPreview(attachment, holder.binding.media);
199 }
200 holder.binding.getRoot().setOnClickListener(v -> ViewUtil.view(activity, attachment));
201 }
202
203 public void setAttachments(final List<Attachment> attachments) {
204 this.attachments.clear();
205 this.attachments.addAll(attachments);
206 notifyDataSetChanged();
207 }
208
209 private void setMediaSize(int mediaSize) {
210 this.mediaSize = mediaSize;
211 }
212
213 private void loadPreview(Attachment attachment, ImageView imageView) {
214 if (cancelPotentialWork(attachment, imageView)) {
215 final Bitmap bm =
216 activity.xmppConnectionService
217 .getFileBackend()
218 .getPreviewForUri(attachment, mediaSize, true);
219 if (bm != null) {
220 cancelPotentialWork(attachment, imageView);
221 imageView.setImageBitmap(bm);
222 imageView.setBackgroundColor(Color.TRANSPARENT);
223 } else {
224 // TODO consider if this is still a good, general purpose loading color
225 imageView.setBackgroundColor(0xff333333);
226 imageView.setImageDrawable(null);
227 final BitmapWorkerTask task = new BitmapWorkerTask(mediaSize, imageView);
228 final AsyncDrawable asyncDrawable =
229 new AsyncDrawable(activity.getResources(), null, task);
230 imageView.setImageDrawable(asyncDrawable);
231 try {
232 task.execute(attachment);
233 } catch (final RejectedExecutionException ignored) {
234 }
235 }
236 }
237 }
238
239 @Override
240 public int getItemCount() {
241 return attachments.size();
242 }
243
244 static class AsyncDrawable extends BitmapDrawable {
245 private final WeakReference<BitmapWorkerTask> bitmapWorkerTaskReference;
246
247 AsyncDrawable(Resources res, Bitmap bitmap, BitmapWorkerTask bitmapWorkerTask) {
248 super(res, bitmap);
249 bitmapWorkerTaskReference = new WeakReference<>(bitmapWorkerTask);
250 }
251
252 BitmapWorkerTask getBitmapWorkerTask() {
253 return bitmapWorkerTaskReference.get();
254 }
255 }
256
257 static class MediaViewHolder extends RecyclerView.ViewHolder {
258
259 private final ItemMediaBinding binding;
260
261 MediaViewHolder(ItemMediaBinding binding) {
262 super(binding.getRoot());
263 this.binding = binding;
264 }
265 }
266
267 private static class BitmapWorkerTask extends AsyncTask<Attachment, Void, Bitmap> {
268 private final WeakReference<ImageView> imageViewReference;
269 private Attachment attachment = null;
270 private final int mediaSize;
271
272 BitmapWorkerTask(int mediaSize, ImageView imageView) {
273 this.mediaSize = mediaSize;
274 imageViewReference = new WeakReference<>(imageView);
275 }
276
277 @Override
278 protected Bitmap doInBackground(final Attachment... params) {
279 this.attachment = params[0];
280 final XmppActivity activity = XmppActivity.find(imageViewReference);
281 if (activity == null) {
282 return null;
283 }
284 return activity.xmppConnectionService
285 .getFileBackend()
286 .getPreviewForUri(this.attachment, mediaSize, false);
287 }
288
289 @Override
290 protected void onPostExecute(Bitmap bitmap) {
291 if (bitmap != null && !isCancelled()) {
292 final ImageView imageView = imageViewReference.get();
293 if (imageView != null) {
294 imageView.setImageBitmap(bitmap);
295 imageView.setBackgroundColor(0x00000000);
296 }
297 }
298 }
299 }
300}