/*
 * bjb-controller.c
 * Copyright (C) Pierre-Yves Luyten 2012, 2013 <py@luyten.fr>
 * Copyright 2017 Mohammed Sadiq <sadiq@sadiqpk.org>
 *
 * bijiben 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.
 *
 * bijiben 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, see <http://www.gnu.org/licenses/>.
 */

/*
 * bjb-controler is window-wide.
 * mainly useful for BjbMainView,
 * it controls the window behaviour.
 */

#include "bjb-application.h"
#include "bjb-controller.h"
#include "bjb-main-view.h"
#include "bjb-window-base.h"

/*
 * The start-up number of items to show,
 * and its incrementation
 */

#define BJB_ITEMS_SLICE 48


struct _BjbController
{
  GObject parent_instance;

  /* needle, notebook and group define what the controller shows */
  BijiManager    *manager;
  gchar          *needle;
  BijiNotebook   *notebook;
  BijiItemsGroup  group;
  GtkTreeModel   *model;

  BjbWindowBase  *window;
  BjbSettings    *settings;

  GList          *items_to_show;
  gint            n_items_to_show;
  gboolean        remaining_items;
  GMutex          mutex;

  gboolean        selection_mode;

  gboolean        connected;
  gulong          manager_change;
};


enum {
  PROP_0,
  PROP_BOOK,
  PROP_WINDOW,
  PROP_NEEDLE,
  PROP_MODEL,
  NUM_PROPERTIES
};

static GParamSpec *properties[NUM_PROPERTIES] = { NULL, };

/* Currently used by toolbars */
enum {
  SEARCH_CHANGED,
  DISPLAY_NOTES_CHANGED, // either search or manager change
  BJB_CONTROLLER_SIGNALS
};

static guint bjb_controller_signals [BJB_CONTROLLER_SIGNALS] = { 0 };

G_DEFINE_TYPE (BjbController, bjb_controller, G_TYPE_OBJECT)

/* GObject */

static void
bjb_controller_init (BjbController *self)
{
  GtkListStore *store;

  /* Create the columns */
  store = gtk_list_store_new (BJB_MODEL_COLUMN_LAST,
                              G_TYPE_STRING,   // BJB_MODEL_COLUMN_UUID
                              G_TYPE_STRING,   // BJB_MODEL_COLUMN_TITLE
                              G_TYPE_STRING,   // BJB_MODEL_COLUMN_TEXT
                              G_TYPE_INT64,    // BJB_MODEL_COLUMN_MTIME
                              G_TYPE_STRING,   // BJB_MODEL_COLUMN_COLOR
                              G_TYPE_BOOLEAN); // BJB_MODEL_COLUMN_SELECTED

  self->model = GTK_TREE_MODEL (store);
  self->n_items_to_show = BJB_ITEMS_SLICE;
  self->group = BIJI_LIVING_ITEMS;
}

static void
free_items_store (BjbController *self)
{
  gtk_list_store_clear (GTK_LIST_STORE (self->model));
}

static void
bjb_controller_finalize (GObject *object)
{
  BjbController *self = BJB_CONTROLLER(object);

  g_object_unref (self->model);

  g_free (self->needle);
  g_list_free (self->items_to_show);

  G_OBJECT_CLASS (bjb_controller_parent_class)->finalize (object);
}

static void
bjb_controller_get_property (GObject  *object,
               guint     property_id,
               GValue   *value,
               GParamSpec *pspec)
{
  BjbController *self = BJB_CONTROLLER (object);

  switch (property_id)
  {
  case PROP_BOOK:
    g_value_set_object (value, self->manager);
    break;
  case PROP_WINDOW:
    g_value_set_object (value, self->window);
    break;
  case PROP_NEEDLE:
    g_value_set_string (value, self->needle);
    break;
  case PROP_MODEL:
    g_value_set_object (value, self->model);
    break;
  default:
    G_OBJECT_WARN_INVALID_PROPERTY_ID (object, property_id, pspec);
    break;
  }
}

static void
bjb_controller_set_property (GObject  *object,
               guint     property_id,
               const GValue *value,
               GParamSpec *pspec)
{
  BjbController *self = BJB_CONTROLLER (object);

  switch (property_id)
  {
  case PROP_BOOK:
    bjb_controller_set_manager(self,g_value_get_object(value));
    break;
  case PROP_WINDOW:
    self->window = g_value_get_object (value);
    break;
  case PROP_NEEDLE:
    self->needle = g_strdup (g_value_get_string (value));
    break;
  default:
    G_OBJECT_WARN_INVALID_PROPERTY_ID (object, property_id, pspec);
    break;
  }
}

/* get iter at this item
 * or get iter at first note if item == NULL*/
static gboolean
bjb_controller_get_iter (BjbController *self,
                         BijiItem *item,
                         GtkTreeIter **iter)
{
  gboolean retval = FALSE;
  gboolean try;
  const gchar *needle = NULL;

  if (item && BIJI_IS_ITEM (item))
      needle = biji_item_get_uuid (item);

  try = gtk_tree_model_get_iter_first (self->model, *iter);

  if (!try)
    *iter = NULL;

  while (try)
  {
    gchar *item_path;
    gtk_tree_model_get (self->model, *iter, BJB_MODEL_COLUMN_UUID, &item_path, -1);

    /* If we look for the item, check by uid */
    if (needle && g_strcmp0 (item_path, needle) == 0)
      retval = TRUE;

    /* Else we check for the first note */
    else if (!needle && BIJI_IS_NOTE_OBJ (
              biji_manager_get_item_at_path (self->manager, item_path)))
      retval = TRUE;

    g_free (item_path);

    if (retval)
      break;

    else
      try = gtk_tree_model_iter_next (self->model, *iter);
  }

  return retval;
}

/* Either append, prepend, or, if iter provider,
 * prepend just before this iter */
static void
bjb_controller_add_item (BjbController *self,
                         BijiItem      *item,
                         gboolean       prepend,
                         GtkTreeIter   *sibling)
{
  GtkTreeIter      iter;
  GtkListStore    *store;
  GdkRGBA          note_color;
  const char      *text       = _("Notebook");
  g_autofree char *color      = NULL;

  g_return_if_fail (BIJI_IS_ITEM (item));
  store = GTK_LIST_STORE (self->model);

  /* Only append notes which are not templates. Currently useless */
  if (BIJI_IS_NOTE_OBJ (item)
      && biji_note_obj_is_template (BIJI_NOTE_OBJ (item)))
    return;


  if (sibling)
    gtk_list_store_insert_before (store, &iter, sibling);

  else if (prepend)
    gtk_list_store_insert (store, &iter, 0);

  else
    gtk_list_store_append (store, &iter);

  if (BIJI_IS_NOTE_OBJ (item))
    {
      text = biji_note_obj_get_raw_text (BIJI_NOTE_OBJ (item));
      if (biji_note_obj_get_rgba (BIJI_NOTE_OBJ (item), &note_color))
        color = gdk_rgba_to_string (&note_color);
    }

  if (!color)
    color = g_strdup (bjb_settings_get_default_color (self->settings));

  gtk_list_store_set (store,
                      &iter,
                      BJB_MODEL_COLUMN_UUID,  biji_item_get_uuid (item),
                      BJB_MODEL_COLUMN_TITLE, biji_item_get_title (item),
                      BJB_MODEL_COLUMN_TEXT,  text,
                      BJB_MODEL_COLUMN_MTIME, biji_item_get_mtime (item),
                      BJB_MODEL_COLUMN_COLOR, color,
                      -1);
}

/* If the user searches for notes, is the note searched? */
static void
bjb_controller_add_item_if_needed (BjbController *self,
                                   BijiItem      *item,
                                   gboolean       prepend,
                                   GtkTreeIter   *sibling)
{
  gboolean need_to_add_item = FALSE;
  const gchar *content;
  const gchar *title;

  /* No note... */
  if (!item || !BIJI_IS_ITEM (item))
    return;

  /* No search - we add the note */
  if (!self->needle || g_strcmp0 (self->needle, "") == 0)
  {
    if (!self->notebook)
      need_to_add_item = TRUE;

    /* To do : we might have a notebook
     * but have to udpate the view */
  }

  /* a search.. we test...*/
  else
  {
    title = biji_item_get_title (item);

    /* matching title... */
    if (g_strrstr (title, self->needle) != NULL)
      need_to_add_item = TRUE;

    /* matching content */
    else if (BIJI_IS_NOTE_OBJ (item))
    {
      content = biji_note_obj_get_raw_text (BIJI_NOTE_OBJ (item));
      if (g_strrstr (content, self->needle) != NULL)
        need_to_add_item = TRUE;
    }
  }

  if (need_to_add_item)
    bjb_controller_add_item (self, item, prepend, sibling);
}


static gint
most_recent_item_first (gconstpointer a, gconstpointer b)
{
  const BijiItem *one = a;
  const BijiItem *other = b;
  glong result = 0;

  /* Always sort notebooks before notes */
  if (BIJI_IS_NOTEBOOK ((gpointer) a))
  {
    if (BIJI_IS_NOTE_OBJ ((gpointer) b))
      result = -1;
  }

  else if (BIJI_IS_NOTEBOOK ((gpointer) b))
  {
    result = 1;
  }

  /* If comparing two notes or
   * two notebooks, use the most recent cookbook */
  else
  {
    result = biji_item_get_mtime ((gpointer) other)
      - biji_item_get_mtime ((gpointer) one);
  }

  return result;
}



static void
sort_items (GList **to_show)
{
  *to_show = g_list_sort (*to_show, most_recent_item_first);
}


void
bjb_controller_update_view (BjbController *self)
{
  GList *l;
  BjbWindowViewType type;

  /* Do not update if nothing to show */
  type = bjb_window_base_get_view_type (self->window);
  if (! (type == BJB_WINDOW_BASE_MAIN_VIEW
         || type == BJB_WINDOW_BASE_ARCHIVE_VIEW))
    return;

  free_items_store (self);

  sort_items (&self->items_to_show);

  for (l = self->items_to_show; l != NULL; l = l->next)
  {
    bjb_controller_add_item (self, l->data, FALSE, NULL);
  }
}


static void
notify_displayed_items_changed (BjbController *self)
{
  g_signal_emit (G_OBJECT (self),
                 bjb_controller_signals[DISPLAY_NOTES_CHANGED],
                 0,
                 (self->items_to_show != NULL),
                 self->remaining_items);
}

static void
update (BjbController *self)
{
  BjbWindowViewType type;

  /* If the user already edits a note, he does not want the view
   * to go back */
  if (bjb_window_base_get_view_type (self->window) != BJB_WINDOW_BASE_NOTE_VIEW)
  {
    if (self->group == BIJI_LIVING_ITEMS)
      type = BJB_WINDOW_BASE_MAIN_VIEW;
    else
      type = BJB_WINDOW_BASE_ARCHIVE_VIEW;

    bjb_window_base_switch_to (self->window, type);
  }


  bjb_controller_update_view (self);
  notify_displayed_items_changed (self);
}



static void
_add_if_group_match (BjbController  *self,
                    GList         **to_show,
                    gpointer       *item,
                    gint           *count)
{
  gboolean trashed, match;
  BijiNoteObj *note;

  if (BIJI_IS_NOTE_OBJ (*item))
  {
    match = FALSE;
    note = BIJI_NOTE_OBJ (*item);
    trashed = biji_note_obj_is_trashed (note);

    if ((trashed==FALSE && self->group == BIJI_LIVING_ITEMS) ||
        (trashed==TRUE  && self->group == BIJI_ARCHIVED_ITEMS))
      match = TRUE;
  }

  else
  {
    match = TRUE;
  }

  if (match)
  {
    *to_show = g_list_prepend (*to_show, *item);
    *count = *count + 1;
  }
}


static void
update_controller_callback (GList *result,
                            gpointer user_data)
{
  BjbController *self;
  GList *l;
  gint i;

  self = BJB_CONTROLLER (user_data);
  self->remaining_items = FALSE;

  if (!result)
  {
    bjb_window_base_switch_to (self->window, BJB_WINDOW_BASE_NO_RESULT);
    return;
  }

  sort_items (&result);
  i = 0;

  /* We are supposed to show n items. While not reach,
   * add to the list. As soon as reached, keep in mind that */

  for (l=result ; l!= NULL ; l=l->next)
  {
    if (i < self->n_items_to_show)
    {
      _add_if_group_match (self,
                           &self->items_to_show,
                           &l->data,
			   &i);
    }

    else if (l->next != NULL)
    {
      self->remaining_items = TRUE;
      break;
    }
  }

  self->items_to_show = g_list_reverse (self->items_to_show);
  update (self);

  switch (self->group)
  {
    case BIJI_ARCHIVED_ITEMS:
      bjb_window_base_switch_to (self->window, BJB_WINDOW_BASE_ARCHIVE_VIEW);
      break;
    case BIJI_LIVING_ITEMS:
    default:
      break;
  }
}

void
bjb_controller_apply_needle (BjbController *self)
{
  GList *result;
  gchar *needle;

  needle = self->needle;
  g_clear_pointer (&self->items_to_show, g_list_free);

  /* Show all items
   * If no items, tell it - unless trash is visited */
  if (needle == NULL || g_strcmp0 (needle,"") == 0)
  {
    result = biji_manager_get_items (self->manager, self->group);

    if (result == NULL)
      {
        if (!self->notebook && self->group == BIJI_LIVING_ITEMS)
          bjb_window_base_switch_to (self->window, BJB_WINDOW_BASE_NO_NOTE);
        else
          bjb_window_base_switch_to (self->window, BJB_WINDOW_BASE_NO_RESULT);
      }
    else
      update_controller_callback (result, self);

    g_list_free (result);

    return;
  }

  /* There is a research, apply lookup */
  biji_get_items_matching_async (self->manager, self->group, needle, update_controller_callback, self);
}

static void
on_needle_changed (BjbController *self)
{
  bjb_controller_apply_needle (self);
  g_signal_emit (self, bjb_controller_signals[SEARCH_CHANGED], 0);
}


/* Return FALSE to end timeout, see below on_manager_changed */
static gboolean
bjb_controller_set_window_active (BjbController *self)
{
  bjb_window_base_set_active (self->window, TRUE);
  return FALSE;
}


/* Depending on the change at data level,
 * the view has to be totaly refreshed or just amended */
static void
on_manager_changed (BijiManager            *manager,
                    BijiItemsGroup          group,
                    BijiManagerChangeFlag   flag,
                    gpointer               *biji_item,
                    BjbController          *self)
{
  BijiItem    *item = BIJI_ITEM (biji_item);
  GtkTreeIter iter;
  GtkTreeIter *p_iter = &iter;

  if (group != self->group)
  {
    g_debug ("Controller received signal for group %i while %i",
             group, self->group);
    return;
  }

  g_mutex_lock (&self->mutex);


  switch (flag)
  {
    case BIJI_MANAGER_MASS_CHANGE:
      bjb_controller_apply_needle (self);
      g_timeout_add (1, (GSourceFunc) bjb_controller_set_window_active, self);
      break;

    /* If this is a *new* item, per def prepend */
    case BIJI_MANAGER_ITEM_ADDED:
          if (BIJI_IS_NOTE_OBJ (item))
            bjb_controller_get_iter (self, NULL, &p_iter);

          else if (BIJI_IS_NOTEBOOK (item))
            p_iter = NULL;

          bjb_controller_add_item_if_needed (self, item, TRUE, p_iter);
          self->n_items_to_show++;
          self->items_to_show = g_list_prepend (self->items_to_show, item);
          notify_displayed_items_changed (self);
      break;

    /* Same comment, prepend but notebook before note */
    case BIJI_MANAGER_NOTE_AMENDED:
    case BIJI_MANAGER_ITEM_ICON_CHANGED:
      if (bjb_controller_get_iter (self, item, &p_iter))
      {
        gtk_list_store_remove (GTK_LIST_STORE (self->model), p_iter);

        if (BIJI_IS_NOTE_OBJ (item))
          bjb_controller_get_iter (self, NULL, &p_iter);

        else if (BIJI_IS_NOTEBOOK (item))
          p_iter = NULL;

        bjb_controller_add_item_if_needed (self, item, TRUE, p_iter);
      }
      break;

    case BIJI_MANAGER_ITEM_DELETED:
    case BIJI_MANAGER_ITEM_TRASHED:
    case BIJI_MANAGER_ITEM_RESTORED:
      if (bjb_controller_get_iter (self, item, &p_iter))
        gtk_list_store_remove (GTK_LIST_STORE (self->model), p_iter);

      self->items_to_show = g_list_remove (self->items_to_show, item);
      if (self->items_to_show == NULL)
        {
          if (!self->notebook && group == BIJI_LIVING_ITEMS)
            bjb_window_base_switch_to (self->window, BJB_WINDOW_BASE_NO_NOTE);
          else
            bjb_window_base_switch_to (self->window, BJB_WINDOW_BASE_NO_RESULT);
        }
      else
        notify_displayed_items_changed (self);

      break;

    /* Apply the needle to display the relevant items.
     * The window will ping to tell it's now active
     * Use another thread for this, because controller is now up to date,
     * and we need to unlock mutex,
     * since activating window can call this function! */

    case BIJI_MANAGER_CHANGE_FLAG:
    default:
      bjb_controller_apply_needle (self);
  }

  g_mutex_unlock (&self->mutex);
}

static void
bjb_controller_connect (BjbController *self)
{
  if (!self->connected)
  {
    self->manager_change = g_signal_connect (self->manager, "changed",
                                     G_CALLBACK(on_manager_changed), self);
    self->connected = TRUE;
  }

  bjb_controller_update_view (self);
}

void
bjb_controller_disconnect (BjbController *self)
{
  g_signal_handler_disconnect (self->manager, self->manager_change);
  self->manager_change = 0;
}

static void
bjb_controller_constructed (GObject *obj)
{
  BjbController *self = BJB_CONTROLLER (obj);
  G_OBJECT_CLASS(bjb_controller_parent_class)->constructed(obj);

  self->settings = bjb_app_get_settings (g_application_get_default ());

  bjb_controller_connect (self);
}

static void
bjb_controller_class_init (BjbControllerClass *klass)
{
  GObjectClass* object_class = G_OBJECT_CLASS (klass);

  object_class->get_property = bjb_controller_get_property;
  object_class->set_property = bjb_controller_set_property;
  object_class->finalize = bjb_controller_finalize;
  object_class->constructed = bjb_controller_constructed;

  bjb_controller_signals[SEARCH_CHANGED] = g_signal_new ( "search-changed" ,
                                                  G_OBJECT_CLASS_TYPE (klass),
                                                  G_SIGNAL_RUN_LAST,
                                                  0,
                                                  NULL,
                                                  NULL,
                                                  g_cclosure_marshal_VOID__VOID,
                                                  G_TYPE_NONE,
                                                  0);

  bjb_controller_signals[DISPLAY_NOTES_CHANGED] = g_signal_new ( "display-items-changed" ,
                                                  G_OBJECT_CLASS_TYPE (klass),
                                                  G_SIGNAL_RUN_LAST,
                                                  0,
                                                  NULL,
                                                  NULL,
                                                  _biji_marshal_VOID__BOOLEAN_BOOLEAN,
                                                  G_TYPE_NONE,
                                                  2,
                                                  G_TYPE_BOOLEAN,
                                                  G_TYPE_BOOLEAN);

  properties[PROP_BOOK] = g_param_spec_object ("manager",
                                               "Book",
                                               "The BijiManager",
                                               BIJI_TYPE_MANAGER,
                                               G_PARAM_READWRITE |
                                               G_PARAM_CONSTRUCT |
                                               G_PARAM_STATIC_STRINGS);

  properties[PROP_WINDOW] = g_param_spec_object ("window",
                                                 "GtkWindow",
                                                 "BjbWindowBase",
                                                 BJB_TYPE_WINDOW_BASE,
                                                 G_PARAM_READWRITE |
                                                 G_PARAM_CONSTRUCT |
                                                 G_PARAM_STATIC_STRINGS);

  properties[PROP_NEEDLE] = g_param_spec_string ("needle",
                                                 "Needle",
                                                 "String to search notes",
                                                 NULL,
                                                 G_PARAM_READWRITE |
                                                 G_PARAM_CONSTRUCT |
                                                 G_PARAM_STATIC_STRINGS);


  properties[PROP_MODEL] = g_param_spec_object ("model",
                                                "Model",
                                                "The GtkTreeModel",
                                                GTK_TYPE_TREE_MODEL,
                                                G_PARAM_READABLE  |
                                                G_PARAM_STATIC_STRINGS);

  g_object_class_install_properties (object_class, NUM_PROPERTIES, properties);

}

BjbController *
bjb_controller_new (BijiManager  *manager,
                    GtkWindow     *window,
                    gchar         *needle)
{
  return g_object_new ( BJB_TYPE_CONTROLLER,
              "manager", manager,
              "window", window,
              "needle", needle,
              NULL);
}

void
bjb_controller_set_manager (BjbController *self, BijiManager  *manager )
{
  self->manager = manager;
}

void
bjb_controller_set_needle (BjbController *self, const gchar *needle )
{
  if (self->needle)
    g_free (self->needle);

  self->needle = g_strdup (needle);
  on_needle_changed (self);
}

gchar *
bjb_controller_get_needle (BjbController *self)
{
  if (!self->needle)
    return NULL;

  return self->needle;
}

GtkTreeModel *
bjb_controller_get_model  (BjbController *self)
{
  return self->model;
}


gboolean
bjb_controller_shows_item (BjbController *self)
{
  return (self->items_to_show != NULL);
}

BijiNotebook *
bjb_controller_get_notebook (BjbController *self)
{
  return self->notebook;
}


void
bjb_controller_set_notebook (BjbController *self,
                             BijiNotebook *coll)
{
  /* Going back from a notebook */
  if (!coll)
  {
    if (!self->notebook)
      return;

    bjb_window_base_switch_to (self->window, BJB_WINDOW_BASE_SPINNER_VIEW);
    self->notebook = NULL;
    bjb_controller_apply_needle (self);
    return;
  }

  /* Opening an __existing__ notebook */
  bjb_window_base_switch_to (self->window, BJB_WINDOW_BASE_SPINNER_VIEW);
  g_clear_pointer (&self->items_to_show, g_list_free);
  g_clear_pointer (&self->needle, g_free);

  self->needle = g_strdup ("");
  self->notebook = coll;
  biji_get_items_with_notebook_async (self->manager,
                                      biji_item_get_title (BIJI_ITEM (coll)),
                                      update_controller_callback,
                                      self);
}



BijiItemsGroup
bjb_controller_get_group (BjbController *self)
{
  return self->group;
}



void
bjb_controller_set_group (BjbController   *self,
                          BijiItemsGroup   group)
{
  if (self->group == group)
    return;

  g_clear_pointer (&self->items_to_show, g_list_free);
  self->group = group;

  /* Living group : refresh the ui */
  if (group == BIJI_LIVING_ITEMS)
  {
    if (self->notebook != NULL)
      bjb_controller_set_notebook (self, self->notebook);

    else
      bjb_controller_apply_needle (self);

    return;
  }
  else /* Archives */
  {
    bjb_controller_apply_needle (self);
  }
}



void
bjb_controller_show_more (BjbController *self)
{
  self->n_items_to_show += BJB_ITEMS_SLICE;

  /* FIXME: this method to refresh is just non sense */

  if (self->notebook != NULL)
    bjb_controller_set_notebook (self, self->notebook);

  else
    on_needle_changed (self);
}



gboolean
bjb_controller_get_remaining_items (BjbController *self)
{
  return self->remaining_items;
}

static gboolean
bjb_controller_build_selection_list_foreach (GtkTreeModel *model,
                                             GtkTreePath  *path,
                                             GtkTreeIter  *iter,
                                             gpointer      user_data)
{
  GList    **sel;
  gboolean   is_selected;

  sel = user_data;

  gtk_tree_model_get (model,
                      iter,
                      BJB_MODEL_COLUMN_SELECTED, &is_selected,
                      -1);

  if (is_selected)
    {
      *sel = g_list_prepend (*sel, gtk_tree_path_copy (path));
    }

  return FALSE;
}

GList *
bjb_controller_get_selection (BjbController *self)
{
  GList *retval = NULL;

  gtk_tree_model_foreach (self->model,
                          bjb_controller_build_selection_list_foreach,
                          &retval);

  return g_list_reverse (retval);
}

static void
bjb_controller_set_selection (GtkListStore *store,
                              GtkTreeIter  *iter,
                              gboolean      selection)
{
  gtk_list_store_set (store, iter,
                      BJB_MODEL_COLUMN_SELECTED, selection,
                      -1);
}

static gboolean
bjb_controller_set_selection_foreach (GtkTreeModel *model,
                                      GtkTreePath  *path,
                                      GtkTreeIter  *iter,
                                      gpointer      user_data)
{
  gboolean selection = GPOINTER_TO_INT (user_data);

  bjb_controller_set_selection (GTK_LIST_STORE (model), iter, selection);

  return FALSE;
}

static void
bjb_controller_set_all_selection (BjbController *self,
                                  gboolean       selection)
{
  gtk_tree_model_foreach (self->model,
                          bjb_controller_set_selection_foreach,
                          GINT_TO_POINTER (selection));

  notify_displayed_items_changed (self);
}

void
bjb_controller_select_item (BjbController *self,
                            const char    *iter_string)
{
  GtkTreeIter iter;
  if (!gtk_tree_model_get_iter_from_string (self->model, &iter, iter_string))
    return;
  bjb_controller_set_selection (GTK_LIST_STORE (self->model), &iter, TRUE);
}

void
bjb_controller_unselect_item (BjbController *self,
                              const char    *iter_string)
{
  GtkTreeIter iter;
  if (!gtk_tree_model_get_iter_from_string (self->model, &iter, iter_string))
    return;
  bjb_controller_set_selection (GTK_LIST_STORE (self->model), &iter, FALSE);
}

void
bjb_controller_select_all (BjbController *self)
{
  bjb_controller_set_all_selection (self, TRUE);
}

void
bjb_controller_unselect_all (BjbController *self)
{
  bjb_controller_set_all_selection (self, FALSE);
}

gboolean
bjb_controller_get_selection_mode (BjbController *self)
{
  return self->selection_mode;
}

void
bjb_controller_set_selection_mode (BjbController *self,
                                   gboolean       selection_mode)
{
  if (self->selection_mode != selection_mode)
    {
      self->selection_mode = selection_mode;
      bjb_controller_unselect_all (self);
    }
}

gboolean
bjb_controller_is_all_selected (BjbController *self)
{
  g_autoptr (GList) selection_list = bjb_controller_get_selection (self);
  return g_list_length (selection_list) == g_list_length (self->items_to_show);
}

