247 lines
10 KiB
Vala
247 lines
10 KiB
Vala
|
using Gee;
|
||
|
|
|
||
|
|
using Dino.Entities;
|
||
|
|
using Qlite;
|
||
|
|
using Xmpp;
|
||
|
|
|
||
|
|
namespace Dino {
|
||
|
|
|
||
|
|
public class ContentItemStore : StreamInteractionModule, Object {
|
||
|
|
public static ModuleIdentity<ContentItemStore> IDENTITY = new ModuleIdentity<ContentItemStore>("content_item_store");
|
||
|
|
public string id { get { return IDENTITY.id; } }
|
||
|
|
|
||
|
|
public signal void new_item(ContentItem item, Conversation conversation);
|
||
|
|
|
||
|
|
private StreamInteractor stream_interactor;
|
||
|
|
private Database db;
|
||
|
|
private Gee.List<ContentFilter> filters = new ArrayList<ContentFilter>();
|
||
|
|
private HashMap<Conversation, ContentItemCollection> collection_conversations = new HashMap<Conversation, ContentItemCollection>(Conversation.hash_func, Conversation.equals_func);
|
||
|
|
|
||
|
|
public static void start(StreamInteractor stream_interactor, Database db) {
|
||
|
|
ContentItemStore m = new ContentItemStore(stream_interactor, db);
|
||
|
|
stream_interactor.add_module(m);
|
||
|
|
}
|
||
|
|
|
||
|
|
public ContentItemStore(StreamInteractor stream_interactor, Database db) {
|
||
|
|
this.stream_interactor = stream_interactor;
|
||
|
|
this.db = db;
|
||
|
|
|
||
|
|
stream_interactor.get_module(MessageProcessor.IDENTITY).message_received.connect(on_new_message);
|
||
|
|
stream_interactor.get_module(MessageProcessor.IDENTITY).message_sent.connect(on_new_message);
|
||
|
|
stream_interactor.get_module(FileManager.IDENTITY).received_file.connect(insert_file_transfer);
|
||
|
|
}
|
||
|
|
|
||
|
|
public void init(Conversation conversation, ContentItemCollection item_collection) {
|
||
|
|
collection_conversations[conversation] = item_collection;
|
||
|
|
}
|
||
|
|
|
||
|
|
public void uninit(Conversation conversation, ContentItemCollection item_collection) {
|
||
|
|
collection_conversations.unset(conversation);
|
||
|
|
}
|
||
|
|
|
||
|
|
public Gee.List<ContentItem> get_items_from_query(QueryBuilder select, Conversation conversation) {
|
||
|
|
Gee.TreeSet<ContentItem> items = new Gee.TreeSet<ContentItem>(ContentItem.compare);
|
||
|
|
|
||
|
|
foreach (var row in select) {
|
||
|
|
int provider = row[db.content.content_type];
|
||
|
|
int foreign_id = row[db.content.foreign_id];
|
||
|
|
switch (provider) {
|
||
|
|
case 1:
|
||
|
|
RowOption row_option = db.message.select().with(db.message.id, "=", foreign_id).row();
|
||
|
|
if (row_option.is_present()) {
|
||
|
|
Message message = stream_interactor.get_module(MessageStorage.IDENTITY).get_message_by_id(foreign_id, conversation);
|
||
|
|
if (message == null) {
|
||
|
|
message = new Message.from_row(db, row_option.inner);
|
||
|
|
}
|
||
|
|
items.add(new MessageItem(message, conversation, row[db.content.id]));
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case 2:
|
||
|
|
RowOption row_option = db.file_transfer.select().with(db.file_transfer.id, "=", foreign_id).row();
|
||
|
|
if (row_option.is_present()) {
|
||
|
|
string storage_dir = stream_interactor.get_module(FileManager.IDENTITY).get_storage_dir();
|
||
|
|
FileTransfer file_transfer = new FileTransfer.from_row(db, row_option.inner, storage_dir);
|
||
|
|
items.add(new FileItem(file_transfer, row[db.content.id]));
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
Gee.List<ContentItem> ret = new ArrayList<ContentItem>();
|
||
|
|
foreach (ContentItem item in items) {
|
||
|
|
ret.add(item);
|
||
|
|
}
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
public Gee.List<ContentItem> get_latest(Conversation conversation, int count) {
|
||
|
|
QueryBuilder select = db.content.select()
|
||
|
|
.with(db.content.conversation_id, "=", conversation.id)
|
||
|
|
.order_by(db.content.local_time, "DESC")
|
||
|
|
.order_by(db.content.time, "DESC")
|
||
|
|
.limit(count);
|
||
|
|
|
||
|
|
return get_items_from_query(select, conversation);
|
||
|
|
}
|
||
|
|
|
||
|
|
public Gee.List<ContentItem> get_before(Conversation conversation, ContentItem item, int count) {
|
||
|
|
long local_time = (long) item.sort_time.to_unix();
|
||
|
|
long time = (long) item.display_time.to_unix();
|
||
|
|
QueryBuilder select = db.content.select()
|
||
|
|
.where(@"local_time < ? OR (local_time = ? AND time < ?) OR (local_time = ? AND time = ? AND id < ?)", { local_time.to_string(), local_time.to_string(), time.to_string(), local_time.to_string(), time.to_string(), item.id.to_string() })
|
||
|
|
.with(db.content.conversation_id, "=", conversation.id)
|
||
|
|
.order_by(db.content.local_time, "DESC")
|
||
|
|
.order_by(db.content.time, "DESC")
|
||
|
|
.limit(count);
|
||
|
|
|
||
|
|
return get_items_from_query(select, conversation);
|
||
|
|
}
|
||
|
|
|
||
|
|
public Gee.List<ContentItem> get_after(Conversation conversation, ContentItem item, int count) {
|
||
|
|
long local_time = (long) item.sort_time.to_unix();
|
||
|
|
long time = (long) item.display_time.to_unix();
|
||
|
|
QueryBuilder select = db.content.select()
|
||
|
|
.where(@"local_time > ? OR (local_time = ? AND time > ?) OR (local_time = ? AND time = ? AND id > ?)", { local_time.to_string(), local_time.to_string(), time.to_string(), local_time.to_string(), time.to_string(), item.id.to_string() })
|
||
|
|
.with(db.content.conversation_id, "=", conversation.id)
|
||
|
|
.order_by(db.content.local_time, "ASC")
|
||
|
|
.order_by(db.content.time, "ASC")
|
||
|
|
.limit(count);
|
||
|
|
|
||
|
|
return get_items_from_query(select, conversation);
|
||
|
|
}
|
||
|
|
|
||
|
|
public void add_filter(ContentFilter content_filter) {
|
||
|
|
filters.add(content_filter);
|
||
|
|
}
|
||
|
|
|
||
|
|
private void on_new_message(Message message, Conversation conversation) {
|
||
|
|
MessageItem item = new MessageItem(message, conversation, -1);
|
||
|
|
if (!discard(item)) {
|
||
|
|
item.id = db.add_content_item(conversation, message.time, message.local_time, 1, message.id);
|
||
|
|
|
||
|
|
if (collection_conversations.has_key(conversation)) {
|
||
|
|
collection_conversations.get(conversation).insert_item(item);
|
||
|
|
}
|
||
|
|
new_item(item, conversation);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private void insert_file_transfer(FileTransfer file_transfer) {
|
||
|
|
FileItem item = new FileItem(file_transfer, -1);
|
||
|
|
if (!discard(item)) {
|
||
|
|
Conversation? conversation = stream_interactor.get_module(ConversationManager.IDENTITY).get_conversation(file_transfer.counterpart, file_transfer.account);
|
||
|
|
item.id = db.add_content_item(conversation, file_transfer.time, file_transfer.local_time, 2, file_transfer.id);
|
||
|
|
|
||
|
|
if (collection_conversations.has_key(conversation)) {
|
||
|
|
collection_conversations.get(conversation).insert_item(item);
|
||
|
|
}
|
||
|
|
new_item(item, conversation);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private bool discard(ContentItem content_item) {
|
||
|
|
foreach (ContentFilter filter in filters) {
|
||
|
|
if (filter.discard(content_item)) {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public interface ContentItemCollection : Object {
|
||
|
|
public abstract void insert_item(ContentItem item);
|
||
|
|
public abstract void remove_item(ContentItem item);
|
||
|
|
}
|
||
|
|
|
||
|
|
public interface ContentFilter : Object {
|
||
|
|
public abstract bool discard(ContentItem content_item);
|
||
|
|
}
|
||
|
|
|
||
|
|
public abstract class ContentItem : Object {
|
||
|
|
public int id { get; set; }
|
||
|
|
public string type_ { get; set; }
|
||
|
|
public Jid? jid { get; set; default=null; }
|
||
|
|
public DateTime? sort_time { get; set; default=null; }
|
||
|
|
public double seccondary_sort_indicator { get; set; }
|
||
|
|
public DateTime? display_time { get; set; default=null; }
|
||
|
|
public Encryption? encryption { get; set; default=null; }
|
||
|
|
public Entities.Message.Marked? mark { get; set; default=null; }
|
||
|
|
|
||
|
|
public ContentItem(int id, string ty, Jid jid, DateTime sort_time, double seccondary_sort_indicator, DateTime display_time, Encryption encryption, Entities.Message.Marked mark) {
|
||
|
|
this.id = id;
|
||
|
|
this.type_ = ty;
|
||
|
|
this.jid = jid;
|
||
|
|
this.sort_time = sort_time;
|
||
|
|
this.seccondary_sort_indicator = seccondary_sort_indicator;
|
||
|
|
this.display_time = display_time;
|
||
|
|
this.encryption = encryption;
|
||
|
|
this.mark = mark;
|
||
|
|
}
|
||
|
|
|
||
|
|
public static int compare(ContentItem a, ContentItem b) {
|
||
|
|
int res = a.sort_time.compare(b.sort_time);
|
||
|
|
if (res == 0) {
|
||
|
|
res = a.display_time.compare(b.display_time);
|
||
|
|
}
|
||
|
|
if (res == 0) {
|
||
|
|
res = a.seccondary_sort_indicator - b.seccondary_sort_indicator > 0 ? 1 : -1;
|
||
|
|
}
|
||
|
|
return res;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public class MessageItem : ContentItem {
|
||
|
|
public const string TYPE = "message";
|
||
|
|
|
||
|
|
public Message message;
|
||
|
|
public Conversation conversation;
|
||
|
|
|
||
|
|
public MessageItem(Message message, Conversation conversation, int id) {
|
||
|
|
base(id, TYPE, message.from, message.local_time, message.id + 0.0845, message.time, message.encryption, message.marked);
|
||
|
|
this.message = message;
|
||
|
|
this.conversation = conversation;
|
||
|
|
|
||
|
|
WeakRef weak_message = WeakRef(message);
|
||
|
|
message.notify["marked"].connect(() => {
|
||
|
|
Message? m = weak_message.get() as Message;
|
||
|
|
if (m == null) return;
|
||
|
|
mark = m.marked;
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public class FileItem : ContentItem {
|
||
|
|
public const string TYPE = "file";
|
||
|
|
|
||
|
|
public FileTransfer file_transfer;
|
||
|
|
public Conversation conversation;
|
||
|
|
|
||
|
|
public FileItem(FileTransfer file_transfer, int id) {
|
||
|
|
Jid jid = file_transfer.direction == FileTransfer.DIRECTION_SENT ? file_transfer.account.bare_jid.with_resource(file_transfer.account.resourcepart) : file_transfer.counterpart;
|
||
|
|
base(id, TYPE, jid, file_transfer.local_time, file_transfer.id + 0.0845, file_transfer.time, file_transfer.encryption, file_to_message_state(file_transfer.state));
|
||
|
|
|
||
|
|
this.file_transfer = file_transfer;
|
||
|
|
|
||
|
|
file_transfer.notify["state"].connect_after(() => {
|
||
|
|
this.mark = file_to_message_state(file_transfer.state);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
private static Entities.Message.Marked file_to_message_state(FileTransfer.State state) {
|
||
|
|
switch (state) {
|
||
|
|
case FileTransfer.State.IN_PROCESS:
|
||
|
|
return Entities.Message.Marked.UNSENT;
|
||
|
|
case FileTransfer.State.COMPLETE:
|
||
|
|
return Entities.Message.Marked.NONE;
|
||
|
|
case FileTransfer.State.NOT_STARTED:
|
||
|
|
return Entities.Message.Marked.UNSENT;
|
||
|
|
case FileTransfer.State.FAILED:
|
||
|
|
return Entities.Message.Marked.WONTSEND;
|
||
|
|
}
|
||
|
|
assert_not_reached();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|