1package eu.siacs.conversations.persistance;
2
3import java.util.ArrayList;
4import java.util.List;
5import java.util.concurrent.CopyOnWriteArrayList;
6
7import eu.siacs.conversations.entities.Account;
8import eu.siacs.conversations.entities.Contact;
9import eu.siacs.conversations.entities.Conversation;
10import eu.siacs.conversations.entities.Message;
11import eu.siacs.conversations.entities.Roster;
12import eu.siacs.conversations.xmpp.jid.Jid;
13
14import android.content.Context;
15import android.database.Cursor;
16import android.database.sqlite.SQLiteCantOpenDatabaseException;
17import android.database.sqlite.SQLiteDatabase;
18import android.database.sqlite.SQLiteOpenHelper;
19
20public class DatabaseBackend extends SQLiteOpenHelper {
21
22 private static DatabaseBackend instance = null;
23
24 private static final String DATABASE_NAME = "history";
25 private static final int DATABASE_VERSION = 13;
26
27 private static String CREATE_CONTATCS_STATEMENT = "create table "
28 + Contact.TABLENAME + "(" + Contact.ACCOUNT + " TEXT, "
29 + Contact.SERVERNAME + " TEXT, " + Contact.SYSTEMNAME + " TEXT,"
30 + Contact.JID + " TEXT," + Contact.KEYS + " TEXT,"
31 + Contact.PHOTOURI + " TEXT," + Contact.OPTIONS + " NUMBER,"
32 + Contact.SYSTEMACCOUNT + " NUMBER, " + Contact.AVATAR + " TEXT, "
33 + Contact.LAST_PRESENCE + " TEXT, " + Contact.LAST_TIME + " NUMBER, "
34 + Contact.GROUPS + " TEXT, FOREIGN KEY(" + Contact.ACCOUNT + ") REFERENCES "
35 + Account.TABLENAME + "(" + Account.UUID
36 + ") ON DELETE CASCADE, UNIQUE(" + Contact.ACCOUNT + ", "
37 + Contact.JID + ") ON CONFLICT REPLACE);";
38
39 private DatabaseBackend(Context context) {
40 super(context, DATABASE_NAME, null, DATABASE_VERSION);
41 }
42
43 @Override
44 public void onCreate(SQLiteDatabase db) {
45 db.execSQL("PRAGMA foreign_keys=ON;");
46 db.execSQL("create table " + Account.TABLENAME + "(" + Account.UUID
47 + " TEXT PRIMARY KEY," + Account.USERNAME + " TEXT,"
48 + Account.SERVER + " TEXT," + Account.PASSWORD + " TEXT,"
49 + Account.ROSTERVERSION + " TEXT," + Account.OPTIONS
50 + " NUMBER, " + Account.AVATAR + " TEXT, " + Account.KEYS
51 + " TEXT)");
52 db.execSQL("create table " + Conversation.TABLENAME + " ("
53 + Conversation.UUID + " TEXT PRIMARY KEY, " + Conversation.NAME
54 + " TEXT, " + Conversation.CONTACT + " TEXT, "
55 + Conversation.ACCOUNT + " TEXT, " + Conversation.CONTACTJID
56 + " TEXT, " + Conversation.CREATED + " NUMBER, "
57 + Conversation.STATUS + " NUMBER, " + Conversation.MODE
58 + " NUMBER, " + Conversation.ATTRIBUTES + " TEXT, FOREIGN KEY("
59 + Conversation.ACCOUNT + ") REFERENCES " + Account.TABLENAME
60 + "(" + Account.UUID + ") ON DELETE CASCADE);");
61 db.execSQL("create table " + Message.TABLENAME + "( " + Message.UUID
62 + " TEXT PRIMARY KEY, " + Message.CONVERSATION + " TEXT, "
63 + Message.TIME_SENT + " NUMBER, " + Message.COUNTERPART
64 + " TEXT, " + Message.TRUE_COUNTERPART + " TEXT,"
65 + Message.BODY + " TEXT, " + Message.ENCRYPTION + " NUMBER, "
66 + Message.STATUS + " NUMBER," + Message.TYPE + " NUMBER, "
67 + Message.RELATIVE_FILE_PATH + " TEXT, "
68 + Message.SERVER_MSG_ID + " TEXT, "
69 + Message.REMOTE_MSG_ID + " TEXT, FOREIGN KEY("
70 + Message.CONVERSATION + ") REFERENCES "
71 + Conversation.TABLENAME + "(" + Conversation.UUID
72 + ") ON DELETE CASCADE);");
73
74 db.execSQL(CREATE_CONTATCS_STATEMENT);
75 }
76
77 @Override
78 public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {
79 if (oldVersion < 2 && newVersion >= 2) {
80 db.execSQL("update " + Account.TABLENAME + " set "
81 + Account.OPTIONS + " = " + Account.OPTIONS + " | 8");
82 }
83 if (oldVersion < 3 && newVersion >= 3) {
84 db.execSQL("ALTER TABLE " + Message.TABLENAME + " ADD COLUMN "
85 + Message.TYPE + " NUMBER");
86 }
87 if (oldVersion < 5 && newVersion >= 5) {
88 db.execSQL("DROP TABLE " + Contact.TABLENAME);
89 db.execSQL(CREATE_CONTATCS_STATEMENT);
90 db.execSQL("UPDATE " + Account.TABLENAME + " SET "
91 + Account.ROSTERVERSION + " = NULL");
92 }
93 if (oldVersion < 6 && newVersion >= 6) {
94 db.execSQL("ALTER TABLE " + Message.TABLENAME + " ADD COLUMN "
95 + Message.TRUE_COUNTERPART + " TEXT");
96 }
97 if (oldVersion < 7 && newVersion >= 7) {
98 db.execSQL("ALTER TABLE " + Message.TABLENAME + " ADD COLUMN "
99 + Message.REMOTE_MSG_ID + " TEXT");
100 db.execSQL("ALTER TABLE " + Contact.TABLENAME + " ADD COLUMN "
101 + Contact.AVATAR + " TEXT");
102 db.execSQL("ALTER TABLE " + Account.TABLENAME + " ADD COLUMN "
103 + Account.AVATAR + " TEXT");
104 }
105 if (oldVersion < 8 && newVersion >= 8) {
106 db.execSQL("ALTER TABLE " + Conversation.TABLENAME + " ADD COLUMN "
107 + Conversation.ATTRIBUTES + " TEXT");
108 }
109 if (oldVersion < 9 && newVersion >= 9) {
110 db.execSQL("ALTER TABLE " + Contact.TABLENAME + " ADD COLUMN "
111 + Contact.LAST_TIME + " NUMBER");
112 db.execSQL("ALTER TABLE " + Contact.TABLENAME + " ADD COLUMN "
113 + Contact.LAST_PRESENCE + " TEXT");
114 }
115 if (oldVersion < 10 && newVersion >= 10) {
116 db.execSQL("ALTER TABLE " + Message.TABLENAME + " ADD COLUMN "
117 + Message.RELATIVE_FILE_PATH + " TEXT");
118 }
119 if (oldVersion < 11 && newVersion >= 11) {
120 db.execSQL("ALTER TABLE " + Contact.TABLENAME + " ADD COLUMN "
121 + Contact.GROUPS + " TEXT");
122 db.execSQL("delete from "+Contact.TABLENAME);
123 db.execSQL("update "+Account.TABLENAME+" set "+Account.ROSTERVERSION+" = NULL");
124 }
125 if (oldVersion < 12 && newVersion >= 12) {
126 db.execSQL("ALTER TABLE " + Message.TABLENAME + " ADD COLUMN "
127 + Message.SERVER_MSG_ID + " TEXT");
128 }
129 if (oldVersion < 13 && newVersion >= 13) {
130 db.execSQL("delete from "+Contact.TABLENAME);
131 db.execSQL("update "+Account.TABLENAME+" set "+Account.ROSTERVERSION+" = NULL");
132 }
133 }
134
135 public static synchronized DatabaseBackend getInstance(Context context) {
136 if (instance == null) {
137 instance = new DatabaseBackend(context);
138 }
139 return instance;
140 }
141
142 public void createConversation(Conversation conversation) {
143 SQLiteDatabase db = this.getWritableDatabase();
144 db.insert(Conversation.TABLENAME, null, conversation.getContentValues());
145 }
146
147 public void createMessage(Message message) {
148 SQLiteDatabase db = this.getWritableDatabase();
149 db.insert(Message.TABLENAME, null, message.getContentValues());
150 }
151
152 public void createAccount(Account account) {
153 SQLiteDatabase db = this.getWritableDatabase();
154 db.insert(Account.TABLENAME, null, account.getContentValues());
155 }
156
157 public void createContact(Contact contact) {
158 SQLiteDatabase db = this.getWritableDatabase();
159 db.insert(Contact.TABLENAME, null, contact.getContentValues());
160 }
161
162 public int getConversationCount() {
163 SQLiteDatabase db = this.getReadableDatabase();
164 Cursor cursor = db.rawQuery("select count(uuid) as count from "
165 + Conversation.TABLENAME + " where " + Conversation.STATUS
166 + "=" + Conversation.STATUS_AVAILABLE, null);
167 cursor.moveToFirst();
168 int count = cursor.getInt(0);
169 cursor.close();
170 return count;
171 }
172
173 public CopyOnWriteArrayList<Conversation> getConversations(int status) {
174 CopyOnWriteArrayList<Conversation> list = new CopyOnWriteArrayList<>();
175 SQLiteDatabase db = this.getReadableDatabase();
176 String[] selectionArgs = { Integer.toString(status) };
177 Cursor cursor = db.rawQuery("select * from " + Conversation.TABLENAME
178 + " where " + Conversation.STATUS + " = ? order by "
179 + Conversation.CREATED + " desc", selectionArgs);
180 while (cursor.moveToNext()) {
181 list.add(Conversation.fromCursor(cursor));
182 }
183 cursor.close();
184 return list;
185 }
186
187 public ArrayList<Message> getMessages(Conversation conversations, int limit) {
188 return getMessages(conversations, limit, -1);
189 }
190
191 public ArrayList<Message> getMessages(Conversation conversation, int limit,
192 long timestamp) {
193 ArrayList<Message> list = new ArrayList<>();
194 SQLiteDatabase db = this.getReadableDatabase();
195 Cursor cursor;
196 if (timestamp == -1) {
197 String[] selectionArgs = { conversation.getUuid() };
198 cursor = db.query(Message.TABLENAME, null, Message.CONVERSATION
199 + "=?", selectionArgs, null, null, Message.TIME_SENT
200 + " DESC", String.valueOf(limit));
201 } else {
202 String[] selectionArgs = { conversation.getUuid(),
203 Long.toString(timestamp) };
204 cursor = db.query(Message.TABLENAME, null, Message.CONVERSATION
205 + "=? and " + Message.TIME_SENT + "<?", selectionArgs,
206 null, null, Message.TIME_SENT + " DESC",
207 String.valueOf(limit));
208 }
209 if (cursor.getCount() > 0) {
210 cursor.moveToLast();
211 do {
212 Message message = Message.fromCursor(cursor);
213 message.setConversation(conversation);
214 list.add(message);
215 } while (cursor.moveToPrevious());
216 }
217 cursor.close();
218 return list;
219 }
220
221 public Conversation findConversation(final Account account, final Jid contactJid) {
222 SQLiteDatabase db = this.getReadableDatabase();
223 String[] selectionArgs = { account.getUuid(),
224 contactJid.toBareJid().toString() + "/%",
225 contactJid.toBareJid().toString()
226 };
227 Cursor cursor = db.query(Conversation.TABLENAME, null,
228 Conversation.ACCOUNT + "=? AND (" + Conversation.CONTACTJID
229 + " like ? OR "+Conversation.CONTACTJID+"=?)", selectionArgs, null, null, null);
230 if (cursor.getCount() == 0)
231 return null;
232 cursor.moveToFirst();
233 Conversation conversation = Conversation.fromCursor(cursor);
234 cursor.close();
235 return conversation;
236 }
237
238 public void updateConversation(final Conversation conversation) {
239 final SQLiteDatabase db = this.getWritableDatabase();
240 final String[] args = { conversation.getUuid() };
241 db.update(Conversation.TABLENAME, conversation.getContentValues(),
242 Conversation.UUID + "=?", args);
243 }
244
245 public List<Account> getAccounts() {
246 List<Account> list = new ArrayList<>();
247 SQLiteDatabase db = this.getReadableDatabase();
248 Cursor cursor = db.query(Account.TABLENAME, null, null, null, null,
249 null, null);
250 while (cursor.moveToNext()) {
251 list.add(Account.fromCursor(cursor));
252 }
253 cursor.close();
254 return list;
255 }
256
257 public void updateAccount(Account account) {
258 SQLiteDatabase db = this.getWritableDatabase();
259 String[] args = { account.getUuid() };
260 db.update(Account.TABLENAME, account.getContentValues(), Account.UUID
261 + "=?", args);
262 }
263
264 public void deleteAccount(Account account) {
265 SQLiteDatabase db = this.getWritableDatabase();
266 String[] args = { account.getUuid() };
267 db.delete(Account.TABLENAME, Account.UUID + "=?", args);
268 }
269
270 public boolean hasEnabledAccounts() {
271 SQLiteDatabase db = this.getReadableDatabase();
272 Cursor cursor = db.rawQuery("select count(" + Account.UUID + ") from "
273 + Account.TABLENAME + " where not options & (1 <<1)", null);
274 try {
275 cursor.moveToFirst();
276 int count = cursor.getInt(0);
277 cursor.close();
278 return (count > 0);
279 } catch (SQLiteCantOpenDatabaseException e) {
280 return true; // better safe than sorry
281 } catch (RuntimeException e) {
282 return true; // better safe than sorry
283 }
284 }
285
286 @Override
287 public SQLiteDatabase getWritableDatabase() {
288 SQLiteDatabase db = super.getWritableDatabase();
289 db.execSQL("PRAGMA foreign_keys=ON;");
290 return db;
291 }
292
293 public void updateMessage(Message message) {
294 SQLiteDatabase db = this.getWritableDatabase();
295 String[] args = { message.getUuid() };
296 db.update(Message.TABLENAME, message.getContentValues(), Message.UUID
297 + "=?", args);
298 }
299
300 public void readRoster(Roster roster) {
301 SQLiteDatabase db = this.getReadableDatabase();
302 Cursor cursor;
303 String args[] = { roster.getAccount().getUuid() };
304 cursor = db.query(Contact.TABLENAME, null, Contact.ACCOUNT + "=?",
305 args, null, null, null);
306 while (cursor.moveToNext()) {
307 roster.initContact(Contact.fromCursor(cursor));
308 }
309 cursor.close();
310 }
311
312 public void writeRoster(final Roster roster) {
313 final Account account = roster.getAccount();
314 final SQLiteDatabase db = this.getWritableDatabase();
315 for (Contact contact : roster.getContacts()) {
316 if (contact.getOption(Contact.Options.IN_ROSTER)) {
317 db.insert(Contact.TABLENAME, null, contact.getContentValues());
318 } else {
319 String where = Contact.ACCOUNT + "=? AND " + Contact.JID + "=?";
320 String[] whereArgs = { account.getUuid(), contact.getJid().toString() };
321 db.delete(Contact.TABLENAME, where, whereArgs);
322 }
323 }
324 account.setRosterVersion(roster.getVersion());
325 updateAccount(account);
326 }
327
328 public void deleteMessage(Message message) {
329 SQLiteDatabase db = this.getWritableDatabase();
330 String[] args = { message.getUuid() };
331 db.delete(Message.TABLENAME, Message.UUID + "=?", args);
332 }
333
334 public void deleteMessagesInConversation(Conversation conversation) {
335 SQLiteDatabase db = this.getWritableDatabase();
336 String[] args = { conversation.getUuid() };
337 db.delete(Message.TABLENAME, Message.CONVERSATION + "=?", args);
338 }
339
340 public Conversation findConversationByUuid(String conversationUuid) {
341 SQLiteDatabase db = this.getReadableDatabase();
342 String[] selectionArgs = { conversationUuid };
343 Cursor cursor = db.query(Conversation.TABLENAME, null,
344 Conversation.UUID + "=?", selectionArgs, null, null, null);
345 if (cursor.getCount() == 0) {
346 return null;
347 }
348 cursor.moveToFirst();
349 Conversation conversation = Conversation.fromCursor(cursor);
350 cursor.close();
351 return conversation;
352 }
353
354 public Message findMessageByUuid(String messageUuid) {
355 SQLiteDatabase db = this.getReadableDatabase();
356 String[] selectionArgs = { messageUuid };
357 Cursor cursor = db.query(Message.TABLENAME, null, Message.UUID + "=?",
358 selectionArgs, null, null, null);
359 if (cursor.getCount() == 0) {
360 return null;
361 }
362 cursor.moveToFirst();
363 Message message = Message.fromCursor(cursor);
364 cursor.close();
365 return message;
366 }
367
368 public Account findAccountByUuid(String accountUuid) {
369 SQLiteDatabase db = this.getReadableDatabase();
370 String[] selectionArgs = { accountUuid };
371 Cursor cursor = db.query(Account.TABLENAME, null, Account.UUID + "=?",
372 selectionArgs, null, null, null);
373 if (cursor.getCount() == 0) {
374 return null;
375 }
376 cursor.moveToFirst();
377 Account account = Account.fromCursor(cursor);
378 cursor.close();
379 return account;
380 }
381
382 public List<Message> getImageMessages(Conversation conversation) {
383 ArrayList<Message> list = new ArrayList<>();
384 SQLiteDatabase db = this.getReadableDatabase();
385 Cursor cursor;
386 String[] selectionArgs = { conversation.getUuid(), String.valueOf(Message.TYPE_IMAGE) };
387 cursor = db.query(Message.TABLENAME, null, Message.CONVERSATION
388 + "=? AND "+Message.TYPE+"=?", selectionArgs, null, null,null);
389 if (cursor.getCount() > 0) {
390 cursor.moveToLast();
391 do {
392 Message message = Message.fromCursor(cursor);
393 message.setConversation(conversation);
394 list.add(message);
395 } while (cursor.moveToPrevious());
396 }
397 cursor.close();
398 return list;
399 }
400}