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.startsWith("audio/")) {
78 attr = R.attr.media_preview_audio;
79 } else if (mime.equals("text/calendar") || (mime.equals("text/x-vcalendar"))) {
80 attr = R.attr.media_preview_calendar;
81 } else if (mime.equals("text/x-vcard")) {
82 attr = R.attr.media_preview_contact;
83 } else if (mime.equals("application/vnd.android.package-archive")) {
84 attr = R.attr.media_preview_app;
85 } else if (mime.equals("application/zip") || mime.equals("application/rar")) {
86 attr = R.attr.media_preview_archive;
87 } else if (mime.equals("application/epub+zip") || mime.equals("application/vnd.amazon.mobi8-ebook")) {
88 attr = R.attr.media_preview_ebook;
89 } else if (mime.equals(ExportBackupService.MIME_TYPE)) {
90 attr = R.attr.media_preview_backup;
91 } else if (DOCUMENT_MIMES.contains(mime)) {
92 attr = R.attr.media_preview_document;
93 } else if (mime.equals("application/gpx+xml")) {
94 attr = R.attr.media_preview_tour;
95 } else if (mime.startsWith("image/")) {
96 attr = R.attr.media_preview_image;
97 } else {
98 attr = R.attr.media_preview_unknown;
99 }
100 }
101 return attr;
102 }
103
104 static void renderPreview(Context context, Attachment attachment, ImageView imageView) {
105 imageView.setBackgroundColor(StyledAttributes.getColor(context, R.attr.color_background_tertiary));
106 imageView.setImageAlpha(Math.round(StyledAttributes.getFloat(context, R.attr.icon_alpha) * 255));
107 imageView.setImageDrawable(StyledAttributes.getDrawable(context, getImageAttr(attachment)));
108 }
109
110 private static boolean cancelPotentialWork(Attachment attachment, ImageView imageView) {
111 final BitmapWorkerTask bitmapWorkerTask = getBitmapWorkerTask(imageView);
112
113 if (bitmapWorkerTask != null) {
114 final Attachment oldAttachment = bitmapWorkerTask.attachment;
115 if (oldAttachment == null || !oldAttachment.equals(attachment)) {
116 bitmapWorkerTask.cancel(true);
117 } else {
118 return false;
119 }
120 }
121 return true;
122 }
123
124 private static BitmapWorkerTask getBitmapWorkerTask(ImageView imageView) {
125 if (imageView != null) {
126 final Drawable drawable = imageView.getDrawable();
127 if (drawable instanceof AsyncDrawable) {
128 final AsyncDrawable asyncDrawable = (AsyncDrawable) drawable;
129 return asyncDrawable.getBitmapWorkerTask();
130 }
131 }
132 return null;
133 }
134
135 @NonNull
136 @Override
137 public MediaViewHolder onCreateViewHolder(@NonNull ViewGroup parent, int viewType) {
138 LayoutInflater layoutInflater = LayoutInflater.from(parent.getContext());
139 MediaBinding binding = DataBindingUtil.inflate(layoutInflater, R.layout.media, parent, false);
140 return new MediaViewHolder(binding);
141 }
142
143 @Override
144 public void onBindViewHolder(@NonNull MediaViewHolder holder, int position) {
145 final Attachment attachment = attachments.get(position);
146 if (attachment.renderThumbnail()) {
147 holder.binding.media.setImageAlpha(255);
148 loadPreview(attachment, holder.binding.media);
149 } else {
150 cancelPotentialWork(attachment, holder.binding.media);
151 renderPreview(activity, attachment, holder.binding.media);
152 }
153 holder.binding.getRoot().setOnClickListener(v -> ViewUtil.view(activity, attachment));
154 }
155
156 public void setAttachments(List<Attachment> attachments) {
157 this.attachments.clear();
158 this.attachments.addAll(attachments);
159 notifyDataSetChanged();
160 }
161
162 private void setMediaSize(int mediaSize) {
163 this.mediaSize = mediaSize;
164 }
165
166 private void loadPreview(Attachment attachment, ImageView imageView) {
167 if (cancelPotentialWork(attachment, imageView)) {
168 final Bitmap bm = activity.xmppConnectionService.getFileBackend().getPreviewForUri(attachment, mediaSize, true);
169 if (bm != null) {
170 cancelPotentialWork(attachment, imageView);
171 imageView.setImageBitmap(bm);
172 imageView.setBackgroundColor(0x00000000);
173 } else {
174 imageView.setBackgroundColor(0xff333333);
175 imageView.setImageDrawable(null);
176 final BitmapWorkerTask task = new BitmapWorkerTask(mediaSize, imageView);
177 final AsyncDrawable asyncDrawable = new AsyncDrawable(activity.getResources(), null, task);
178 imageView.setImageDrawable(asyncDrawable);
179 try {
180 task.execute(attachment);
181 } catch (final RejectedExecutionException ignored) {
182 }
183 }
184 }
185 }
186
187 @Override
188 public int getItemCount() {
189 return attachments.size();
190 }
191
192 static class AsyncDrawable extends BitmapDrawable {
193 private final WeakReference<BitmapWorkerTask> bitmapWorkerTaskReference;
194
195 AsyncDrawable(Resources res, Bitmap bitmap, BitmapWorkerTask bitmapWorkerTask) {
196 super(res, bitmap);
197 bitmapWorkerTaskReference = new WeakReference<>(bitmapWorkerTask);
198 }
199
200 BitmapWorkerTask getBitmapWorkerTask() {
201 return bitmapWorkerTaskReference.get();
202 }
203 }
204
205 class MediaViewHolder extends RecyclerView.ViewHolder {
206
207 private final MediaBinding binding;
208
209 MediaViewHolder(MediaBinding binding) {
210 super(binding.getRoot());
211 this.binding = binding;
212 }
213 }
214
215 private static class BitmapWorkerTask extends AsyncTask<Attachment, Void, Bitmap> {
216 private final WeakReference<ImageView> imageViewReference;
217 private Attachment attachment = null;
218 private final int mediaSize;
219
220 BitmapWorkerTask(int mediaSize, ImageView imageView) {
221 this.mediaSize = mediaSize;
222 imageViewReference = new WeakReference<>(imageView);
223 }
224
225 @Override
226 protected Bitmap doInBackground(Attachment... params) {
227 this.attachment = params[0];
228 final XmppActivity activity = XmppActivity.find(imageViewReference);
229 if (activity == null) {
230 return null;
231 }
232 return activity.xmppConnectionService.getFileBackend().getPreviewForUri(this.attachment, mediaSize, false);
233 }
234
235 @Override
236 protected void onPostExecute(Bitmap bitmap) {
237 if (bitmap != null && !isCancelled()) {
238 final ImageView imageView = imageViewReference.get();
239 if (imageView != null) {
240 imageView.setImageBitmap(bitmap);
241 imageView.setBackgroundColor(0x00000000);
242 }
243 }
244 }
245 }
246}