/*
 * rosterwin.c
 *
 * Copyright (C) 2012 - 2015 James Booth <boothj5@gmail.com>
 *
 * 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 <http://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 <assert.h>
#include <stdlib.h>

#include "contact.h"
#include "ui/ui.h"
#include "ui/window.h"
#include "window_list.h"
#include "config/preferences.h"
#include "roster_list.h"

static void
_rosterwin_presence(ProfLayoutSplit *layout, int indent, theme_item_t colour, const char *presence, const char *status)
{
    // don't show presence for offline contacts
    gboolean is_offline = g_strcmp0(presence, "offline") == 0;
    if (is_offline) {
        return;
    }

    char *by = prefs_get_string(PREF_ROSTER_BY);
    gboolean by_presence = g_strcmp0(by, "presence") == 0;

    // show only status when grouped by presence
    if (by_presence) {
        if (status && prefs_get_boolean(PREF_ROSTER_STATUS)) {
            wattron(layout->subwin, theme_attrs(colour));
            GString *msg = g_string_new(" ");
            while (indent > 0) {
                g_string_append(msg, " ");
                indent--;
            }
            g_string_append_printf(msg, "\"%s\"", status);
            win_printline_nowrap(layout->subwin, msg->str);
            g_string_free(msg, TRUE);
            wattroff(layout->subwin, theme_attrs(colour));
        }

    // show both presence and status when not grouped by presence
    } else {
        wattron(layout->subwin, theme_attrs(colour));
        GString *msg = g_string_new(" ");
        while (indent > 0) {
            g_string_append(msg, " ");
            indent--;
        }
        if (prefs_get_boolean(PREF_ROSTER_PRESENCE)) {
            g_string_append(msg, presence);
            if (status && prefs_get_boolean(PREF_ROSTER_STATUS)) {
                g_string_append_printf(msg, " \"%s\"", status);
            }
        } else if (status && prefs_get_boolean(PREF_ROSTER_STATUS)) {
            g_string_append_printf(msg, "\"%s\"", status);
        }
        win_printline_nowrap(layout->subwin, msg->str);
        g_string_free(msg, TRUE);
        wattroff(layout->subwin, theme_attrs(colour));
    }
}

static void
_rosterwin_resource(ProfLayoutSplit *layout, PContact contact)
{
    GList *resources = p_contact_get_available_resources(contact);
    if (resources) {
        GList *curr_resource = resources;
        while (curr_resource) {
            Resource *resource = curr_resource->data;
            const char *resource_presence = string_from_resource_presence(resource->presence);
            theme_item_t resource_presence_colour = theme_main_presence_attrs(resource_presence);

            wattron(layout->subwin, theme_attrs(resource_presence_colour));
            GString *msg = g_string_new("     ");
            g_string_append(msg, resource->name);
            if (prefs_get_boolean(PREF_ROSTER_PRIORITY)) {
                g_string_append_printf(msg, " [%d]", resource->priority);
            }
            win_printline_nowrap(layout->subwin, msg->str);
            g_string_free(msg, TRUE);
            wattroff(layout->subwin, theme_attrs(resource_presence_colour));

            if (prefs_get_boolean(PREF_ROSTER_PRESENCE) || prefs_get_boolean(PREF_ROSTER_STATUS)) {
                _rosterwin_presence(layout, 6, resource_presence_colour, resource_presence, resource->status);
            }

            curr_resource = g_list_next(curr_resource);
        }
    } else if (prefs_get_boolean(PREF_ROSTER_PRESENCE) || prefs_get_boolean(PREF_ROSTER_STATUS)) {
        const char *presence = p_contact_presence(contact);
        const char *status = p_contact_status(contact);
        theme_item_t presence_colour = theme_main_presence_attrs(presence);
        _rosterwin_presence(layout, 4, presence_colour, presence, status);
    }
    g_list_free(resources);

}

static void
_rosterwin_contact(ProfLayoutSplit *layout, PContact contact)
{
    const char *name = p_contact_name_or_jid(contact);
    const char *presence = p_contact_presence(contact);
    const char *status = p_contact_status(contact);

    theme_item_t presence_colour = theme_main_presence_attrs(presence);

    wattron(layout->subwin, theme_attrs(presence_colour));
    GString *msg = g_string_new("   ");
    char ch = prefs_get_roster_contact_char();
    if (ch) {
        g_string_append_printf(msg, "%c", ch);
    }
    g_string_append(msg, name);
    win_printline_nowrap(layout->subwin, msg->str);
    g_string_free(msg, TRUE);
    wattroff(layout->subwin, theme_attrs(presence_colour));

    if (prefs_get_boolean(PREF_ROSTER_RESOURCE)) {
        _rosterwin_resource(layout, contact);
    } else if (prefs_get_boolean(PREF_ROSTER_PRESENCE) || prefs_get_boolean(PREF_ROSTER_STATUS)) {
        _rosterwin_presence(layout, 4, presence_colour, presence, status);
    }
}

static void
_rosterwin_contacts_by_presence(ProfLayoutSplit *layout, const char *const presence, char *title)
{
    GSList *contacts = roster_get_contacts_by_presence(presence);

    // if this group has contacts, or if we want to show empty groups
    if (contacts || prefs_get_boolean(PREF_ROSTER_EMPTY)) {
        wattron(layout->subwin, theme_attrs(THEME_ROSTER_HEADER));
        GString *title_str = g_string_new(" ");
        char ch = prefs_get_roster_header_char();
        if (ch) {
            g_string_append_printf(title_str, "%c", ch);
        }
        g_string_append(title_str, title);
        if (prefs_get_boolean(PREF_ROSTER_COUNT)) {
            g_string_append_printf(title_str, " (%d)", g_slist_length(contacts));
        }
        win_printline_nowrap(layout->subwin, title_str->str);
        g_string_free(title_str, TRUE);
        wattroff(layout->subwin, theme_attrs(THEME_ROSTER_HEADER));
    }

    if (contacts) {
        GSList *curr_contact = contacts;
        while (curr_contact) {
            PContact contact = curr_contact->data;
            _rosterwin_contact(layout, contact);
            curr_contact = g_slist_next(curr_contact);
        }
    }
    g_slist_free(contacts);
}

static void
_rosterwin_contacts_by_group(ProfLayoutSplit *layout, char *group)
{
    GSList *contacts = NULL;

    char *order = prefs_get_string(PREF_ROSTER_ORDER);
    gboolean offline = prefs_get_boolean(PREF_ROSTER_OFFLINE);
    if (g_strcmp0(order, "presence") == 0) {
        contacts = roster_get_group(group, ROSTER_ORD_PRESENCE, offline);
    } else {
        contacts = roster_get_group(group, ROSTER_ORD_NAME, offline);
    }

    if (contacts || prefs_get_boolean(PREF_ROSTER_EMPTY)) {
        wattron(layout->subwin, theme_attrs(THEME_ROSTER_HEADER));
        GString *title = g_string_new(" ");
        char ch = prefs_get_roster_header_char();
        if (ch) {
            g_string_append_printf(title, "%c", ch);
        }
        g_string_append(title, group);
        if (prefs_get_boolean(PREF_ROSTER_COUNT)) {
            g_string_append_printf(title, " (%d)", g_slist_length(contacts));
        }
        win_printline_nowrap(layout->subwin, title->str);
        g_string_free(title, TRUE);
        wattroff(layout->subwin, theme_attrs(THEME_ROSTER_HEADER));

        GSList *curr_contact = contacts;
        while (curr_contact) {
            PContact contact = curr_contact->data;
            _rosterwin_contact(layout, contact);
            curr_contact = g_slist_next(curr_contact);
        }
    }
    g_slist_free(contacts);
}

static void
_rosterwin_contacts_by_no_group(ProfLayoutSplit *layout)
{
    GSList *contacts = NULL;

    char *order = prefs_get_string(PREF_ROSTER_ORDER);
    gboolean offline = prefs_get_boolean(PREF_ROSTER_OFFLINE);
    if (g_strcmp0(order, "presence") == 0) {
        contacts = roster_get_nogroup(ROSTER_ORD_PRESENCE, offline);
    } else {
        contacts = roster_get_nogroup(ROSTER_ORD_NAME, offline);
    }

    if (contacts || prefs_get_boolean(PREF_ROSTER_EMPTY)) {
        wattron(layout->subwin, theme_attrs(THEME_ROSTER_HEADER));
        GString *title = g_string_new(" ");
        char ch = prefs_get_roster_header_char();
        if (ch) {
            g_string_append_printf(title, "%c", ch);
        }
        g_string_append(title, "no group");

        if (prefs_get_boolean(PREF_ROSTER_COUNT)) {
            g_string_append_printf(title, " (%d)", g_slist_length(contacts));
        }
        win_printline_nowrap(layout->subwin, title->str);
        g_string_free(title, TRUE);
        wattroff(layout->subwin, theme_attrs(THEME_ROSTER_HEADER));

        GSList *curr_contact = contacts;
        while (curr_contact) {
            PContact contact = curr_contact->data;
            _rosterwin_contact(layout, contact);
            curr_contact = g_slist_next(curr_contact);
        }
    }
    g_slist_free(contacts);
}

void
rosterwin_roster(void)
{
    ProfWin *console = wins_get_console();
    if (console) {
        ProfLayoutSplit *layout = (ProfLayoutSplit*)console->layout;
        assert(layout->memcheck == LAYOUT_SPLIT_MEMCHECK);

       char *by = prefs_get_string(PREF_ROSTER_BY);
        if (g_strcmp0(by, "presence") == 0) {
            werase(layout->subwin);
            _rosterwin_contacts_by_presence(layout, "chat", "Available for chat");
            _rosterwin_contacts_by_presence(layout, "online", "Online");
            _rosterwin_contacts_by_presence(layout, "away", "Away");
            _rosterwin_contacts_by_presence(layout, "xa", "Extended Away");
            _rosterwin_contacts_by_presence(layout, "dnd", "Do not disturb");
            if (prefs_get_boolean(PREF_ROSTER_OFFLINE)) {
                _rosterwin_contacts_by_presence(layout, "offline", "Offline");
            }
        } else if (g_strcmp0(by, "group") == 0) {
            werase(layout->subwin);
            GSList *groups = roster_get_groups();
            GSList *curr_group = groups;
            while (curr_group) {
                _rosterwin_contacts_by_group(layout, curr_group->data);
                curr_group = g_slist_next(curr_group);
            }
            g_slist_free_full(groups, free);
            _rosterwin_contacts_by_no_group(layout);
        } else {
            GSList *contacts = NULL;

            char *order = prefs_get_string(PREF_ROSTER_ORDER);
            gboolean offline = prefs_get_boolean(PREF_ROSTER_OFFLINE);
            if (g_strcmp0(order, "presence") == 0) {
                contacts = roster_get_contacts(ROSTER_ORD_PRESENCE, offline);
            } else {
                contacts = roster_get_contacts(ROSTER_ORD_NAME, offline);
            }

            werase(layout->subwin);

            wattron(layout->subwin, theme_attrs(THEME_ROSTER_HEADER));
            GString *title = g_string_new(" ");
            char ch = prefs_get_roster_header_char();
            if (ch) {
                g_string_append_printf(title, "%c", ch);
            }
            g_string_append(title, "Roster");
            if (prefs_get_boolean(PREF_ROSTER_COUNT)) {
                g_string_append_printf(title, " (%d)", g_slist_length(contacts));
            }
            win_printline_nowrap(layout->subwin, title->str);
            g_string_free(title, TRUE);
            wattroff(layout->subwin, theme_attrs(THEME_ROSTER_HEADER));

            if (contacts) {
                GSList *curr_contact = contacts;
                while (curr_contact) {
                    PContact contact = curr_contact->data;
                    _rosterwin_contact(layout, contact);
                    curr_contact = g_slist_next(curr_contact);
                }
            }
            g_slist_free(contacts);
        }
        free(by);
    }
}