src/eu/alefzero/owncloud/datamodel/FileDataStorageManager.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 eu.alefzero.owncloud.datamodel;
  20
  21 import java.util.Vector;
  22
  23 import eu.alefzero.owncloud.db.ProviderMeta.ProviderTableMeta;
  24 import android.accounts.Account;
  25 import android.content.ContentProviderClient;
  26 import android.content.ContentResolver;
  27 import android.content.ContentValues;
  28 import android.database.Cursor;
  29 import android.net.Uri;
  30 import android.os.RemoteException;
  31 import android.util.Log;
  32
  33 public class FileDataStorageManager implements DataStorageManager {
  34
  35   private ContentResolver mContentResolver;
  36   private ContentProviderClient mContentProvider;
  37   private Account mAccount;
  38
  39   private static String TAG = "FileDataStorageManager";
  40
  41   public FileDataStorageManager(Account account, ContentResolver cr) {
  42     mContentProvider = null;
  43     mContentResolver = cr;
  44     mAccount = account;
  45   }
  46
  47   public FileDataStorageManager(Account account, ContentProviderClient cp) {
  48     mContentProvider = cp;
  49     mContentResolver = null;
  50     mAccount = account;
  51   }
  52
  53   @Override
  54   public OCFile getFileByPath(String path) {
  55     Cursor c = getCursorForValue(ProviderTableMeta.FILE_PATH, path);
  56     OCFile file = null;
  57     if (c.moveToFirst()) {
  58       file = createFileInstance(c);
  59       c.close();
  60     }
  61     return file;
  62   }
  63
  64   @Override
  65   public OCFile getFileById(long id) {
  66     Cursor c = getCursorForValue(ProviderTableMeta._ID, String.valueOf(id));
  67     OCFile file = null;
  68     if (c.moveToFirst()) {
  69       file = createFileInstance(c);
  70       c.close();
  71     }
  72     return file;
  73   }
  74
  75   @Override
  76   public boolean fileExists(long id) {
  77     return fileExists(ProviderTableMeta._ID, String.valueOf(id));
  78   }
  79
  80   @Override
  81   public boolean fileExists(String path) {
  82     return fileExists(ProviderTableMeta.FILE_PATH, path);
  83   }
  84
  85   @Override
  86   public boolean saveFile(OCFile file) {
  87     boolean overriden = false;
  88     ContentValues cv = new ContentValues();
  89     cv.put(ProviderTableMeta.FILE_MODIFIED, file.getModificationTimestamp());
  90     cv.put(ProviderTableMeta.FILE_CREATION, file.getCreationTimestamp());
  91     cv.put(ProviderTableMeta.FILE_CONTENT_LENGTH, file.getFileLength());
  92     cv.put(ProviderTableMeta.FILE_CONTENT_TYPE, file.getMimetype());
  93     cv.put(ProviderTableMeta.FILE_NAME, file.getFileName());
  94     if (file.getParentId() != 0)
  95       cv.put(ProviderTableMeta.FILE_PARENT, file.getParentId());
  96     cv.put(ProviderTableMeta.FILE_PATH, file.getPath());
  97     cv.put(ProviderTableMeta.FILE_STORAGE_PATH, file.getStoragePath());
  98     cv.put(ProviderTableMeta.FILE_ACCOUNT_OWNER, mAccount.name);
  99
 100     if (fileExists(file.getPath())) {
 101       file.setFileId(getFileByPath(file.getPath()).getFileId());
 102       overriden = true;
 103       if (getContentResolver() != null) {
 104         getContentResolver().update(ProviderTableMeta.CONTENT_URI,
 105                                     cv,
 106                                     ProviderTableMeta._ID + "=?",
 107                                     new String[] {String.valueOf(file.getFileId())});
 108       } else {
 109         try {
 110           getContentProvider().update(ProviderTableMeta.CONTENT_URI,
 111                                       cv,
 112                                       ProviderTableMeta._ID + "=?",
 113                                       new String[] {String.valueOf(file.getFileId())});
 114         } catch (RemoteException e) {
 115           Log.e(TAG, "Fail to insert insert file to database " + e.getMessage());
 116         }
 117       }
 118     } else {
 119       Uri result_uri = null;
 120       if (getContentResolver() != null) {
 121         result_uri = getContentResolver().insert(ProviderTableMeta.CONTENT_URI_FILE, cv);
 122       } else {
 123         try {
 124           result_uri = getContentProvider().insert(ProviderTableMeta.CONTENT_URI_FILE, cv);
 125         } catch (RemoteException e) {
 126           Log.e(TAG, "Fail to insert insert file to database " + e.getMessage());
 127         }
 128       }
 129       if (result_uri != null) {
 130         long new_id = Long.parseLong(result_uri.getPathSegments().get(1));
 131         file.setFileId(new_id);
 132       }
 133     }
 134
 135     if (file.isDirectory() && file.needsUpdatingWhileSaving())
 136       for (OCFile f : getDirectoryContent(file))
 137         saveFile(f);
 138
 139     return overriden;
 140   }
 141
 142   public void setAccount(Account account) {
 143     mAccount = account;
 144   }
 145
 146   public Account getAccount() {
 147     return mAccount;
 148   }
 149
 150   public void setContentResolver(ContentResolver cr) {
 151     mContentResolver = cr;
 152   }
 153
 154   public ContentResolver getContentResolver() {
 155     return mContentResolver;
 156   }
 157
 158   public void setContentProvider(ContentProviderClient cp) {
 159     mContentProvider = cp;
 160   }
 161
 162   public ContentProviderClient getContentProvider() {
 163     return mContentProvider;
 164   }
 165
 166   public Vector<OCFile> getDirectoryContent(OCFile f) {
 167     if (f != null && f.isDirectory() && f.getFileId() != -1) {
 168       Vector<OCFile> ret = new Vector<OCFile>();
 169
 170       Uri req_uri = Uri.withAppendedPath(
 171           ProviderTableMeta.CONTENT_URI_DIR, String.valueOf(f.getFileId()));
 172       Cursor c = null;
 173
 174       if (getContentProvider() != null) {
 175         try {
 176           c = getContentProvider().query(req_uri, null, null, null, null);
 177         } catch (RemoteException e) {
 178           Log.e(TAG, e.getMessage());
 179           return ret;
 180         }
 181       } else {
 182         c = getContentResolver().query(req_uri, null, null, null, null);
 183       }
 184
 185       if (c.moveToFirst()) {
 186         do {
 187           OCFile child = createFileInstance(c);
 188           ret.add(child);
 189         } while (c.moveToNext());
 190       }
 191
 192       c.close();
 193       return ret;
 194     }
 195     return null;
 196   }
 197
 198
 199   private boolean fileExists(String cmp_key, String value) {
 200     Cursor c;
 201     if (getContentResolver() != null) {
 202       c = getContentResolver().query(ProviderTableMeta.CONTENT_URI,
 203                                     null,
 204                                     cmp_key + "=?",
 205                                     new String[] {value},
 206                                     null);
 207     } else {
 208       try {
 209         c = getContentProvider().query(ProviderTableMeta.CONTENT_URI,
 210                                       null,
 211                                       cmp_key + "=?",
 212                                       new String[] {value},
 213                                       null);
 214       } catch (RemoteException e) {
 215         Log.e(TAG, "Couldn't determine file existance, assuming non existance: " + e.getMessage());
 216         return false;
 217       }
 218     }
 219     return c.moveToFirst();
 220   }
 221
 222   private Cursor getCursorForValue(String key, String value) {
 223     Cursor c = null;
 224     if (getContentResolver() != null) {
 225       c = getContentResolver().query(ProviderTableMeta.CONTENT_URI,
 226                                      null,
 227                                      key + "=?",
 228                                      new String[] {value},
 229                                      null);
 230     } else {
 231       try {
 232         c = getContentProvider().query(ProviderTableMeta.CONTENT_URI,
 233                                        null,
 234                                        key + "=?",
 235                                        new String[]{value},
 236                                        null);
 237       } catch (RemoteException e) {
 238         Log.e(TAG, "Could not get file details: " + e.getMessage());
 239         c = null;
 240       }
 241     }
 242     return c;
 243   }
 244
 245   private OCFile createFileInstance(Cursor c) {
 246     OCFile file = null;
 247     if (c != null) {
 248       file = new OCFile(c.getString(c.getColumnIndex(ProviderTableMeta.FILE_PATH)));
 249       file.setFileId(c.getLong(c.getColumnIndex(ProviderTableMeta._ID)));
 250       file.setParentId(c.getLong(c.getColumnIndex(ProviderTableMeta.FILE_PARENT)));
 251       file.setStoragePath(c.getString(c.getColumnIndex(ProviderTableMeta.FILE_STORAGE_PATH)));
 252       file.setMimetype(c.getString(c.getColumnIndex(ProviderTableMeta.FILE_CONTENT_TYPE)));
 253       file.setFileLength(c.getLong(c.getColumnIndex(ProviderTableMeta.FILE_CONTENT_LENGTH)));
 254       file.setCreationTimestamp(c.getLong(c.getColumnIndex(ProviderTableMeta.FILE_CREATION)));
 255       file.setModificationTimestamp(c.getLong(c.getColumnIndex(ProviderTableMeta.FILE_MODIFIED)));
 256     }
 257     return file;
 258   }
 259
 260 }