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