diff options
author | fiaxh <git@mx.ax.lt> | 2017-03-10 19:53:24 +0100 |
---|---|---|
committer | fiaxh <git@mx.ax.lt> | 2017-03-10 19:53:24 +0100 |
commit | 24b55d83a1f9e21b028c8d5ac9778e75565f417d (patch) | |
tree | 195b0f332e32d08e1a18fc3d8a5a6725c5ed11d7 /client/src/service/database.vala | |
parent | 4c48c5c16d5f5b010c2e5dfb1affcb59f937510c (diff) | |
parent | 29ca70a6d534e1cd79963718c793ae740318cff1 (diff) | |
download | dino-24b55d83a1f9e21b028c8d5ac9778e75565f417d.tar.gz dino-24b55d83a1f9e21b028c8d5ac9778e75565f417d.zip |
Merge branch 'master' of github.com:dino/dino
Diffstat (limited to 'client/src/service/database.vala')
-rw-r--r-- | client/src/service/database.vala | 466 |
1 files changed, 0 insertions, 466 deletions
diff --git a/client/src/service/database.vala b/client/src/service/database.vala deleted file mode 100644 index 13be6222..00000000 --- a/client/src/service/database.vala +++ /dev/null @@ -1,466 +0,0 @@ -using Gee; -using Sqlite; -using Qlite; - -using Dino.Entities; - -namespace Dino { - -public class Database : Qlite.Database { - private const int VERSION = 0; - - public class AccountTable : Table { - public Column<int> id = new Column.Integer("id") { primary_key = true, auto_increment = true }; - public Column<string> bare_jid = new Column.Text("bare_jid") { unique = true, not_null = true }; - public Column<string> resourcepart = new Column.Text("resourcepart"); - public Column<string> password = new Column.Text("password"); - public Column<string> alias = new Column.Text("alias"); - public Column<bool> enabled = new Column.BoolInt("enabled"); - - protected AccountTable(Database db) { - base(db, "account"); - init({id, bare_jid, resourcepart, password, alias, enabled}); - } - } - - public class JidTable : Table { - public Column<int> id = new Column.Integer("id") { primary_key = true, auto_increment = true }; - public Column<string> bare_jid = new Column.Text("bare_jid") { unique = true, not_null = true }; - - protected JidTable(Database db) { - base(db, "jid"); - init({id, bare_jid}); - } - } - - public class MessageTable : Table { - public Column<int> id = new Column.Integer("id") { primary_key = true, auto_increment = true }; - public Column<string> stanza_id = new Column.Text("stanza_id"); - public Column<int> account_id = new Column.Integer("account_id") { not_null = true }; - public Column<int> counterpart_id = new Column.Integer("counterpart_id") { not_null = true }; - public Column<string> counterpart_resource = new Column.Text("counterpart_resource"); - public Column<string> our_resource = new Column.Text("our_resource"); - public Column<bool> direction = new Column.BoolInt("direction") { not_null = true }; - public Column<int> type_ = new Column.Integer("type"); - public Column<long> time = new Column.Long("time"); - public Column<long> local_time = new Column.Long("local_time"); - public Column<string> body = new Column.Text("body"); - public Column<int> encryption = new Column.Integer("encryption"); - public Column<int> marked = new Column.Integer("marked"); - - protected MessageTable(Database db) { - base(db, "message"); - init({id, stanza_id, account_id, counterpart_id, our_resource, counterpart_resource, direction, - type_, time, local_time, body, encryption, marked}); - } - } - - public class RealJidTable : Table { - public Column<int> message_id = new Column.Integer("message_id") { primary_key = true }; - public Column<string> real_jid = new Column.Text("real_jid"); - - protected RealJidTable(Database db) { - base(db, "real_jid"); - init({message_id, real_jid}); - } - } - - public class UndecryptedTable : Table { - public Column<int> message_id = new Column.Integer("message_id"); - public Column<int> type_ = new Column.Integer("type"); - public Column<string> data = new Column.Text("data"); - - protected UndecryptedTable(Database db) { - base(db, "undecrypted"); - init({message_id, type_, data}); - } - } - - public class ConversationTable : Table { - public Column<int> id = new Column.Integer("id") { primary_key = true, auto_increment = true }; - public Column<int> account_id = new Column.Integer("account_id") { not_null = true }; - public Column<int> jid_id = new Column.Integer("jid_id") { not_null = true }; - public Column<bool> active = new Column.BoolInt("active"); - public Column<long> last_active = new Column.Long("last_active"); - public Column<int> type_ = new Column.Integer("type"); - public Column<int> encryption = new Column.Integer("encryption"); - public Column<int> read_up_to = new Column.Integer("read_up_to"); - - protected ConversationTable(Database db) { - base(db, "conversation"); - init({id, account_id, jid_id, active, last_active, type_, encryption, read_up_to}); - } - } - - public class AvatarTable : Table { - public Column<string> jid = new Column.Text("jid"); - public Column<string> hash = new Column.Text("hash"); - public Column<int> type_ = new Column.Integer("type"); - - protected AvatarTable(Database db) { - base(db, "avatar"); - init({jid, hash, type_}); - } - } - - public class PgpTable : Table { - public Column<string> jid = new Column.Text("jid") { primary_key = true }; - public Column<string> key = new Column.Text("key") { not_null = true }; - - protected PgpTable(Database db) { - base(db, "pgp"); - init({jid, key}); - } - } - - public class EntityFeatureTable : Table { - public Column<string> entity = new Column.Text("entity"); - public Column<string> feature = new Column.Text("feature"); - - protected EntityFeatureTable(Database db) { - base(db, "entity_feature"); - init({entity, feature}); - } - } - - public AccountTable account { get; private set; } - public JidTable jid { get; private set; } - public MessageTable message { get; private set; } - public RealJidTable real_jid { get; private set; } - public ConversationTable conversation { get; private set; } - public AvatarTable avatar { get; private set; } - public PgpTable pgp { get; private set; } - public EntityFeatureTable entity_feature { get; private set; } - - public Database(string fileName) { - base(fileName, VERSION); - account = new AccountTable(this); - jid = new JidTable(this); - message = new MessageTable(this); - real_jid = new RealJidTable(this); - conversation = new ConversationTable(this); - avatar = new AvatarTable(this); - pgp = new PgpTable(this); - entity_feature = new EntityFeatureTable(this); - init({ account, jid, message, real_jid, conversation, avatar, pgp, entity_feature }); - } - - public override void migrate(long oldVersion) { - // new table columns are added, outdated columns are still present - } - - public void add_account(Account new_account) { - new_account.id = (int) account.insert() - .value(account.bare_jid, new_account.bare_jid.to_string()) - .value(account.resourcepart, new_account.resourcepart) - .value(account.password, new_account.password) - .value(account.alias, new_account.alias) - .value(account.enabled, new_account.enabled) - .perform(); - new_account.notify.connect(on_account_update); - } - - private void on_account_update(Object o, ParamSpec sp) { - Account changed_account = (Account) o; - account.update().with(account.id, "=", changed_account.id) - .set(account.bare_jid, changed_account.bare_jid.to_string()) - .set(account.resourcepart, changed_account.resourcepart) - .set(account.password, changed_account.password) - .set(account.alias, changed_account.alias) - .set(account.enabled, changed_account.enabled) - .perform(); - } - - public void remove_account(Account to_delete) { - account.delete().with(account.bare_jid, "=", to_delete.bare_jid.to_string()).perform(); - } - - public ArrayList<Account> get_accounts() { - ArrayList<Account> ret = new ArrayList<Account>(); - foreach(Row row in account.select()) { - Account account = get_account_from_row(row); - account.notify.connect(on_account_update); - ret.add(account); - } - return ret; - } - - private Account? get_account_by_id(int id) { - Row? row = account.row_with(account.id, id); - if (row != null) { - return get_account_from_row(row); - } - return null; - } - - private Account get_account_from_row(Row row) { - Account new_account = new Account.from_bare_jid(row[account.bare_jid]); - - new_account.id = row[account.id]; - new_account.resourcepart = row[account.resourcepart]; - new_account.password = row[account.password]; - new_account.alias = row[account.alias]; - new_account.enabled = row[account.enabled]; - return new_account; - } - - public void add_message(Message new_message, Account account) { - InsertBuilder builder = message.insert() - .value(message.account_id, new_message.account.id) - .value(message.counterpart_id, get_jid_id(new_message.counterpart)) - .value(message.counterpart_resource, new_message.counterpart.resourcepart) - .value(message.our_resource, new_message.ourpart.resourcepart) - .value(message.direction, new_message.direction) - .value(message.type_, new_message.type_) - .value(message.time, (long) new_message.time.to_unix()) - .value(message.local_time, (long) new_message.local_time.to_unix()) - .value(message.body, new_message.body) - .value(message.encryption, new_message.encryption) - .value(message.marked, new_message.marked); - if (new_message.stanza_id != null) builder.value(message.stanza_id, new_message.stanza_id); - new_message.id = (int) builder.perform(); - - if (new_message.real_jid != null) { - real_jid.insert() - .value(real_jid.message_id, new_message.id) - .value(real_jid.real_jid, new_message.real_jid) - .perform(); - } - new_message.notify.connect(on_message_update); - } - - private void on_message_update(Object o, ParamSpec sp) { - Message changed_message = (Message) o; - UpdateBuilder update_builder = message.update().with(message.id, "=", changed_message.id); - switch (sp.get_name()) { - case "stanza_id": - update_builder.set(message.stanza_id, changed_message.stanza_id); break; - case "counterpart": - update_builder.set(message.counterpart_id, get_jid_id(changed_message.counterpart)); - update_builder.set(message.counterpart_resource, changed_message.counterpart.resourcepart); break; - case "ourpart": - update_builder.set(message.our_resource, changed_message.ourpart.resourcepart); break; - case "direction": - update_builder.set(message.direction, changed_message.direction); break; - case "type_": - update_builder.set(message.type_, changed_message.type_); break; - case "time": - update_builder.set(message.time, (long) changed_message.time.to_unix()); break; - case "local_time": - update_builder.set(message.local_time, (long) changed_message.local_time.to_unix()); break; - case "body": - update_builder.set(message.body, changed_message.body); break; - case "encryption": - update_builder.set(message.encryption, changed_message.encryption); break; - case "marked": - update_builder.set(message.marked, changed_message.marked); break; - } - update_builder.perform(); - - if (sp.get_name() == "real_jid") { - real_jid.insert() - .value(real_jid.message_id, changed_message.id) - .value(real_jid.real_jid, changed_message.real_jid) - .perform(); - } - } - - public Gee.List<Message> get_messages(Jid jid, Account account, int count, Message? before) { - string jid_id = get_jid_id(jid).to_string(); - - QueryBuilder select = message.select() - .with(message.counterpart_id, "=", get_jid_id(jid)) - .with(message.account_id, "=", account.id) - .order_by(message.id, "DESC") - .limit(count); - if (before != null) { - select.with(message.time, "<", (long) before.time.to_unix()); - } - - LinkedList<Message> ret = new LinkedList<Message>(); - foreach (Row row in select) { - ret.insert(0, get_message_from_row(row)); - } - return ret; - } - - public Gee.List<Message> get_unsend_messages(Account account) { - Gee.List<Message> ret = new ArrayList<Message>(); - foreach (Row row in message.select().with(message.marked, "=", (int) Message.Marked.UNSENT)) { - ret.add(get_message_from_row(row)); - } - return ret; - } - - public bool contains_message(Message query_message, Account account) { - int jid_id = get_jid_id(query_message.counterpart); - return message.select() - .with(message.account_id, "=", account.id) - .with(message.stanza_id, "=", query_message.stanza_id) - .with(message.counterpart_id, "=", jid_id) - .with(message.counterpart_resource, "=", query_message.counterpart.resourcepart) - .with(message.body, "=", query_message.body) - .with(message.time, "<", (long) query_message.time.add_minutes(1).to_unix()) - .with(message.time, ">", (long) query_message.time.add_minutes(-1).to_unix()) - .count() > 0; - } - - public bool contains_message_by_stanza_id(string stanza_id) { - return message.select() - .with(message.stanza_id, "=", stanza_id) - .count() > 0; - } - - public Message? get_message_by_id(int id) { - Row? row = message.row_with(message.id, id); - if (row != null) { - return get_message_from_row(row); - } - return null; - } - - public Message get_message_from_row(Row row) { - Message new_message = new Message(); - - new_message.id = row[message.id]; - new_message.stanza_id = row[message.stanza_id]; - string from = get_jid_by_id(row[message.counterpart_id]); - string from_resource = row[message.counterpart_resource]; - if (from_resource != null) { - new_message.counterpart = new Jid(from + "/" + from_resource); - } else { - new_message.counterpart = new Jid(from); - } - new_message.direction = row[message.direction]; - new_message.type_ = (Message.Type) row[message.type_]; - new_message.time = new DateTime.from_unix_utc(row[message.time]); - new_message.body = row[message.body]; - new_message.account = get_account_by_id(row[message.account_id]); // TODO dont have to generate acc new - new_message.marked = (Message.Marked) row[message.marked]; - new_message.encryption = (Message.Encryption) row[message.encryption]; - new_message.real_jid = get_real_jid_for_message(new_message); - - new_message.notify.connect(on_message_update); - return new_message; - } - - public string? get_real_jid_for_message(Message message) { - return real_jid.select({real_jid.real_jid}).with(real_jid.message_id, "=", message.id)[real_jid.real_jid]; - } - - public void add_conversation(Conversation new_conversation) { - var insert = conversation.insert() - .value(conversation.jid_id, get_jid_id(new_conversation.counterpart)) - .value(conversation.account_id, new_conversation.account.id) - .value(conversation.type_, new_conversation.type_) - .value(conversation.encryption, new_conversation.encryption) - //.value(conversation.read_up_to, new_conversation.read_up_to) - .value(conversation.active, new_conversation.active); - if (new_conversation.last_active != null) { - insert.value(conversation.last_active, (long) new_conversation.last_active.to_unix()); - } else { - insert.value_null(conversation.last_active); - } - new_conversation.id = (int) insert.perform(); - new_conversation.notify.connect(on_conversation_update); - } - - public ArrayList<Conversation> get_conversations(Account account) { - ArrayList<Conversation> ret = new ArrayList<Conversation>(); - foreach (Row row in conversation.select().with(conversation.account_id, "=", account.id)) { - ret.add(get_conversation_from_row(row)); - } - return ret; - } - - private void on_conversation_update(Object o, ParamSpec sp) { - Conversation changed_conversation = (Conversation) o; - var update = conversation.update().with(conversation.jid_id, "=", get_jid_id(changed_conversation.counterpart)).with(conversation.account_id, "=", changed_conversation.account.id) - .set(conversation.type_, changed_conversation.type_) - .set(conversation.encryption, changed_conversation.encryption) - //.set(conversation.read_up_to, changed_conversation.read_up_to) - .set(conversation.active, changed_conversation.active); - if (changed_conversation.last_active != null) { - update.set(conversation.last_active, (long) changed_conversation.last_active.to_unix()); - } else { - update.set_null(conversation.last_active); - } - update.perform(); - } - - private Conversation get_conversation_from_row(Row row) { - Conversation new_conversation = new Conversation(new Jid(get_jid_by_id(row[conversation.jid_id])), get_account_by_id(row[conversation.account_id])); - - new_conversation.id = row[conversation.id]; - new_conversation.active = row[conversation.active]; - int64? last_active = row[conversation.last_active]; - if (last_active != null) new_conversation.last_active = new DateTime.from_unix_utc(last_active); - new_conversation.type_ = (Conversation.Type) row[conversation.type_]; - new_conversation.encryption = (Conversation.Encryption) row[conversation.encryption]; - int? read_up_to = row[conversation.read_up_to]; - if (read_up_to != null) new_conversation.read_up_to = get_message_by_id(read_up_to); - - new_conversation.notify.connect(on_conversation_update); - return new_conversation; - } - - public void set_avatar_hash(Jid jid, string hash, int type) { - avatar.insert().or("REPLACE") - .value(avatar.jid, jid.to_string()) - .value(avatar.hash, hash) - .value(avatar.type_, type) - .perform(); - } - - public HashMap<Jid, string> get_avatar_hashes(int type) { - HashMap<Jid, string> ret = new HashMap<Jid, string>(Jid.hash_func, Jid.equals_func); - foreach (Row row in avatar.select({avatar.jid, avatar.hash}).with(avatar.type_, "=", type)) { - ret[new Jid(row[avatar.jid])] = row[avatar.hash]; - } - return ret; - } - - public void set_pgp_key(Jid jid, string key) { - pgp.insert().or("REPLACE") - .value(pgp.jid, jid.to_string()) - .value(pgp.key, key) - .perform(); - } - - public string? get_pgp_key(Jid jid) { - return pgp.select({pgp.key}).with(pgp.jid, "=", jid.to_string())[pgp.key]; - } - - public void add_entity_features(string entity, ArrayList<string> features) { - foreach (string feature in features) { - entity_feature.insert() - .value(entity_feature.entity, entity) - .value(entity_feature.feature, feature) - .perform(); - } - } - - public ArrayList<string> get_entity_features(string entity) { - ArrayList<string> ret = new ArrayList<string>(); - foreach (Row row in entity_feature.select({entity_feature.feature}).with(entity_feature.entity, "=", entity)) { - ret.add(row[entity_feature.feature]); - } - return ret; - } - - - private int get_jid_id(Jid jid_obj) { - Row? row = jid.row_with(jid.bare_jid, jid_obj.bare_jid.to_string()); - return row != null ? row[jid.id] : add_jid(jid_obj); - } - - private string? get_jid_by_id(int id) { - return jid.select({jid.bare_jid}).with(jid.id, "=", id)[jid.bare_jid]; - } - - private int add_jid(Jid jid_obj) { - return (int) jid.insert().value(jid.bare_jid, jid_obj.bare_jid.to_string()).perform(); - } -} - -}
\ No newline at end of file |