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