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