src/main/java/net/bionicmessage/objects/MultipleSourceVCardObjectStore.java

   1 /*
   2  * MultipleSourceVCardObjectStore.java
   3  *
   4  * Original file created on December 13, 2006, 10:44 PM
   5  * Copyright (C) 2006-2010 Mathew McBride
   6  *
   7  * Permission is hereby granted, free of charge, to any person obtaining a
   8  * copy of this software and associated documentation files (the "Software"),
   9  * to deal in the Software without restriction, including without limitation
  10  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  11  * and/or sell copies of the Software, and to permit persons to whom the
  12  * Software is furnished to do so, subject to the following conditions:
  13  *
  14  * The above copyright notice and this permission notice shall be included in
  15  * all copies or substantial portions of the Software.
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  19  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  22  * IN THE SOFTWARE.
  23  */
  24 package net.bionicmessage.objects;
  25
  26 import com.funambol.common.pim.contact.Contact;
  27 import com.funambol.common.pim.contact.Name;
  28 import com.funambol.common.pim.vcard.VcardParser;
  29 import java.io.ByteArrayInputStream;
  30 import java.io.PrintWriter;
  31 import java.io.StringWriter;
  32 import java.net.URI;
  33 import java.util.Hashtable;
  34 import java.util.List;
  35 import net.bionicmessage.groupdav.GroupDAVObject;
  36
  37 /**
  38  * Storage and synchronization mechanism for vCard objects
  39  * @author matt
  40  */
  41 public class MultipleSourceVCardObjectStore extends AbstractObjectStore {
  42
  43     Hashtable<String, Contact> objCache = new Hashtable();
  44
  45     public MultipleSourceVCardObjectStore(String storedir, int options) {
  46         super(storedir, options);
  47     }
  48
  49     String addFromServerToStore(String storeName, GroupDAVObject obj, List toAdd) throws Exception {
  50         Contact ct = constructContactObject(obj.getContent());
  51         String uid = ct.getUid();
  52         this.objCache.put(uid, ct);
  53         Name name = ct.getName();
  54         String n = name.getDisplayName().getPropertyValueAsString();
  55         if (name.getDisplayName().getPropertyValueAsString() == null) {
  56             n = name.getFirstName() + " " + name.getLastName();
  57         }
  58         n = n.trim();
  59         ContactWrapper cw = new ContactWrapper(obj.getContent());
  60         this.db.createObject(storeName, uid, obj.getLocation(), obj.getEtag(), n, cw, null, null);
  61         toAdd.add(uid);
  62         this.objCache.put(uid, ct);
  63         return uid;
  64     }
  65
  66     public String addObject(String sourceName, String uid, String name, String contents)
  67             throws Exception {
  68         if (contents.indexOf("\nUID:") == -1) {
  69             int endLoc = contents.indexOf("END:VCARD");
  70             StringBuffer sb = new StringBuffer();
  71             sb.append(contents.substring(0, endLoc));
  72             sb.append("UID:");
  73             sb.append(uid);
  74             sb.append("\r\n");
  75             sb.append(contents.substring(endLoc));
  76             contents = sb.toString();
  77             sb = null;
  78         }
  79         URI sourceURI = this.hostURI.resolve((String) this.sources.get(sourceName));
  80         URI newURI = sourceURI.resolve(uid + ".vcf");
  81
  82         GroupDAVObject gbo = this.server.putObject(newURI.getPath(), "text/x-vcard", contents.getBytes());
  83         return addFromServerToStore(sourceName, gbo.getLocation(), this.addedToServer);
  84     }
  85
  86     public int replaceObject(String storeName, String uid, String name, String contents)
  87             throws Exception {
  88         String[] urletag = this.db.getUrlEtagForUid(uid);
  89         GroupDAVObject gbo = this.server.modifyObject(urletag[0], urletag[1], "text/x-vcard", contents.getBytes());
  90
  91         if (gbo == null) {
  92             return 403;
  93         }
  94         this.updatedOnServer.add(uid);
  95         this.db.deleteObjectByUid(uid);
  96         addFromServerToStore(storeName, urletag[0], this.updatedOnServer);
  97         return 0;
  98     }
  99
 100     public int mergeObject(String newuid, String contents1, String contents2)
 101             throws Exception {
 102         return -1;
 103     }
 104
 105     public Contact getObjectFromStore(String uid) throws ObjectStoreException {
 106         if (this.objCache.get(uid) != null) {
 107             this.log.fine("Object " + uid + "has been retrieved from store");
 108             Contact ct = (Contact) this.objCache.get(uid);
 109             return ct;
 110         }
 111         try {
 112             ContactWrapper cw = (ContactWrapper) this.db.getObjectByUid(uid);
 113             if (cw == null) {
 114                 return null;
 115             }
 116             return constructContactObject(cw.getData());
 117         } catch (Exception ex) {
 118             throw new ObjectStoreException(uid, ObjectStoreException.OP_DOWNLOAD_ADD, new byte[0], ex);
 119         }
 120     }
 121
 122     private Contact constructContactObject(byte[] content)
 123             throws Exception {
 124         ByteArrayInputStream bis = new ByteArrayInputStream(content);
 125         VcardParser vcp = new VcardParser(bis, null, null);
 126         return vcp.vCard();
 127     }
 128
 129     public void printDebugReport() throws Exception {
 130         StringWriter sw = new StringWriter();
 131         PrintWriter lw = new PrintWriter(sw);
 132         List data = this.db.getAllData();
 133         for (int i = 0; i < data.size(); ++i) {
 134             Object[] dataarray = (Object[]) data.get(i);
 135             String uid = (String) dataarray[0];
 136             String url = (String) dataarray[1];
 137             if (url == null) {
 138                 throw new Exception("URL for " + uid + " is null");
 139             }
 140             String source = (String) dataarray[2];
 141             String etag = (String) dataarray[3];
 142             String name = (String) dataarray[4];
 143             ContactWrapper cw = (ContactWrapper) dataarray[7];
 144             String objData = new String(cw.getData());
 145             lw.println("----------");
 146             lw.println("UID:" + uid);
 147             lw.println("URL:" + url);
 148             lw.println("ETAG:" + etag);
 149             lw.println("NAME:" + name);
 150             lw.println("DATA FOLLOWS:");
 151             lw.println(objData);
 152             lw.println("----------");
 153         }
 154         lw.println("Objects added to store: ");
 155         String uid;
 156         int i;
 157         for (i = 0; i < this.addedToStore.size(); ++i) {
 158             uid = (String) this.addedToStore.get(i);
 159             lw.println("A:    " + uid);
 160         }
 161         lw.println("Objects updated from server: ");
 162         for (i = 0; i < this.updatedInStore.size(); ++i) {
 163             uid = (String) this.updatedInStore.get(i);
 164             lw.println("U:   " + uid);
 165         }
 166         lw.println("Objects deleted from store: ");
 167         for (i = 0; i < this.deletedFromStore.size(); ++i) {
 168             uid = (String) this.deletedFromStore.get(i);
 169             lw.println("D:    " + uid);
 170         }
 171         lw.println("Objects added to the server: ");
 172         for (i = 0; i < this.addedToServer.size(); ++i) {
 173             uid = (String) this.addedToServer.get(i);
 174             lw.println("SA:   " + uid);
 175         }
 176         lw.println("Objects merged to server: ");
 177         for (i = 0; i < this.updatedOnServer.size(); ++i) {
 178             uid = (String) this.updatedOnServer.get(i);
 179             lw.println("SM:   " + uid);
 180         }
 181         lw.println("Objects deleted from server: ");
 182         for (i = 0; i < this.deletedFromServer.size(); ++i) {
 183             uid = (String) this.deletedFromServer.get(i);
 184             lw.println("SD:   " + uid);
 185         }
 186         lw.flush();
 187         lw.close();
 188         this.log.info(sw.toString());
 189     }
 190
 191     private int getDownloaderParameter() {
 192         return 0;
 193     }
 194
 195     void updateObjectFromServer(String uid, GroupDAVObject obj) throws Exception {
 196         Contact ct = constructContactObject(obj.getContent());
 197         Name name = ct.getName();
 198         this.objCache.put(uid, ct);
 199         String nameValue = name.getDisplayName().getPropertyValueAsString();
 200         if (nameValue == null) {
 201             nameValue = name.getFirstName() + " " + name.getLastName();
 202         }
 203         nameValue = nameValue.trim();
 204         String strContent = new String(obj.getContent());
 205         this.db.updateObject(uid, obj.getEtag(), nameValue, null, null, strContent);
 206
 207         this.updatedInStore.add(uid);
 208     }
 209
 210     public String searchUids(String name, long dstime, long detime) throws Exception {
 211         return this.db.findUidForNameAndDate(name, null, null);
 212     }
 213 }