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