2019-12-18 04:14:46 -05:00
|
|
|
/*
|
2019-12-18 07:42:21 -05:00
|
|
|
* avatar.c
|
2019-12-18 04:14:46 -05:00
|
|
|
* vim: expandtab:ts=4:sts=4:sw=4
|
|
|
|
*
|
|
|
|
* Copyright (C) 2019 Michael Vetter <jubalh@iodoru.org>
|
|
|
|
*
|
|
|
|
* This file is part of Profanity.
|
|
|
|
*
|
|
|
|
* Profanity is free software: you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU General Public License as published by
|
|
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* Profanity is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with Profanity. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
*
|
|
|
|
* In addition, as a special exception, the copyright holders give permission to
|
|
|
|
* link the code of portions of this program with the OpenSSL library under
|
|
|
|
* certain conditions as described in each individual source file, and
|
|
|
|
* distribute linked combinations including the two.
|
|
|
|
*
|
|
|
|
* You must obey the GNU General Public License in all respects for all of the
|
|
|
|
* code used other than OpenSSL. If you modify file(s) with this exception, you
|
|
|
|
* may extend this exception to your version of the file(s), but you are not
|
|
|
|
* obligated to do so. If you do not wish to do so, delete this exception
|
|
|
|
* statement from your version. If you delete this exception statement from all
|
|
|
|
* source files in the program, then also delete it here.
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include <glib.h>
|
2019-12-18 06:46:03 -05:00
|
|
|
#include <stdio.h>
|
2019-12-18 07:29:28 -05:00
|
|
|
#include <errno.h>
|
|
|
|
#include <sys/stat.h>
|
2019-12-18 04:14:46 -05:00
|
|
|
|
|
|
|
#include "log.h"
|
|
|
|
#include "xmpp/connection.h"
|
|
|
|
#include "xmpp/form.h"
|
|
|
|
#include "xmpp/iq.h"
|
|
|
|
#include "xmpp/message.h"
|
|
|
|
#include "xmpp/stanza.h"
|
2019-12-18 05:24:50 -05:00
|
|
|
#include "ui/ui.h"
|
2019-12-18 06:46:03 -05:00
|
|
|
#include "config/files.h"
|
2019-12-18 05:24:50 -05:00
|
|
|
|
|
|
|
char *looking_for = NULL;
|
2019-12-18 04:14:46 -05:00
|
|
|
|
|
|
|
static int _avatar_metadata_nofication(xmpp_stanza_t *const stanza, void *const userdata);
|
2019-12-18 06:46:03 -05:00
|
|
|
void avatar_request_item_by_id(const char *jid, const char *id);
|
|
|
|
int avatar_request_item_handler(xmpp_stanza_t *const stanza, void *const userdata);
|
2019-12-18 04:14:46 -05:00
|
|
|
|
|
|
|
void
|
|
|
|
avatar_pep_subscribe(void)
|
|
|
|
{
|
|
|
|
message_pubsub_event_handler_add(STANZA_NS_USER_AVATAR_METADATA, _avatar_metadata_nofication, NULL, NULL);
|
|
|
|
message_pubsub_event_handler_add(STANZA_NS_USER_AVATAR_DATA, _avatar_metadata_nofication, NULL, NULL);
|
|
|
|
|
2019-12-18 05:24:50 -05:00
|
|
|
//caps_add_feature(XMPP_FEATURE_USER_AVATAR_METADATA_NOTIFY);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
avatar_get_by_nick(const char* nick)
|
|
|
|
{
|
2019-12-18 07:52:48 -05:00
|
|
|
caps_remove_feature(XMPP_FEATURE_USER_AVATAR_METADATA_NOTIFY);
|
2019-12-18 07:29:28 -05:00
|
|
|
free(looking_for);
|
2019-12-18 07:52:48 -05:00
|
|
|
|
2019-12-18 05:24:50 -05:00
|
|
|
looking_for = strdup(nick);
|
2019-12-18 07:52:48 -05:00
|
|
|
|
2019-12-18 04:14:46 -05:00
|
|
|
caps_add_feature(XMPP_FEATURE_USER_AVATAR_METADATA_NOTIFY);
|
2019-12-18 07:52:48 -05:00
|
|
|
|
2019-12-18 05:24:50 -05:00
|
|
|
return TRUE;
|
2019-12-18 04:14:46 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
|
|
|
_avatar_metadata_nofication(xmpp_stanza_t *const stanza, void *const userdata)
|
|
|
|
{
|
|
|
|
const char *from = xmpp_stanza_get_attribute(stanza, STANZA_ATTR_FROM);
|
|
|
|
|
2019-12-18 05:24:50 -05:00
|
|
|
if (!(looking_for &&
|
|
|
|
(g_strcmp0(looking_for, from) == 0))) {
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2019-12-18 06:46:03 -05:00
|
|
|
// free(looking_for);
|
|
|
|
// looking_for = NULL;
|
2019-12-18 05:24:50 -05:00
|
|
|
|
2019-12-18 04:14:46 -05:00
|
|
|
xmpp_stanza_t *root = NULL;
|
|
|
|
xmpp_stanza_t *event = xmpp_stanza_get_child_by_ns(stanza, STANZA_NS_PUBSUB_EVENT);
|
|
|
|
if (event) {
|
|
|
|
root = event;
|
|
|
|
}
|
|
|
|
|
|
|
|
xmpp_stanza_t *pubsub = xmpp_stanza_get_child_by_ns(stanza, STANZA_NS_PUBSUB);
|
|
|
|
if (pubsub) {
|
|
|
|
root = pubsub;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!root) {
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
xmpp_stanza_t *items = xmpp_stanza_get_child_by_name(root, "items");
|
|
|
|
if (!items) {
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
xmpp_stanza_t *item = xmpp_stanza_get_child_by_name(items, "item");
|
|
|
|
if (item) {
|
|
|
|
xmpp_stanza_t *metadata = xmpp_stanza_get_child_by_name(item, "metadata");
|
|
|
|
if (!metadata)
|
|
|
|
return 1;
|
|
|
|
|
|
|
|
xmpp_stanza_t *info = xmpp_stanza_get_child_by_name(metadata, "info");
|
|
|
|
const char *id = xmpp_stanza_get_id(info);
|
2019-12-18 05:24:50 -05:00
|
|
|
|
2019-12-18 08:35:47 -05:00
|
|
|
log_debug("Avatar ID for %s is: %s", from, id);
|
2019-12-18 06:46:03 -05:00
|
|
|
avatar_request_item_by_id(from, id);
|
|
|
|
}
|
|
|
|
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
avatar_request_item_by_id(const char *jid, const char *id)
|
|
|
|
{
|
2019-12-18 07:14:17 -05:00
|
|
|
caps_remove_feature(XMPP_FEATURE_USER_AVATAR_METADATA_NOTIFY);
|
|
|
|
|
2019-12-18 06:46:03 -05:00
|
|
|
xmpp_ctx_t * const ctx = connection_get_ctx();
|
|
|
|
|
|
|
|
xmpp_stanza_t *iq = stanza_create_avatar_retrieve_data_request(ctx, id, jid);
|
|
|
|
iq_id_handler_add("retrieve1", avatar_request_item_handler, free, NULL);
|
|
|
|
|
|
|
|
iq_send_stanza(iq);
|
|
|
|
|
|
|
|
xmpp_stanza_release(iq);
|
|
|
|
}
|
|
|
|
|
|
|
|
int
|
|
|
|
avatar_request_item_handler(xmpp_stanza_t *const stanza, void *const userdata)
|
|
|
|
{
|
|
|
|
const char *from_attr = xmpp_stanza_get_attribute(stanza, STANZA_ATTR_FROM);
|
|
|
|
|
|
|
|
if (!from_attr) {
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (g_strcmp0(from_attr, looking_for) != 0) {
|
|
|
|
return 1;
|
2019-12-18 04:14:46 -05:00
|
|
|
}
|
2019-12-18 06:46:03 -05:00
|
|
|
free(looking_for);
|
|
|
|
looking_for = NULL;
|
|
|
|
|
|
|
|
xmpp_stanza_t *pubsub = xmpp_stanza_get_child_by_ns(stanza, STANZA_NS_PUBSUB);
|
|
|
|
if (!pubsub) {
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
xmpp_stanza_t *items = xmpp_stanza_get_child_by_name(pubsub, "items");
|
|
|
|
if (!items) {
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
xmpp_stanza_t *item = xmpp_stanza_get_child_by_name(items, "item");
|
|
|
|
if (!item) {
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
xmpp_stanza_t *data = stanza_get_child_by_name_and_ns(item, "data", STANZA_NS_USER_AVATAR_DATA);
|
|
|
|
if (!data) {
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
char *buf = xmpp_stanza_get_text(data);
|
|
|
|
gsize size;
|
|
|
|
gchar *de = (gchar*)g_base64_decode(buf, &size);
|
|
|
|
free(buf);
|
2019-12-18 07:29:28 -05:00
|
|
|
|
2019-12-18 06:46:03 -05:00
|
|
|
char *path = files_get_data_path("");
|
|
|
|
GString *filename = g_string_new(path);
|
2019-12-18 07:29:28 -05:00
|
|
|
free(path);
|
|
|
|
|
|
|
|
g_string_append(filename, "avatars/");
|
|
|
|
|
|
|
|
errno = 0;
|
|
|
|
int res = g_mkdir_with_parents(filename->str, S_IRWXU);
|
|
|
|
if (res == -1) {
|
|
|
|
char *errmsg = strerror(errno);
|
|
|
|
if (errmsg) {
|
|
|
|
log_error("Avatar: error creating directory: %s, %s", filename->str, errmsg);
|
|
|
|
} else {
|
|
|
|
log_error("Avatar: creating directory: %s", filename->str);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-12-18 08:36:18 -05:00
|
|
|
gchar *from = str_replace(from_attr, "@", "_at_");
|
|
|
|
g_string_append(filename, from);
|
2019-12-18 06:46:03 -05:00
|
|
|
g_string_append(filename, ".png");
|
2019-12-18 08:36:18 -05:00
|
|
|
free(from);
|
2019-12-18 06:46:03 -05:00
|
|
|
|
2019-12-18 07:29:28 -05:00
|
|
|
GError *err = NULL;
|
2019-12-18 06:46:03 -05:00
|
|
|
if (g_file_set_contents (filename->str, de, size, &err) == FALSE) {
|
|
|
|
log_error("Unable to save picture: %s", err->message);
|
|
|
|
cons_show("Unable to save picture %s", err->message);
|
|
|
|
g_error_free(err);
|
2019-12-18 07:14:17 -05:00
|
|
|
} else {
|
|
|
|
cons_show("Avatar saved as %s", filename->str);
|
2019-12-18 06:46:03 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
g_string_free(filename, TRUE);
|
|
|
|
free(de);
|
2019-12-18 04:14:46 -05:00
|
|
|
|
|
|
|
return 1;
|
|
|
|
}
|