src/com/owncloud/android/operations/SynchronizeFolderOperation.java

   1 /* ownCloud Android client application
   2  *   Copyright (C) 2012 Bartek Przybylski
   3  *
   4  *   This program is free software: you can redistribute it and/or modify
   5  *   it under the terms of the GNU General Public License as published by
   6  *   the Free Software Foundation, either version 3 of the License, or
   7  *   (at your option) any later version.
   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.operations;
  20
  21 import java.io.File;
  22 import java.io.FileInputStream;
  23 import java.io.FileOutputStream;
  24 import java.io.IOException;
  25 import java.io.InputStream;
  26 import java.io.OutputStream;
  27 import java.util.HashMap;
  28 import java.util.List;
  29 import java.util.Map;
  30 import java.util.Vector;
  31
  32 import org.apache.http.HttpStatus;
  33 import org.apache.jackrabbit.webdav.MultiStatus;
  34 import org.apache.jackrabbit.webdav.client.methods.PropFindMethod;
  35
  36 import android.accounts.Account;
  37 import android.content.Context;
  38 import android.util.Log;
  39
  40 import com.owncloud.android.datamodel.DataStorageManager;
  41 import com.owncloud.android.datamodel.OCFile;
  42 import com.owncloud.android.operations.RemoteOperationResult.ResultCode;
  43 import com.owncloud.android.utils.FileStorageUtils;
  44
  45 import eu.alefzero.webdav.WebdavClient;
  46 import eu.alefzero.webdav.WebdavEntry;
  47 import eu.alefzero.webdav.WebdavUtils;
  48
  49
  50 /**
  51  * Remote operation performing the synchronization a the contents of a remote folder with the local database
  52  *
  53  * @author David A. Velasco
  54  */
  55 public class SynchronizeFolderOperation extends RemoteOperation {
  56
  57     private static final String TAG = SynchronizeFolderOperation.class.getSimpleName();
  58
  59     /** Remote folder to synchronize */
  60     private String mRemotePath;
  61
  62     /** Timestamp for the synchronization in progress */
  63     private long mCurrentSyncTime;
  64
  65     /** Id of the folder to synchronize in the local database */
  66     private long mParentId;
  67
  68     /** Access to the local database */
  69     private DataStorageManager mStorageManager;
  70
  71     /** Account where the file to synchronize belongs */
  72     private Account mAccount;
  73
  74     /** Android context; necessary to send requests to the download service; maybe something to refactor */
  75     private Context mContext;
  76
  77     /** Files and folders contained in the synchronized folder */
  78     private List<OCFile> mChildren;
  79
  80     private int mConflictsFound;
  81
  82     private int mFailsInFavouritesFound;
  83
  84     private Map<String, String> mForgottenLocalFiles;
  85
  86
  87     public SynchronizeFolderOperation(  String remotePath,
  88                                         long currentSyncTime,
  89                                         long parentId,
  90                                         DataStorageManager dataStorageManager,
  91                                         Account account,
  92                                         Context context ) {
  93         mRemotePath = remotePath;
  94         mCurrentSyncTime = currentSyncTime;
  95         mParentId = parentId;
  96         mStorageManager = dataStorageManager;
  97         mAccount = account;
  98         mContext = context;
  99         mForgottenLocalFiles = new HashMap<String, String>();
 100     }
 101
 102
 103     public int getConflictsFound() {
 104         return mConflictsFound;
 105     }
 106
 107     public int getFailsInFavouritesFound() {
 108         return mFailsInFavouritesFound;
 109     }
 110
 111     public Map<String, String> getForgottenLocalFiles() {
 112         return mForgottenLocalFiles;
 113     }
 114
 115     /**
 116      * Returns the list of files and folders contained in the synchronized folder, if called after synchronization is complete.
 117      *
 118      * @return      List of files and folders contained in the synchronized folder.
 119      */
 120     public List<OCFile> getChildren() {
 121         return mChildren;
 122     }
 123
 124
 125     @Override
 126     protected RemoteOperationResult run(WebdavClient client) {
 127         RemoteOperationResult result = null;
 128         mFailsInFavouritesFound = 0;
 129         mConflictsFound = 0;
 130         mForgottenLocalFiles.clear();
 131
 132         // code before in FileSyncAdapter.fetchData
 133         PropFindMethod query = null;
 134         try {
 135             Log.d(TAG, "Synchronizing " + mAccount.name + ", fetching files in " + mRemotePath);
 136
 137             // remote request
 138             query = new PropFindMethod(client.getBaseUri() + WebdavUtils.encodePath(mRemotePath));
 139             int status = client.executeMethod(query);
 140
 141             // check and process response   - /// TODO take into account all the possible status per child-resource
 142             if (isMultiStatus(status)) {
 143                 MultiStatus resp = query.getResponseBodyAsMultiStatus();
 144
 145                 // synchronize properties of the parent folder, if necessary
 146                 if (mParentId == DataStorageManager.ROOT_PARENT_ID) {
 147                     WebdavEntry we = new WebdavEntry(resp.getResponses()[0], client.getBaseUri().getPath());
 148                     OCFile parent = fillOCFile(we);
 149                     mStorageManager.saveFile(parent);
 150                     mParentId = parent.getFileId();
 151                 }
 152
 153                 // read contents in folder
 154                 List<OCFile> updatedFiles = new Vector<OCFile>(resp.getResponses().length - 1);
 155                 List<SynchronizeFileOperation> filesToSyncContents = new Vector<SynchronizeFileOperation>();
 156                 for (int i = 1; i < resp.getResponses().length; ++i) {
 157                     /// new OCFile instance with the data from the server
 158                     WebdavEntry we = new WebdavEntry(resp.getResponses()[i], client.getBaseUri().getPath());
 159                     OCFile file = fillOCFile(we);
 160
 161                     /// set data about local state, keeping unchanged former data if existing
 162                     file.setLastSyncDateForProperties(mCurrentSyncTime);
 163                     OCFile oldFile = mStorageManager.getFileByPath(file.getRemotePath());
 164                     if (oldFile != null) {
 165                         file.setKeepInSync(oldFile.keepInSync());
 166                         file.setLastSyncDateForData(oldFile.getLastSyncDateForData());
 167                         checkAndFixForeignStoragePath(oldFile);
 168                         file.setStoragePath(oldFile.getStoragePath());
 169                     }
 170
 171                     /// scan default location if local copy of file is not linked in OCFile instance
 172                     if (file.getStoragePath() == null && !file.isDirectory()) {
 173                         File f = new File(FileStorageUtils.getDefaultSavePathFor(mAccount.name, file));
 174                         if (f.exists()) {
 175                             file.setStoragePath(f.getAbsolutePath());
 176                             file.setLastSyncDateForData(f.lastModified());
 177                         }
 178                     }
 179
 180                     /// prepare content synchronization for kept-in-sync files
 181                     if (file.keepInSync()) {
 182                         SynchronizeFileOperation operation = new SynchronizeFileOperation(  oldFile,
 183                                                                                             file,
 184                                                                                             mStorageManager,
 185                                                                                             mAccount,
 186                                                                                             true,
 187                                                                                             false,
 188                                                                                             mContext
 189                                                                                             );
 190                         filesToSyncContents.add(operation);
 191                     }
 192
 193                     updatedFiles.add(file);
 194                 }
 195
 196                 // save updated contents in local database; all at once, trying to get a best performance in database update (not a big deal, indeed)
 197                 mStorageManager.saveFiles(updatedFiles);
 198
 199                 // request for the synchronization of files AFTER saving last properties
 200                 SynchronizeFileOperation op = null;
 201                 RemoteOperationResult contentsResult = null;
 202                 for (int i=0; i < filesToSyncContents.size(); i++) {
 203                     op = filesToSyncContents.get(i);
 204                     contentsResult = op.execute(client);   // returns without waiting for upload or download finishes
 205                     if (!contentsResult.isSuccess()) {
 206                         if (contentsResult.getCode() == ResultCode.SYNC_CONFLICT) {
 207                             mConflictsFound++;
 208                         } else {
 209                             mFailsInFavouritesFound++;
 210                             if (contentsResult.getException() != null) {
 211                                 Log.d(TAG, "Error while synchronizing favourites : " +  contentsResult.getLogMessage(), contentsResult.getException());
 212                             } else {
 213                                 Log.d(TAG, "Error while synchronizing favourites : " + contentsResult.getLogMessage());
 214                             }
 215                         }
 216                     }   // won't let these fails break the synchronization process
 217                 }
 218
 219
 220                 // removal of obsolete files
 221                 mChildren = mStorageManager.getDirectoryContent(mStorageManager.getFileById(mParentId));
 222                 OCFile file;
 223                 String currentSavePath = FileStorageUtils.getSavePath(mAccount.name);
 224                 for (int i=0; i < mChildren.size(); ) {
 225                     file = mChildren.get(i);
 226                     if (file.getLastSyncDateForProperties() != mCurrentSyncTime) {
 227                         Log.d(TAG, "removing file: " + file);
 228                         mStorageManager.removeFile(file, (file.isDown() && file.getStoragePath().startsWith(currentSavePath)));
 229                         mChildren.remove(i);
 230                     } else {
 231                         i++;
 232                     }
 233                 }
 234
 235             } else {
 236                 client.exhaustResponse(query.getResponseBodyAsStream());
 237             }
 238
 239             // prepare result object
 240             if (isMultiStatus(status)) {
 241                 if (mConflictsFound > 0  || mFailsInFavouritesFound > 0) {
 242                     result = new RemoteOperationResult(ResultCode.SYNC_CONFLICT);   // should be different result, but will do the job
 243
 244                 } else {
 245                     result = new RemoteOperationResult(true, status);
 246                 }
 247             } else {
 248                 result = new RemoteOperationResult(false, status);
 249             }
 250             Log.i(TAG, "Synchronizing " + mAccount.name + ", folder " + mRemotePath + ": " + result.getLogMessage());
 251
 252
 253         } catch (Exception e) {
 254             result = new RemoteOperationResult(e);
 255             Log.e(TAG, "Synchronizing " + mAccount.name + ", folder " + mRemotePath + ": " + result.getLogMessage(), result.getException());
 256
 257         } finally {
 258             if (query != null)
 259                 query.releaseConnection();  // let the connection available for other methods
 260         }
 261
 262         return result;
 263     }
 264
 265
 266     public boolean isMultiStatus(int status) {
 267         return (status == HttpStatus.SC_MULTI_STATUS);
 268     }
 269
 270
 271     /**
 272      * Creates and populates a new {@link OCFile} object with the data read from the server.
 273      *
 274      * @param we        WebDAV entry read from the server for a WebDAV resource (remote file or folder).
 275      * @return          New OCFile instance representing the remote resource described by we.
 276      */
 277     private OCFile fillOCFile(WebdavEntry we) {
 278         OCFile file = new OCFile(we.decodedPath());
 279         file.setCreationTimestamp(we.createTimestamp());
 280         file.setFileLength(we.contentLength());
 281         file.setMimetype(we.contentType());
 282         file.setModificationTimestamp(we.modifiedTimestamp());
 283         file.setParentId(mParentId);
 284         return file;
 285     }
 286
 287
 288     /**
 289      * Checks the storage path of the OCFile received as parameter. If it's out of the local ownCloud folder,
 290      * tries to copy the file inside it.
 291      *
 292      * If the copy fails, the link to the local file is nullified. The account of forgotten files is kept in
 293      * {@link #mForgottenLocalFiles}
 294      *
 295      * @param file      File to check and fix.
 296      */
 297     private void checkAndFixForeignStoragePath(OCFile file) {
 298         String storagePath = file.getStoragePath();
 299         String expectedPath = FileStorageUtils.getDefaultSavePathFor(mAccount.name, file);
 300         if (storagePath != null && !storagePath.equals(expectedPath)) {
 301             /// fix storagePaths out of the local ownCloud folder
 302             File originalFile = new File(storagePath);
 303             if (FileStorageUtils.getUsableSpace(mAccount.name) < originalFile.length()) {
 304                 mForgottenLocalFiles.put(file.getRemotePath(), storagePath);
 305                 file.setStoragePath(null);
 306
 307             } else {
 308                 InputStream in = null;
 309                 OutputStream out = null;
 310                 try {
 311                     File expectedFile = new File(expectedPath);
 312                     File expectedParent = expectedFile.getParentFile();
 313                     expectedParent.mkdirs();
 314                     if (!expectedParent.isDirectory()) {
 315                         throw new IOException("Unexpected error: parent directory could not be created");
 316                     }
 317                     expectedFile.createNewFile();
 318                     if (!expectedFile.isFile()) {
 319                         throw new IOException("Unexpected error: target file could not be created");
 320                     }
 321                     in = new FileInputStream(originalFile);
 322                     out = new FileOutputStream(expectedFile);
 323                     byte[] buf = new byte[1024];
 324                     int len;
 325                     while ((len = in.read(buf)) > 0){
 326                         out.write(buf, 0, len);
 327                     }
 328                     file.setStoragePath(expectedPath);
 329
 330                 } catch (Exception e) {
 331                     Log.e(TAG, "Exception while copying foreign file " + expectedPath, e);
 332                     mForgottenLocalFiles.put(file.getRemotePath(), storagePath);
 333                     file.setStoragePath(null);
 334
 335                 } finally {
 336                     try {
 337                         if (in != null) in.close();
 338                     } catch (Exception e) {
 339                         Log.d(TAG, "Weird exception while closing input stream for " + storagePath + " (ignoring)", e);
 340                     }
 341                     try {
 342                         if (out != null) out.close();
 343                     } catch (Exception e) {
 344                         Log.d(TAG, "Weird exception while closing output stream for " + expectedPath + " (ignoring)", e);
 345                     }
 346                 }
 347             }
 348         }
 349     }
 350
 351
 352 }