/*
 * Copyright (C) 2013 Red Hat, Inc.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library. If not, see <http://www.gnu.org/licenses/>.
 */

#include "config.h"

#include "gtklistboxaccessibleprivate.h"

#include "gtk/gtklistbox.h"

static void atk_selection_interface_init (AtkSelectionIface *iface);

G_DEFINE_TYPE_WITH_CODE (GtkListBoxAccessible, gtk_list_box_accessible, GTK_TYPE_WIDGET_ACCESSIBLE,
                         G_IMPLEMENT_INTERFACE (ATK_TYPE_SELECTION, atk_selection_interface_init))

static void
gtk_list_box_accessible_init (GtkListBoxAccessible *accessible)
{
}

static void
gtk_list_box_accessible_initialize (AtkObject *obj,
                                    gpointer   data)
{
  ATK_OBJECT_CLASS (gtk_list_box_accessible_parent_class)->initialize (obj, data);

  obj->role = ATK_ROLE_LIST_BOX;
}

static AtkStateSet*
gtk_list_box_accessible_ref_state_set (AtkObject *obj)
{
  AtkStateSet *state_set;
  GtkWidget *widget;

  state_set = ATK_OBJECT_CLASS (gtk_list_box_accessible_parent_class)->ref_state_set (obj);
  widget = gtk_accessible_get_widget (GTK_ACCESSIBLE (obj));

  if (widget != NULL)
    atk_state_set_add_state (state_set, ATK_STATE_MANAGES_DESCENDANTS);

  return state_set;
}

static void
gtk_list_box_accessible_class_init (GtkListBoxAccessibleClass *klass)
{
  AtkObjectClass *object_class = ATK_OBJECT_CLASS (klass);

  object_class->initialize = gtk_list_box_accessible_initialize;
  object_class->ref_state_set = gtk_list_box_accessible_ref_state_set;
}

static gboolean
gtk_list_box_accessible_add_selection (AtkSelection *selection,
                                       gint          idx)
{
  GtkWidget *box;
  GtkListBoxRow *row;

  box = gtk_accessible_get_widget (GTK_ACCESSIBLE (selection));
  if (box == NULL)
    return FALSE;

  row = gtk_list_box_get_row_at_index (GTK_LIST_BOX (box), idx);
  if (row)
    {
      gtk_list_box_select_row (GTK_LIST_BOX (box), row);
      return TRUE;
    }
  return FALSE;
}

static gboolean
gtk_list_box_accessible_remove_selection (AtkSelection *selection,
                                          gint          idx)
{
  GtkWidget *box;
  GtkListBoxRow *row;

  box = gtk_accessible_get_widget (GTK_ACCESSIBLE (selection));
  if (box == NULL)
    return FALSE;

  row = gtk_list_box_get_row_at_index (GTK_LIST_BOX (box), idx);
  if (row)
    {
      gtk_list_box_unselect_row (GTK_LIST_BOX (box), row);
      return TRUE;
    }
  return FALSE;
}

static gboolean
gtk_list_box_accessible_clear_selection (AtkSelection *selection)
{
  GtkWidget *box;

  box = gtk_accessible_get_widget (GTK_ACCESSIBLE (selection));
  if (box == NULL)
    return FALSE;

  gtk_list_box_unselect_all (GTK_LIST_BOX (box));
  return TRUE;
}

static gboolean
gtk_list_box_accessible_select_all (AtkSelection *selection)
{
  GtkWidget *box;

  box = gtk_accessible_get_widget (GTK_ACCESSIBLE (selection));
  if (box == NULL)
    return FALSE;

  gtk_list_box_select_all (GTK_LIST_BOX (box));
  return TRUE;
}

typedef struct
{
  gint idx;
  GtkWidget *row;
} FindSelectedData;

static void
find_selected_row (GtkListBox    *box,
                   GtkListBoxRow *row,
                   gpointer       data)
{
  FindSelectedData *d = data;

  if (d->idx == 0)
    {
      if (d->row == NULL)
        d->row = GTK_WIDGET (row);
    }
  else
    d->idx -= 1;
}

static AtkObject *
gtk_list_box_accessible_ref_selection (AtkSelection *selection,
                                       gint          idx)
{
  GtkWidget *box;
  AtkObject *accessible;
  FindSelectedData data;

  box = gtk_accessible_get_widget (GTK_ACCESSIBLE (selection));
  if (box == NULL)
    return NULL;

  data.idx = idx;
  data.row = NULL;
  gtk_list_box_selected_foreach (GTK_LIST_BOX (box), find_selected_row, &data);

  if (data.row == NULL)
    return NULL;

  accessible = gtk_widget_get_accessible (data.row);
  g_object_ref (accessible);
  return accessible;
}

static void
count_selected (GtkListBox    *box,
                GtkListBoxRow *row,
                gpointer       data)
{
  gint *count = data;
  *count += 1;
}

static gint
gtk_list_box_accessible_get_selection_count (AtkSelection *selection)
{
  GtkWidget *box;
  gint count;

  box = gtk_accessible_get_widget (GTK_ACCESSIBLE (selection));
  if (box == NULL)
    return 0;

  count = 0;
  gtk_list_box_selected_foreach (GTK_LIST_BOX (box), count_selected, &count);

  return count;
}

static gboolean
gtk_list_box_accessible_is_child_selected (AtkSelection *selection,
                                           gint          idx)
{
  GtkWidget *box;
  GtkListBoxRow *row;

  box = gtk_accessible_get_widget (GTK_ACCESSIBLE (selection));
  if (box == NULL)
    return FALSE;

  row = gtk_list_box_get_row_at_index (GTK_LIST_BOX (box), idx);

  return gtk_list_box_row_is_selected (row);
}

static void atk_selection_interface_init (AtkSelectionIface *iface)
{
  iface->add_selection = gtk_list_box_accessible_add_selection;
  iface->remove_selection = gtk_list_box_accessible_remove_selection;
  iface->clear_selection = gtk_list_box_accessible_clear_selection;
  iface->ref_selection = gtk_list_box_accessible_ref_selection;
  iface->get_selection_count = gtk_list_box_accessible_get_selection_count;
  iface->is_child_selected = gtk_list_box_accessible_is_child_selected;
  iface->select_all_selection = gtk_list_box_accessible_select_all;
}

void
_gtk_list_box_accessible_selection_changed (GtkListBox *box)
{
  AtkObject *accessible;
  accessible = gtk_widget_get_accessible (GTK_WIDGET (box));
  g_signal_emit_by_name (accessible, "selection-changed");
}

void
_gtk_list_box_accessible_update_cursor (GtkListBox *box,
                                        GtkListBoxRow *row)
{
  AtkObject *accessible;
  AtkObject *descendant;
  accessible = gtk_widget_get_accessible (GTK_WIDGET (box));
  descendant = row ? gtk_widget_get_accessible (GTK_WIDGET (row)) : NULL;
  g_signal_emit_by_name (accessible, "active-descendant-changed", descendant);
}