/*
 * Copyright (c) 2017 Håvard Pettersson <mail@haavard.me>
 *
 * This file is part of Tox-WeeChat.
 *
 * Tox-WeeChat 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.
 *
 * Tox-WeeChat 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 Tox-WeeChat.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <string.h>
#include <inttypes.h>

#include <weechat/weechat-plugin.h>
#include <tox/tox.h>

#ifdef TOXAV_ENABLED
    #include <tox/toxav.h>
#endif // TOXAV_ENABLED

#include "twc.h"
#include "twc-profile.h"
#include "twc-chat.h"
#include "twc-friend-request.h"
#include "twc-group-invite.h"
#include "twc-message-queue.h"
#include "twc-utils.h"

#include "twc-tox-callbacks.h"

int
twc_do_timer_cb(const void *pointer, void *data,
                int remaining_calls)
{
    /* TODO: don't strip the const */
    struct t_twc_profile *profile = (void *)pointer;

    tox_iterate(profile->tox, profile);
    struct t_hook *hook = weechat_hook_timer(tox_iteration_interval(profile->tox),
                                             0, 1, twc_do_timer_cb, profile, NULL);
    profile->tox_do_timer = hook;

    // check connection status
    TOX_CONNECTION connection = tox_self_get_connection_status(profile->tox);
    bool is_connected = connection == TOX_CONNECTION_TCP
                        || connection == TOX_CONNECTION_UDP;
    twc_profile_set_online_status(profile, is_connected);

    return WEECHAT_RC_OK;
}

void
twc_friend_message_callback(Tox *tox, uint32_t friend_number,
                            TOX_MESSAGE_TYPE type,
                            const uint8_t *message, size_t length,
                            void *data)
{
    struct t_twc_profile *profile = data;
    struct t_twc_chat *chat = twc_chat_search_friend(profile,
                                                     friend_number,
                                                     true);

    char *name = twc_get_name_nt(profile->tox, friend_number);
    char *message_nt = twc_null_terminate(message, length);

    twc_chat_print_message(chat, "notify_private", weechat_color("chat_nick_other"), name,
                           message_nt, type);

    free(name);
    free(message_nt);
}

void
twc_connection_status_callback(Tox *tox, uint32_t friend_number,
                               TOX_CONNECTION status, void *data)
{
    struct t_twc_profile *profile = data;
    char *name = twc_get_name_nt(profile->tox, friend_number);
    struct t_gui_nick *nick = NULL;
    struct t_twc_chat *chat = twc_chat_search_friend(profile,
                                                     friend_number,
                                                     false);

    // TODO: print in friend's buffer if it exists
    if (status == 0)
    {
        nick = weechat_nicklist_search_nick(profile->buffer,
                                            profile->nicklist_group, name);
        if (nick)
            weechat_nicklist_remove_nick(profile->buffer, nick);

        weechat_printf(profile->buffer,
                       "%s%s just went offline.",
                       weechat_prefix("network"),
                       name);
        if (chat)
        {
            weechat_printf(chat->buffer,
                           "%s%s just went offline.",
                           weechat_prefix("network"),
                           name);
        }
    }
    else if (status == 1)
    {
        weechat_nicklist_add_nick(profile->buffer, profile->nicklist_group,
                                  name, NULL, NULL, NULL, 1);

        weechat_printf(profile->buffer,
                       "%s%s just came online.",
                       weechat_prefix("network"),
                       name);
        if (chat)
        {
            weechat_printf(chat->buffer,
                           "%s%s just came online.",
                           weechat_prefix("network"),
                           name);
        }
        twc_message_queue_flush_friend(profile, friend_number);
    }
    free(name);
}

void
twc_name_change_callback(Tox *tox, uint32_t friend_number,
                         const uint8_t *name, size_t length,
                         void *data)
{
    struct t_twc_profile *profile = data;
    struct t_gui_nick *nick = NULL;
    struct t_twc_chat *chat = twc_chat_search_friend(profile,
                                                     friend_number,
                                                     false);

    char *old_name = twc_get_name_nt(profile->tox, friend_number);
    char *new_name = twc_null_terminate(name, length);

    if (strcmp(old_name, new_name) != 0)
    {
        if (chat)
        {
            twc_chat_queue_refresh(chat);

            weechat_printf(chat->buffer,
                           "%s%s is now known as %s",
                           weechat_prefix("network"),
                           old_name, new_name);
        }

        nick = weechat_nicklist_search_nick(profile->buffer,
                                            profile->nicklist_group, old_name);
        if (nick)
            weechat_nicklist_remove_nick(profile->buffer, nick);

        weechat_nicklist_add_nick(profile->buffer, profile->nicklist_group,
                                  new_name, NULL, NULL, NULL, 1);

        weechat_printf(profile->buffer,
                       "%s%s is now known as %s",
                       weechat_prefix("network"),
                       old_name, new_name);
    }

    free(old_name);
    free(new_name);
}

void
twc_user_status_callback(Tox *tox, uint32_t friend_number,
                         TOX_USER_STATUS status, void *data)
{
    struct t_twc_profile *profile = data;
    struct t_twc_chat *chat = twc_chat_search_friend(profile,
                                                     friend_number,
                                                     false);
    if (chat)
        twc_chat_queue_refresh(chat);
}

void
twc_status_message_callback(Tox *tox, uint32_t friend_number,
                            const uint8_t *message, size_t length,
                            void *data)
{
    struct t_twc_profile *profile = data;
    struct t_twc_chat *chat = twc_chat_search_friend(profile,
                                                     friend_number,
                                                     false);
    if (chat)
        twc_chat_queue_refresh(chat);
}

void
twc_friend_request_callback(Tox *tox, const uint8_t *public_key,
                            const uint8_t *message, size_t length,
                            void *data)
{
    struct t_twc_profile *profile = data;

    char *message_nt = twc_null_terminate(message, length);
    int rc = twc_friend_request_add(profile, public_key, message_nt);

    if (rc == -1)
    {
        weechat_printf_date_tags(profile->buffer, 0, "notify_highlight",
                       "%sReceived a friend request, but your friend request list is full!",
                       weechat_prefix("warning"));
    }
    else
    {
        char hex_address[TOX_PUBLIC_KEY_SIZE * 2 + 1];
        twc_bin2hex(public_key, TOX_PUBLIC_KEY_SIZE, hex_address);

        weechat_printf_date_tags(profile->buffer, 0, "notify_highlight",
                       "%sReceived a friend request from %s with message \"%s\"; "
                       "accept it with \"/friend accept %d\"",
                       weechat_prefix("network"),
                       hex_address, message_nt, rc);

        if (rc == -2)
        {
            weechat_printf_date_tags(profile->buffer, 0, "notify_highlight",
                           "%sFailed to save friend request, try manually "
                           "accepting with /friend add",
                           weechat_prefix("error"));
        }
    }

    free(message_nt);
}

void
twc_group_invite_callback(Tox *tox,
                          uint32_t friend_number, TOX_CONFERENCE_TYPE type,
                          const uint8_t *invite_data, size_t length,
                          void *data)
{
    TOX_ERR_CONFERENCE_JOIN err = TOX_ERR_CONFERENCE_JOIN_OK;
    struct t_twc_profile *profile = data;
    char *friend_name = twc_get_name_nt(profile->tox, friend_number);
    struct t_twc_chat *friend_chat
        = twc_chat_search_friend(profile, friend_number, false);
    int64_t rc;
    char *tags;

    char *type_str;
    switch (type)
    {
        case TOX_CONFERENCE_TYPE_TEXT:
            type_str = "a text-only group chat"; break;
        case TOX_CONFERENCE_TYPE_AV:
            type_str = "an audio/vikdeo group chat"; break;
        default:
            type_str = "a group chat"; break;
    }

    if (TWC_PROFILE_OPTION_BOOLEAN(profile, TWC_PROFILE_OPTION_AUTOJOIN))
    {
        switch (type)
        {
            case TOX_CONFERENCE_TYPE_TEXT:
                rc = tox_conference_join(tox, friend_number,
                                         invite_data, length, &err);
                break;
#ifdef TOXAV_ENABLED
            case TOX_CONFERENCE_TYPE_AV:
                rc = toxav_join_av_groupchat(tox, friend_number,
                                             invite_data, length,
                                             NULL, NULL);
                break;
#endif
            default:
                rc = -1;
                break;
        }

        if (rc >= 0 && err == TOX_ERR_CONFERENCE_JOIN_OK)
        {
            tags = "notify_private";
            if (friend_chat)
            {
                weechat_printf_date_tags(friend_chat->buffer, 0, tags,
                                         "%sWe joined the %s%s%s's invite to %s.",
                                         weechat_prefix("network"),
                                         weechat_color("chat_nick_other"), friend_name,
                                         weechat_color("reset"), type_str);
                tags = "";
            }
            weechat_printf_date_tags(profile->buffer, 0, tags,
                                     "%sWe joined the %s%s%s's invite to %s.",
                                     weechat_prefix("network"),
                                     weechat_color("chat_nick_other"), friend_name,
                                     weechat_color("reset"), type_str);
        }
        else
        {
            tags = "notify_highlight";
            if (friend_chat)
            {
                weechat_printf_date_tags(friend_chat->buffer, 0, tags,
                               "%s%s%s%s invites you to join %s, but we failed to "
                               "process the invite. Please try again.",
                               weechat_prefix("network"),
                               weechat_color("chat_nick_other"), friend_name,
                               weechat_color("reset"));
                tags = "";
            }
            weechat_printf_date_tags(profile->buffer, 0, tags,
                                     "%s%s%s%s invites you to join %s, but we failed to "
                                     "process the invite. Please try again.",
                                     weechat_prefix("network"),
                                     weechat_color("chat_nick_other"), friend_name,
                                     weechat_color("reset"));
        }
    }
    else
    {
        rc = twc_group_chat_invite_add(profile, friend_number, type,
                                       (uint8_t *)invite_data, length);

        if (rc >= 0)
        {
            tags = "notify_highlight";
            if (friend_chat)
            {
                weechat_printf_date_tags(friend_chat->buffer, 0, tags,
                               "%s%s%s%s invites you to join %s. Type "
                               "\"/group join %d\" to accept.",
                               weechat_prefix("network"),
                               weechat_color("chat_nick_other"), friend_name,
                               weechat_color("reset"), type_str, rc);
                tags = "";
            }
            weechat_printf_date_tags(profile->buffer, 0, tags,
                                     "%s%s%s%s invites you to join %s. Type "
                                     "\"/group join %d\" to accept.",
                                     weechat_prefix("network"),
                                     weechat_color("chat_nick_other"), friend_name,
                                     weechat_color("reset"), type_str, rc);
        }
        else
        {
            tags = "notify_highlight";
            if (friend_chat)
            {
                weechat_printf_date_tags(friend_chat->buffer, 0, tags,
                               "%s%s%s%s invites you to join %s, but we failed to "
                               "process the invite with error %d. Please try again.",
                               weechat_prefix("network"),
                               weechat_color("chat_nick_other"), friend_name,
                               weechat_color("reset"), rc, err);
                tags = "";
            }
            weechat_printf_date_tags(profile->buffer, 0, tags,
                                     "%s%s%s%s invites you to join %s, but we failed to "
                                     "process the invite. Please try again.",
                                     weechat_prefix("network"),
                                     weechat_color("chat_nick_other"), friend_name,
                                     weechat_color("reset"), rc);
        }
    }
    free(friend_name);
}

void
twc_handle_group_message(Tox *tox,
                         int32_t group_number, int32_t peer_number,
                         const uint8_t *message, uint16_t length,
                         void *data,
                         TOX_MESSAGE_TYPE message_type)
{
    TOX_ERR_CONFERENCE_PEER_QUERY err = TOX_ERR_CONFERENCE_PEER_QUERY_OK;
    bool rc;
    struct t_twc_profile *profile = data;

    struct t_twc_chat *chat = twc_chat_search_group(profile,
                                                    group_number,
                                                    true);

    char *myname = twc_get_self_name_nt(profile->tox);
    char *name = twc_get_peer_name_nt(profile->tox, group_number, peer_number);
    char *tags = "notify_message";
    char *message_nt = twc_null_terminate(message, length);

    const char *nick_color;
    rc = tox_conference_peer_number_is_ours(tox, group_number, peer_number, &err);
    if (rc && (err == TOX_ERR_CONFERENCE_PEER_QUERY_OK))
        nick_color = weechat_color("chat_nick_self");
    else
        nick_color = weechat_info_get("nick_color", name);

    if (weechat_string_has_highlight(message_nt, myname))
        tags = "notify_highlight";
    twc_chat_print_message(chat, tags, nick_color, name,
                           message_nt, message_type);

    free(name);
    free(myname);
    free(message_nt);
}

void
twc_group_message_callback(Tox *tox,
                           uint32_t group_number, uint32_t peer_number,
                           TOX_MESSAGE_TYPE type, const uint8_t *message,
                           size_t length, void *data)
{
    twc_handle_group_message(tox,
                             group_number,
                             peer_number,
                             message,
                             length,
                             data,
                             type);
}

void
twc_group_namelist_change_callback(Tox *tox,
                                   uint32_t group_number, uint32_t peer_number,
                                   TOX_CONFERENCE_STATE_CHANGE change_type,
                                   void *data)
{
    struct t_twc_profile *profile = data;
    struct t_twc_chat *chat = twc_chat_search_group(profile,
                                                    group_number,
                                                    true);

    struct t_gui_nick *nick = NULL;
    int i, npeers;
    TOX_ERR_CONFERENCE_PEER_QUERY err = TOX_ERR_CONFERENCE_PEER_QUERY_OK;

    struct t_weelist *new_nicks;
    struct t_weelist_item *n;

    npeers = tox_conference_peer_count(profile->tox, group_number, &err);

    if (err == TOX_ERR_CONFERENCE_PEER_QUERY_OK)
    {
        new_nicks = weechat_list_new();
        for (i = 0; i < npeers; i++)
        {
            char *name = twc_get_peer_name_nt(profile->tox, group_number, i);
            weechat_list_add(new_nicks, name, WEECHAT_LIST_POS_SORT, NULL);
            free(name);
        }
    }
    else
        return;

    // searching for exits
    n = weechat_list_get(chat->nicks, 0);

    while (n)
    {
        const char *name = weechat_list_string(n);
        if (!weechat_list_search(new_nicks, name))
        {
            weechat_printf(chat->buffer, "%s%s just left the group chat",
                           weechat_prefix("quit"), name);
            nick = weechat_nicklist_search_nick(chat->buffer,
                                                chat->nicklist_group,
                                                name);
            weechat_nicklist_remove_nick(chat->buffer, nick);
        }
        n = weechat_list_next(n);
    }

    // searching for joins
    n = weechat_list_get(new_nicks, 0);

    while (n)
    {
        const char *name = weechat_list_string(n);
        if (!weechat_list_search(chat->nicks, name))
        {
            weechat_printf(chat->buffer, "%s%s just joined the group chat",
                           weechat_prefix("join"), name);
            weechat_nicklist_add_nick(chat->buffer, chat->nicklist_group,
                                      name, NULL, NULL, NULL, 1);
        }
        n = weechat_list_next(n);
    }

    weechat_list_remove_all(chat->nicks);
    weechat_list_free(chat->nicks);
    chat->nicks = new_nicks;
}

void
twc_group_title_callback(Tox *tox,
                         uint32_t group_number, uint32_t peer_number,
                         const uint8_t *title, size_t length,
                         void *data)
{
    struct t_twc_profile *profile = data;
    struct t_twc_chat *chat = twc_chat_search_group(profile,
                                                    group_number,
                                                    true);
    twc_chat_queue_refresh(chat);

    char *name = twc_get_peer_name_nt(profile->tox, group_number, peer_number);

    char *topic = strndup((char *)title, length);
    weechat_printf(chat->buffer, "%s%s has changed the topic to \"%s\"",
                   weechat_prefix("network"), name, topic);
    free(topic);
}

#ifndef NDEBUG
void
twc_tox_log_callback(Tox *tox,
                     TOX_LOG_LEVEL level,
                     const char *file, uint32_t line, const char *func,
                     const char *message, void *user_data)
{
    struct t_twc_profile *const profile = user_data;

    char const *color;
    switch (level)
    {
        case TOX_LOG_LEVEL_TRACE:
            color = weechat_color("gray");
            break;
        case TOX_LOG_LEVEL_DEBUG:
            color = weechat_color("white");
            break;
        case TOX_LOG_LEVEL_INFO:
            color = weechat_color("lightblue");
            break;
        case TOX_LOG_LEVEL_WARNING:
            color = weechat_color("yellow");
            break;
        case TOX_LOG_LEVEL_ERROR:
            color = weechat_color("red");
            break;
        default:
            color = weechat_color("reset");
    }

    weechat_printf(profile->buffer, "%stox\t%s%s:%"PRIu32" [%s]%s %s",
            color, weechat_color("reset"), file, line, func,
            weechat_color("lightred"), message);
}
#endif /* !NDEBUG */