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