refactoring + fixed memory leak
[psensor.git] / src / ui_appindicator.c
1 /*
2  * Copyright (C) 2010-2013 jeanfi@gmail.com
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License as
6  * published by the Free Software Foundation; either version 2 of the
7  * License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
17  * 02110-1301 USA
18  */
19 #include <stdio.h>
20 #include <stdlib.h>
21 #include <string.h>
22
23 #include <gtk/gtk.h>
24 #include <libappindicator/app-indicator.h>
25
26 #include "cfg.h"
27 #include "psensor.h"
28 #include "ui.h"
29 #include "ui_appindicator.h"
30 #include "ui_sensorpref.h"
31 #include "ui_status.h"
32 #include "ui_pref.h"
33
34 static const char *ICON = "psensor_normal";
35 static const char *ATTENTION_ICON = "psensor_hot";
36
37 static struct psensor **sensors;
38 static GtkMenuItem **menu_items;
39 static int appindicator_supported = 1;
40 static AppIndicator *indicator;
41 static struct ui_psensor *ui_psensor;
42
43 static void cb_menu_show(GtkMenuItem *mi, gpointer data)
44 {
45         ui_window_show((struct ui_psensor *)data);
46 }
47
48 static void cb_menu_quit(GtkMenuItem *mi, gpointer data)
49 {
50         ui_psensor_quit(data);
51 }
52
53 static void cb_menu_preferences(GtkMenuItem *mi, gpointer data)
54 {
55 #ifdef HAVE_APPINDICATOR_029
56         gdk_threads_enter();
57 #endif
58
59         ui_pref_dialog_run((struct ui_psensor *)data);
60
61 #ifdef HAVE_APPINDICATOR_029
62         gdk_threads_leave();
63 #endif
64 }
65
66 static void cb_sensor_preferences(GtkMenuItem *mi, gpointer data)
67 {
68         struct ui_psensor *ui = data;
69
70 #ifdef HAVE_APPINDICATOR_029
71         gdk_threads_enter();
72 #endif
73
74         if (ui->sensors && *ui->sensors)
75                 ui_sensorpref_dialog_run(*ui->sensors, ui);
76
77 #ifdef HAVE_APPINDICATOR_029
78         gdk_threads_leave();
79 #endif
80 }
81
82 static void cb_about(GtkMenuItem *mi, gpointer data)
83 {
84         ui_show_about_dialog();
85 }
86
87 static const char *menu_desc =
88 "<ui>"
89 "  <popup name='MainMenu'>"
90 "      <menuitem name='Show' action='ShowAction' />"
91 "      <separator />"
92 "      <separator />"
93 "      <menuitem name='Preferences' action='PreferencesAction' />"
94 "      <menuitem name='SensorPreferences' action='SensorPreferencesAction' />"
95 "      <separator />"
96 "      <menuitem name='About' action='AboutAction' />"
97 "      <separator />"
98 "      <menuitem name='Quit' action='QuitAction' />"
99 "  </popup>"
100 "</ui>";
101
102 static GtkActionEntry entries[] = {
103         { "PsensorMenuAction", NULL, "_Psensor" },
104
105         { "ShowAction", NULL,
106           N_("_Show"), NULL,
107           N_("Show"),
108           G_CALLBACK(cb_menu_show) },
109
110         { "PreferencesAction", GTK_STOCK_PREFERENCES,
111           N_("_Preferences"), NULL,
112           N_("Preferences"),
113           G_CALLBACK(cb_menu_preferences) },
114
115         { "SensorPreferencesAction", GTK_STOCK_PREFERENCES,
116           N_("S_ensor Preferences"),
117           NULL,
118           N_("SensorPreferences"),
119           G_CALLBACK(cb_sensor_preferences) },
120
121         { "AboutAction", NULL,
122           N_("_About"),
123           NULL,
124           N_("About"),
125           G_CALLBACK(cb_about) },
126
127         { "QuitAction",
128           GTK_STOCK_QUIT, "_Quit", NULL, "Quit", G_CALLBACK(cb_menu_quit) }
129 };
130 static guint n_entries = G_N_ELEMENTS(entries);
131
132 static void
133 update_menu_item(GtkMenuItem *item, struct psensor *s, int use_celcius)
134 {
135         gchar *str;
136         char *v;
137
138         v = psensor_current_value_to_str(s, use_celcius);
139
140         str = g_strdup_printf("%s: %s", s->name, v);
141
142         gtk_menu_item_set_label(item, str);
143
144         free(v);
145         g_free(str);
146 }
147
148 static void update_menu_items(int use_celcius)
149 {
150         struct psensor **s;
151         GtkMenuItem **m;
152
153         if (!sensors)
154                 return ;
155
156         for (s = sensors, m = menu_items; *s; s++, m++)
157                 update_menu_item(*m, *s, use_celcius);
158 }
159
160 static void
161 build_sensor_menu_items(const struct ui_psensor *ui,
162                         GtkMenu *menu)
163 {
164         int i, j, n, celcius;
165         const char *name;
166
167         free(menu_items);
168
169         celcius  = ui->config->temperature_unit == CELCIUS;
170
171         n = psensor_list_size(ui->sensors);
172         menu_items = malloc(n * sizeof(GtkWidget *));
173         sensors = malloc((n + 1) * sizeof(struct psensor *));
174         for (i = 0, j = 0; i < n; i++) {
175                 if (config_is_appindicator_enabled(ui->sensors[i]->id)) {
176                         sensors[j] = ui->sensors[i];
177                         name = sensors[j]->name;
178
179                         menu_items[j] = GTK_MENU_ITEM
180                                 (gtk_menu_item_new_with_label(name));
181
182                         gtk_menu_shell_insert(GTK_MENU_SHELL(menu),
183                                               GTK_WIDGET(menu_items[j]),
184                                               j+2);
185
186                         update_menu_item(menu_items[j], sensors[j], celcius);
187
188                         j++;
189                 }
190         }
191
192         sensors[j] = NULL;
193 }
194
195 static GtkWidget *get_menu(struct ui_psensor *ui)
196 {
197         GtkActionGroup *action_group;
198         GtkUIManager *menu_manager;
199         GError *error;
200         GtkMenu *menu;
201
202         action_group = gtk_action_group_new("PsensorActions");
203         gtk_action_group_set_translation_domain(action_group, PACKAGE);
204         menu_manager = gtk_ui_manager_new();
205
206         gtk_action_group_add_actions(action_group, entries, n_entries, ui);
207         gtk_ui_manager_insert_action_group(menu_manager, action_group, 0);
208
209         error = NULL;
210         gtk_ui_manager_add_ui_from_string(menu_manager, menu_desc, -1, &error);
211
212         if (error)
213                 g_error(_("building menus failed: %s"), error->message);
214
215         menu = GTK_MENU(gtk_ui_manager_get_widget(menu_manager, "/MainMenu"));
216
217         build_sensor_menu_items(ui, menu);
218
219         return GTK_WIDGET(menu);
220 }
221
222 void ui_appindicator_update(struct ui_psensor *ui, unsigned int attention)
223 {
224         AppIndicatorStatus status;
225
226         if (!indicator)
227                 return;
228
229         status = app_indicator_get_status(indicator);
230
231         if (!attention && status == APP_INDICATOR_STATUS_ATTENTION)
232                 app_indicator_set_status(indicator,
233                                          APP_INDICATOR_STATUS_ACTIVE);
234
235         if (attention && status == APP_INDICATOR_STATUS_ACTIVE)
236                 app_indicator_set_status(indicator,
237                                          APP_INDICATOR_STATUS_ATTENTION);
238
239         update_menu_items(ui->config->temperature_unit == CELCIUS);
240 }
241
242 static GtkStatusIcon *unity_fallback(AppIndicator *indicator)
243 {
244         GtkStatusIcon *ico;
245
246         log_debug("ui_appindicator.unity_fallback()");
247
248         appindicator_supported = 0;
249
250         ico = ui_status_get_icon(ui_psensor);
251
252         ui_status_set_visible(1);
253
254         return ico;
255 }
256
257 static void
258 unity_unfallback(AppIndicator *indicator, GtkStatusIcon *status_icon)
259 {
260         log_debug("ui_appindicator.unity_unfallback()");
261
262         ui_status_set_visible(0);
263
264         appindicator_supported = 1;
265 }
266
267 void ui_appindicator_update_menu(struct ui_psensor *ui)
268 {
269         GtkWidget *menu;
270
271         menu = get_menu(ui);
272         app_indicator_set_menu(indicator, GTK_MENU(menu));
273
274         gtk_widget_show_all(menu);
275 }
276
277 void ui_appindicator_init(struct ui_psensor *ui)
278 {
279         ui_psensor = ui;
280
281         indicator = app_indicator_new
282                 ("psensor",
283                  ICON,
284                  APP_INDICATOR_CATEGORY_APPLICATION_STATUS);
285
286         APP_INDICATOR_GET_CLASS(indicator)->fallback = unity_fallback;
287         APP_INDICATOR_GET_CLASS(indicator)->unfallback = unity_unfallback;
288
289         app_indicator_set_status(indicator, APP_INDICATOR_STATUS_ACTIVE);
290         app_indicator_set_attention_icon(indicator, ATTENTION_ICON);
291
292         ui_appindicator_update_menu(ui);
293 }
294
295 int is_appindicator_supported()
296 {
297         return appindicator_supported;
298 }
299
300 void ui_appindicator_cleanup()
301 {
302         free(sensors);
303         /* TODO: cleanup menu items. */
304 }