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;
29
30import java.io.File;
31import java.lang.ref.WeakReference;
32import java.util.ArrayList;
33import java.util.Arrays;
34import java.util.List;
35import java.util.concurrent.RejectedExecutionException;
36
37public class MediaAdapter extends RecyclerView.Adapter<MediaAdapter.MediaViewHolder> {
38
39 public static final List<String> DOCUMENT_MIMES =
40 new ImmutableList.Builder<String>()
41 .add("application/pdf")
42 .add("text/x-tex")
43 .add("text/plain")
44 .addAll(MimeUtils.WORD_DOCUMENT_MIMES)
45 .build();
46 public static final List<String> SPREAD_SHEET_MIMES =
47 Arrays.asList(
48 "text/comma-separated-values",
49 "application/vnd.ms-excel",
50 "application/vnd.stardivision.calc",
51 "application/vnd.oasis.opendocument.spreadsheet",
52 "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
53
54 public static final List<String> SLIDE_SHOW_MIMES =
55 Arrays.asList(
56 "application/vnd.ms-powerpoint",
57 "application/vnd.stardivision.impress",
58 "application/vnd.oasis.opendocument.presentation",
59 "application/vnd.openxmlformats-officedocument.presentationml.presentation",
60 "application/vnd.openxmlformats-officedocument.presentationml.slideshow");
61
62 private static final List<String> ARCHIVE_MIMES =
63 Arrays.asList(
64 "application/x-7z-compressed",
65 "application/zip",
66 "application/rar",
67 "application/x-gtar",
68 "application/x-tar");
69 public static final List<String> CODE_MIMES = Arrays.asList("text/html", "text/xml");
70
71 private final ArrayList<Attachment> attachments = new ArrayList<>();
72
73 private final XmppActivity activity;
74
75 private int mediaSize = 0;
76
77 public MediaAdapter(XmppActivity activity, @DimenRes int mediaSize) {
78 this.activity = activity;
79 this.mediaSize = Math.round(activity.getResources().getDimension(mediaSize));
80 }
81
82 public static void setMediaSize(final RecyclerView recyclerView, final int mediaSize) {
83 if (recyclerView.getAdapter() 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 if (mime.equals("application/webxdc+zip")) {
141 return R.drawable.toys_and_games_24dp;
142 } else if (Arrays.asList("application/x-pcapng", "application/vnd.tcpdump.pcap")
143 .contains(mime)) {
144 return R.drawable.ic_lan_24dp;
145 } else {
146 return R.drawable.ic_help_center_48dp;
147 }
148 }
149
150 static void renderPreview(final Attachment attachment, final ImageView imageView) {
151 ImageViewCompat.setImageTintList(
152 imageView,
153 ColorStateList.valueOf(
154 MaterialColors.getColor(
155 imageView, com.google.android.material.R.attr.colorOnSurface)));
156 imageView.setImageResource(getImageDrawable(attachment));
157 imageView.setBackgroundColor(
158 MaterialColors.getColor(
159 imageView,
160 com.google.android.material.R.attr.colorSurfaceContainerHighest));
161 }
162
163 private static boolean cancelPotentialWork(Attachment attachment, ImageView imageView) {
164 final BitmapWorkerTask bitmapWorkerTask = getBitmapWorkerTask(imageView);
165
166 if (bitmapWorkerTask != null) {
167 final Attachment oldAttachment = bitmapWorkerTask.attachment;
168 if (oldAttachment == null || !oldAttachment.equals(attachment)) {
169 bitmapWorkerTask.cancel(true);
170 } else {
171 return false;
172 }
173 }
174 return true;
175 }
176
177 private static BitmapWorkerTask getBitmapWorkerTask(ImageView imageView) {
178 if (imageView != null) {
179 final Drawable drawable = imageView.getDrawable();
180 if (drawable instanceof AsyncDrawable asyncDrawable) {
181 return asyncDrawable.getBitmapWorkerTask();
182 }
183 }
184 return null;
185 }
186
187 @NonNull
188 @Override
189 public MediaViewHolder onCreateViewHolder(@NonNull ViewGroup parent, int viewType) {
190 final LayoutInflater layoutInflater = LayoutInflater.from(parent.getContext());
191 ItemMediaBinding binding =
192 DataBindingUtil.inflate(layoutInflater, R.layout.item_media, parent, false);
193 return new MediaViewHolder(binding);
194 }
195
196 @Override
197 public void onBindViewHolder(@NonNull MediaViewHolder holder, int position) {
198 final Attachment attachment = attachments.get(position);
199 if (attachment.renderThumbnail()) {
200 loadPreview(attachment, holder.binding.media);
201 } else {
202 cancelPotentialWork(attachment, holder.binding.media);
203 renderPreview(attachment, holder.binding.media);
204 }
205 holder.binding.getRoot().setOnClickListener(v -> ViewUtil.view(activity, attachment));
206 holder.binding.getRoot().setOnCreateContextMenuListener((menu, v, menuInfo) -> {
207 final var path = activity.xmppConnectionService.getFileBackend().getOriginalPath(attachment.getUri());
208 if (path == null) return;
209 final var file = new File(path);
210 if (!file.canWrite()) return;
211
212 menu.add("Delete File").setOnMenuItemClickListener((x) -> {
213 if (file.delete()) {
214 activity.xmppConnectionService.evictPreview(file);
215 attachments.remove(attachment);
216 notifyDataSetChanged();
217 }
218 return true;
219 });
220 });
221 }
222
223 public void setAttachments(final List<Attachment> attachments) {
224 this.attachments.clear();
225 this.attachments.addAll(attachments);
226 notifyDataSetChanged();
227 }
228
229 private void setMediaSize(int mediaSize) {
230 this.mediaSize = mediaSize;
231 }
232
233 private void loadPreview(Attachment attachment, ImageView imageView) {
234 if (cancelPotentialWork(attachment, imageView)) {
235 final Bitmap bm =
236 activity.xmppConnectionService
237 .getFileBackend()
238 .getPreviewForUri(attachment, mediaSize, true);
239 if (bm != null) {
240 cancelPotentialWork(attachment, imageView);
241 imageView.setImageBitmap(bm);
242 imageView.setBackgroundColor(Color.TRANSPARENT);
243 } else {
244 // TODO consider if this is still a good, general purpose loading color
245 imageView.setBackgroundColor(0xff333333);
246 imageView.setImageDrawable(null);
247 final BitmapWorkerTask task = new BitmapWorkerTask(mediaSize, imageView);
248 final AsyncDrawable asyncDrawable =
249 new AsyncDrawable(activity.getResources(), null, task);
250 imageView.setImageDrawable(asyncDrawable);
251 try {
252 task.execute(attachment);
253 } catch (final RejectedExecutionException ignored) {
254 }
255 }
256 }
257 }
258
259 @Override
260 public int getItemCount() {
261 return attachments.size();
262 }
263
264 static class AsyncDrawable extends BitmapDrawable {
265 private final WeakReference<BitmapWorkerTask> bitmapWorkerTaskReference;
266
267 AsyncDrawable(Resources res, Bitmap bitmap, BitmapWorkerTask bitmapWorkerTask) {
268 super(res, bitmap);
269 bitmapWorkerTaskReference = new WeakReference<>(bitmapWorkerTask);
270 }
271
272 BitmapWorkerTask getBitmapWorkerTask() {
273 return bitmapWorkerTaskReference.get();
274 }
275 }
276
277 public static class MediaViewHolder extends RecyclerView.ViewHolder {
278
279 private final ItemMediaBinding binding;
280
281 MediaViewHolder(ItemMediaBinding binding) {
282 super(binding.getRoot());
283 this.binding = binding;
284 }
285 }
286
287 private static class BitmapWorkerTask extends AsyncTask<Attachment, Void, Bitmap> {
288 private final WeakReference<ImageView> imageViewReference;
289 private Attachment attachment = null;
290 private final int mediaSize;
291
292 BitmapWorkerTask(int mediaSize, ImageView imageView) {
293 this.mediaSize = mediaSize;
294 imageViewReference = new WeakReference<>(imageView);
295 }
296
297 @Override
298 protected Bitmap doInBackground(final Attachment... params) {
299 this.attachment = params[0];
300 final XmppActivity activity = XmppActivity.find(imageViewReference);
301 if (activity == null) {
302 return null;
303 }
304 return activity.xmppConnectionService
305 .getFileBackend()
306 .getPreviewForUri(this.attachment, mediaSize, false);
307 }
308
309 @Override
310 protected void onPostExecute(Bitmap bitmap) {
311 if (bitmap != null && !isCancelled()) {
312 final ImageView imageView = imageViewReference.get();
313 if (imageView != null) {
314 imageView.setImageBitmap(bitmap);
315 imageView.setBackgroundColor(0x00000000);
316 }
317 }
318 }
319 }
320}