1package eu.siacs.conversations.ui.adapter;
2
3import android.content.Context;
4import android.content.res.Resources;
5import android.graphics.Bitmap;
6import android.graphics.drawable.BitmapDrawable;
7import android.graphics.drawable.Drawable;
8import android.os.AsyncTask;
9import android.util.Log;
10import android.view.LayoutInflater;
11import android.view.ViewGroup;
12import android.widget.ImageView;
13
14import androidx.annotation.AttrRes;
15import androidx.annotation.DimenRes;
16import androidx.annotation.NonNull;
17import androidx.databinding.DataBindingUtil;
18import androidx.recyclerview.widget.RecyclerView;
19
20import java.lang.ref.WeakReference;
21import java.util.ArrayList;
22import java.util.Arrays;
23import java.util.List;
24import java.util.concurrent.RejectedExecutionException;
25
26import eu.siacs.conversations.Config;
27import eu.siacs.conversations.R;
28import eu.siacs.conversations.databinding.MediaBinding;
29import eu.siacs.conversations.services.ExportBackupService;
30import eu.siacs.conversations.ui.XmppActivity;
31import eu.siacs.conversations.ui.util.Attachment;
32import eu.siacs.conversations.ui.util.StyledAttributes;
33import eu.siacs.conversations.ui.util.ViewUtil;
34
35public class MediaAdapter extends RecyclerView.Adapter<MediaAdapter.MediaViewHolder> {
36
37 public static final List<String> DOCUMENT_MIMES = Arrays.asList(
38 "application/pdf",
39 "application/vnd.oasis.opendocument.text",
40 "application/msword",
41 "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
42 "text/x-tex",
43 "text/plain"
44 );
45
46 private final ArrayList<Attachment> attachments = new ArrayList<>();
47
48 private final XmppActivity activity;
49
50 private int mediaSize = 0;
51
52 public MediaAdapter(XmppActivity activity, @DimenRes int mediaSize) {
53 this.activity = activity;
54 this.mediaSize = Math.round(activity.getResources().getDimension(mediaSize));
55 }
56
57 @SuppressWarnings("rawtypes")
58 public static void setMediaSize(RecyclerView recyclerView, int mediaSize) {
59 final RecyclerView.Adapter adapter = recyclerView.getAdapter();
60 if (adapter instanceof MediaAdapter) {
61 ((MediaAdapter) adapter).setMediaSize(mediaSize);
62 }
63 }
64
65 private static @AttrRes
66 int getImageAttr(Attachment attachment) {
67 final @AttrRes int attr;
68 if (attachment.getType() == Attachment.Type.LOCATION) {
69 attr = R.attr.media_preview_location;
70 } else if (attachment.getType() == Attachment.Type.RECORDING) {
71 attr = R.attr.media_preview_recording;
72 } else {
73 final String mime = attachment.getMime();
74 Log.d(Config.LOGTAG, "mime=" + mime);
75 if (mime == null) {
76 attr = R.attr.media_preview_unknown;
77 } else if (mime.equals("audio/x-m4b")) {
78 attr = R.attr.media_preview_audiobook;
79 } else if (mime.startsWith("audio/")) {
80 attr = R.attr.media_preview_audio;
81 } else if (mime.equals("text/calendar") || (mime.equals("text/x-vcalendar"))) {
82 attr = R.attr.media_preview_calendar;
83 } else if (mime.equals("text/x-vcard")) {
84 attr = R.attr.media_preview_contact;
85 } else if (mime.equals("application/vnd.android.package-archive")) {
86 attr = R.attr.media_preview_app;
87 } else if (mime.equals("application/zip") || mime.equals("application/rar")) {
88 attr = R.attr.media_preview_archive;
89 } else if (mime.equals("application/epub+zip") || mime.equals("application/vnd.amazon.mobi8-ebook")) {
90 attr = R.attr.media_preview_ebook;
91 } else if (mime.equals(ExportBackupService.MIME_TYPE)) {
92 attr = R.attr.media_preview_backup;
93 } else if (DOCUMENT_MIMES.contains(mime)) {
94 attr = R.attr.media_preview_document;
95 } else if (mime.equals("application/gpx+xml")) {
96 attr = R.attr.media_preview_tour;
97 } else if (mime.startsWith("image/")) {
98 attr = R.attr.media_preview_image;
99 } else {
100 attr = R.attr.media_preview_unknown;
101 }
102 }
103 return attr;
104 }
105
106 static void renderPreview(Context context, Attachment attachment, ImageView imageView) {
107 imageView.setBackgroundColor(StyledAttributes.getColor(context, R.attr.color_background_tertiary));
108 imageView.setImageAlpha(Math.round(StyledAttributes.getFloat(context, R.attr.icon_alpha) * 255));
109 imageView.setImageDrawable(StyledAttributes.getDrawable(context, getImageAttr(attachment)));
110 }
111
112 private static boolean cancelPotentialWork(Attachment attachment, ImageView imageView) {
113 final BitmapWorkerTask bitmapWorkerTask = getBitmapWorkerTask(imageView);
114
115 if (bitmapWorkerTask != null) {
116 final Attachment oldAttachment = bitmapWorkerTask.attachment;
117 if (oldAttachment == null || !oldAttachment.equals(attachment)) {
118 bitmapWorkerTask.cancel(true);
119 } else {
120 return false;
121 }
122 }
123 return true;
124 }
125
126 private static BitmapWorkerTask getBitmapWorkerTask(ImageView imageView) {
127 if (imageView != null) {
128 final Drawable drawable = imageView.getDrawable();
129 if (drawable instanceof AsyncDrawable) {
130 final AsyncDrawable asyncDrawable = (AsyncDrawable) drawable;
131 return asyncDrawable.getBitmapWorkerTask();
132 }
133 }
134 return null;
135 }
136
137 @NonNull
138 @Override
139 public MediaViewHolder onCreateViewHolder(@NonNull ViewGroup parent, int viewType) {
140 LayoutInflater layoutInflater = LayoutInflater.from(parent.getContext());
141 MediaBinding binding = DataBindingUtil.inflate(layoutInflater, R.layout.media, parent, false);
142 return new MediaViewHolder(binding);
143 }
144
145 @Override
146 public void onBindViewHolder(@NonNull MediaViewHolder holder, int position) {
147 final Attachment attachment = attachments.get(position);
148 if (attachment.renderThumbnail()) {
149 holder.binding.media.setImageAlpha(255);
150 loadPreview(attachment, holder.binding.media);
151 } else {
152 cancelPotentialWork(attachment, holder.binding.media);
153 renderPreview(activity, attachment, holder.binding.media);
154 }
155 holder.binding.getRoot().setOnClickListener(v -> ViewUtil.view(activity, attachment));
156 }
157
158 public void setAttachments(List<Attachment> attachments) {
159 this.attachments.clear();
160 this.attachments.addAll(attachments);
161 notifyDataSetChanged();
162 }
163
164 private void setMediaSize(int mediaSize) {
165 this.mediaSize = mediaSize;
166 }
167
168 private void loadPreview(Attachment attachment, ImageView imageView) {
169 if (cancelPotentialWork(attachment, imageView)) {
170 final Bitmap bm = activity.xmppConnectionService.getFileBackend().getPreviewForUri(attachment, mediaSize, true);
171 if (bm != null) {
172 cancelPotentialWork(attachment, imageView);
173 imageView.setImageBitmap(bm);
174 imageView.setBackgroundColor(0x00000000);
175 } else {
176 imageView.setBackgroundColor(0xff333333);
177 imageView.setImageDrawable(null);
178 final BitmapWorkerTask task = new BitmapWorkerTask(mediaSize, imageView);
179 final AsyncDrawable asyncDrawable = new AsyncDrawable(activity.getResources(), null, task);
180 imageView.setImageDrawable(asyncDrawable);
181 try {
182 task.execute(attachment);
183 } catch (final RejectedExecutionException ignored) {
184 }
185 }
186 }
187 }
188
189 @Override
190 public int getItemCount() {
191 return attachments.size();
192 }
193
194 static class AsyncDrawable extends BitmapDrawable {
195 private final WeakReference<BitmapWorkerTask> bitmapWorkerTaskReference;
196
197 AsyncDrawable(Resources res, Bitmap bitmap, BitmapWorkerTask bitmapWorkerTask) {
198 super(res, bitmap);
199 bitmapWorkerTaskReference = new WeakReference<>(bitmapWorkerTask);
200 }
201
202 BitmapWorkerTask getBitmapWorkerTask() {
203 return bitmapWorkerTaskReference.get();
204 }
205 }
206
207 class MediaViewHolder extends RecyclerView.ViewHolder {
208
209 private final MediaBinding binding;
210
211 MediaViewHolder(MediaBinding binding) {
212 super(binding.getRoot());
213 this.binding = binding;
214 }
215 }
216
217 private static class BitmapWorkerTask extends AsyncTask<Attachment, Void, Bitmap> {
218 private final WeakReference<ImageView> imageViewReference;
219 private Attachment attachment = null;
220 private final int mediaSize;
221
222 BitmapWorkerTask(int mediaSize, ImageView imageView) {
223 this.mediaSize = mediaSize;
224 imageViewReference = new WeakReference<>(imageView);
225 }
226
227 @Override
228 protected Bitmap doInBackground(Attachment... params) {
229 this.attachment = params[0];
230 final XmppActivity activity = XmppActivity.find(imageViewReference);
231 if (activity == null) {
232 return null;
233 }
234 return activity.xmppConnectionService.getFileBackend().getPreviewForUri(this.attachment, mediaSize, false);
235 }
236
237 @Override
238 protected void onPostExecute(Bitmap bitmap) {
239 if (bitmap != null && !isCancelled()) {
240 final ImageView imageView = imageViewReference.get();
241 if (imageView != null) {
242 imageView.setImageBitmap(bitmap);
243 imageView.setBackgroundColor(0x00000000);
244 }
245 }
246 }
247 }
248}