1 files changed, 147 insertions, 0 deletions
diff --git a/src/terminal-menu-button.c b/src/terminal-menu-button.c
new file mode 100644
index 0000000..fbe30f2
--- /dev/null
+++ b/src/terminal-menu-button.c
@@ -0,0 +1,147 @@
+/*
+ * Copyright © 2017 Christian Persch
+ *
+ * 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 3 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
+ * MERCHANMENUILITY 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/>.
+ */
+
+#include "config.h"
+
+#include "terminal-menu-button.h"
+#include "terminal-intl.h"
+#include "terminal-libgsystem.h"
+
+/* All this just because GtkToggleButton:toggled is RUN_FIRST (and the
+ * notify::active comes after the toggled signal). :-(
+ */
+
+enum
+{
+  UPDATE_MENU,
+  LAST_SIGNAL
+};
+
+static guint signals[LAST_SIGNAL];
+
+static void popup_menu_selection_done_cb (GtkMenu *menu,
+                                          GtkMenuButton *button);
+
+/* The menu button sets itself insensitive when it has no menu.
+ * Work around this by using an empty menu.
+ */
+static void
+set_empty_menu (GtkMenuButton *button)
+{
+  gs_unref_object GMenu *menu = g_menu_new ();
+  gtk_menu_button_set_menu_model (button, G_MENU_MODEL (menu));
+}
+
+static void
+disconnect_popup_menu (GtkMenuButton *button)
+{
+  GtkMenu *popup_menu = gtk_menu_button_get_popup (button);
+
+  if (popup_menu)
+    g_signal_handlers_disconnect_by_func
+      (popup_menu, G_CALLBACK (popup_menu_selection_done_cb), button);
+}
+
+static void
+popup_menu_selection_done_cb (GtkMenu *menu,
+                              GtkMenuButton *button)
+{
+  disconnect_popup_menu (button);
+  set_empty_menu (button);
+}
+
+/* Class implementation */
+
+G_DEFINE_TYPE (TerminalMenuButton, terminal_menu_button, GTK_TYPE_MENU_BUTTON);
+
+static void
+terminal_menu_button_init (TerminalMenuButton *button_)
+{
+  GtkButton *button = GTK_BUTTON (button_);
+  GtkMenuButton *menu_button = GTK_MENU_BUTTON (button_);
+
+  gtk_button_set_relief (button, GTK_RELIEF_NONE);
+  gtk_button_set_focus_on_click (button, FALSE);
+  gtk_menu_button_set_use_popover (menu_button, FALSE);
+  set_empty_menu (menu_button);
+}
+
+static void
+terminal_menu_button_toggled (GtkToggleButton *button)
+{
+  gboolean active = gtk_toggle_button_get_active (button); /* this is already the new state */
+
+  /* On activate, update the menu */
+  if (active)
+    g_signal_emit (button, signals[UPDATE_MENU], 0);
+
+  GTK_TOGGLE_BUTTON_CLASS (terminal_menu_button_parent_class)->toggled (button);
+}
+
+static void
+terminal_menu_button_update_menu (TerminalMenuButton *button)
+{
+  GtkMenuButton *gtk_button = GTK_MENU_BUTTON (button);
+  GtkMenu *popup_menu = gtk_menu_button_get_popup (gtk_button);
+
+  if (popup_menu)
+    g_signal_connect (popup_menu, "selection-done",
+                      G_CALLBACK (popup_menu_selection_done_cb), button);
+}
+
+static void
+terminal_menu_button_dispose (GObject *object)
+{
+  disconnect_popup_menu (GTK_MENU_BUTTON (object));
+
+  G_OBJECT_CLASS (terminal_menu_button_parent_class)->dispose (object);
+}
+
+static void
+terminal_menu_button_class_init (TerminalMenuButtonClass *klass)
+{
+  klass->update_menu = terminal_menu_button_update_menu;
+
+  GObjectClass *object_class = G_OBJECT_CLASS (klass);
+  object_class->dispose = terminal_menu_button_dispose;
+
+  GtkToggleButtonClass *toggle_button_class = GTK_TOGGLE_BUTTON_CLASS (klass);
+  toggle_button_class->toggled = terminal_menu_button_toggled;
+
+  signals[UPDATE_MENU] =
+    g_signal_new (I_("update-menu"),
+                  G_OBJECT_CLASS_TYPE (klass),
+                  G_SIGNAL_RUN_LAST,
+                  G_STRUCT_OFFSET (TerminalMenuButtonClass, update_menu),
+                  NULL, NULL,
+                  g_cclosure_marshal_VOID__VOID,
+                  G_TYPE_NONE,
+                  0);
+}
+
+/* public API */
+
+/**
+ * terminal_menu_button_new:
+ *
+ * Returns: a new #TerminalMenuButton
+ */
+GtkWidget *
+terminal_menu_button_new (void)
+{
+  return g_object_new (TERMINAL_TYPE_MENU_BUTTON, NULL);
+}