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