Initial commit

1 files changed, 134 insertions, 0 deletions

diff --git a/client/src/service/avatar_manager.vala b/client/src/service/avatar_manager.vala
new file mode 100644
index 00000000..de44c419
--- /dev/null
+++ b/client/src/service/avatar_manager.vala
@@ -0,0 +1,134 @@
+using Gdk;
+using Gee;
+
+using Xmpp;
+using Dino.Entities;
+
+namespace Dino {
+
+public class AvatarManager : StreamInteractionModule, Object {
+    public const string id = "avatar_manager";
+
+    public signal void received_avatar(Pixbuf avatar, Jid jid, Account account);
+
+    private enum Source {
+        USER_AVATARS,
+        VCARD
+    }
+
+    private StreamInteractor stream_interactor;
+    private Database db;
+    private HashMap<Jid, string> user_avatars = new HashMap<Jid, string>(Jid.hash_func, Jid.equals_func);
+    private HashMap<Jid, string> vcard_avatars = new HashMap<Jid, string>(Jid.hash_func, Jid.equals_func);
+    private AvatarStorage avatar_storage = new AvatarStorage("./"); // TODO ihh
+    private const int MAX_PIXEL = 192;
+
+    public static void start(StreamInteractor stream_interactor, Database db) {
+        AvatarManager m = new AvatarManager(stream_interactor, db);
+        stream_interactor.add_module(m);
+    }
+
+    private AvatarManager(StreamInteractor stream_interactor, Database db) {
+        this.stream_interactor = stream_interactor;
+        this.db = db;
+        stream_interactor.account_added.connect(on_account_added);
+    }
+
+    public Pixbuf? get_avatar(Account account, Jid jid) {
+        Jid jid_ = jid;
+        if (!MucManager.get_instance(stream_interactor).is_groupchat_occupant(jid, account)) {
+            jid_ = jid.bare_jid;
+        }
+        string? user_avatars_id = user_avatars[jid_];
+        if (user_avatars_id != null) {
+            return avatar_storage.get_image(user_avatars_id);
+        }
+        string? vcard_avatars_id = vcard_avatars[jid_];
+        if (vcard_avatars_id != null) {
+            return avatar_storage.get_image(vcard_avatars_id);
+        }
+        return null;
+    }
+
+    public void publish(Account account, string file) {
+        print(file + "\n");
+        try {
+            Pixbuf pixbuf = new Pixbuf.from_file(file);
+            if (pixbuf.width >= pixbuf.height && pixbuf.width > MAX_PIXEL) {
+                int dest_height = (int) ((float) MAX_PIXEL / pixbuf.width * pixbuf.height);
+                pixbuf = pixbuf.scale_simple(MAX_PIXEL, dest_height, InterpType.BILINEAR);
+            } else if (pixbuf.height > pixbuf.width && pixbuf.width > MAX_PIXEL) {
+                int dest_width = (int) ((float) MAX_PIXEL / pixbuf.height * pixbuf.width);
+                pixbuf = pixbuf.scale_simple(dest_width, MAX_PIXEL, InterpType.BILINEAR);
+            }
+            uint8[] buffer;
+            pixbuf.save_to_buffer(out buffer, "png");
+            Core.XmppStream stream = stream_interactor.get_stream(account);
+            if (stream != null) {
+                Xep.UserAvatars.Module.get_module(stream).publish_png(stream, buffer, pixbuf.width, pixbuf.height);
+                on_user_avatar_received(account, account.bare_jid, Base64.encode(buffer));
+            }
+        } catch (Error e) {
+            print("error " + e.message + "\n");
+        }
+    }
+
+    private class PublishResponseListenerImpl : Object {
+        public void on_success(Core.XmppStream stream) {
+
+        }
+        public void on_error(Core.XmppStream stream) { }
+    }
+
+    public static AvatarManager? get_instance(StreamInteractor stream_interaction) {
+        return (AvatarManager) stream_interaction.get_module(id);
+    }
+
+    internal string get_id() {
+        return id;
+    }
+
+    private void on_account_added(Account account) {
+        stream_interactor.module_manager.user_avatars_modules[account].received_avatar.connect((stream, jid, id) =>
+            on_user_avatar_received(account, new Jid(jid), id)
+        );
+        stream_interactor.module_manager.vcard_modules[account].received_avatar.connect((stream, jid, id) =>
+            on_vcard_avatar_received(account, new Jid(jid), id)
+        );
+
+        user_avatars = db.get_avatar_hashes(Source.USER_AVATARS);
+        foreach (Jid jid in user_avatars.keys) {
+            on_user_avatar_received(account, jid, user_avatars[jid]);
+        }
+        vcard_avatars = db.get_avatar_hashes(Source.VCARD);
+        foreach (Jid jid in vcard_avatars.keys) {
+            on_vcard_avatar_received(account, jid, vcard_avatars[jid]);
+        }
+    }
+
+    private void on_user_avatar_received(Account account, Jid jid, string id) {
+        if (!user_avatars.has_key(jid) || user_avatars[jid] != id) {
+            user_avatars[jid] = id;
+            db.set_avatar_hash(jid, id, Source.USER_AVATARS);
+        }
+        Pixbuf? avatar = avatar_storage.get_image(id);
+        if (avatar != null) {
+            received_avatar(avatar, jid, account);
+        }
+    }
+
+    private void on_vcard_avatar_received(Account account, Jid jid, string id) {
+        if (!vcard_avatars.has_key(jid) || vcard_avatars[jid] != id) {
+            vcard_avatars[jid] = id;
+            if (!jid.is_full()) { // don't save muc avatars
+                db.set_avatar_hash(jid, id, Source.VCARD);
+            }
+        }
+        Pixbuf? avatar = avatar_storage.get_image(id);
+        if (avatar != null) {
+            received_avatar(avatar, jid, account);
+        }
+    }
+}
+
+}
\ No newline at end of file