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