#include <glib.h>
#include <glib-object.h>

#include <gladeui/glade-app.h>

typedef void (* AssertParentedFunc) (GObject *parent, GObject *child);

typedef struct {
  GType parent_type;
  GType child_type;
  AssertParentedFunc func;
} TestData;

/* Avoid warnings from GVFS-RemoteVolumeMonitor */
static gboolean
ignore_gvfs_warning (const gchar *log_domain,
		     GLogLevelFlags log_level,
		     const gchar *message,
		     gpointer user_data)
{
  if (g_strcmp0 (log_domain, "GVFS-RemoteVolumeMonitor") == 0)
    return FALSE;

  return TRUE;
}

static gboolean
main_loop_quit_cb (gpointer data)
{
  gtk_main_quit ();

  return FALSE;
}

static void
check_finalized (gpointer data,
		 GObject *where_the_object_was)
{
  gboolean *did_finalize = (gboolean *)data;

  *did_finalize = TRUE;
}

static void
test_add_child (gconstpointer data)
{
  TestData *test = (TestData *)data;
  GladeWidgetAdaptor *parent_adaptor;
  GladeWidgetAdaptor *child_adaptor;
  GladeWidget        *gparent, *gchild;
  GObject            *parent, *child;
  gboolean            parent_finalized = FALSE;
  gboolean            child_finalized = FALSE;
  gboolean            gparent_finalized = FALSE;
  gboolean            gchild_finalized = FALSE;

  g_test_log_set_fatal_handler (ignore_gvfs_warning, NULL);

  parent_adaptor = glade_widget_adaptor_get_by_type (test->parent_type);
  child_adaptor  = glade_widget_adaptor_get_by_type (test->child_type);

  gparent = glade_widget_adaptor_create_widget (parent_adaptor, FALSE, NULL);
  gchild  = glade_widget_adaptor_create_widget (child_adaptor, FALSE, NULL);

  parent = glade_widget_get_object (gparent);
  child  = glade_widget_get_object (gchild);

  glade_widget_add_child (gparent, gchild, FALSE);

  /* Pass ownership to the parent */
  g_object_unref (gchild);

  g_assert (glade_widget_get_parent (gchild) == gparent);

  if (test->func)
    test->func (parent, child);

  /* filechoosers hold a reference until an async operation is complete */
  if (GTK_IS_FILE_CHOOSER (parent) || GTK_IS_FILE_CHOOSER (child))
    {
      g_timeout_add (2000, main_loop_quit_cb, NULL);
      gtk_main();
    }
  /* Our plugin code adds an idle when cell renderers are created */
  else if (GTK_IS_CELL_RENDERER (child))
    {
      g_timeout_add (50, main_loop_quit_cb, NULL);
      gtk_main();
    }

  /* Unreffing the parent should finalize the parent and child runtime objects */
  g_object_weak_ref (G_OBJECT (gparent), check_finalized, &gparent_finalized);
  g_object_weak_ref (G_OBJECT (gchild),  check_finalized, &gchild_finalized);
  g_object_weak_ref (G_OBJECT (parent), check_finalized, &parent_finalized);
  g_object_weak_ref (G_OBJECT (child),  check_finalized, &child_finalized);

  g_object_unref (gparent);

  g_assert (gparent_finalized);
  g_assert (gchild_finalized);
  g_assert (parent_finalized);
  g_assert (child_finalized);
}

static void
add_test (GType parent_type,
	  GType child_type,
	  AssertParentedFunc func)
{
  gchar *test_path;
  TestData *data = g_new (TestData, 1);

  test_path = g_strdup_printf ("/AddChild/%s/%s",
			       g_type_name (parent_type),
			       g_type_name (child_type));

  data->parent_type = parent_type;
  data->child_type  = child_type;
  data->func        = func;

  g_test_add_data_func_full (test_path, data, test_add_child, g_free);
  g_free (test_path);
}

static void
assert_widget_parented (GObject *parent,
			GObject *child)
{
  g_assert (gtk_widget_get_parent (GTK_WIDGET (child)) == GTK_WIDGET (parent));
}

static void
assert_submenu (GObject *parent,
		GObject *child)
{
  if (GTK_IS_MENU_ITEM (parent))
    g_assert (gtk_menu_item_get_submenu (GTK_MENU_ITEM (parent)) == GTK_WIDGET (child));
  else if (GTK_IS_MENU_TOOL_BUTTON (parent))
    g_assert (gtk_menu_tool_button_get_menu (GTK_MENU_TOOL_BUTTON (parent)) == GTK_WIDGET (child));
  else
    g_assert_not_reached ();
}

static void
assert_cell_parented (GObject *parent,
		      GObject *child)
{
  GList *cells;

  cells = gtk_cell_layout_get_cells (GTK_CELL_LAYOUT (parent));
  g_assert (g_list_find (cells, child) != NULL);
  g_list_free (cells);
}

static void
assert_column_parented (GObject *parent,
			GObject *child)
{
  g_assert (gtk_tree_view_get_column (GTK_TREE_VIEW (parent), 0) == GTK_TREE_VIEW_COLUMN (child));
}

/* Ignore deprecated classes, we test them regardless */
G_GNUC_BEGIN_IGNORE_DEPRECATIONS

static void
add_child_widgets (GType parent_type)
{
  add_test (parent_type, GTK_TYPE_IMAGE,               assert_widget_parented);
  add_test (parent_type, GTK_TYPE_LABEL,               assert_widget_parented);
  add_test (parent_type, GTK_TYPE_ACCEL_LABEL,         assert_widget_parented);
  add_test (parent_type, GTK_TYPE_ENTRY,               assert_widget_parented);
  add_test (parent_type, GTK_TYPE_SEARCH_ENTRY,        assert_widget_parented);
  add_test (parent_type, GTK_TYPE_SPIN_BUTTON,         assert_widget_parented);
  add_test (parent_type, GTK_TYPE_SWITCH,              assert_widget_parented);
  add_test (parent_type, GTK_TYPE_SEPARATOR,           assert_widget_parented);
  add_test (parent_type, GTK_TYPE_ARROW,               assert_widget_parented);
  add_test (parent_type, GTK_TYPE_DRAWING_AREA,        assert_widget_parented);
  add_test (parent_type, GTK_TYPE_SPINNER,             assert_widget_parented);
  add_test (parent_type, GTK_TYPE_LEVEL_BAR,           assert_widget_parented);
  add_test (parent_type, GTK_TYPE_PROGRESS_BAR,        assert_widget_parented);
  add_test (parent_type, GTK_TYPE_SCALE,               assert_widget_parented);
  add_test (parent_type, GTK_TYPE_SCROLLBAR,           assert_widget_parented);
  add_test (parent_type, GTK_TYPE_BUTTON,              assert_widget_parented);
  add_test (parent_type, GTK_TYPE_TOGGLE_BUTTON,       assert_widget_parented);
  add_test (parent_type, GTK_TYPE_CHECK_BUTTON,        assert_widget_parented);
  add_test (parent_type, GTK_TYPE_RADIO_BUTTON,        assert_widget_parented);
  add_test (parent_type, GTK_TYPE_LINK_BUTTON,         assert_widget_parented);
  add_test (parent_type, GTK_TYPE_MENU_BUTTON,         assert_widget_parented);

  add_test (parent_type, GTK_TYPE_COMBO_BOX,           assert_widget_parented);
  add_test (parent_type, GTK_TYPE_COMBO_BOX_TEXT,      assert_widget_parented);
  add_test (parent_type, GTK_TYPE_SCALE_BUTTON,        assert_widget_parented);
  add_test (parent_type, GTK_TYPE_VOLUME_BUTTON,       assert_widget_parented);
  add_test (parent_type, GTK_TYPE_FONT_BUTTON,         assert_widget_parented);
  add_test (parent_type, GTK_TYPE_COLOR_BUTTON,        assert_widget_parented);

  /* FIXME: FileChooserButton leaks a GTask which will crash in the following test */
  /* add_test (parent_type, GTK_TYPE_FILE_CHOOSER_BUTTON, assert_widget_parented); */
  add_test (parent_type, GTK_TYPE_APP_CHOOSER_BUTTON,  assert_widget_parented);
  add_test (parent_type, GTK_TYPE_TEXT_VIEW,           assert_widget_parented);
  add_test (parent_type, GTK_TYPE_TREE_VIEW,           assert_widget_parented);
  add_test (parent_type, GTK_TYPE_ICON_VIEW,           assert_widget_parented);
  add_test (parent_type, GTK_TYPE_CALENDAR,            assert_widget_parented);
  add_test (parent_type, GTK_TYPE_BOX,                 assert_widget_parented);
  add_test (parent_type, GTK_TYPE_NOTEBOOK,            assert_widget_parented);
  add_test (parent_type, GTK_TYPE_FRAME,               assert_widget_parented);
  add_test (parent_type, GTK_TYPE_ASPECT_FRAME,        assert_widget_parented);
  add_test (parent_type, GTK_TYPE_OVERLAY,             assert_widget_parented);
  add_test (parent_type, GTK_TYPE_MENU_BAR,            assert_widget_parented);
  add_test (parent_type, GTK_TYPE_TOOLBAR,             assert_widget_parented);
  add_test (parent_type, GTK_TYPE_TOOL_PALETTE,        assert_widget_parented);
  add_test (parent_type, GTK_TYPE_PANED,               assert_widget_parented);
  add_test (parent_type, GTK_TYPE_BUTTON_BOX,          assert_widget_parented);
  add_test (parent_type, GTK_TYPE_LAYOUT,              assert_widget_parented);
  add_test (parent_type, GTK_TYPE_FIXED,               assert_widget_parented);
  add_test (parent_type, GTK_TYPE_EVENT_BOX,           assert_widget_parented);
  add_test (parent_type, GTK_TYPE_EXPANDER,            assert_widget_parented);
  add_test (parent_type, GTK_TYPE_VIEWPORT,            assert_widget_parented);
  add_test (parent_type, GTK_TYPE_ALIGNMENT,           assert_widget_parented);
  add_test (parent_type, GTK_TYPE_GRID,                assert_widget_parented);
  add_test (parent_type, GTK_TYPE_SCROLLED_WINDOW,     assert_widget_parented);
  add_test (parent_type, GTK_TYPE_INFO_BAR,            assert_widget_parented);
  add_test (parent_type, GTK_TYPE_STATUSBAR,           assert_widget_parented);
}

static void
add_child_cells (GType parent_type)
{
  add_test (parent_type, GTK_TYPE_CELL_RENDERER_TEXT, assert_cell_parented);
  add_test (parent_type, GTK_TYPE_CELL_RENDERER_ACCEL, assert_cell_parented);
  add_test (parent_type, GTK_TYPE_CELL_RENDERER_TOGGLE, assert_cell_parented);
  add_test (parent_type, GTK_TYPE_CELL_RENDERER_COMBO, assert_cell_parented);
  add_test (parent_type, GTK_TYPE_CELL_RENDERER_SPIN, assert_cell_parented);
  add_test (parent_type, GTK_TYPE_CELL_RENDERER_PIXBUF, assert_cell_parented);
  add_test (parent_type, GTK_TYPE_CELL_RENDERER_PROGRESS, assert_cell_parented);
  add_test (parent_type, GTK_TYPE_CELL_RENDERER_SPINNER, assert_cell_parented);
}

int
main (int   argc,
      char *argv[])
{
  gtk_test_init (&argc, &argv, NULL);

  glade_init ();
  glade_app_get ();

  /* Normal GtkContainer / GtkWidget parenting */
  add_child_widgets (GTK_TYPE_WINDOW);
  add_child_widgets (GTK_TYPE_BOX);
  add_child_widgets (GTK_TYPE_GRID);
  add_child_widgets (GTK_TYPE_NOTEBOOK);
  add_child_widgets (GTK_TYPE_OVERLAY);
  add_child_widgets (GTK_TYPE_PANED);
  add_child_widgets (GTK_TYPE_BUTTON_BOX);
  add_child_widgets (GTK_TYPE_LAYOUT);
  add_child_widgets (GTK_TYPE_FIXED);
  add_child_widgets (GTK_TYPE_EVENT_BOX);
  add_child_widgets (GTK_TYPE_EXPANDER);
  add_child_widgets (GTK_TYPE_VIEWPORT);
  add_child_widgets (GTK_TYPE_ALIGNMENT);

  /* Actions */
  add_test (GTK_TYPE_ACTION_GROUP, GTK_TYPE_ACTION, NULL);
  add_test (GTK_TYPE_ACTION_GROUP, GTK_TYPE_TOGGLE_ACTION, NULL);
  add_test (GTK_TYPE_ACTION_GROUP, GTK_TYPE_RADIO_ACTION, NULL);
  add_test (GTK_TYPE_ACTION_GROUP, GTK_TYPE_RECENT_ACTION, NULL);

  /* Menus */
  add_test (GTK_TYPE_MENU_BAR, GTK_TYPE_MENU_ITEM, assert_widget_parented);
  add_test (GTK_TYPE_MENU_BAR, GTK_TYPE_IMAGE_MENU_ITEM, assert_widget_parented);
  add_test (GTK_TYPE_MENU_BAR, GTK_TYPE_CHECK_MENU_ITEM, assert_widget_parented);
  add_test (GTK_TYPE_MENU_BAR, GTK_TYPE_RADIO_MENU_ITEM, assert_widget_parented);
  add_test (GTK_TYPE_MENU_BAR, GTK_TYPE_SEPARATOR_MENU_ITEM, assert_widget_parented);
  add_test (GTK_TYPE_MENU, GTK_TYPE_MENU_ITEM, assert_widget_parented);
  add_test (GTK_TYPE_MENU, GTK_TYPE_IMAGE_MENU_ITEM, assert_widget_parented);
  add_test (GTK_TYPE_MENU, GTK_TYPE_CHECK_MENU_ITEM, assert_widget_parented);
  add_test (GTK_TYPE_MENU, GTK_TYPE_RADIO_MENU_ITEM, assert_widget_parented);
  add_test (GTK_TYPE_MENU, GTK_TYPE_SEPARATOR_MENU_ITEM, assert_widget_parented);
  add_test (GTK_TYPE_MENU_ITEM, GTK_TYPE_MENU, assert_submenu);
  add_test (GTK_TYPE_IMAGE_MENU_ITEM, GTK_TYPE_MENU, assert_submenu);
  add_test (GTK_TYPE_CHECK_MENU_ITEM, GTK_TYPE_MENU, assert_submenu);
  add_test (GTK_TYPE_RADIO_MENU_ITEM, GTK_TYPE_MENU, assert_submenu);

  /* Toolbars / ToolPalette */
  add_test (GTK_TYPE_TOOLBAR, GTK_TYPE_TOOL_BUTTON, assert_widget_parented);
  add_test (GTK_TYPE_TOOLBAR, GTK_TYPE_TOGGLE_TOOL_BUTTON, assert_widget_parented);
  add_test (GTK_TYPE_TOOLBAR, GTK_TYPE_RADIO_TOOL_BUTTON, assert_widget_parented);
  add_test (GTK_TYPE_TOOLBAR, GTK_TYPE_MENU_TOOL_BUTTON, assert_widget_parented);
  add_test (GTK_TYPE_TOOL_ITEM_GROUP, GTK_TYPE_TOOL_BUTTON, assert_widget_parented);
  add_test (GTK_TYPE_TOOL_ITEM_GROUP, GTK_TYPE_TOGGLE_TOOL_BUTTON, assert_widget_parented);
  add_test (GTK_TYPE_TOOL_ITEM_GROUP, GTK_TYPE_RADIO_TOOL_BUTTON, assert_widget_parented);
  add_test (GTK_TYPE_TOOL_ITEM_GROUP, GTK_TYPE_MENU_TOOL_BUTTON, assert_widget_parented);
  add_test (GTK_TYPE_TOOL_PALETTE, GTK_TYPE_TOOL_ITEM_GROUP, assert_widget_parented);
  add_test (GTK_TYPE_MENU_TOOL_BUTTON, GTK_TYPE_MENU, assert_submenu);

  /* Cell layouts */
  /* FIXME: GtkTreeViewColumn gives a critical when creating it */
  /* add_test (GTK_TYPE_TREE_VIEW, GTK_TYPE_TREE_VIEW_COLUMN, assert_column_parented); */
  add_child_cells (GTK_TYPE_TREE_VIEW_COLUMN);
  add_child_cells (GTK_TYPE_ICON_VIEW);
  add_child_cells (GTK_TYPE_COMBO_BOX);

  /* TextTag */
  add_test (GTK_TYPE_TEXT_TAG_TABLE, GTK_TYPE_TEXT_TAG, NULL);

  return g_test_run ();
}