Store file on db with 'downloading' state after adding it to pending downloads
[pub/Android/ownCloud.git] / src / com / owncloud / android / files / services / FileDownloader.java
1 /* ownCloud Android client application
2 * Copyright (C) 2012 Bartek Przybylski
3 * Copyright (C) 2012-2013 ownCloud Inc.
4 *
5 * This program is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License version 2,
7 * as published by the Free Software Foundation.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
16 *
17 */
18
19 package com.owncloud.android.files.services;
20
21 import java.io.File;
22 import java.io.IOException;
23 import java.util.AbstractList;
24 import java.util.ArrayList;
25 import java.util.HashMap;
26 import java.util.Iterator;
27 import java.util.Map;
28 import java.util.Vector;
29 import java.util.concurrent.ConcurrentHashMap;
30 import java.util.concurrent.ConcurrentMap;
31
32 import com.owncloud.android.R;
33 import com.owncloud.android.authentication.AuthenticatorActivity;
34 import com.owncloud.android.datamodel.FileDataStorageManager;
35 import com.owncloud.android.datamodel.OCFile;
36
37 import com.owncloud.android.lib.common.network.OnDatatransferProgressListener;
38 import com.owncloud.android.lib.common.OwnCloudAccount;
39 import com.owncloud.android.lib.common.OwnCloudClient;
40 import com.owncloud.android.lib.common.OwnCloudClientManagerFactory;
41 import com.owncloud.android.notifications.NotificationBuilderWithProgressBar;
42 import com.owncloud.android.notifications.NotificationDelayer;
43 import com.owncloud.android.lib.common.operations.RemoteOperationResult;
44 import com.owncloud.android.lib.common.operations.RemoteOperationResult.ResultCode;
45 import com.owncloud.android.lib.common.utils.Log_OC;
46 import com.owncloud.android.lib.resources.files.FileUtils;
47 import com.owncloud.android.operations.DownloadFileOperation;
48 import com.owncloud.android.ui.activity.FileActivity;
49 import com.owncloud.android.ui.activity.FileDisplayActivity;
50 import com.owncloud.android.ui.preview.PreviewImageActivity;
51 import com.owncloud.android.ui.preview.PreviewImageFragment;
52 import com.owncloud.android.utils.ErrorMessageAdapter;
53
54 import android.accounts.Account;
55 import android.accounts.AccountsException;
56 import android.app.NotificationManager;
57 import android.app.PendingIntent;
58 import android.app.Service;
59 import android.content.Intent;
60 import android.os.Binder;
61 import android.os.Handler;
62 import android.os.HandlerThread;
63 import android.os.IBinder;
64 import android.os.Looper;
65 import android.os.Message;
66 import android.os.Process;
67 import android.support.v4.app.NotificationCompat;
68
69 public class FileDownloader extends Service implements OnDatatransferProgressListener {
70
71 public static final String EXTRA_ACCOUNT = "ACCOUNT";
72 public static final String EXTRA_FILE = "FILE";
73
74 public static final String ACTION_CANCEL_FILE_DOWNLOAD = "CANCEL_FILE_DOWNLOAD";
75
76 private static final String DOWNLOAD_ADDED_MESSAGE = "DOWNLOAD_ADDED";
77 private static final String DOWNLOAD_FINISH_MESSAGE = "DOWNLOAD_FINISH";
78 public static final String EXTRA_DOWNLOAD_RESULT = "RESULT";
79 public static final String EXTRA_FILE_PATH = "FILE_PATH";
80 public static final String EXTRA_REMOTE_PATH = "REMOTE_PATH";
81 public static final String ACCOUNT_NAME = "ACCOUNT_NAME";
82
83 private static final String TAG = "FileDownloader";
84
85 private Looper mServiceLooper;
86 private ServiceHandler mServiceHandler;
87 private IBinder mBinder;
88 private OwnCloudClient mDownloadClient = null;
89 private Account mLastAccount = null;
90 private FileDataStorageManager mStorageManager;
91
92 private ConcurrentMap<String, DownloadFileOperation> mPendingDownloads = new ConcurrentHashMap<String, DownloadFileOperation>();
93 private DownloadFileOperation mCurrentDownload = null;
94
95 private NotificationManager mNotificationManager;
96 private NotificationCompat.Builder mNotificationBuilder;
97 private int mLastPercent;
98
99
100 public static String getDownloadAddedMessage() {
101 return FileDownloader.class.getName().toString() + DOWNLOAD_ADDED_MESSAGE;
102 }
103
104 public static String getDownloadFinishMessage() {
105 return FileDownloader.class.getName().toString() + DOWNLOAD_FINISH_MESSAGE;
106 }
107
108 /**
109 * Builds a key for mPendingDownloads from the account and file to download
110 *
111 * @param account Account where the file to download is stored
112 * @param file File to download
113 */
114 private String buildRemoteName(Account account, OCFile file) {
115 return account.name + file.getRemotePath();
116 }
117
118
119 /**
120 * Service initialization
121 */
122 @Override
123 public void onCreate() {
124 super.onCreate();
125 mNotificationManager = (NotificationManager) getSystemService(NOTIFICATION_SERVICE);
126 HandlerThread thread = new HandlerThread("FileDownloaderThread",
127 Process.THREAD_PRIORITY_BACKGROUND);
128 thread.start();
129 mServiceLooper = thread.getLooper();
130 mServiceHandler = new ServiceHandler(mServiceLooper, this);
131 mBinder = new FileDownloaderBinder();
132 }
133
134 /**
135 * Entry point to add one or several files to the queue of downloads.
136 *
137 * New downloads are added calling to startService(), resulting in a call to this method. This ensures the service will keep on working
138 * although the caller activity goes away.
139 */
140 @Override
141 public int onStartCommand(Intent intent, int flags, int startId) {
142 if ( !intent.hasExtra(EXTRA_ACCOUNT) ||
143 !intent.hasExtra(EXTRA_FILE)
144 /*!intent.hasExtra(EXTRA_FILE_PATH) ||
145 !intent.hasExtra(EXTRA_REMOTE_PATH)*/
146 ) {
147 Log_OC.e(TAG, "Not enough information provided in intent");
148 return START_NOT_STICKY;
149 } else {
150 final Account account = intent.getParcelableExtra(EXTRA_ACCOUNT);
151 final OCFile file = intent.getParcelableExtra(EXTRA_FILE);
152
153 if (ACTION_CANCEL_FILE_DOWNLOAD.equals(intent.getAction())) {
154
155 new Thread(new Runnable() {
156 public void run() {
157 // Cancel the download
158 cancel(account, file);
159 }
160 }).start();
161
162 } else {
163
164 AbstractList<String> requestedDownloads = new Vector<String>(); // dvelasco: now this always contains just one element, but that can change in a near future (download of multiple selection)
165 String downloadKey = buildRemoteName(account, file);
166 try {
167 DownloadFileOperation newDownload = new DownloadFileOperation(account, file);
168 mPendingDownloads.putIfAbsent(downloadKey, newDownload);
169 newDownload.addDatatransferProgressListener(this);
170 newDownload.addDatatransferProgressListener((FileDownloaderBinder) mBinder);
171 requestedDownloads.add(downloadKey);
172
173 // Store file on db with state 'downloading'
174 FileDataStorageManager storageManager = new FileDataStorageManager(account, getContentResolver());
175 file.setDownloading(true);
176 storageManager.saveFile(file);
177
178 sendBroadcastNewDownload(newDownload);
179
180 } catch (IllegalArgumentException e) {
181 Log_OC.e(TAG, "Not enough information provided in intent: " + e.getMessage());
182 return START_NOT_STICKY;
183 }
184
185 if (requestedDownloads.size() > 0) {
186 Message msg = mServiceHandler.obtainMessage();
187 msg.arg1 = startId;
188 msg.obj = requestedDownloads;
189 mServiceHandler.sendMessage(msg);
190 }
191 }
192 }
193
194 return START_NOT_STICKY;
195 }
196
197
198 /**
199 * Provides a binder object that clients can use to perform operations on the queue of downloads, excepting the addition of new files.
200 *
201 * Implemented to perform cancellation, pause and resume of existing downloads.
202 */
203 @Override
204 public IBinder onBind(Intent arg0) {
205 return mBinder;
206 }
207
208
209 /**
210 * Called when ALL the bound clients were onbound.
211 */
212 @Override
213 public boolean onUnbind(Intent intent) {
214 ((FileDownloaderBinder)mBinder).clearListeners();
215 return false; // not accepting rebinding (default behaviour)
216 }
217
218
219 /**
220 * Binder to let client components to perform operations on the queue of downloads.
221 *
222 * It provides by itself the available operations.
223 */
224 public class FileDownloaderBinder extends Binder implements OnDatatransferProgressListener {
225
226 /**
227 * Map of listeners that will be reported about progress of downloads from a {@link FileDownloaderBinder} instance
228 */
229 private Map<String, OnDatatransferProgressListener> mBoundListeners = new HashMap<String, OnDatatransferProgressListener>();
230
231
232 /**
233 * Cancels a pending or current download of a remote file.
234 *
235 * @param account Owncloud account where the remote file is stored.
236 * @param file A file in the queue of pending downloads
237 */
238 public void cancel(Account account, OCFile file) {
239 DownloadFileOperation download = null;
240 synchronized (mPendingDownloads) {
241 download = mPendingDownloads.remove(buildRemoteName(account, file));
242 }
243 if (download != null) {
244 download.cancel();
245 }
246 }
247
248
249 public void clearListeners() {
250 mBoundListeners.clear();
251 }
252
253
254 /**
255 * Returns True when the file described by 'file' in the ownCloud account 'account' is downloading or waiting to download.
256 *
257 * If 'file' is a directory, returns 'true' if some of its descendant files is downloading or waiting to download.
258 *
259 * @param account Owncloud account where the remote file is stored.
260 * @param file A file that could be in the queue of downloads.
261 */
262 public boolean isDownloading(Account account, OCFile file) {
263 if (account == null || file == null) return false;
264 String targetKey = buildRemoteName(account, file);
265 synchronized (mPendingDownloads) {
266 if (file.isFolder()) {
267 // this can be slow if there are many downloads :(
268 Iterator<String> it = mPendingDownloads.keySet().iterator();
269 boolean found = false;
270 while (it.hasNext() && !found) {
271 found = it.next().startsWith(targetKey);
272 }
273 return found;
274 } else {
275 return (mPendingDownloads.containsKey(targetKey));
276 }
277 }
278 }
279
280
281 /**
282 * Adds a listener interested in the progress of the download for a concrete file.
283 *
284 * @param listener Object to notify about progress of transfer.
285 * @param account ownCloud account holding the file of interest.
286 * @param file {@link OCfile} of interest for listener.
287 */
288 public void addDatatransferProgressListener (OnDatatransferProgressListener listener, Account account, OCFile file) {
289 if (account == null || file == null || listener == null) return;
290 String targetKey = buildRemoteName(account, file);
291 mBoundListeners.put(targetKey, listener);
292 }
293
294
295 /**
296 * Removes a listener interested in the progress of the download for a concrete file.
297 *
298 * @param listener Object to notify about progress of transfer.
299 * @param account ownCloud account holding the file of interest.
300 * @param file {@link OCfile} of interest for listener.
301 */
302 public void removeDatatransferProgressListener (OnDatatransferProgressListener listener, Account account, OCFile file) {
303 if (account == null || file == null || listener == null) return;
304 String targetKey = buildRemoteName(account, file);
305 if (mBoundListeners.get(targetKey) == listener) {
306 mBoundListeners.remove(targetKey);
307 }
308 }
309
310 @Override
311 public void onTransferProgress(long progressRate, long totalTransferredSoFar, long totalToTransfer,
312 String fileName) {
313 String key = buildRemoteName(mCurrentDownload.getAccount(), mCurrentDownload.getFile());
314 OnDatatransferProgressListener boundListener = mBoundListeners.get(key);
315 if (boundListener != null) {
316 boundListener.onTransferProgress(progressRate, totalTransferredSoFar, totalToTransfer, fileName);
317 }
318 }
319
320 }
321
322
323 /**
324 * Download worker. Performs the pending downloads in the order they were requested.
325 *
326 * Created with the Looper of a new thread, started in {@link FileUploader#onCreate()}.
327 */
328 private static class ServiceHandler extends Handler {
329 // don't make it a final class, and don't remove the static ; lint will warn about a possible memory leak
330 FileDownloader mService;
331 public ServiceHandler(Looper looper, FileDownloader service) {
332 super(looper);
333 if (service == null)
334 throw new IllegalArgumentException("Received invalid NULL in parameter 'service'");
335 mService = service;
336 }
337
338 @Override
339 public void handleMessage(Message msg) {
340 @SuppressWarnings("unchecked")
341 AbstractList<String> requestedDownloads = (AbstractList<String>) msg.obj;
342 if (msg.obj != null) {
343 Iterator<String> it = requestedDownloads.iterator();
344 while (it.hasNext()) {
345 mService.downloadFile(it.next());
346 }
347 }
348 mService.stopSelf(msg.arg1);
349 }
350 }
351
352
353 /**
354 * Core download method: requests a file to download and stores it.
355 *
356 * @param downloadKey Key to access the download to perform, contained in mPendingDownloads
357 */
358 private void downloadFile(String downloadKey) {
359
360 synchronized(mPendingDownloads) {
361 mCurrentDownload = mPendingDownloads.get(downloadKey);
362 }
363
364 if (mCurrentDownload != null) {
365
366 notifyDownloadStart(mCurrentDownload);
367
368 RemoteOperationResult downloadResult = null;
369 try {
370 /// prepare client object to send the request to the ownCloud server
371 if (mDownloadClient == null || !mLastAccount.equals(mCurrentDownload.getAccount())) {
372 mLastAccount = mCurrentDownload.getAccount();
373 mStorageManager =
374 new FileDataStorageManager(mLastAccount, getContentResolver());
375 OwnCloudAccount ocAccount = new OwnCloudAccount(mLastAccount, this);
376 mDownloadClient = OwnCloudClientManagerFactory.getDefaultSingleton().
377 getClientFor(ocAccount, this);
378 }
379
380 /// perform the download
381 downloadResult = mCurrentDownload.execute(mDownloadClient);
382 if (downloadResult.isSuccess()) {
383 saveDownloadedFile();
384 }
385
386 } catch (AccountsException e) {
387 Log_OC.e(TAG, "Error while trying to get autorization for " + mLastAccount.name, e);
388 downloadResult = new RemoteOperationResult(e);
389 } catch (IOException e) {
390 Log_OC.e(TAG, "Error while trying to get autorization for " + mLastAccount.name, e);
391 downloadResult = new RemoteOperationResult(e);
392
393 } finally {
394 synchronized(mPendingDownloads) {
395 mPendingDownloads.remove(downloadKey);
396 }
397 }
398
399
400 /// notify result
401 notifyDownloadResult(mCurrentDownload, downloadResult);
402
403 sendBroadcastDownloadFinished(mCurrentDownload, downloadResult);
404 }
405 }
406
407
408 /**
409 * Updates the OC File after a successful download.
410 */
411 private void saveDownloadedFile() {
412 OCFile file = mStorageManager.getFileById(mCurrentDownload.getFile().getFileId());
413 long syncDate = System.currentTimeMillis();
414 file.setLastSyncDateForProperties(syncDate);
415 file.setLastSyncDateForData(syncDate);
416 file.setNeedsUpdateThumbnail(true);
417 file.setModificationTimestamp(mCurrentDownload.getModificationTimestamp());
418 file.setModificationTimestampAtLastSyncForData(mCurrentDownload.getModificationTimestamp());
419 // file.setEtag(mCurrentDownload.getEtag()); // TODO Etag, where available
420 file.setMimetype(mCurrentDownload.getMimeType());
421 file.setStoragePath(mCurrentDownload.getSavePath());
422 file.setFileLength((new File(mCurrentDownload.getSavePath()).length()));
423 file.setRemoteId(mCurrentDownload.getFile().getRemoteId());
424 file.setDownloading(false);
425 mStorageManager.saveFile(file);
426 mStorageManager.triggerMediaScan(file.getStoragePath());
427 }
428
429
430 /**
431 * Creates a status notification to show the download progress
432 *
433 * @param download Download operation starting.
434 */
435 private void notifyDownloadStart(DownloadFileOperation download) {
436 /// create status notification with a progress bar
437 mLastPercent = 0;
438 mNotificationBuilder =
439 NotificationBuilderWithProgressBar.newNotificationBuilderWithProgressBar(this);
440 mNotificationBuilder
441 .setSmallIcon(R.drawable.notification_icon)
442 .setTicker(getString(R.string.downloader_download_in_progress_ticker))
443 .setContentTitle(getString(R.string.downloader_download_in_progress_ticker))
444 .setOngoing(true)
445 .setProgress(100, 0, download.getSize() < 0)
446 .setContentText(
447 String.format(getString(R.string.downloader_download_in_progress_content), 0,
448 new File(download.getSavePath()).getName())
449 );
450
451 /// includes a pending intent in the notification showing the details view of the file
452 Intent showDetailsIntent = null;
453 if (PreviewImageFragment.canBePreviewed(download.getFile())) {
454 showDetailsIntent = new Intent(this, PreviewImageActivity.class);
455 } else {
456 showDetailsIntent = new Intent(this, FileDisplayActivity.class);
457 }
458 showDetailsIntent.putExtra(FileActivity.EXTRA_FILE, download.getFile());
459 showDetailsIntent.putExtra(FileActivity.EXTRA_ACCOUNT, download.getAccount());
460 showDetailsIntent.setFlags(Intent.FLAG_ACTIVITY_CLEAR_TOP);
461
462 mNotificationBuilder.setContentIntent(PendingIntent.getActivity(
463 this, (int) System.currentTimeMillis(), showDetailsIntent, 0
464 ));
465
466 mNotificationManager.notify(R.string.downloader_download_in_progress_ticker, mNotificationBuilder.build());
467 }
468
469
470 /**
471 * Callback method to update the progress bar in the status notification.
472 */
473 @Override
474 public void onTransferProgress(long progressRate, long totalTransferredSoFar, long totalToTransfer, String filePath) {
475 int percent = (int)(100.0*((double)totalTransferredSoFar)/((double)totalToTransfer));
476 if (percent != mLastPercent) {
477 mNotificationBuilder.setProgress(100, percent, totalToTransfer < 0);
478 String fileName = filePath.substring(filePath.lastIndexOf(FileUtils.PATH_SEPARATOR) + 1);
479 String text = String.format(getString(R.string.downloader_download_in_progress_content), percent, fileName);
480 mNotificationBuilder.setContentText(text);
481 mNotificationManager.notify(R.string.downloader_download_in_progress_ticker, mNotificationBuilder.build());
482 }
483 mLastPercent = percent;
484 }
485
486
487 /**
488 * Updates the status notification with the result of a download operation.
489 *
490 * @param downloadResult Result of the download operation.
491 * @param download Finished download operation
492 */
493 private void notifyDownloadResult(DownloadFileOperation download, RemoteOperationResult downloadResult) {
494 mNotificationManager.cancel(R.string.downloader_download_in_progress_ticker);
495 if (!downloadResult.isCancelled()) {
496 int tickerId = (downloadResult.isSuccess()) ? R.string.downloader_download_succeeded_ticker :
497 R.string.downloader_download_failed_ticker;
498
499 boolean needsToUpdateCredentials = (
500 downloadResult.getCode() == ResultCode.UNAUTHORIZED ||
501 downloadResult.isIdPRedirection()
502 );
503 tickerId = (needsToUpdateCredentials) ?
504 R.string.downloader_download_failed_credentials_error : tickerId;
505
506 mNotificationBuilder
507 .setTicker(getString(tickerId))
508 .setContentTitle(getString(tickerId))
509 .setAutoCancel(true)
510 .setOngoing(false)
511 .setProgress(0, 0, false);
512
513 if (needsToUpdateCredentials) {
514
515 // let the user update credentials with one click
516 Intent updateAccountCredentials = new Intent(this, AuthenticatorActivity.class);
517 updateAccountCredentials.putExtra(AuthenticatorActivity.EXTRA_ACCOUNT, download.getAccount());
518 updateAccountCredentials.putExtra(AuthenticatorActivity.EXTRA_ACTION, AuthenticatorActivity.ACTION_UPDATE_EXPIRED_TOKEN);
519 updateAccountCredentials.addFlags(Intent.FLAG_ACTIVITY_NEW_TASK);
520 updateAccountCredentials.addFlags(Intent.FLAG_ACTIVITY_EXCLUDE_FROM_RECENTS);
521 updateAccountCredentials.addFlags(Intent.FLAG_FROM_BACKGROUND);
522 mNotificationBuilder
523 .setContentIntent(PendingIntent.getActivity(
524 this, (int) System.currentTimeMillis(), updateAccountCredentials, PendingIntent.FLAG_ONE_SHOT));
525
526 mDownloadClient = null; // grant that future retries on the same account will get the fresh credentials
527
528 } else {
529 // TODO put something smart in showDetailsIntent
530 Intent showDetailsIntent = new Intent();
531 mNotificationBuilder
532 .setContentIntent(PendingIntent.getActivity(
533 this, (int) System.currentTimeMillis(), showDetailsIntent, 0));
534 }
535
536 mNotificationBuilder.setContentText(ErrorMessageAdapter.getErrorCauseMessage(downloadResult, download, getResources()));
537 mNotificationManager.notify(tickerId, mNotificationBuilder.build());
538
539 // Remove success notification
540 if (downloadResult.isSuccess()) {
541 // Sleep 2 seconds, so show the notification before remove it
542 NotificationDelayer.cancelWithDelay(
543 mNotificationManager,
544 R.string.downloader_download_succeeded_ticker,
545 2000);
546 }
547
548 }
549 }
550
551
552 /**
553 * Sends a broadcast when a download finishes in order to the interested activities can update their view
554 *
555 * @param download Finished download operation
556 * @param downloadResult Result of the download operation
557 */
558 private void sendBroadcastDownloadFinished(DownloadFileOperation download, RemoteOperationResult downloadResult) {
559 Intent end = new Intent(getDownloadFinishMessage());
560 end.putExtra(EXTRA_DOWNLOAD_RESULT, downloadResult.isSuccess());
561 end.putExtra(ACCOUNT_NAME, download.getAccount().name);
562 end.putExtra(EXTRA_REMOTE_PATH, download.getRemotePath());
563 end.putExtra(EXTRA_FILE_PATH, download.getSavePath());
564 sendStickyBroadcast(end);
565 }
566
567
568 /**
569 * Sends a broadcast when a new download is added to the queue.
570 *
571 * @param download Added download operation
572 */
573 private void sendBroadcastNewDownload(DownloadFileOperation download) {
574 Intent added = new Intent(getDownloadAddedMessage());
575 added.putExtra(ACCOUNT_NAME, download.getAccount().name);
576 added.putExtra(EXTRA_REMOTE_PATH, download.getRemotePath());
577 added.putExtra(EXTRA_FILE_PATH, download.getSavePath());
578 sendStickyBroadcast(added);
579 }
580
581 /**
582 * Cancel operation
583 * @param account Owncloud account where the remote file is stored.
584 * @param file File OCFile
585 */
586 public void cancel(Account account, OCFile file){
587 DownloadFileOperation download = null;
588 String targetKey = buildRemoteName(account, file);
589 ArrayList<String> keyItems = new ArrayList<String>();
590 synchronized (mPendingDownloads) {
591 if (file.isFolder()) {
592 Log_OC.d(TAG, "Folder download. Canceling pending downloads (from folder)");
593 Iterator<String> it = mPendingDownloads.keySet().iterator();
594 boolean found = false;
595 while (it.hasNext()) {
596 String keyDownloadOperation = it.next();
597 found = keyDownloadOperation.startsWith(targetKey);
598 if (found) {
599 keyItems.add(keyDownloadOperation);
600 }
601 }
602 } else {
603 // this is not really expected...
604 Log_OC.d(TAG, "Canceling file download");
605 keyItems.add(buildRemoteName(account, file));
606 }
607 }
608 for (String item: keyItems) {
609 download = mPendingDownloads.remove(item);
610 Log_OC.d(TAG, "Key removed: " + item);
611
612 if (download != null) {
613 download.cancel();
614 }
615 }
616 }
617 }