/*
 * Sylpheed -- a GTK+ based, lightweight, and fast e-mail client
 * Copyright (C) 1999-2005 Hiroyuki Yamamoto
 *
 * This program 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 2 of the License, or
 * (at your option) any later version.
 *
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */

#ifdef HAVE_CONFIG_H
#  include "config.h"
#endif

#include "defs.h"

#include <glib.h>
#include <glib/gi18n.h>
#include <gdk/gdkkeysyms.h>
#include <gtk/gtkmain.h>
#include <gtk/gtkwidget.h>
#include <gtk/gtkdialog.h>
#include <gtk/gtkwindow.h>
#include <gtk/gtksignal.h>
#include <gtk/gtkvbox.h>
#include <gtk/gtkhbox.h>
#include <gtk/gtklabel.h>
#include <gtk/gtkentry.h>
#include <gtk/gtktreestore.h>
#include <gtk/gtktreeview.h>
#include <gtk/gtktreeselection.h>
#include <gtk/gtkcellrenderertoggle.h>
#include <gtk/gtkcellrenderertext.h>
#include <gtk/gtkscrolledwindow.h>
#include <gtk/gtkbutton.h>
#include <gtk/gtkhbbox.h>
#include <gtk/gtkstock.h>
#include <string.h>

#include "subscribedialog.h"
#include "manage_window.h"
#include "mainwindow.h"
#include "gtkutils.h"
#include "utils.h"
#include "news.h"
#include "folder.h"
#include "alertpanel.h"
#include "statusbar.h"
#include "recv.h"
#include "socket.h"

enum {
	SUBSCRIBE_TOGGLE,
	SUBSCRIBE_NAME,
	SUBSCRIBE_NUM,
	SUBSCRIBE_TYPE,
	SUBSCRIBE_INFO,
	SUBSCRIBE_CAN_TOGGLE,
	N_SUBSCRIBE_COLUMNS
};

#define SUBSCRIBE_DIALOG_WIDTH		450
#define SUBSCRIBE_DIALOG_HEIGHT		400
#define SUBSCRIBE_COL_NAME_WIDTH	250

static gboolean ack;
static gboolean locked;

static GtkWidget *dialog;
static GtkWidget *entry;
static GtkWidget *treeview;
static GtkWidget *status_label;
static GtkWidget *ok_button;

static GtkTreeStore *tree_store;

static GSList *group_list;
static GSList *subscribe_list;
static Folder *news_folder;

static void subscribe_dialog_create	(void);
static void subscribe_dialog_set_list	(const gchar	*pattern,
					 gboolean	 refresh);
static void subscribe_search		(void);
static void subscribe_clear		(void);

static gboolean subscribe_recv_func	(SockInfo	*sock,
					 gint		 count,
					 gint		 read_bytes,
					 gpointer	 data);

static gint window_deleted	(GtkWidget	*widget,
				 GdkEventAny	*event,
				 gpointer	 data);
static void ok_clicked		(GtkWidget	*widget,
				 gpointer	 data);
static void cancel_clicked	(GtkWidget	*widget,
				 gpointer	 data);
static void refresh_clicked	(GtkWidget	*widget,
				 gpointer	 data);
static gboolean key_pressed	(GtkWidget	*widget,
				 GdkEventKey	*event,
				 gpointer	 data);

static gboolean subscribe_selected	(GtkTreeSelection	*selection,
					 GtkTreeModel		*model,
					 GtkTreePath		*path,
					 gboolean		 cur_selected,
					 gpointer		 data);

static void subscribe_toggled	(GtkCellRenderer	*cell,
				 gchar			*path,
				 gpointer		 data);

static void entry_activated	(GtkEditable	*editable);
static void search_clicked	(GtkWidget	*widget,
				 gpointer	 data);

GSList *subscribe_dialog(Folder *folder)
{
	GNode *node;
	FolderItem *item;

	if (dialog && GTK_WIDGET_VISIBLE(dialog)) return NULL;

	if (!dialog)
		subscribe_dialog_create();

	news_folder = folder;

	gtk_widget_show(dialog);
	gtk_window_set_modal(GTK_WINDOW(dialog), TRUE);
	manage_window_set_transient(GTK_WINDOW(dialog));
	gtk_widget_grab_focus(ok_button);
	gtk_widget_grab_focus(treeview);
	GTK_EVENTS_FLUSH();

	subscribe_list = NULL;
	for (node = folder->node->children; node != NULL; node = node->next) {
		item = FOLDER_ITEM(node->data);
		subscribe_list = g_slist_append(subscribe_list,
						g_strdup(item->path));
	}

	subscribe_dialog_set_list(NULL, TRUE);

	if (ack) gtk_main();

	manage_window_focus_out(dialog, NULL, NULL);
	gtk_widget_hide(dialog);
	main_window_popup(main_window_get());

	if (ack) {
		GSList *cur;

		slist_free_strings(subscribe_list);
		subscribe_list = NULL;
		for (cur = group_list; cur != NULL; cur = cur->next) {
			NewsGroupInfo *ginfo = (NewsGroupInfo *)cur->data;

			if (ginfo->subscribed)
				subscribe_list = g_slist_append
					(subscribe_list, g_strdup(ginfo->name));
		}
	}

	subscribe_clear();

	return subscribe_list;
}

static void subscribe_dialog_create(void)
{
	GtkWidget *vbox;
	GtkWidget *hbox;
	GtkWidget *msg_label;
	GtkWidget *search_button;
	GtkWidget *confirm_area;
	GtkWidget *cancel_button;	
	GtkWidget *refresh_button;	
	GtkWidget *scrolledwin;
	GtkTreeViewColumn *column;
	GtkCellRenderer *renderer;
	GtkTreeSelection *selection;

	dialog = gtk_dialog_new();
	gtk_window_set_policy(GTK_WINDOW(dialog), FALSE, TRUE, FALSE);
	gtk_widget_set_size_request(dialog,
				    SUBSCRIBE_DIALOG_WIDTH,
				    SUBSCRIBE_DIALOG_HEIGHT);
	gtk_container_set_border_width
		(GTK_CONTAINER(GTK_DIALOG(dialog)->action_area), 5);
	gtk_window_set_position(GTK_WINDOW(dialog), GTK_WIN_POS_CENTER);
	gtk_window_set_title(GTK_WINDOW(dialog), _("Subscribe to newsgroup"));
	g_signal_connect(G_OBJECT(dialog), "delete_event",
			 G_CALLBACK(window_deleted), NULL);
	g_signal_connect(G_OBJECT(dialog), "key_press_event",
			 G_CALLBACK(key_pressed), NULL);
	MANAGE_WINDOW_SIGNALS_CONNECT(dialog);

	gtk_widget_realize(dialog);

	vbox = gtk_vbox_new(FALSE, 8);
	gtk_container_add(GTK_CONTAINER(GTK_DIALOG(dialog)->vbox), vbox);
	gtk_container_set_border_width(GTK_CONTAINER(vbox), 8);

	hbox = gtk_hbox_new(FALSE, 0);
	gtk_box_pack_start(GTK_BOX(vbox), hbox, FALSE, FALSE, 0);

	msg_label = gtk_label_new(_("Select newsgroups to subscribe."));
	gtk_box_pack_start(GTK_BOX(hbox), msg_label, FALSE, FALSE, 0);

	hbox = gtk_hbox_new(FALSE, 8);
	gtk_box_pack_start(GTK_BOX(vbox), hbox, FALSE, FALSE, 0);

	msg_label = gtk_label_new(_("Find groups:"));
	gtk_box_pack_start(GTK_BOX(hbox), msg_label, FALSE, FALSE, 0);

	entry = gtk_entry_new();
	gtk_box_pack_start(GTK_BOX(hbox), entry, TRUE, TRUE, 0);
	g_signal_connect(G_OBJECT(entry), "activate",
			 G_CALLBACK(entry_activated), NULL);

	search_button = gtk_button_new_with_label(_(" Search "));
	gtk_box_pack_start(GTK_BOX(hbox), search_button, FALSE, FALSE, 0);

	g_signal_connect(G_OBJECT(search_button), "clicked",
			 G_CALLBACK(search_clicked), NULL);

	scrolledwin = gtk_scrolled_window_new(NULL, NULL);
	gtk_box_pack_start(GTK_BOX (vbox), scrolledwin, TRUE, TRUE, 0);
	gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(scrolledwin),
				       GTK_POLICY_AUTOMATIC,
				       GTK_POLICY_AUTOMATIC);
	gtk_scrolled_window_set_shadow_type(GTK_SCROLLED_WINDOW(scrolledwin),
					    GTK_SHADOW_IN);

	tree_store = gtk_tree_store_new(N_SUBSCRIBE_COLUMNS,
					G_TYPE_BOOLEAN,
					G_TYPE_STRING,
					G_TYPE_STRING,
					G_TYPE_STRING,
					G_TYPE_POINTER,
					G_TYPE_BOOLEAN);
	gtk_tree_sortable_set_sort_column_id(GTK_TREE_SORTABLE(tree_store),
					     SUBSCRIBE_NAME,
					     GTK_SORT_ASCENDING);

	treeview = gtk_tree_view_new_with_model(GTK_TREE_MODEL(tree_store));
	g_object_unref(G_OBJECT(tree_store));
	gtk_tree_view_set_rules_hint(GTK_TREE_VIEW(treeview), TRUE);
	gtk_tree_view_set_search_column(GTK_TREE_VIEW(treeview),
					SUBSCRIBE_NAME);

	selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(treeview));
	gtk_tree_selection_set_mode(selection, GTK_SELECTION_MULTIPLE);
	gtk_tree_selection_set_select_function(selection, subscribe_selected,
					       NULL, NULL);

	gtk_container_add(GTK_CONTAINER(scrolledwin), treeview);

	renderer = gtk_cell_renderer_toggle_new();
	column = gtk_tree_view_column_new_with_attributes
		(NULL, renderer, "active", SUBSCRIBE_TOGGLE,
		 "activatable", SUBSCRIBE_CAN_TOGGLE,
		 "visible", SUBSCRIBE_CAN_TOGGLE, NULL);
	gtk_tree_view_column_set_min_width(column, 20);
	gtk_tree_view_append_column(GTK_TREE_VIEW(treeview), column);
	g_signal_connect(renderer, "toggled", G_CALLBACK(subscribe_toggled),
			 NULL);

	renderer = gtk_cell_renderer_text_new();
	column = gtk_tree_view_column_new_with_attributes
		(_("Newsgroup name"), renderer, "text", SUBSCRIBE_NAME, NULL);
	gtk_tree_view_append_column(GTK_TREE_VIEW(treeview), column);
	gtk_tree_view_set_expander_column(GTK_TREE_VIEW(treeview), column);

	renderer = gtk_cell_renderer_text_new();
	column = gtk_tree_view_column_new_with_attributes
		(_("Messages"), renderer, "text", SUBSCRIBE_NUM, NULL);
	gtk_tree_view_append_column(GTK_TREE_VIEW(treeview), column);

	renderer = gtk_cell_renderer_text_new();
	column = gtk_tree_view_column_new_with_attributes
		(_("Type"), renderer, "text", SUBSCRIBE_TYPE, NULL);
	gtk_tree_view_append_column(GTK_TREE_VIEW(treeview), column);

	hbox = gtk_hbox_new(FALSE, 0);
	gtk_box_pack_start(GTK_BOX(vbox), hbox, FALSE, FALSE, 0);

	status_label = gtk_label_new("");
	gtk_box_pack_start(GTK_BOX(hbox), status_label, FALSE, FALSE, 0);

	gtkut_stock_button_set_create(&confirm_area,
				      &ok_button, GTK_STOCK_OK,
				      &cancel_button, GTK_STOCK_CANCEL,
				      &refresh_button, GTK_STOCK_REFRESH);
	gtk_container_add(GTK_CONTAINER(GTK_DIALOG(dialog)->action_area),
			  confirm_area);
	gtk_widget_grab_default(ok_button);

	g_signal_connect(G_OBJECT(ok_button), "clicked",
			 G_CALLBACK(ok_clicked), NULL);
	g_signal_connect(G_OBJECT(cancel_button), "clicked",
			 G_CALLBACK(cancel_clicked), NULL);
	g_signal_connect(G_OBJECT(refresh_button), "clicked",
			 G_CALLBACK(refresh_clicked), NULL);

	gtk_widget_show_all(GTK_DIALOG(dialog)->vbox);
}

static GHashTable *branch_node_table;

static void subscribe_hash_init(void)
{
	branch_node_table = g_hash_table_new_full
		(g_str_hash, g_str_equal, NULL,
		 (GDestroyNotify)gtk_tree_iter_free);
}

static void subscribe_hash_free(void)
{
	hash_free_strings(branch_node_table);
	g_hash_table_destroy(branch_node_table);
}

static gboolean subscribe_hash_get_branch_node(const gchar *name,
					       GtkTreeIter *iter)
{
	GtkTreeIter *iter_;

	iter_ = g_hash_table_lookup(branch_node_table, name);
	if (iter_) {
		*iter = *iter_;
		return TRUE;
	}

	return FALSE;
}

static void subscribe_hash_set_branch_node(const gchar *name,
					   GtkTreeIter *iter)
{
	GtkTreeIter *iter_;

	iter_ = gtk_tree_iter_copy(iter);
	g_hash_table_replace(branch_node_table, g_strdup(name), iter_);
}

static gchar *subscribe_get_parent_name(const gchar *name)
{
	gchar *p;

	p = strrchr(name, '.');
	if (!p)
		return g_strdup("");
	return g_strndup(name, p - name);
}

static gboolean subscribe_create_parent(const gchar *name,
					const gchar *pattern,
					GtkTreeIter *parent)
{
	GtkTreeIter parent_;
	GtkTreeIter iter;
	gchar *parent_name;
	gboolean has_parent;

	if (*name == '\0') return FALSE;
	if (subscribe_hash_get_branch_node(name, &iter)) {
		*parent = iter;
		return TRUE;
	}

	parent_name = subscribe_get_parent_name(name);
	has_parent = subscribe_create_parent(parent_name, pattern, &parent_);

	if (has_parent) {
		gtk_tree_store_append(tree_store, &iter, &parent_);
		if (g_pattern_match_simple(pattern, parent_name) == FALSE)
			gtkut_tree_view_expand_parent_all
				(GTK_TREE_VIEW(treeview), &iter);
	} else
		gtk_tree_store_append(tree_store, &iter, NULL);
	gtk_tree_store_set(tree_store, &iter, SUBSCRIBE_NAME, name, -1);

	subscribe_hash_set_branch_node(name, &iter);

	g_free(parent_name);

	*parent = iter;

	return TRUE;
}

static gboolean subscribe_create_branch(NewsGroupInfo *ginfo,
					const gchar *pattern,
					GtkTreeIter *iter)
{
	GtkTreeIter iter_;
	GtkTreeIter parent;
	const gchar *name = ginfo->name;
	gchar *parent_name;
	gint count;
	const gchar *count_str;
	const gchar *type_str;
	gboolean has_parent;

	count = ginfo->last - ginfo->first;
	if (count < 0)
		count = 0;
	count_str = itos(count);

	if (ginfo->type == 'y')
		type_str = "";
	else if (ginfo->type == 'm')
		type_str = _("moderated");
	else if (ginfo->type == 'n')
		type_str = _("readonly");
	else
		type_str = _("unknown");

	parent_name = subscribe_get_parent_name(name);
	has_parent = subscribe_create_parent(parent_name, pattern, &parent);
	if (!subscribe_hash_get_branch_node(name, &iter_)) {
		if (has_parent) {
			gtk_tree_store_append(tree_store, &iter_, &parent);
			if (g_pattern_match_simple(pattern, parent_name) == FALSE)
				gtkut_tree_view_expand_parent_all
					(GTK_TREE_VIEW(treeview), &iter_);
		} else
			gtk_tree_store_append(tree_store, &iter_, NULL);
	}

	gtk_tree_store_set(tree_store, &iter_,
			   SUBSCRIBE_NAME, name,
			   SUBSCRIBE_NUM, count_str,
			   SUBSCRIBE_TYPE, type_str,
			   SUBSCRIBE_INFO, ginfo,
			   SUBSCRIBE_CAN_TOGGLE, TRUE,
			   -1);

	g_free(parent_name);

	*iter = iter_;

	return TRUE;
}

static void subscribe_dialog_set_list(const gchar *pattern, gboolean refresh)
{
	gchar *pattern_;
	GSList *cur;
	GPatternSpec *pspec;

	if (locked) return;
	locked = TRUE;

	if (!pattern || *pattern == '\0')
		pattern_ = g_strdup("*");
	else if (strchr(pattern, '*') == NULL)
		pattern_ = g_strconcat("*", pattern, "*", NULL);
	else
		pattern_ = g_strdup(pattern);

	if (refresh) {
		ack = TRUE;
		subscribe_clear();
		if (pattern)
			gtk_entry_set_text(GTK_ENTRY(entry), pattern);
		gtk_label_set_text(GTK_LABEL(status_label),
				   _("Getting newsgroup list..."));
		GTK_EVENTS_FLUSH();
		recv_set_ui_func(subscribe_recv_func, NULL);
		group_list = news_get_group_list(news_folder);
		group_list = g_slist_reverse(group_list);
		recv_set_ui_func(NULL, NULL);
		statusbar_pop_all();
		if (group_list == NULL && ack == TRUE) {
			alertpanel_error(_("Can't retrieve newsgroup list."));
			g_free(pattern_);
			locked = FALSE;
			return;
		}
	} else {
		gtk_tree_store_clear(tree_store);
	}

	subscribe_hash_init();

	pspec = g_pattern_spec_new(pattern_);

	for (cur = group_list; cur != NULL ; cur = cur->next) {
		NewsGroupInfo *ginfo = (NewsGroupInfo *)cur->data;
		GtkTreeIter iter;

		if (!ginfo->name || !is_ascii_str(ginfo->name))
			continue;

		if (g_slist_find_custom(subscribe_list, ginfo->name,
					(GCompareFunc)g_ascii_strcasecmp)
		    != NULL)
			ginfo->subscribed = TRUE;

		if (g_pattern_match_string(pspec, ginfo->name)) {
			subscribe_create_branch(ginfo, pattern_, &iter);
			if (ginfo->subscribed)
				gtk_tree_store_set(tree_store, &iter,
						   SUBSCRIBE_TOGGLE, TRUE, -1);
		}
	}

	g_pattern_spec_free(pspec);
	subscribe_hash_free();
	g_free(pattern_);

	gtk_label_set_text(GTK_LABEL(status_label), _("Done."));

	locked = FALSE;
}

static void subscribe_search(void)
{
	gchar *str;

	if (locked) return;

	str = gtk_editable_get_chars(GTK_EDITABLE(entry), 0, -1);
	subscribe_dialog_set_list(str, FALSE);
	g_free(str);
}

static void subscribe_clear(void)
{
	gtk_tree_store_clear(tree_store);
	gtk_entry_set_text(GTK_ENTRY(entry), "");
	news_group_list_free(group_list);
	group_list = NULL;
}

static gboolean subscribe_recv_func(SockInfo *sock, gint count, gint read_bytes,
				    gpointer data)
{
	gchar buf[BUFFSIZE];

	g_snprintf(buf, sizeof(buf),
		   _("%d newsgroups received (%s read)"),
		   count, to_human_readable(read_bytes));
	gtk_label_set_text(GTK_LABEL(status_label), buf);
	GTK_EVENTS_FLUSH();
	if (ack == FALSE)
		return FALSE;
	else
		return TRUE;
}

static gint window_deleted(GtkWidget *widget, GdkEventAny *event, gpointer data)
{
	ack = FALSE;
	if (gtk_main_level() > 1)
		gtk_main_quit();

	return TRUE;
}

static void ok_clicked(GtkWidget *widget, gpointer data)
{
	ack = TRUE;
	if (gtk_main_level() > 1)
		gtk_main_quit();
}

static void cancel_clicked(GtkWidget *widget, gpointer data)
{
	ack = FALSE;
	if (gtk_main_level() > 1)
		gtk_main_quit();
}

static void refresh_clicked(GtkWidget *widget, gpointer data)
{ 
	gchar *str;

	if (locked) return;

	news_remove_group_list_cache(news_folder);

	str = gtk_editable_get_chars(GTK_EDITABLE(entry), 0, -1);
	subscribe_dialog_set_list(str, TRUE);
	g_free(str);
}

static gboolean key_pressed(GtkWidget *widget, GdkEventKey *event,
			    gpointer data)
{
	if (event && event->keyval == GDK_Escape)
		cancel_clicked(NULL, NULL);
	return FALSE;
}

static gboolean subscribe_selected(GtkTreeSelection *selection,
				   GtkTreeModel *model, GtkTreePath *path,
				   gboolean cur_selected, gpointer data)
{
	GtkTreeIter iter;
	NewsGroupInfo *ginfo;

	if (!gtk_tree_model_get_iter(GTK_TREE_MODEL(tree_store), &iter, path))
		return FALSE;

	gtk_tree_model_get(GTK_TREE_MODEL(tree_store), &iter,
			   SUBSCRIBE_INFO, &ginfo, -1);
	if (!ginfo)
		return FALSE;

	return TRUE;
}

static void subscribe_toggled(GtkCellRenderer *cell, gchar *path_str,
			      gpointer data)
{
	GtkTreePath *path;
	GtkTreeIter iter;
	gboolean enabled;
	NewsGroupInfo *ginfo;
	gboolean can_toggle;

	path = gtk_tree_path_new_from_string(path_str);
	gtk_tree_model_get_iter(GTK_TREE_MODEL(tree_store), &iter, path);
	gtk_tree_path_free(path);

	gtk_tree_model_get(GTK_TREE_MODEL(tree_store), &iter,
			   SUBSCRIBE_TOGGLE, &enabled,
			   SUBSCRIBE_INFO, &ginfo,
			   SUBSCRIBE_CAN_TOGGLE, &can_toggle,
			   -1);
	if (ginfo && can_toggle) {
		ginfo->subscribed = !enabled;
		gtk_tree_store_set(tree_store, &iter,
				   SUBSCRIBE_TOGGLE, !enabled, -1);
	}
}

static void entry_activated(GtkEditable *editable)
{
	subscribe_search();
}

static void search_clicked(GtkWidget *widget, gpointer data)
{
	subscribe_search();
}