]> Pileus Git - aweather/blob - src/plugins/alert.c
Add warning/watch/alert plugin
[aweather] / src / plugins / alert.c
1 /*
2  * Copyright (C) 2010-2011 Andy Spencer <andy753421@gmail.com>
3  * 
4  * This program is free software: you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation, either version 3 of the License, or
7  * (at your option) any later version.
8  * 
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU 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, see <http://www.gnu.org/licenses/>.
16  */
17
18
19 #include <sys/time.h>
20 #include <grits.h>
21 #include <GL/gl.h>
22 #include <stdio.h>
23 #include <string.h>
24
25 #include "alert.h"
26 #include "alert-info.h"
27
28 #define MSG_INDEX "http://alerts.weather.gov/cap/us.php?x=0"
29
30 /* Format for single cap alert:
31  *   "http://alerts.weather.gov/cap/wwacapget.php?x="
32  *   "AK20111012000500CoastalFloodWarning20111012151500"
33  *   "AKAFGCFWAFG.6258a84eb1e8c34cd888057248224d10" */
34
35 /************
36  * AlertMsg *
37  ************/
38 /* AlertMsg data types */
39 typedef struct {
40         char sep0;
41         char class [1 ]; char sep1;
42         char action[3 ]; char sep2;
43         char office[4 ]; char sep3;
44         char phenom[2 ]; char sep4;
45         char signif[1 ]; char sep5;
46         char event [4 ]; char sep6;
47         char begin [12]; char sep7;
48         char end   [12]; char sep8;
49 } AlertVtec;
50
51 typedef struct {
52         /* Basic info (from alert index) */
53         char *title;   // Winter Weather Advisory issued December 19 at 8:51PM
54         char *link;    // http://www.weather.gov/alerts-beta/wwacapget.php?x=AK20101219205100AFGWinterWeatherAdvisoryAFG20101220030000AK
55         char *summary; // ...WINTER WEATHER ADVISORY REMAINS IN EFFECT UNTIL 6
56         struct {
57                 time_t     effective; // 2010-12-19T20:51:00-09:00
58                 time_t     expires;   // 2010-12-20T03:00:00-09:00
59                 char      *status;    // Actual
60                 char      *urgency;   // Expected
61                 char      *severity;  // Minor
62                 char      *certainty; // Likely
63                 char      *area_desc; // Northeastern Brooks Range; Northwestern Brooks Range
64                 char      *fips6;     // 006015 006023 006045 006105
65                 AlertVtec *vtec;      // /X.CON.PAFG.WW.Y.0064.000000T0000Z-101220T0300Z/
66         } cap;
67
68         /* Advanced info (from CAP file) */
69         char *description;
70         char *instruction;
71         char *polygon;
72
73         /* Internal state */
74         AlertInfo *info;         // Link to info structure for this alert
75         GritsPoly *county_based; // Polygon for county-based warning
76         GritsPoly *storm_based;  // Polygon for storm-based warning
77 } AlertMsg;
78
79 /* AlertMsg parsing */
80 typedef struct {
81         time_t    updated;
82         AlertMsg *msg;
83         GList    *msgs;
84         gchar    *text;
85         gchar    *value_name;
86 } ParseData;
87 time_t msg_parse_time(gchar *iso8601)
88 {
89         GTimeVal tv = {};
90         g_time_val_from_iso8601(iso8601, &tv);
91         return tv.tv_sec;
92 }
93 AlertVtec *msg_parse_vtec(char *buf)
94 {
95         AlertVtec *vtec = g_new0(AlertVtec, 1);
96         strncpy((char*)vtec, buf, sizeof(AlertVtec));
97         vtec->sep0 = vtec->sep1 = vtec->sep2 = '\0';
98         vtec->sep3 = vtec->sep4 = vtec->sep5 = '\0';
99         vtec->sep6 = vtec->sep7 = vtec->sep8 = '\0';
100         return vtec;
101 }
102 void msg_parse_text(GMarkupParseContext *context, const gchar *text,
103                 gsize len, gpointer user_data, GError **error)
104 {
105         //g_debug("text %s", text);
106         ParseData *data = user_data;
107         if (data->text)
108                 g_free(data->text);
109         data->text = g_strndup(text, len);
110 }
111 void msg_parse_index_start(GMarkupParseContext *context, const gchar *name,
112                 const gchar **keys, const gchar **vals,
113                 gpointer user_data, GError **error)
114 {
115         //g_debug("start %s", name);
116         ParseData *data = user_data;
117         if (g_str_equal(name, "entry"))
118                 data->msg  = g_new0(AlertMsg, 1);
119 }
120 void msg_parse_index_end(GMarkupParseContext *context, const gchar *name,
121                 gpointer user_data, GError **error)
122 {
123         //g_debug("end %s", name);
124         ParseData *data = user_data;
125         AlertMsg  *msg  = data->msg;
126         char      *text = data->text;
127
128         if (g_str_equal(name, "updated") && text && !data->updated)
129                 data->updated = msg_parse_time(text);
130
131         if (g_str_equal(name, "entry"))
132                 data->msgs = g_list_prepend(data->msgs, data->msg);
133
134         if (!text || !msg) return;
135         else if (g_str_equal(name, "title"))         msg->title         = g_strdup(text);
136         else if (g_str_equal(name, "id"))            msg->link          = g_strdup(text); // hack
137         else if (g_str_equal(name, "summary"))       msg->summary       = g_strdup(text);
138         else if (g_str_equal(name, "cap:effective")) msg->cap.effective = msg_parse_time(text);
139         else if (g_str_equal(name, "cap:expires"))   msg->cap.expires   = msg_parse_time(text);
140         else if (g_str_equal(name, "cap:status"))    msg->cap.status    = g_strdup(text);
141         else if (g_str_equal(name, "cap:urgency"))   msg->cap.urgency   = g_strdup(text);
142         else if (g_str_equal(name, "cap:severity"))  msg->cap.severity  = g_strdup(text);
143         else if (g_str_equal(name, "cap:certainty")) msg->cap.certainty = g_strdup(text);
144         else if (g_str_equal(name, "cap:areaDesc"))  msg->cap.area_desc = g_strdup(text);
145
146         if (g_str_equal(name, "title"))
147                 msg->info = alert_info_find(msg->title);
148
149         if (g_str_equal(name, "valueName")) {
150                 if (data->value_name)
151                         g_free(data->value_name);
152                 data->value_name = g_strdup(text);
153         }
154
155         if (g_str_equal(name, "value") && data->value_name) {
156                 if (g_str_equal(data->value_name, "FIPS6")) msg->cap.fips6 = g_strdup(text);
157                 if (g_str_equal(data->value_name, "VTEC"))  msg->cap.vtec  = msg_parse_vtec(text);
158         }
159 }
160 void msg_parse_cap_end(GMarkupParseContext *context, const gchar *name,
161                 gpointer user_data, GError **error)
162 {
163         ParseData *data = user_data;
164         AlertMsg  *msg  = data->msg;
165         char      *text = data->text;
166         if (!text || !msg) return;
167         if      (g_str_equal(name, "description")) msg->description = g_strdup(text);
168         else if (g_str_equal(name, "instruction")) msg->instruction = g_strdup(text);
169         else if (g_str_equal(name, "polygon"))     msg->polygon     = g_strdup(text);
170 }
171
172 /* AlertMsg methods */
173 GList *msg_parse_index(gchar *text, gsize len, time_t *updated)
174 {
175         g_debug("GritsPluginAlert: msg_parse");
176         GMarkupParser parser = {
177                 .start_element = msg_parse_index_start,
178                 .end_element   = msg_parse_index_end,
179                 .text          = msg_parse_text,
180         };
181         ParseData data = {};
182         GMarkupParseContext *context =
183                 g_markup_parse_context_new(&parser, 0, &data, NULL);
184         g_markup_parse_context_parse(context, text, len, NULL);
185         g_markup_parse_context_free(context);
186         if (data.text)
187                 g_free(data.text);
188         if (data.value_name)
189                 g_free(data.value_name);
190         *updated = data.updated;
191         return data.msgs;
192 }
193
194 void msg_parse_cap(AlertMsg *msg, gchar *text, gsize len)
195 {
196         g_debug("GritsPluginAlert: msg_parse_cap");
197         GMarkupParser parser = {
198                 .end_element   = msg_parse_cap_end,
199                 .text          = msg_parse_text,
200         };
201         ParseData data = { .msg = msg };
202         GMarkupParseContext *context =
203                 g_markup_parse_context_new(&parser, 0, &data, NULL);
204         g_markup_parse_context_parse(context, text, len, NULL);
205         g_markup_parse_context_free(context);
206         if (data.text)
207                 g_free(data.text);
208
209         /* Add spaces to description... */
210         static GRegex *regex = NULL;
211         if (regex == NULL)
212                 regex = g_regex_new("\\.\\n", 0, G_REGEX_MATCH_NEWLINE_ANY, NULL);
213         if (msg->description && regex) {
214                 char *old = msg->description;
215                 msg->description = g_regex_replace_literal(
216                                 regex, old, -1, 0, ".\n\n", 0, NULL);
217                 g_free(old);
218         }
219 }
220
221 void msg_free(AlertMsg *msg)
222 {
223         g_free(msg->title);
224         g_free(msg->link);
225         g_free(msg->summary);
226         g_free(msg->cap.status);
227         g_free(msg->cap.urgency);
228         g_free(msg->cap.severity);
229         g_free(msg->cap.certainty);
230         g_free(msg->cap.area_desc);
231         g_free(msg->cap.fips6);
232         g_free(msg->cap.vtec);
233         g_free(msg->description);
234         g_free(msg->instruction);
235         g_free(msg->polygon);
236         g_free(msg);
237 }
238
239 void msg_print(GList *msgs)
240 {
241         g_message("msg_print");
242         for (GList *cur = msgs; cur; cur = cur->next) {
243                 AlertMsg *msg = cur->data;
244                 g_message("alert:");
245                 g_message("     title         = %s",  msg->title        );
246                 g_message("     link          = %s",  msg->link         );
247                 g_message("     summary       = %s",  msg->summary      );
248                 g_message("     cat.effective = %lu", msg->cap.effective);
249                 g_message("     cat.expires   = %lu", msg->cap.expires  );
250                 g_message("     cat.status    = %s",  msg->cap.status   );
251                 g_message("     cat.urgency   = %s",  msg->cap.urgency  );
252                 g_message("     cat.severity  = %s",  msg->cap.severity );
253                 g_message("     cat.certainty = %s",  msg->cap.certainty);
254                 g_message("     cat.area_desc = %s",  msg->cap.area_desc);
255                 g_message("     cat.fips6     = %s",  msg->cap.fips6    );
256                 g_message("     cat.vtec      = %p",  msg->cap.vtec     );
257         }
258 }
259
260 gchar *msg_find_nearest(GritsHttp *http, time_t when, gboolean offline)
261 {
262         GList *files = grits_http_available(http,
263                         "^[0-9]*.xml$", "index", NULL, NULL);
264
265         time_t  this_time    = 0;
266         time_t  nearest_time = offline ? 0 : time(NULL);
267         char   *nearest_file = NULL;
268
269         for (GList *cur = files; cur; cur = cur->next) {
270                 gchar *file = cur->data;
271                 sscanf(file, "%ld", &this_time);
272                 if (ABS(when - this_time) <
273                     ABS(when - nearest_time)) {
274                         nearest_file = file;
275                         nearest_time = this_time;
276                 }
277         }
278
279         if (nearest_file)
280                 return g_strconcat("index/", nearest_file, NULL);
281         else if (!offline)
282                 return g_strdup_printf("index/%ld.xml", time(NULL));
283         else
284                 return NULL;
285 }
286
287 GList *msg_load_index(GritsHttp *http, time_t when, time_t *updated, gboolean offline)
288 {
289         /* Fetch current alerts */
290         gchar *tmp = msg_find_nearest(http, when, offline);
291         if (!tmp)
292                 return NULL;
293         gchar *file = grits_http_fetch(http, MSG_INDEX, tmp, GRITS_ONCE, NULL, NULL);
294         g_free(tmp);
295         if (!file)
296                 return NULL;
297
298         /* Load file */
299         gchar *text; gsize len;
300         g_file_get_contents(file, &text, &len, NULL);
301         GList *msgs = msg_parse_index(text, len, updated);
302         //msg_print(msgs);
303         g_free(file);
304         g_free(text);
305
306         /* Delete unrecognized messages */
307         GList *dead = NULL;
308         for (GList *cur = msgs; cur; cur = cur->next)
309                 if (!((AlertMsg*)cur->data)->info)
310                         dead = g_list_prepend(dead, cur->data);
311         for (GList *cur = dead; cur; cur = cur->next) {
312                 AlertMsg *msg = cur->data;
313                 g_warning("GritsPluginAlert: unknown msg type - %s", msg->title);
314                 msgs = g_list_remove(msgs, msg);
315                 msg_free(msg);
316         }
317         g_list_free(dead);
318
319         return msgs;
320 }
321
322 void msg_load_cap(GritsHttp *http, AlertMsg *msg)
323 {
324         if (msg->description || msg->instruction || msg->polygon)
325                 return;
326         g_debug("GritsPlguinAlert: update_cap");
327
328         /* Download */
329         gchar *id = strrchr(msg->link, '=');
330         if (!id) return; id++;
331         gchar *dir  = g_strdelimit(g_strdup(msg->info->abbr), " ", '_');
332         gchar *tmp  = g_strdup_printf("%s/%s.xml", dir, id);
333         gchar *file = grits_http_fetch(http, msg->link, tmp, GRITS_ONCE, NULL, NULL);
334         g_free(tmp);
335         g_free(dir);
336         if (!file)
337                 return;
338
339         /* Parse */
340         gchar *text; gsize len;
341         g_file_get_contents(file, &text, &len, NULL);
342         msg_parse_cap(msg, text, len);
343         g_free(file);
344         g_free(text);
345 }
346
347
348 /********
349  * FIPS *
350  ********/
351 int fips_compare(int a, int b)
352 {
353         return (a <  b) ? -1 :
354                (a == b) ?  0 : 1;
355 }
356
357 GritsPoly *fips_combine(GList *polys)
358 {
359         /* Create points list */
360         GPtrArray *array = g_ptr_array_new();
361         for (GList *cur = polys; cur; cur = cur->next) {
362                 GritsPoly *poly       = cur->data;
363                 gdouble (**points)[3] = poly->points;
364                 for (int i = 0; points[i]; i++)
365                         g_ptr_array_add(array, points[i]);
366         }
367         g_ptr_array_add(array, NULL);
368         gdouble (**points)[3] = (gpointer)g_ptr_array_free(array, FALSE);
369
370         /* Calculate center */
371         GritsBounds bounds = {-90, 90, -180, 180};
372         for (GList *cur = polys; cur; cur = cur->next) {
373                 GritsObject *poly = cur->data;
374                 gdouble lat = poly->center.lat;
375                 gdouble lon = poly->center.lon;
376                 if (lat > bounds.n) bounds.n = lat;
377                 if (lat < bounds.s) bounds.s = lat;
378                 if (lon > bounds.e) bounds.e = lon;
379                 if (lon < bounds.w) bounds.w = lon;
380         }
381         GritsPoint center = {
382                 .lat = (bounds.n + bounds.s)/2,
383                 .lon = lon_avg(bounds.e, bounds.w),
384         };
385
386         /* Create polygon */
387         GritsPoly *poly = grits_poly_new(points);
388         GRITS_OBJECT(poly)->skip  |= GRITS_SKIP_CENTER;
389         GRITS_OBJECT(poly)->skip  |= GRITS_SKIP_STATE;
390         GRITS_OBJECT(poly)->center = center;
391         g_object_weak_ref(G_OBJECT(poly), (GWeakNotify)g_free, points);
392         return poly;
393 }
394
395 gboolean fips_group_state(gpointer key, gpointer value, gpointer data)
396 {
397         GList  *counties = value;
398         GList **states   = data;
399         GritsPoly *poly  = fips_combine(counties);
400         GRITS_OBJECT(poly)->lod = EARTH_R/10;
401         *states = g_list_prepend(*states, poly);
402         g_list_free(counties);
403         return FALSE;
404 }
405
406 void fips_parse(gchar *text, GTree **_counties, GList **_states)
407 {
408         g_debug("GritsPluginAlert: fips_parse");
409         GTree *counties = g_tree_new((GCompareFunc)fips_compare);
410         GTree *states   = g_tree_new_full((GCompareDataFunc)g_strcmp0,
411                         NULL, g_free, NULL);
412         gchar **lines = g_strsplit(text, "\n", -1);
413         for (gint li = 0; lines[li]; li++) {
414                 /* Split line */
415                 gchar **sparts = g_strsplit(lines[li], "\t", 4);
416                 int     nparts = g_strv_length(sparts);
417                 if (nparts < 4) {
418                         g_strfreev(sparts);
419                         continue;
420                 }
421
422                 /* Create GritsPoly */
423                 GritsPoly *poly = grits_poly_parse(sparts[3], "\t", " ", ",");
424
425                 /* Insert polys into the tree */
426                 gint id = g_ascii_strtoll(sparts[0], NULL, 10);
427                 g_tree_insert(counties, (gpointer)id, poly);
428
429                 /* Insert into states list */
430                 GList *list = g_tree_lookup(states, sparts[2]);
431                 list = g_list_prepend(list, poly);
432                 g_tree_replace(states, g_strdup(sparts[2]), list);
433
434                 g_strfreev(sparts);
435         }
436         g_strfreev(lines);
437
438         /* Group state counties */
439         *_counties = counties;
440         *_states   = NULL;
441         g_tree_foreach(states, (GTraverseFunc)fips_group_state, _states);
442         g_tree_destroy(states);
443 }
444
445 /********************
446  * GritsPluginAlert *
447  ********************/
448 static GtkWidget *_make_msg_details(AlertMsg *msg)
449 {
450
451         GtkWidget *title     = gtk_label_new("");
452         gchar     *title_str = g_markup_printf_escaped(
453                         "<big><b>%s</b></big>", msg->title);
454         gtk_label_set_use_markup(GTK_LABEL(title), TRUE);
455         gtk_label_set_markup(GTK_LABEL(title), title_str);
456         gtk_label_set_line_wrap(GTK_LABEL(title), TRUE);
457         gtk_misc_set_alignment(GTK_MISC(title), 0, 0);
458         gtk_widget_set_size_request(GTK_WIDGET(title), 500, -1);
459         g_free(title_str);
460
461         GtkWidget     *alert      = gtk_scrolled_window_new(NULL, NULL);
462         GtkWidget     *alert_view = gtk_text_view_new();
463         GtkTextBuffer *alert_buf  = gtk_text_buffer_new(NULL);
464         gchar         *alert_str  = g_markup_printf_escaped(
465                         "%s\n\n%s", msg->description, msg->instruction);
466         PangoFontDescription *alert_font = pango_font_description_from_string(
467                         "monospace");
468         gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(alert),
469                         GTK_POLICY_AUTOMATIC, GTK_POLICY_AUTOMATIC);
470         gtk_scrolled_window_set_shadow_type(GTK_SCROLLED_WINDOW(alert),
471                         GTK_SHADOW_IN);
472         gtk_text_buffer_set_text(alert_buf, alert_str, -1);
473         gtk_text_view_set_buffer(GTK_TEXT_VIEW(alert_view), alert_buf);
474         gtk_widget_modify_font(GTK_WIDGET(alert_view), alert_font);
475         gtk_container_add(GTK_CONTAINER(alert), alert_view);
476         g_free(alert_str);
477
478         GtkWidget *align = gtk_alignment_new(0, 0, 1, 1);
479         GtkWidget *box   = gtk_vbox_new(FALSE, 10);
480         gtk_alignment_set_padding(GTK_ALIGNMENT(align), 10, 10, 10, 10);
481         gtk_container_add(GTK_CONTAINER(align), box);
482         gtk_box_pack_start(GTK_BOX(box), title, FALSE, FALSE, 0);
483         gtk_box_pack_start(GTK_BOX(box), alert, TRUE,  TRUE,  0);
484
485         return align;
486 }
487
488 static GtkWidget *_find_details(GtkNotebook *notebook, AlertMsg *msg)
489 {
490         int pages = gtk_notebook_get_n_pages(GTK_NOTEBOOK(notebook));
491         for (int i = 0; i < pages; i++) {
492                 GtkWidget *page = gtk_notebook_get_nth_page(GTK_NOTEBOOK(notebook), i);
493                 if (msg == g_object_get_data(G_OBJECT(page), "msg"))
494                         return page;
495         }
496         return NULL;
497 }
498
499 static void _show_details(GritsPoly *county, GritsPluginAlert *alert)
500 {
501         /* Add details for this messages */
502         AlertMsg *msg = g_object_get_data(G_OBJECT(county), "msg");
503         msg_load_cap(alert->http, msg);
504
505         GtkWidget *dialog   = alert->details;
506         GtkWidget *content  = gtk_dialog_get_content_area(GTK_DIALOG(dialog));
507         GList     *list     = gtk_container_get_children(GTK_CONTAINER(content));
508         GtkWidget *notebook = list->data;
509         GtkWidget *details  = _find_details(GTK_NOTEBOOK(notebook), msg);
510         g_list_free(list);
511
512         if (details == NULL) {
513                 details          = _make_msg_details(msg);
514                 GtkWidget *label = gtk_label_new(msg->info->abbr);
515                 g_object_set_data(G_OBJECT(details), "msg", msg);
516                 gtk_notebook_append_page(GTK_NOTEBOOK(notebook), details, label);
517         }
518
519         gtk_widget_show_all(dialog);
520         gint num = gtk_notebook_page_num(GTK_NOTEBOOK(notebook), details);
521         gtk_notebook_set_current_page(GTK_NOTEBOOK(notebook), num);
522 }
523
524 /* Update counties */
525 static void _alert_leave(GritsPoly *county, GritsPluginAlert *alert)
526 {
527         g_debug("_alert_leave");
528         if (county->width == 3) {
529                 county->color[3]  = 0;
530         } else {
531                 county->border[3] = 0.25;
532                 county->width     = 1;
533         }
534         grits_object_queue_draw(GRITS_OBJECT(county));
535 }
536
537 static void _alert_enter(GritsPoly *county, GritsPluginAlert *alert)
538 {
539         g_debug("_alert_enter");
540         if (county->width == 3) {
541                 county->color[3]  = 0.25;
542         } else {
543                 county->border[3] = 1.0;
544                 county->width     = 2;
545         }
546         grits_object_queue_draw(GRITS_OBJECT(county));
547 }
548
549 /* Update polygons */
550 static void _load_common(GritsPluginAlert *alert, AlertMsg *msg, GritsPoly *poly,
551                 float color, float border, int width)
552 {
553         g_object_set_data(G_OBJECT(poly), "msg", msg);
554         poly->color[0]  = poly->border[0] = (float)msg->info->color[0] / 256;
555         poly->color[1]  = poly->border[1] = (float)msg->info->color[1] / 256;
556         poly->color[2]  = poly->border[2] = (float)msg->info->color[2] / 256;
557         poly->color[3]  = color;
558         poly->border[3] = border;
559         poly->width     = width;
560         GRITS_OBJECT(poly)->lod    = 0;
561         GRITS_OBJECT(poly)->hidden = msg->info->hidden;
562         g_signal_connect(poly, "enter",   G_CALLBACK(_alert_enter),  alert);
563         g_signal_connect(poly, "leave",   G_CALLBACK(_alert_leave),  alert);
564         g_signal_connect(poly, "clicked", G_CALLBACK(_show_details), alert);
565 }
566
567 static GritsPoly *_load_storm_based(GritsPluginAlert *alert, AlertMsg *msg)
568 {
569         if (!msg->info->ispoly)
570                 return NULL;
571
572         msg_load_cap(alert->http, msg);
573
574         GritsPoly *poly = grits_poly_parse(msg->polygon, "\t", " ", ",");
575         _load_common(alert, msg, poly, 0, 1, 3);
576         grits_viewer_add(alert->viewer, GRITS_OBJECT(poly), GRITS_LEVEL_WORLD+2, TRUE);
577
578         return poly;
579 }
580
581 static GritsPoly *_load_county_based(GritsPluginAlert *alert, AlertMsg *msg)
582 {
583         /* Locate counties in the path of the storm */
584         gchar **fipses  = g_strsplit(msg->cap.fips6, " ", -1);
585         GList *counties = NULL;
586         for (int i = 0; fipses[i]; i++) {
587                 gint fips = g_ascii_strtoll(fipses[i], NULL, 10);
588                 GritsPoly *county = g_tree_lookup(alert->counties, (gpointer)fips);
589                 if (!county)
590                         continue;
591                 counties = g_list_prepend(counties, county);
592         }
593         g_strfreev(fipses);
594
595         /* No county based warning.. */
596         if (!counties)
597                 return NULL;
598
599         /* Create new county based warning */
600         GritsPoly *poly = fips_combine(counties);
601         _load_common(alert, msg, poly, 0.25, 0.25, 0);
602         grits_viewer_add(alert->viewer, GRITS_OBJECT(poly), GRITS_LEVEL_WORLD+1, FALSE);
603
604         g_list_free(counties);
605         return poly;
606 }
607
608 /* Update buttons */
609 static void _button_click(GtkToggleButton *button, gpointer _alert)
610 {
611         g_debug("GritsPluginAlert: _button_click");
612         GritsPluginAlert *alert = _alert;
613         AlertInfo *info = g_object_get_data(G_OBJECT(button), "info");
614         info->hidden = !gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(button));
615
616         /* Show/hide each polygons */
617         for (GList *cur = alert->msgs; cur; cur = cur->next) {
618                 AlertMsg *msg = cur->data;
619                 if (msg->info != info)
620                         continue;
621                 if (msg->county_based) GRITS_OBJECT(msg->county_based)->hidden = info->hidden;
622                 if (msg->storm_based)  GRITS_OBJECT(msg->storm_based)->hidden  = info->hidden;
623         }
624         gtk_widget_queue_draw(GTK_WIDGET(alert->viewer));
625 }
626
627 static GtkWidget *_button_new(AlertInfo *info)
628 {
629         g_debug("GritsPluginAlert: _button_new - %s", info->title);
630         GdkColor black = {0, 0, 0, 0};
631         GdkColor color = {0, info->color[0]<<8, info->color[1]<<8, info->color[2]<<8};
632
633         gchar text[64];
634         g_snprintf(text, sizeof(text), "<b>%.10s</b>", info->abbr);
635
636         GtkWidget *button = gtk_toggle_button_new();
637         GtkWidget *align  = gtk_alignment_new(0.5, 0.5, 1, 1);
638         GtkWidget *cbox   = gtk_event_box_new();
639         GtkWidget *label  = gtk_label_new(text);
640         for (int state = 0; state < GTK_STATE_INSENSITIVE; state++) {
641                 gtk_widget_modify_fg(label, state, &black);
642                 gtk_widget_modify_bg(cbox,  state, &color);
643         } /* Yuck.. */
644         g_object_set_data(G_OBJECT(button), "info", info);
645         gtk_label_set_use_markup(GTK_LABEL(label), TRUE);
646         gtk_alignment_set_padding(GTK_ALIGNMENT(align), 2, 2, 4, 4);
647         gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), !info->hidden);
648         gtk_widget_set_tooltip_text(GTK_WIDGET(button), info->title);
649         gtk_container_add(GTK_CONTAINER(cbox), label);
650         gtk_container_add(GTK_CONTAINER(align), cbox);
651         gtk_container_add(GTK_CONTAINER(button), align);
652         return button;
653 }
654
655 static gboolean _update_buttons(GritsPluginAlert *alert)
656 {
657         g_debug("GritsPluginAlert: _update_buttons");
658         GtkWidget *alerts  = g_object_get_data(G_OBJECT(alert->config), "alerts");
659         GtkWidget *updated = g_object_get_data(G_OBJECT(alert->config), "updated");
660
661         /* Determine which buttons to show */
662         for (int i = 0; alert_info[i].title; i++)
663                 alert_info[i].current = FALSE;
664         for (GList *cur = alert->msgs; cur; cur = cur->next) {
665                 AlertMsg *msg = cur->data;
666                 msg->info->current = TRUE;
667         }
668
669         /* Delete old buttons */
670         GList *frames = gtk_container_get_children(GTK_CONTAINER(alerts));
671         for (GList *frame = frames; frame; frame = frame->next) {
672                 GtkWidget *table = gtk_bin_get_child(GTK_BIN(frame->data));
673                 GList *btns = gtk_container_get_children(GTK_CONTAINER(table));
674                 g_list_foreach(btns, (GFunc)gtk_widget_destroy, NULL);
675                 g_list_free(btns);
676         }
677         g_list_free(frames);
678
679         /* Add new buttons */
680         for (int i = 0; alert_info[i].title; i++) {
681                 if (!alert_info[i].current)
682                         continue;
683
684                 GtkWidget *table = g_object_get_data(G_OBJECT(alerts),
685                                 alert_info[i].category);
686                 GList *kids = gtk_container_get_children(GTK_CONTAINER(table));
687                 gint nkids = g_list_length(kids);
688                 guint rows, cols;
689                 gtk_table_get_size(GTK_TABLE(table), &rows, &cols);
690                 gint x = nkids % cols;
691                 gint y = nkids / cols;
692                 g_list_free(kids);
693
694                 GtkWidget *button = _button_new(&alert_info[i]);
695                 gtk_table_attach(GTK_TABLE(table), button, x, x+1, y, y+1,
696                                 GTK_FILL|GTK_EXPAND, GTK_FILL, 0, 0);
697                 g_signal_connect(button, "clicked",
698                                 G_CALLBACK(_button_click), alert);
699         }
700
701         /* Set time widget */
702         struct tm tm;
703         gmtime_r(&alert->updated, &tm);
704         gchar *date_str = g_strdup_printf(" <b><i>%04d-%02d-%02d %02d:%02d</i></b>",
705                         tm.tm_year+1900, tm.tm_mon+1, tm.tm_mday,
706                         tm.tm_hour,      tm.tm_min);
707         gtk_label_set_markup(GTK_LABEL(updated), date_str);
708         g_free(date_str);
709
710         gtk_widget_show_all(GTK_WIDGET(alert->config));
711         return FALSE;
712 }
713
714 static gint _sort_warnings(gconstpointer _a, gconstpointer _b)
715 {
716         const AlertMsg *a=_a, *b=_b;
717         return (a->info->prior <  b->info->prior) ? -1 :
718                (a->info->prior == b->info->prior) ?  0 : 1;
719 }
720
721 static void _update_warnings(GritsPluginAlert *alert, GList *old)
722 {
723         g_debug("GritsPluginAlert: _update_warnings");
724
725         /* Sort by priority:
726          *   reversed here since they're added
727          *   to the viewer in reverse order */
728         alert->msgs = g_list_sort(alert->msgs, _sort_warnings);
729
730         /* Remove old messages */
731         for (GList *cur = old; cur; cur = cur->next) {
732                 AlertMsg *msg = cur->data;
733                 if (msg->county_based) grits_viewer_remove(alert->viewer,
734                                 GRITS_OBJECT(msg->county_based));
735                 if (msg->storm_based) grits_viewer_remove(alert->viewer,
736                                 GRITS_OBJECT(msg->storm_based));
737         }
738
739         /* Add new messages */
740         for (GList *cur = alert->msgs; cur; cur = cur->next) {
741                 AlertMsg *msg = cur->data;
742                 msg->county_based = _load_county_based(alert, msg);
743                 msg->storm_based  = _load_storm_based(alert, msg);
744         }
745
746         g_debug("GritsPluginAlert: _load_warnings - end");
747 }
748
749 /* Callbacks */
750 static void _update(gpointer _, gpointer _alert)
751 {
752         GritsPluginAlert *alert = _alert;
753         GList *old = alert->msgs;
754         g_debug("GritsPluginAlert: _update");
755
756         time_t   when    = grits_viewer_get_time(alert->viewer);
757         gboolean offline = grits_viewer_get_offline(alert->viewer);
758         if (!(alert->msgs = msg_load_index(alert->http, when, &alert->updated, offline)))
759                 return;
760
761         g_idle_add((GSourceFunc)_update_buttons, alert);
762         _update_warnings(alert, old);
763
764         g_list_foreach(old, (GFunc)msg_free, NULL);
765         g_list_free(old);
766
767         gtk_widget_queue_draw(GTK_WIDGET(alert->viewer));
768         g_debug("GritsPluginAlert: _update - end");
769 }
770
771 static void _on_update(GritsPluginAlert *alert)
772 {
773         g_thread_pool_push(alert->threads, NULL+1, NULL);
774 }
775
776 /* Init helpers */
777 static GtkWidget *_make_config(void)
778 {
779         GtkWidget *config = gtk_vbox_new(FALSE, 0);
780
781         /* Setup tools area */
782         GtkWidget *tools   = gtk_hbox_new(FALSE, 10);
783         GtkWidget *updated = gtk_label_new(" Loading...");
784         gtk_label_set_use_markup(GTK_LABEL(updated), TRUE);
785         gtk_box_pack_start(GTK_BOX(tools), updated, FALSE, FALSE, 0);
786         gtk_box_pack_start(GTK_BOX(config), tools, FALSE, FALSE, 0);
787         g_object_set_data(G_OBJECT(config), "updated", updated);
788
789         /* Setup alerts */
790         gchar *labels[] = {"<b>Warnings</b>", "<b>Watches</b>",
791                            "<b>Advisories</b>", "<b>Other</b>"};
792         gchar *keys[]   = {"warning",  "watch",   "advisory",   "other"};
793         gint   cols[]   = {2, 2, 3, 2};
794         GtkWidget *alerts = gtk_hbox_new(FALSE, 10);
795         for (int i = 0; i < G_N_ELEMENTS(labels); i++) {
796                 GtkWidget *frame = gtk_frame_new(labels[i]);
797                 GtkWidget *table = gtk_table_new(1, cols[i], TRUE);
798                 GtkWidget *label = gtk_frame_get_label_widget(GTK_FRAME(frame));
799                 gtk_label_set_use_markup(GTK_LABEL(label), TRUE);
800                 gtk_frame_set_shadow_type(GTK_FRAME(frame), GTK_SHADOW_NONE);
801                 gtk_container_add(GTK_CONTAINER(frame), table);
802                 gtk_box_pack_start(GTK_BOX(alerts), frame, TRUE, TRUE, 0);
803                 g_object_set_data(G_OBJECT(alerts), keys[i], table);
804         }
805         gtk_box_pack_start(GTK_BOX(config), alerts, TRUE, TRUE, 0);
806         g_object_set_data(G_OBJECT(config), "alerts",  alerts);
807
808         return config;
809 }
810
811 static gboolean _clear_details(GtkWidget *dialog)
812 {
813         GtkWidget *content  = gtk_dialog_get_content_area(GTK_DIALOG(dialog));
814         GList     *list     = gtk_container_get_children(GTK_CONTAINER(content));
815         GtkWidget *notebook = list->data;
816         g_list_free(list);
817         gtk_widget_hide(dialog);
818         while (gtk_notebook_get_n_pages(GTK_NOTEBOOK(notebook)))
819                 gtk_notebook_remove_page(GTK_NOTEBOOK(notebook), 0);
820         return TRUE;
821 }
822
823 static gboolean _set_details_uri(GtkWidget *notebook, GtkNotebookPage *_,
824                 guint num, GtkWidget *button)
825 {
826         g_debug("_set_details_uri");
827         GtkWidget *page = gtk_notebook_get_nth_page(GTK_NOTEBOOK(notebook), num);
828         AlertMsg  *msg  = g_object_get_data(G_OBJECT(page), "msg");
829         gtk_link_button_set_uri(GTK_LINK_BUTTON(button), msg->link);
830         return FALSE;
831 }
832
833 static GtkWidget *_make_details(GritsViewer *viewer)
834 {
835         GtkWidget *dialog   = gtk_dialog_new();
836         GtkWidget *action   = gtk_dialog_get_action_area (GTK_DIALOG(dialog));
837         GtkWidget *content  = gtk_dialog_get_content_area(GTK_DIALOG(dialog));
838         GtkWidget *notebook = gtk_notebook_new();
839         GtkWidget *win      = gtk_widget_get_toplevel(GTK_WIDGET(viewer));
840         GtkWidget *link     = gtk_link_button_new_with_label("", "Full Text");
841
842         gtk_window_set_transient_for(GTK_WINDOW(dialog), GTK_WINDOW(win));
843         gtk_window_set_title(GTK_WINDOW(dialog), "Alert Details - AWeather");
844         gtk_window_set_default_size(GTK_WINDOW(dialog), 625, 500);
845         gtk_notebook_set_scrollable(GTK_NOTEBOOK(notebook), TRUE);
846         gtk_container_add(GTK_CONTAINER(content), notebook);
847         gtk_box_pack_end(GTK_BOX(action), link, 0, 0, 0);
848         gtk_dialog_add_button(GTK_DIALOG(dialog), GTK_STOCK_CLOSE, GTK_RESPONSE_CLOSE);
849
850         g_signal_connect(dialog,   "response",     G_CALLBACK(_clear_details),   NULL);
851         g_signal_connect(dialog,   "delete-event", G_CALLBACK(_clear_details),   NULL);
852         g_signal_connect(notebook, "switch-page",  G_CALLBACK(_set_details_uri), link);
853
854         return dialog;
855 }
856
857 /* Methods */
858 GritsPluginAlert *grits_plugin_alert_new(GritsViewer *viewer, GritsPrefs *prefs)
859 {
860         g_debug("GritsPluginAlert: new");
861         GritsPluginAlert *alert = g_object_new(GRITS_TYPE_PLUGIN_ALERT, NULL);
862         alert->details = _make_details(viewer);
863         alert->viewer  = g_object_ref(viewer);
864         alert->prefs   = g_object_ref(prefs);
865
866         alert->refresh_id      = g_signal_connect_swapped(alert->viewer, "refresh",
867                         G_CALLBACK(_on_update), alert);
868         alert->time_changed_id = g_signal_connect_swapped(alert->viewer, "time_changed",
869                         G_CALLBACK(_on_update), alert);
870
871         for (GList *cur = alert->states; cur; cur = cur->next)
872                 grits_viewer_add(viewer, cur->data, GRITS_LEVEL_WORLD+1, FALSE);
873
874         _on_update(alert);
875         return alert;
876 }
877
878 static GtkWidget *grits_plugin_alert_get_config(GritsPlugin *_alert)
879 {
880         GritsPluginAlert *alert = GRITS_PLUGIN_ALERT(_alert);
881         return alert->config;
882 }
883
884
885 /* GObject code */
886 static void grits_plugin_alert_plugin_init(GritsPluginInterface *iface);
887 G_DEFINE_TYPE_WITH_CODE(GritsPluginAlert, grits_plugin_alert, G_TYPE_OBJECT,
888                 G_IMPLEMENT_INTERFACE(GRITS_TYPE_PLUGIN,
889                         grits_plugin_alert_plugin_init));
890 static void grits_plugin_alert_plugin_init(GritsPluginInterface *iface)
891 {
892         g_debug("GritsPluginAlert: plugin_init");
893         /* Add methods to the interface */
894         iface->get_config = grits_plugin_alert_get_config;
895 }
896 static void grits_plugin_alert_init(GritsPluginAlert *alert)
897 {
898         g_debug("GritsPluginAlert: init");
899         /* Set defaults */
900         alert->threads = g_thread_pool_new(_update, alert, 1, TRUE, NULL);
901         alert->config  = _make_config();
902         alert->http    = grits_http_new(G_DIR_SEPARATOR_S
903                         "alerts" G_DIR_SEPARATOR_S
904                         "cap"    G_DIR_SEPARATOR_S);
905
906         /* Load counties */
907         gchar *text; gsize len;
908         const gchar *file = PKGDATADIR G_DIR_SEPARATOR_S "fips.txt";
909         if (!g_file_get_contents(file, &text, &len, NULL))
910                 g_error("GritsPluginAlert: init - error loading fips polygons");
911         fips_parse(text, &alert->counties, &alert->states);
912         g_free(text);
913 }
914 static void grits_plugin_alert_dispose(GObject *gobject)
915 {
916         g_debug("GritsPluginAlert: dispose");
917         GritsPluginAlert *alert = GRITS_PLUGIN_ALERT(gobject);
918         /* Drop references */
919         if (alert->viewer) {
920                 GritsViewer *viewer = alert->viewer;
921                 alert->viewer       = NULL;
922                 g_signal_handler_disconnect(viewer, alert->refresh_id);
923                 g_signal_handler_disconnect(viewer, alert->time_changed_id);
924                 soup_session_abort(alert->http->soup);
925                 g_thread_pool_free(alert->threads, TRUE, TRUE);
926                 gtk_widget_destroy(alert->details);
927                 while (gtk_events_pending())
928                         gtk_main_iteration();
929                 for (GList *cur = alert->msgs; cur; cur = cur->next) {
930                         AlertMsg *msg = cur->data;
931                         if (msg->county_based) grits_viewer_remove(viewer,
932                                         GRITS_OBJECT(msg->county_based));
933                         if (msg->storm_based) grits_viewer_remove(viewer,
934                                         GRITS_OBJECT(msg->storm_based));
935                 }
936                 for (GList *cur = alert->states; cur; cur = cur->next)
937                         grits_viewer_remove(viewer, cur->data);
938                 g_object_unref(alert->prefs);
939                 g_object_unref(viewer);
940         }
941         G_OBJECT_CLASS(grits_plugin_alert_parent_class)->dispose(gobject);
942 }
943 static gboolean _unref_county(gpointer key, gpointer val, gpointer data)
944 {
945         g_object_unref(val);
946         return FALSE;
947 }
948 static void grits_plugin_alert_finalize(GObject *gobject)
949 {
950         g_debug("GritsPluginAlert: finalize");
951         GritsPluginAlert *alert = GRITS_PLUGIN_ALERT(gobject);
952         g_list_foreach(alert->msgs, (GFunc)msg_free, NULL);
953         g_list_free(alert->msgs);
954         g_list_free(alert->states);
955         g_tree_foreach(alert->counties, (GTraverseFunc)_unref_county, NULL);
956         g_tree_destroy(alert->counties);
957         grits_http_free(alert->http);
958         G_OBJECT_CLASS(grits_plugin_alert_parent_class)->finalize(gobject);
959 }
960 static void grits_plugin_alert_class_init(GritsPluginAlertClass *klass)
961 {
962         g_debug("GritsPluginAlert: class_init");
963         GObjectClass *gobject_class = (GObjectClass*)klass;
964         gobject_class->dispose  = grits_plugin_alert_dispose;
965         gobject_class->finalize = grits_plugin_alert_finalize;
966 }