/*
 * statusbar.c
 *
 * Copyright (C) 2012 - 2018 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 <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 "config.h"

#include <assert.h>
#include <string.h>
#include <stdlib.h>

#ifdef HAVE_NCURSESW_NCURSES_H
#include <ncursesw/ncurses.h>
#elif HAVE_NCURSES_H
#include <ncurses.h>
#endif

#include "config/theme.h"
#include "config/preferences.h"
#include "ui/ui.h"
#include "ui/statusbar.h"
#include "ui/inputwin.h"
#include "ui/screen.h"

typedef struct _status_bar_tab_t {
    char *display_name;
    gboolean highlight;
} StatusBarTab;

typedef struct _status_bar_t {
    gchar *time;
    char *message;
    GHashTable *tabs;
    int current_tab;
} StatusBar;

#define MAX_TABS 10

static GTimeZone *tz;
static StatusBar *statusbar;
static WINDOW *statusbar_win;

static void _status_bar_draw(void);

void
_destroy_tab(StatusBarTab *tab)
{
    if (tab) {
        if (tab->display_name) {
            free(tab->display_name);
        }
        free(tab);
    }
}

void
status_bar_init(void)
{
    tz = g_time_zone_new_local();

    statusbar = malloc(sizeof(StatusBar));
    statusbar->time = NULL;
    statusbar->message = NULL;
    statusbar->tabs = g_hash_table_new_full(g_direct_hash, g_direct_equal, NULL, (GDestroyNotify)_destroy_tab);
    StatusBarTab *console = malloc(sizeof(StatusBarTab));
    console->display_name = strdup("console");
    g_hash_table_insert(statusbar->tabs, GINT_TO_POINTER(1), console);
    statusbar->current_tab = 1;

    int row = screen_statusbar_row();
    int cols = getmaxx(stdscr);
    statusbar_win = newwin(1, cols, row, 0);

    _status_bar_draw();
}

void
status_bar_close(void)
{
    if (tz) {
        g_time_zone_unref(tz);
    }
    if (statusbar) {
        if (statusbar->time) {
            g_free(statusbar->time);
        }
        if (statusbar->message) {
            free(statusbar->message);
        }
        free(statusbar);
    }
}

void
status_bar_update_virtual(void)
{
    _status_bar_draw();
}

void
status_bar_resize(void)
{
    int cols = getmaxx(stdscr);
    werase(statusbar_win);
    int row = screen_statusbar_row();
    mvwin(statusbar_win, row, 0);
    wresize(statusbar_win, 1, cols);

    _status_bar_draw();
}

void
status_bar_set_all_inactive(void)
{
    g_hash_table_remove_all(statusbar->tabs);
}

void
status_bar_current(int i)
{
    if (i == 0) {
        statusbar->current_tab = 10;
    } else {
        statusbar->current_tab = i;
    }

    _status_bar_draw();
}

void
status_bar_inactive(const int win)
{
    int true_win = win;
    if (true_win == 0) {
        true_win = 10;
    }

    g_hash_table_remove(statusbar->tabs, GINT_TO_POINTER(true_win));

    _status_bar_draw();
}

void
status_bar_active(const int win, char *name)
{
    int true_win = win;
    if (true_win == 0) {
        true_win = 10;
    }

    StatusBarTab *tab = malloc(sizeof(StatusBarTab));
    tab->display_name = strdup(name);
    tab->highlight = FALSE;
    g_hash_table_replace(statusbar->tabs, GINT_TO_POINTER(true_win), tab);

    _status_bar_draw();
}

void
status_bar_new(const int win, char* name)
{
    int true_win = win;
    if (true_win == 0) {
        true_win = 10;
    }

    StatusBarTab *tab = malloc(sizeof(StatusBarTab));
    tab->display_name = strdup(name);
    tab->highlight = TRUE;
    g_hash_table_replace(statusbar->tabs, GINT_TO_POINTER(true_win), tab);

    _status_bar_draw();
}

void
status_bar_get_password(void)
{
    status_bar_print_message("Enter password:");
}

void
status_bar_print_message(const char *const msg)
{
    if (statusbar->message) {
        free(statusbar->message);
        statusbar->message = NULL;
    }
    statusbar->message = strdup(msg);

    _status_bar_draw();
}

void
status_bar_clear(void)
{
    if (statusbar->message) {
        free(statusbar->message);
        statusbar->message = NULL;
    }

    _status_bar_draw();
}

void
status_bar_clear_message(void)
{
    if (statusbar->message) {
        free(statusbar->message);
        statusbar->message = NULL;
    }

    _status_bar_draw();
}

static int
_tabs_width(void)
{
    gboolean show_empty = prefs_get_boolean(PREF_STATUSBAR_SHOW_EMPTY);
    gboolean show_name = prefs_get_boolean(PREF_STATUSBAR_SHOW_NAME);

    if (show_name) {
        if (show_empty) {
            int width = 4;
            int i = 0;
            for (i = 1; i <= MAX_TABS; i++) {
                StatusBarTab *tab = g_hash_table_lookup(statusbar->tabs, GINT_TO_POINTER(i));
                if (tab) {
                    width += strlen(tab->display_name);
                    width += 4;
                } else {
                    width += 3;
                }
            }
            return width;
        } else {
            int width = 4;
            int i = 0;
            for (i = 1; i <= MAX_TABS; i++) {
                StatusBarTab *tab = g_hash_table_lookup(statusbar->tabs, GINT_TO_POINTER(i));
                if (tab) {
                    width += strlen(tab->display_name);
                    width += 4;
                }
            }
            return width;
        }
    } else {
        if (show_empty) {
            return MAX_TABS * 3 + 4;
        } else {
            return g_hash_table_size(statusbar->tabs) * 3 + 4;
        }
    }
}

static void
_status_bar_draw(void)
{
    werase(statusbar_win);
    wbkgd(statusbar_win, theme_attrs(THEME_STATUS_TEXT));

    int pos = 1;

    char *time_pref = prefs_get_string(PREF_TIME_STATUSBAR);
    if (g_strcmp0(time_pref, "off") != 0) {
        // time
        if (statusbar->time) {
            g_free(statusbar->time);
            statusbar->time = NULL;
        }

        GDateTime *datetime = g_date_time_new_now(tz);
        statusbar->time  = g_date_time_format(datetime, time_pref);
        assert(statusbar->time != NULL);
        g_date_time_unref(datetime);

        int bracket_attrs = theme_attrs(THEME_STATUS_BRACKET);
        int time_attrs = theme_attrs(THEME_STATUS_TIME);

        size_t len = strlen(statusbar->time);
        wattron(statusbar_win, bracket_attrs);
        mvwaddch(statusbar_win, 0, pos, '[');
        pos++;
        wattroff(statusbar_win, bracket_attrs);
        wattron(statusbar_win, time_attrs);
        mvwprintw(statusbar_win, 0, pos, statusbar->time);
        pos += len;
        wattroff(statusbar_win, time_attrs);
        wattron(statusbar_win, bracket_attrs);
        mvwaddch(statusbar_win, 0, pos, ']');
        wattroff(statusbar_win, bracket_attrs);
        pos += 2;

        // message
        if (statusbar->message) {
            mvwprintw(statusbar_win, 0, pos, statusbar->message);
        }
    } else {
        // message
        if (statusbar->message) {
            mvwprintw(statusbar_win, 0, pos, statusbar->message);
        }
    }
    prefs_free_string(time_pref);

    // tabs
    int cols = getmaxx(stdscr);
    pos = cols - _tabs_width();
    int bracket_attrs = theme_attrs(THEME_STATUS_BRACKET);

    gboolean show_empty = prefs_get_boolean(PREF_STATUSBAR_SHOW_EMPTY);
    gboolean show_name = prefs_get_boolean(PREF_STATUSBAR_SHOW_NAME);

    int i = 1;
    for (i = 1; i <= MAX_TABS; i++) {
        int display_num = i == 10 ? 0 : i;

        StatusBarTab *tab = g_hash_table_lookup(statusbar->tabs, GINT_TO_POINTER(i));
        if (tab || (tab == NULL && show_empty)) {
            wattron(statusbar_win, bracket_attrs);
            if (i == statusbar->current_tab) {
                mvwprintw(statusbar_win, 0, pos, "-");
            } else {
                mvwprintw(statusbar_win, 0, pos, "[");
            }
            wattroff(statusbar_win, bracket_attrs);
            pos++;
            if (tab) {
                if (tab->highlight) {
                    int status_attrs = theme_attrs(THEME_STATUS_NEW);
                    wattron(statusbar_win, status_attrs);
                    mvwprintw(statusbar_win, 0, pos, "%d", display_num);
                    if (show_name) {
                        pos++;
                        mvwprintw(statusbar_win, 0, pos, ":");
                        pos++;
                        mvwprintw(statusbar_win, 0, pos, tab->display_name);
                        pos += strlen(tab->display_name) -1 ;
                    }
                    wattroff(statusbar_win, status_attrs);
                } else {
                    int status_attrs = theme_attrs(THEME_STATUS_ACTIVE);
                    wattron(statusbar_win, status_attrs);
                    mvwprintw(statusbar_win, 0, pos, "%d", display_num);
                    if (show_name) {
                        pos++;
                        mvwprintw(statusbar_win, 0, pos, ":");
                        pos++;
                        mvwprintw(statusbar_win, 0, pos, tab->display_name);
                        pos += strlen(tab->display_name) - 1;
                    }
                    wattroff(statusbar_win, status_attrs);
                }
            } else {
                mvwprintw(statusbar_win, 0, pos, " ");
            }
            pos++;
            wattron(statusbar_win, bracket_attrs);
            if (i == statusbar->current_tab) {
                mvwprintw(statusbar_win, 0, pos, "-");
            } else {
                mvwprintw(statusbar_win, 0, pos, "]");
            }
            pos++;
            wattroff(statusbar_win, bracket_attrs);
        }
    }

    wattron(statusbar_win, bracket_attrs);
    mvwprintw(statusbar_win, 0, pos, "[");
    wattroff(statusbar_win, bracket_attrs);
    pos++;
    mvwprintw(statusbar_win, 0, pos, " ");
    pos++;
    wattron(statusbar_win, bracket_attrs);
    mvwprintw(statusbar_win, 0, pos, "]");
    wattroff(statusbar_win, bracket_attrs);
    pos++;

    wnoutrefresh(statusbar_win);
    inp_put_back();
}