1package eu.siacs.conversations.persistance;
2
3import java.util.ArrayList;
4import java.util.List;
5import java.util.UUID;
6import java.util.concurrent.CopyOnWriteArrayList;
7
8import eu.siacs.conversations.entities.Account;
9import eu.siacs.conversations.entities.Contact;
10import eu.siacs.conversations.entities.Conversation;
11import eu.siacs.conversations.entities.Message;
12import eu.siacs.conversations.entities.Presences;
13import android.content.ContentValues;
14import android.content.Context;
15import android.database.Cursor;
16import android.database.sqlite.SQLiteDatabase;
17import android.database.sqlite.SQLiteOpenHelper;
18import android.os.Bundle;
19import android.util.Log;
20
21public class DatabaseBackend extends SQLiteOpenHelper {
22
23 private static DatabaseBackend instance = null;
24
25 private static final String DATABASE_NAME = "history";
26 private static final int DATABASE_VERSION = 3;
27
28 public DatabaseBackend(Context context) {
29 super(context, DATABASE_NAME, null, DATABASE_VERSION);
30 }
31
32 @Override
33 public void onCreate(SQLiteDatabase db) {
34 db.execSQL("PRAGMA foreign_keys=ON;");
35 db.execSQL("create table " + Account.TABLENAME + "(" + Account.UUID
36 + " TEXT PRIMARY KEY," + Account.USERNAME + " TEXT,"
37 + Account.SERVER + " TEXT," + Account.PASSWORD + " TEXT,"
38 + Account.ROSTERVERSION + " TEXT," + Account.OPTIONS
39 + " NUMBER, "+Account.KEYS+" TEXT)");
40 db.execSQL("create table " + Conversation.TABLENAME + " ("
41 + Conversation.UUID + " TEXT PRIMARY KEY, " + Conversation.NAME
42 + " TEXT, " + Conversation.CONTACT + " TEXT, "
43 + Conversation.ACCOUNT + " TEXT, " + Conversation.CONTACTJID
44 + " TEXT, " + Conversation.CREATED + " NUMBER, "
45 + Conversation.STATUS + " NUMBER," + Conversation.MODE
46 + " NUMBER," + "FOREIGN KEY(" + Conversation.ACCOUNT
47 + ") REFERENCES " + Account.TABLENAME + "(" + Account.UUID
48 + ") ON DELETE CASCADE);");
49 db.execSQL("create table " + Message.TABLENAME + "( " + Message.UUID
50 + " TEXT PRIMARY KEY, " + Message.CONVERSATION + " TEXT, "
51 + Message.TIME_SENT + " NUMBER, " + Message.COUNTERPART
52 + " TEXT, " + Message.BODY + " TEXT, " + Message.ENCRYPTION
53 + " NUMBER, " + Message.STATUS + " NUMBER," +Message.TYPE +" NUMBER, FOREIGN KEY("
54 + Message.CONVERSATION + ") REFERENCES "
55 + Conversation.TABLENAME + "(" + Conversation.UUID
56 + ") ON DELETE CASCADE);");
57 db.execSQL("create table " + Contact.TABLENAME + "(" + Contact.UUID
58 + " TEXT PRIMARY KEY, " + Contact.ACCOUNT + " TEXT, "
59 + Contact.DISPLAYNAME + " TEXT," + Contact.JID + " TEXT,"
60 + Contact.PRESENCES + " TEXT, " + Contact.KEYS
61 + " TEXT," + Contact.PHOTOURI + " TEXT," + Contact.SUBSCRIPTION
62 + " NUMBER," + Contact.SYSTEMACCOUNT + " NUMBER, "
63 + "FOREIGN KEY(" + Contact.ACCOUNT + ") REFERENCES "
64 + Account.TABLENAME + "(" + Account.UUID
65 + ") ON DELETE CASCADE);");
66 }
67
68 @Override
69 public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {
70 if (oldVersion < 2 && newVersion >= 2) {
71 // enable compression by default.
72 db.execSQL("update " + Account.TABLENAME
73 + " set " + Account.OPTIONS + " = " + Account.OPTIONS + " | 8");
74 }
75 if (oldVersion < 3 && newVersion >= 3) {
76 //add field type to message
77 db.execSQL("ALTER TABLE "+Message.TABLENAME+" ADD COLUMN "+Message.TYPE+" NUMBER");;
78 }
79 }
80
81 public static synchronized DatabaseBackend getInstance(Context context) {
82 if (instance == null) {
83 instance = new DatabaseBackend(context);
84 }
85 return instance;
86 }
87
88 public void createConversation(Conversation conversation) {
89 SQLiteDatabase db = this.getWritableDatabase();
90 db.insert(Conversation.TABLENAME, null, conversation.getContentValues());
91 }
92
93 public void createMessage(Message message) {
94 SQLiteDatabase db = this.getWritableDatabase();
95 db.insert(Message.TABLENAME, null, message.getContentValues());
96 }
97
98 public void createAccount(Account account) {
99 SQLiteDatabase db = this.getWritableDatabase();
100 db.insert(Account.TABLENAME, null, account.getContentValues());
101 }
102
103 public void createContact(Contact contact) {
104 SQLiteDatabase db = this.getWritableDatabase();
105 db.insert(Contact.TABLENAME, null, contact.getContentValues());
106 }
107
108 public int getConversationCount() {
109 SQLiteDatabase db = this.getReadableDatabase();
110 Cursor cursor = db.rawQuery("select count(uuid) as count from "
111 + Conversation.TABLENAME + " where " + Conversation.STATUS
112 + "=" + Conversation.STATUS_AVAILABLE, null);
113 cursor.moveToFirst();
114 return cursor.getInt(0);
115 }
116
117 public List<Conversation> getConversations(int status) {
118 List<Conversation> list = new ArrayList<Conversation>();
119 SQLiteDatabase db = this.getReadableDatabase();
120 String[] selectionArgs = { "" + status };
121 Cursor cursor = db.rawQuery("select * from " + Conversation.TABLENAME
122 + " where " + Conversation.STATUS + " = ? order by "
123 + Conversation.CREATED + " desc", selectionArgs);
124 while (cursor.moveToNext()) {
125 list.add(Conversation.fromCursor(cursor));
126 }
127 return list;
128 }
129
130 public List<Message> getMessages(Conversation conversation, int limit) {
131 List<Message> list = new CopyOnWriteArrayList<Message>();
132 SQLiteDatabase db = this.getReadableDatabase();
133 String[] selectionArgs = { conversation.getUuid() };
134 Cursor cursor = db.query(Message.TABLENAME, null, Message.CONVERSATION
135 + "=?", selectionArgs, null, null, Message.TIME_SENT + " DESC",
136 String.valueOf(limit));
137 if (cursor.getCount() > 0) {
138 cursor.moveToLast();
139 do {
140 list.add(Message.fromCursor(cursor));
141 } while (cursor.moveToPrevious());
142 }
143 return list;
144 }
145
146 public Conversation findConversation(Account account, String contactJid) {
147 SQLiteDatabase db = this.getReadableDatabase();
148 String[] selectionArgs = { account.getUuid(), contactJid+"%" };
149 Cursor cursor = db.query(Conversation.TABLENAME, null,
150 Conversation.ACCOUNT + "=? AND " + Conversation.CONTACTJID + " like ?",
151 selectionArgs, null, null, null);
152 if (cursor.getCount() == 0)
153 return null;
154 cursor.moveToFirst();
155 return Conversation.fromCursor(cursor);
156 }
157
158 public void updateConversation(Conversation conversation) {
159 SQLiteDatabase db = this.getWritableDatabase();
160 String[] args = { conversation.getUuid() };
161 db.update(Conversation.TABLENAME, conversation.getContentValues(),
162 Conversation.UUID + "=?", args);
163 }
164
165 public List<Account> getAccounts() {
166 List<Account> list = new ArrayList<Account>();
167 SQLiteDatabase db = this.getReadableDatabase();
168 Cursor cursor = db.query(Account.TABLENAME, null, null, null, null,
169 null, null);
170 Log.d("gultsch", "found " + cursor.getCount() + " accounts");
171 while (cursor.moveToNext()) {
172 list.add(Account.fromCursor(cursor));
173 }
174 return list;
175 }
176
177 public void updateAccount(Account account) {
178 SQLiteDatabase db = this.getWritableDatabase();
179 String[] args = { account.getUuid() };
180 db.update(Account.TABLENAME, account.getContentValues(), Account.UUID
181 + "=?", args);
182 }
183
184 public void deleteAccount(Account account) {
185 SQLiteDatabase db = this.getWritableDatabase();
186 String[] args = { account.getUuid() };
187 db.delete(Account.TABLENAME, Account.UUID + "=?", args);
188 }
189
190 @Override
191 public SQLiteDatabase getWritableDatabase() {
192 SQLiteDatabase db = super.getWritableDatabase();
193 db.execSQL("PRAGMA foreign_keys=ON;");
194 return db;
195 }
196
197 public void updateMessage(Message message) {
198 SQLiteDatabase db = this.getWritableDatabase();
199 String[] args = { message.getUuid() };
200 db.update(Message.TABLENAME, message.getContentValues(), Message.UUID
201 + "=?", args);
202 }
203
204 public void updateContact(Contact contact, boolean updatePresences) {
205 SQLiteDatabase db = this.getWritableDatabase();
206 String[] args = { contact.getUuid() };
207 ContentValues values = contact.getContentValues();
208 if (!updatePresences) {
209 values.remove(Contact.PRESENCES);
210 } else {
211 values.remove(Contact.DISPLAYNAME);
212 values.remove(Contact.PHOTOURI);
213 values.remove(Contact.SYSTEMACCOUNT);
214 }
215 db.update(Contact.TABLENAME, contact.getContentValues(), Contact.UUID
216 + "=?", args);
217 }
218
219 public void clearPresences(Account account) {
220 SQLiteDatabase db = this.getWritableDatabase();
221 String[] args = { account.getUuid() };
222 ContentValues values = new ContentValues();
223 values.put(Contact.PRESENCES,"[]");
224 db.update(Contact.TABLENAME, values, Contact.ACCOUNT
225 + "=?", args);
226 }
227
228 public void mergeContacts(List<Contact> contacts) {
229 SQLiteDatabase db = this.getWritableDatabase();
230 for (int i = 0; i < contacts.size(); i++) {
231 Contact contact = contacts.get(i);
232 String[] columns = {Contact.UUID, Contact.PRESENCES};
233 String[] args = {contact.getAccount().getUuid(), contact.getJid()};
234 Cursor cursor = db.query(Contact.TABLENAME, columns,Contact.ACCOUNT+"=? AND "+Contact.JID+"=?", args, null, null, null);
235 if (cursor.getCount()>=1) {
236 cursor.moveToFirst();
237 contact.setUuid(cursor.getString(0));
238 updateContact(contact,false);
239 } else {
240 contact.setUuid(UUID.randomUUID().toString());
241 createContact(contact);
242 }
243 }
244 }
245
246 public List<Contact> getContactsByAccount(Account account) {
247 List<Contact> list = new ArrayList<Contact>();
248 SQLiteDatabase db = this.getReadableDatabase();
249 Cursor cursor;
250 if (account==null) {
251 cursor = db.query(Contact.TABLENAME, null, null, null, null,
252 null, null);
253 } else {
254 String args[] = {account.getUuid()};
255 cursor = db.query(Contact.TABLENAME, null, Contact.ACCOUNT+"=?", args, null,
256 null, null);
257 }
258 while (cursor.moveToNext()) {
259 list.add(Contact.fromCursor(cursor));
260 }
261 return list;
262 }
263
264 public List<Contact> getContacts(String where) {
265 List<Contact> list = new ArrayList<Contact>();
266 SQLiteDatabase db = this.getReadableDatabase();
267 Cursor cursor = db.query(Contact.TABLENAME, null, where, null, null, null, null);
268 while (cursor.moveToNext()) {
269 list.add(Contact.fromCursor(cursor));
270 }
271 return list;
272 }
273
274 public Contact findContact(Account account, String jid) {
275 SQLiteDatabase db = this.getReadableDatabase();
276 String[] selectionArgs = { account.getUuid(), jid };
277 Cursor cursor = db.query(Contact.TABLENAME, null,
278 Contact.ACCOUNT + "=? AND " + Contact.JID + "=?",
279 selectionArgs, null, null, null);
280 if (cursor.getCount() == 0)
281 return null;
282 cursor.moveToFirst();
283 return Contact.fromCursor(cursor);
284 }
285
286 public void deleteMessage(Message message) {
287 SQLiteDatabase db = this.getWritableDatabase();
288 String[] args = { message.getUuid() };
289 db.delete(Message.TABLENAME, Message.UUID + "=?", args);
290 }
291
292 public void deleteMessagesInConversation(Conversation conversation) {
293 SQLiteDatabase db = this.getWritableDatabase();
294 String[] args = { conversation.getUuid() };
295 db.delete(Message.TABLENAME, Message.CONVERSATION + "=?", args);
296 }
297
298 public void deleteContact(Contact contact) {
299 SQLiteDatabase db = this.getWritableDatabase();
300 String[] args = { contact.getUuid() };
301 db.delete(Contact.TABLENAME, Contact.UUID + "=?", args);
302 }
303
304 public Contact getContact(String uuid) {
305 SQLiteDatabase db = this.getWritableDatabase();
306 String[] args = { uuid };
307 Cursor cursor = db.query(Contact.TABLENAME, null, Contact.UUID + "=?", args, null, null, null);
308 if (cursor.getCount() == 0) {
309 return null;
310 }
311 cursor.moveToFirst();
312 return Contact.fromCursor(cursor);
313 }
314
315 public Conversation findConversationByUuid(String conversationUuid) {
316 SQLiteDatabase db = this.getReadableDatabase();
317 String[] selectionArgs = { conversationUuid };
318 Cursor cursor = db.query(Conversation.TABLENAME, null, Conversation.UUID + "=?", selectionArgs, null, null, null);
319 if (cursor.getCount() == 0) {
320 return null;
321 }
322 cursor.moveToFirst();
323 return Conversation.fromCursor(cursor);
324 }
325
326 public Message findMessageByUuid(String messageUuid) {
327 SQLiteDatabase db = this.getReadableDatabase();
328 String[] selectionArgs = { messageUuid };
329 Cursor cursor = db.query(Message.TABLENAME, null, Message.UUID + "=?", selectionArgs, null, null, null);
330 if (cursor.getCount() == 0) {
331 return null;
332 }
333 cursor.moveToFirst();
334 return Message.fromCursor(cursor);
335 }
336
337 public Account findAccountByUuid(String accountUuid) {
338 SQLiteDatabase db = this.getReadableDatabase();
339 String[] selectionArgs = { accountUuid };
340 Cursor cursor = db.query(Account.TABLENAME, null, Account.UUID + "=?", selectionArgs, null, null, null);
341 if (cursor.getCount() == 0) {
342 return null;
343 }
344 cursor.moveToFirst();
345 return Account.fromCursor(cursor);
346 }
347}