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