Pileus Git - aweather/blob - 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 <time.h>
  20 #include <grits.h>
  21 #include <stdio.h>
  22 #include <string.h>
  23
  24 #include "alert.h"
  25 #include "alert-info.h"
  26
  27 #define MSG_INDEX "http://alerts.weather.gov/cap/us.php?x=0"
  28
  29 /* Format for single cap alert:
  30  *   "http://alerts.weather.gov/cap/wwacapget.php?x="
  31  *   "AK20111012000500CoastalFloodWarning20111012151500"
  32  *   "AKAFGCFWAFG.6258a84eb1e8c34cd888057248224d10" */
  33
  34 /************
  35  * AlertMsg *
  36  ************/
  37 /* AlertMsg data types */
  38 typedef struct {
  39         char sep0;
  40         char class [1 ]; char sep1;
  41         char action[3 ]; char sep2;
  42         char office[4 ]; char sep3;
  43         char phenom[2 ]; char sep4;
  44         char signif[1 ]; char sep5;
  45         char event [4 ]; char sep6;
  46         char begin [12]; char sep7;
  47         char end   [12]; char sep8;
  48 } AlertVtec;
  49
  50 typedef struct {
  51         /* Basic info (from alert index) */
  52         char *title;   // Winter Weather Advisory issued December 19 at 8:51PM
  53         char *link;    // http://www.weather.gov/alerts-beta/wwacapget.php?x=AK20101219205100AFGWinterWeatherAdvisoryAFG20101220030000AK
  54         char *summary; // ...WINTER WEATHER ADVISORY REMAINS IN EFFECT UNTIL 6
  55         struct {
  56                 time_t     effective; // 2010-12-19T20:51:00-09:00
  57                 time_t     expires;   // 2010-12-20T03:00:00-09:00
  58                 char      *status;    // Actual
  59                 char      *urgency;   // Expected
  60                 char      *severity;  // Minor
  61                 char      *certainty; // Likely
  62                 char      *area_desc; // Northeastern Brooks Range; Northwestern Brooks Range
  63                 char      *fips6;     // 006015 006023 006045 006105
  64                 AlertVtec *vtec;      // /X.CON.PAFG.WW.Y.0064.000000T0000Z-101220T0300Z/
  65         } cap;
  66
  67         /* Advanced info (from CAP file) */
  68         char *description;
  69         char *instruction;
  70         char *polygon;
  71
  72         /* Internal state */
  73         AlertInfo *info;         // Link to info structure for this alert
  74         GritsPoly *county_based; // Polygon for county-based warning
  75         GritsPoly *storm_based;  // Polygon for storm-based warning
  76 } AlertMsg;
  77
  78 /* AlertMsg parsing */
  79 typedef struct {
  80         time_t    updated;
  81         AlertMsg *msg;
  82         GList    *msgs;
  83         gchar    *text;
  84         gchar    *value_name;
  85 } ParseData;
  86 time_t msg_parse_time(gchar *iso8601)
  87 {
  88         GTimeVal tv = {};
  89         g_time_val_from_iso8601(iso8601, &tv);
  90         return tv.tv_sec;
  91 }
  92 AlertVtec *msg_parse_vtec(char *buf)
  93 {
  94         AlertVtec *vtec = g_new0(AlertVtec, 1);
  95         strncpy((char*)vtec, buf, sizeof(AlertVtec));
  96         vtec->sep0 = vtec->sep1 = vtec->sep2 = '\0';
  97         vtec->sep3 = vtec->sep4 = vtec->sep5 = '\0';
  98         vtec->sep6 = vtec->sep7 = vtec->sep8 = '\0';
  99         return vtec;
 100 }
 101 void msg_parse_text(GMarkupParseContext *context, const gchar *text,
 102                 gsize len, gpointer user_data, GError **error)
 103 {
 104         //g_debug("text %s", text);
 105         ParseData *data = user_data;
 106         if (data->text)
 107                 g_free(data->text);
 108         data->text = g_strndup(text, len);
 109 }
 110 void msg_parse_index_start(GMarkupParseContext *context, const gchar *name,
 111                 const gchar **keys, const gchar **vals,
 112                 gpointer user_data, GError **error)
 113 {
 114         //g_debug("start %s", name);
 115         ParseData *data = user_data;
 116         if (g_str_equal(name, "entry"))
 117                 data->msg  = g_new0(AlertMsg, 1);
 118 }
 119 void msg_parse_index_end(GMarkupParseContext *context, const gchar *name,
 120                 gpointer user_data, GError **error)
 121 {
 122         //g_debug("end %s", name);
 123         ParseData *data = user_data;
 124         AlertMsg  *msg  = data->msg;
 125         char      *text = data->text;
 126
 127         if (g_str_equal(name, "updated") && text && !data->updated)
 128                 data->updated = msg_parse_time(text);
 129
 130         if (g_str_equal(name, "entry"))
 131                 data->msgs = g_list_prepend(data->msgs, data->msg);
 132
 133         if (!text || !msg) return;
 134         else if (g_str_equal(name, "title"))         msg->title         = g_strdup(text);
 135         else if (g_str_equal(name, "id"))            msg->link          = g_strdup(text); // hack
 136         else if (g_str_equal(name, "summary"))       msg->summary       = g_strdup(text);
 137         else if (g_str_equal(name, "cap:effective")) msg->cap.effective = msg_parse_time(text);
 138         else if (g_str_equal(name, "cap:expires"))   msg->cap.expires   = msg_parse_time(text);
 139         else if (g_str_equal(name, "cap:status"))    msg->cap.status    = g_strdup(text);
 140         else if (g_str_equal(name, "cap:urgency"))   msg->cap.urgency   = g_strdup(text);
 141         else if (g_str_equal(name, "cap:severity"))  msg->cap.severity  = g_strdup(text);
 142         else if (g_str_equal(name, "cap:certainty")) msg->cap.certainty = g_strdup(text);
 143         else if (g_str_equal(name, "cap:areaDesc"))  msg->cap.area_desc = g_strdup(text);
 144
 145         if (g_str_equal(name, "title"))
 146                 msg->info = alert_info_find(msg->title);
 147
 148         if (g_str_equal(name, "valueName")) {
 149                 if (data->value_name)
 150                         g_free(data->value_name);
 151                 data->value_name = g_strdup(text);
 152         }
 153
 154         if (g_str_equal(name, "value") && data->value_name) {
 155                 if (g_str_equal(data->value_name, "FIPS6")) msg->cap.fips6 = g_strdup(text);
 156                 if (g_str_equal(data->value_name, "VTEC"))  msg->cap.vtec  = msg_parse_vtec(text);
 157         }
 158 }
 159 void msg_parse_cap_end(GMarkupParseContext *context, const gchar *name,
 160                 gpointer user_data, GError **error)
 161 {
 162         ParseData *data = user_data;
 163         AlertMsg  *msg  = data->msg;
 164         char      *text = data->text;
 165         if (!text || !msg) return;
 166         if      (g_str_equal(name, "description")) msg->description = g_strdup(text);
 167         else if (g_str_equal(name, "instruction")) msg->instruction = g_strdup(text);
 168         else if (g_str_equal(name, "polygon"))     msg->polygon     = g_strdup(text);
 169 }
 170
 171 /* AlertMsg methods */
 172 GList *msg_parse_index(gchar *text, gsize len, time_t *updated)
 173 {
 174         g_debug("GritsPluginAlert: msg_parse");
 175         GMarkupParser parser = {
 176                 .start_element = msg_parse_index_start,
 177                 .end_element   = msg_parse_index_end,
 178                 .text          = msg_parse_text,
 179         };
 180         ParseData data = {};
 181         GMarkupParseContext *context =
 182                 g_markup_parse_context_new(&parser, 0, &data, NULL);
 183         g_markup_parse_context_parse(context, text, len, NULL);
 184         g_markup_parse_context_free(context);
 185         if (data.text)
 186                 g_free(data.text);
 187         if (data.value_name)
 188                 g_free(data.value_name);
 189         *updated = data.updated;
 190         return data.msgs;
 191 }
 192
 193 void msg_parse_cap(AlertMsg *msg, gchar *text, gsize len)
 194 {
 195         g_debug("GritsPluginAlert: msg_parse_cap");
 196         GMarkupParser parser = {
 197                 .end_element   = msg_parse_cap_end,
 198                 .text          = msg_parse_text,
 199         };
 200         ParseData data = { .msg = msg };
 201         GMarkupParseContext *context =
 202                 g_markup_parse_context_new(&parser, 0, &data, NULL);
 203         g_markup_parse_context_parse(context, text, len, NULL);
 204         g_markup_parse_context_free(context);
 205         if (data.text)
 206                 g_free(data.text);
 207
 208         /* Add spaces to description... */
 209         static GRegex *regex = NULL;
 210         if (regex == NULL)
 211                 regex = g_regex_new("\\.\\n", 0, G_REGEX_MATCH_NEWLINE_ANY, NULL);
 212         if (msg->description && regex) {
 213                 char *old = msg->description;
 214                 msg->description = g_regex_replace_literal(
 215                                 regex, old, -1, 0, ".\n\n", 0, NULL);
 216                 g_free(old);
 217         }
 218 }
 219
 220 void msg_free(AlertMsg *msg)
 221 {
 222         g_free(msg->title);
 223         g_free(msg->link);
 224         g_free(msg->summary);
 225         g_free(msg->cap.status);
 226         g_free(msg->cap.urgency);
 227         g_free(msg->cap.severity);
 228         g_free(msg->cap.certainty);
 229         g_free(msg->cap.area_desc);
 230         g_free(msg->cap.fips6);
 231         g_free(msg->cap.vtec);
 232         g_free(msg->description);
 233         g_free(msg->instruction);
 234         g_free(msg->polygon);
 235         g_free(msg);
 236 }
 237
 238 void msg_print(GList *msgs)
 239 {
 240         g_message("msg_print");
 241         for (GList *cur = msgs; cur; cur = cur->next) {
 242                 AlertMsg *msg = cur->data;
 243                 g_message("alert:");
 244                 g_message("     title         = %s",  msg->title        );
 245                 g_message("     link          = %s",  msg->link         );
 246                 g_message("     summary       = %s",  msg->summary      );
 247                 g_message("     cat.effective = %lu", msg->cap.effective);
 248                 g_message("     cat.expires   = %lu", msg->cap.expires  );
 249                 g_message("     cat.status    = %s",  msg->cap.status   );
 250                 g_message("     cat.urgency   = %s",  msg->cap.urgency  );
 251                 g_message("     cat.severity  = %s",  msg->cap.severity );
 252                 g_message("     cat.certainty = %s",  msg->cap.certainty);
 253                 g_message("     cat.area_desc = %s",  msg->cap.area_desc);
 254                 g_message("     cat.fips6     = %s",  msg->cap.fips6    );
 255                 g_message("     cat.vtec      = %p",  msg->cap.vtec     );
 256         }
 257 }
 258
 259 gchar *msg_find_nearest(GritsHttp *http, time_t when, gboolean offline)
 260 {
 261         GList *files = grits_http_available(http,
 262                         "^[0-9]*.xml$", "index", NULL, NULL);
 263
 264         time_t  this_time    = 0;
 265         time_t  nearest_time = offline ? 0 : time(NULL);
 266         char   *nearest_file = NULL;
 267
 268         for (GList *cur = files; cur; cur = cur->next) {
 269                 gchar *file = cur->data;
 270                 sscanf(file, "%ld", &this_time);
 271                 if (ABS(when - this_time) <
 272                     ABS(when - nearest_time)) {
 273                         nearest_file = file;
 274                         nearest_time = this_time;
 275                 }
 276         }
 277
 278         if (nearest_file)
 279                 return g_strconcat("index/", nearest_file, NULL);
 280         else if (!offline)
 281                 return g_strdup_printf("index/%ld.xml", time(NULL));
 282         else
 283                 return NULL;
 284 }
 285
 286 GList *msg_load_index(GritsHttp *http, time_t when, time_t *updated, gboolean offline)
 287 {
 288         /* Fetch current alerts */
 289         gchar *tmp = msg_find_nearest(http, when, offline);
 290         if (!tmp)
 291                 return NULL;
 292         gchar *file = grits_http_fetch(http, MSG_INDEX, tmp, GRITS_ONCE, NULL, NULL);
 293         g_free(tmp);
 294         if (!file)
 295                 return NULL;
 296
 297         /* Load file */
 298         gchar *text; gsize len;
 299         g_file_get_contents(file, &text, &len, NULL);
 300         GList *msgs = msg_parse_index(text, len, updated);
 301         //msg_print(msgs);
 302         g_free(file);
 303         g_free(text);
 304
 305         /* Delete unrecognized messages */
 306         GList *dead = NULL;
 307         for (GList *cur = msgs; cur; cur = cur->next)
 308                 if (!((AlertMsg*)cur->data)->info)
 309                         dead = g_list_prepend(dead, cur->data);
 310         for (GList *cur = dead; cur; cur = cur->next) {
 311                 AlertMsg *msg = cur->data;
 312                 g_warning("GritsPluginAlert: unknown msg type - %s", msg->title);
 313                 msgs = g_list_remove(msgs, msg);
 314                 msg_free(msg);
 315         }
 316         g_list_free(dead);
 317
 318         return msgs;
 319 }
 320
 321 void msg_load_cap(GritsHttp *http, AlertMsg *msg)
 322 {
 323         if (msg->description || msg->instruction || msg->polygon)
 324                 return;
 325         g_debug("GritsPlguinAlert: update_cap");
 326
 327         /* Download */
 328         gchar *id = strrchr(msg->link, '=');
 329         if (!id) return; id++;
 330         gchar *dir  = g_strdelimit(g_strdup(msg->info->abbr), " ", '_');
 331         gchar *tmp  = g_strdup_printf("%s/%s.xml", dir, id);
 332         gchar *file = grits_http_fetch(http, msg->link, tmp, GRITS_ONCE, NULL, NULL);
 333         g_free(tmp);
 334         g_free(dir);
 335         if (!file)
 336                 return;
 337
 338         /* Parse */
 339         gchar *text; gsize len;
 340         g_file_get_contents(file, &text, &len, NULL);
 341         msg_parse_cap(msg, text, len);
 342         g_free(file);
 343         g_free(text);
 344 }
 345
 346
 347 /********
 348  * FIPS *
 349  ********/
 350 int fips_compare(int a, int b)
 351 {
 352         return (a <  b) ? -1 :
 353                (a == b) ?  0 : 1;
 354 }
 355
 356 GritsPoly *fips_combine(GList *polys)
 357 {
 358         /* Create points list */
 359         GPtrArray *array = g_ptr_array_new();
 360         for (GList *cur = polys; cur; cur = cur->next) {
 361                 GritsPoly *poly       = cur->data;
 362                 gdouble (**points)[3] = poly->points;
 363                 for (int i = 0; points[i]; i++)
 364                         g_ptr_array_add(array, points[i]);
 365         }
 366         g_ptr_array_add(array, NULL);
 367         gdouble (**points)[3] = (gpointer)g_ptr_array_free(array, FALSE);
 368
 369         /* Calculate center */
 370         GritsBounds bounds = {-90, 90, -180, 180};
 371         for (GList *cur = polys; cur; cur = cur->next) {
 372                 GritsObject *poly = cur->data;
 373                 gdouble lat = poly->center.lat;
 374                 gdouble lon = poly->center.lon;
 375                 if (lat > bounds.n) bounds.n = lat;
 376                 if (lat < bounds.s) bounds.s = lat;
 377                 if (lon > bounds.e) bounds.e = lon;
 378                 if (lon < bounds.w) bounds.w = lon;
 379         }
 380         GritsPoint center = {
 381                 .lat = (bounds.n + bounds.s)/2,
 382                 .lon = lon_avg(bounds.e, bounds.w),
 383         };
 384
 385         /* Create polygon */
 386         GritsPoly *poly = grits_poly_new(points);
 387         GRITS_OBJECT(poly)->skip  |= GRITS_SKIP_CENTER;
 388         GRITS_OBJECT(poly)->skip  |= GRITS_SKIP_STATE;
 389         GRITS_OBJECT(poly)->center = center;
 390         g_object_weak_ref(G_OBJECT(poly), (GWeakNotify)g_free, points);
 391         return poly;
 392 }
 393
 394 gboolean fips_group_state(gpointer key, gpointer value, gpointer data)
 395 {
 396         GList  *counties = value;
 397         GList **states   = data;
 398         GritsPoly *poly  = fips_combine(counties);
 399         GRITS_OBJECT(poly)->lod = EARTH_R/10;
 400         *states = g_list_prepend(*states, poly);
 401         g_list_free(counties);
 402         return FALSE;
 403 }
 404
 405 void fips_parse(gchar *text, GTree **_counties, GList **_states)
 406 {
 407         g_debug("GritsPluginAlert: fips_parse");
 408         GTree *counties = g_tree_new((GCompareFunc)fips_compare);
 409         GTree *states   = g_tree_new_full((GCompareDataFunc)g_strcmp0,
 410                         NULL, g_free, NULL);
 411         gchar **lines = g_strsplit(text, "\n", -1);
 412         for (gint li = 0; lines[li]; li++) {
 413                 /* Split line */
 414                 gchar **sparts = g_strsplit(lines[li], "\t", 4);
 415                 int     nparts = g_strv_length(sparts);
 416                 if (nparts < 4) {
 417                         g_strfreev(sparts);
 418                         continue;
 419                 }
 420
 421                 /* Create GritsPoly */
 422                 GritsPoly *poly = grits_poly_parse(sparts[3], "\t", " ", ",");
 423
 424                 /* Insert polys into the tree */
 425                 glong id = g_ascii_strtoll(sparts[0], NULL, 10);
 426                 g_tree_insert(counties, (gpointer)id, poly);
 427
 428                 /* Insert into states list */
 429                 GList *list = g_tree_lookup(states, sparts[2]);
 430                 list = g_list_prepend(list, poly);
 431                 g_tree_replace(states, g_strdup(sparts[2]), list);
 432
 433                 g_strfreev(sparts);
 434         }
 435         g_strfreev(lines);
 436
 437         /* Group state counties */
 438         *_counties = counties;
 439         *_states   = NULL;
 440         g_tree_foreach(states, (GTraverseFunc)fips_group_state, _states);
 441         g_tree_destroy(states);
 442 }
 443
 444 /********************
 445  * GritsPluginAlert *
 446  ********************/
 447 static GtkWidget *_make_msg_details(AlertMsg *msg)
 448 {
 449
 450         GtkWidget *title     = gtk_label_new("");
 451         gchar     *title_str = g_markup_printf_escaped("<big><b>%s</b></big>",
 452                         msg->title ?: "No title provided");
 453         gtk_label_set_use_markup(GTK_LABEL(title), TRUE);
 454         gtk_label_set_markup(GTK_LABEL(title), title_str);
 455         gtk_label_set_line_wrap(GTK_LABEL(title), TRUE);
 456         gtk_misc_set_alignment(GTK_MISC(title), 0, 0);
 457         gtk_widget_set_size_request(GTK_WIDGET(title), 500, -1);
 458         g_free(title_str);
 459
 460         GtkWidget     *alert      = gtk_scrolled_window_new(NULL, NULL);
 461         GtkWidget     *alert_view = gtk_text_view_new();
 462         GtkTextBuffer *alert_buf  = gtk_text_buffer_new(NULL);
 463         gchar         *alert_str  = g_markup_printf_escaped("%s\n\n%s",
 464                         msg->description ?: "No description provided",
 465                         msg->instruction ?: "No instructions provided");
 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                 glong 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 = gmtime(&alert->updated);
 703         gchar *date_str = g_strdup_printf(" <b><i>%04d-%02d-%02d %02d:%02d</i></b>",
 704                         tm->tm_year+1900, tm->tm_mon+1, tm->tm_mday,
 705                         tm->tm_hour,      tm->tm_min);
 706         gtk_label_set_markup(GTK_LABEL(updated), date_str);
 707         g_free(date_str);
 708
 709         gtk_widget_show_all(GTK_WIDGET(alert->config));
 710         return FALSE;
 711 }
 712
 713 static gint _sort_warnings(gconstpointer _a, gconstpointer _b)
 714 {
 715         const AlertMsg *a=_a, *b=_b;
 716         return (a->info->prior <  b->info->prior) ? -1 :
 717                (a->info->prior == b->info->prior) ?  0 : 1;
 718 }
 719
 720 static gboolean _update_end(gpointer _alert)
 721 {
 722         GritsPluginAlert *alert = _alert;
 723         gtk_widget_queue_draw(GTK_WIDGET(alert->viewer));
 724         return FALSE;
 725 }
 726
 727 static void _update_warnings(GritsPluginAlert *alert, GList *old)
 728 {
 729         g_debug("GritsPluginAlert: _update_warnings");
 730
 731         /* Sort by priority:
 732          *   reversed here since they're added
 733          *   to the viewer in reverse order */
 734         alert->msgs = g_list_sort(alert->msgs, _sort_warnings);
 735
 736         /* Remove old messages */
 737         for (GList *cur = old; cur; cur = cur->next) {
 738                 AlertMsg *msg = cur->data;
 739                 if (msg->county_based) grits_viewer_remove(alert->viewer,
 740                                 GRITS_OBJECT(msg->county_based));
 741                 if (msg->storm_based) grits_viewer_remove(alert->viewer,
 742                                 GRITS_OBJECT(msg->storm_based));
 743         }
 744
 745         /* Add new messages */
 746         /* Load counties first since it does not require network access */
 747         for (GList *cur = alert->msgs; cur; cur = cur->next) {
 748                 AlertMsg *msg = cur->data;
 749                 msg->county_based = _load_county_based(alert, msg);
 750         }
 751         g_idle_add(_update_end, alert);
 752         for (GList *cur = alert->msgs; cur; cur = cur->next) {
 753                 AlertMsg *msg = cur->data;
 754                 msg->storm_based  = _load_storm_based(alert, msg);
 755         }
 756         g_idle_add(_update_end, alert);
 757
 758         g_debug("GritsPluginAlert: _load_warnings - end");
 759 }
 760
 761 /* Callbacks */
 762 static void _update(gpointer _, gpointer _alert)
 763 {
 764         GritsPluginAlert *alert = _alert;
 765         GList *old = alert->msgs;
 766         g_debug("GritsPluginAlert: _update");
 767
 768         time_t   when    = grits_viewer_get_time(alert->viewer);
 769         gboolean offline = grits_viewer_get_offline(alert->viewer);
 770         if (!(alert->msgs = msg_load_index(alert->http, when, &alert->updated, offline)))
 771                 return;
 772
 773         g_idle_add((GSourceFunc)_update_buttons, alert);
 774         _update_warnings(alert, old);
 775
 776         g_list_foreach(old, (GFunc)msg_free, NULL);
 777         g_list_free(old);
 778
 779         g_debug("GritsPluginAlert: _update - end");
 780 }
 781
 782 static void _on_update(GritsPluginAlert *alert)
 783 {
 784         g_thread_pool_push(alert->threads, NULL+1, NULL);
 785 }
 786
 787 /* Init helpers */
 788 static GtkWidget *_make_config(void)
 789 {
 790         GtkWidget *config = gtk_vbox_new(FALSE, 0);
 791
 792         /* Setup tools area */
 793         GtkWidget *tools   = gtk_hbox_new(FALSE, 10);
 794         GtkWidget *updated = gtk_label_new(" Loading...");
 795         gtk_label_set_use_markup(GTK_LABEL(updated), TRUE);
 796         gtk_box_pack_start(GTK_BOX(tools), updated, FALSE, FALSE, 0);
 797         gtk_box_pack_start(GTK_BOX(config), tools, FALSE, FALSE, 0);
 798         g_object_set_data(G_OBJECT(config), "updated", updated);
 799
 800         /* Setup alerts */
 801         gchar *labels[] = {"<b>Warnings</b>", "<b>Watches</b>",
 802                            "<b>Advisories</b>", "<b>Other</b>"};
 803         gchar *keys[]   = {"warning",  "watch",   "advisory",   "other"};
 804         gint   cols[]   = {2, 2, 3, 2};
 805         GtkWidget *alerts = gtk_hbox_new(FALSE, 10);
 806         for (int i = 0; i < G_N_ELEMENTS(labels); i++) {
 807                 GtkWidget *frame = gtk_frame_new(labels[i]);
 808                 GtkWidget *table = gtk_table_new(1, cols[i], TRUE);
 809                 GtkWidget *label = gtk_frame_get_label_widget(GTK_FRAME(frame));
 810                 gtk_label_set_use_markup(GTK_LABEL(label), TRUE);
 811                 gtk_frame_set_shadow_type(GTK_FRAME(frame), GTK_SHADOW_NONE);
 812                 gtk_container_add(GTK_CONTAINER(frame), table);
 813                 gtk_box_pack_start(GTK_BOX(alerts), frame, TRUE, TRUE, 0);
 814                 g_object_set_data(G_OBJECT(alerts), keys[i], table);
 815         }
 816         gtk_box_pack_start(GTK_BOX(config), alerts, TRUE, TRUE, 0);
 817         g_object_set_data(G_OBJECT(config), "alerts",  alerts);
 818
 819         return config;
 820 }
 821
 822 static gboolean _clear_details(GtkWidget *dialog)
 823 {
 824         GtkWidget *content  = gtk_dialog_get_content_area(GTK_DIALOG(dialog));
 825         GList     *list     = gtk_container_get_children(GTK_CONTAINER(content));
 826         GtkWidget *notebook = list->data;
 827         g_list_free(list);
 828         gtk_widget_hide(dialog);
 829         while (gtk_notebook_get_n_pages(GTK_NOTEBOOK(notebook)))
 830                 gtk_notebook_remove_page(GTK_NOTEBOOK(notebook), 0);
 831         return TRUE;
 832 }
 833
 834 static gboolean _set_details_uri(GtkWidget *notebook, GtkNotebookPage *_,
 835                 guint num, GtkWidget *button)
 836 {
 837         g_debug("_set_details_uri");
 838         GtkWidget *page = gtk_notebook_get_nth_page(GTK_NOTEBOOK(notebook), num);
 839         AlertMsg  *msg  = g_object_get_data(G_OBJECT(page), "msg");
 840         gtk_link_button_set_uri(GTK_LINK_BUTTON(button), msg->link);
 841         return FALSE;
 842 }
 843
 844 static GtkWidget *_make_details(GritsViewer *viewer)
 845 {
 846         GtkWidget *dialog   = gtk_dialog_new();
 847         GtkWidget *action   = gtk_dialog_get_action_area (GTK_DIALOG(dialog));
 848         GtkWidget *content  = gtk_dialog_get_content_area(GTK_DIALOG(dialog));
 849         GtkWidget *notebook = gtk_notebook_new();
 850         GtkWidget *win      = gtk_widget_get_toplevel(GTK_WIDGET(viewer));
 851         GtkWidget *link     = gtk_link_button_new_with_label("", "Full Text");
 852
 853         gtk_window_set_transient_for(GTK_WINDOW(dialog), GTK_WINDOW(win));
 854         gtk_window_set_title(GTK_WINDOW(dialog), "Alert Details - AWeather");
 855         gtk_window_set_default_size(GTK_WINDOW(dialog), 625, 500);
 856         gtk_notebook_set_scrollable(GTK_NOTEBOOK(notebook), TRUE);
 857         gtk_container_add(GTK_CONTAINER(content), notebook);
 858         gtk_box_pack_end(GTK_BOX(action), link, 0, 0, 0);
 859         gtk_dialog_add_button(GTK_DIALOG(dialog), GTK_STOCK_CLOSE, GTK_RESPONSE_CLOSE);
 860
 861         g_signal_connect(dialog,   "response",     G_CALLBACK(_clear_details),   NULL);
 862         g_signal_connect(dialog,   "delete-event", G_CALLBACK(_clear_details),   NULL);
 863         g_signal_connect(notebook, "switch-page",  G_CALLBACK(_set_details_uri), link);
 864
 865         return dialog;
 866 }
 867
 868 /* Methods */
 869 GritsPluginAlert *grits_plugin_alert_new(GritsViewer *viewer, GritsPrefs *prefs)
 870 {
 871         g_debug("GritsPluginAlert: new");
 872         GritsPluginAlert *alert = g_object_new(GRITS_TYPE_PLUGIN_ALERT, NULL);
 873         alert->details = _make_details(viewer);
 874         alert->viewer  = g_object_ref(viewer);
 875         alert->prefs   = g_object_ref(prefs);
 876
 877         alert->refresh_id      = g_signal_connect_swapped(alert->viewer, "refresh",
 878                         G_CALLBACK(_on_update), alert);
 879         alert->time_changed_id = g_signal_connect_swapped(alert->viewer, "time_changed",
 880                         G_CALLBACK(_on_update), alert);
 881
 882         for (GList *cur = alert->states; cur; cur = cur->next)
 883                 grits_viewer_add(viewer, cur->data, GRITS_LEVEL_WORLD+1, FALSE);
 884
 885         _on_update(alert);
 886         return alert;
 887 }
 888
 889 static GtkWidget *grits_plugin_alert_get_config(GritsPlugin *_alert)
 890 {
 891         GritsPluginAlert *alert = GRITS_PLUGIN_ALERT(_alert);
 892         return alert->config;
 893 }
 894
 895
 896 /* GObject code */
 897 static void grits_plugin_alert_plugin_init(GritsPluginInterface *iface);
 898 G_DEFINE_TYPE_WITH_CODE(GritsPluginAlert, grits_plugin_alert, G_TYPE_OBJECT,
 899                 G_IMPLEMENT_INTERFACE(GRITS_TYPE_PLUGIN,
 900                         grits_plugin_alert_plugin_init));
 901 static void grits_plugin_alert_plugin_init(GritsPluginInterface *iface)
 902 {
 903         g_debug("GritsPluginAlert: plugin_init");
 904         /* Add methods to the interface */
 905         iface->get_config = grits_plugin_alert_get_config;
 906 }
 907 static void grits_plugin_alert_init(GritsPluginAlert *alert)
 908 {
 909         g_debug("GritsPluginAlert: init");
 910         /* Set defaults */
 911         alert->threads = g_thread_pool_new(_update, alert, 1, FALSE, NULL);
 912         alert->config  = _make_config();
 913         alert->http    = grits_http_new(G_DIR_SEPARATOR_S
 914                         "alerts" G_DIR_SEPARATOR_S
 915                         "cap"    G_DIR_SEPARATOR_S);
 916
 917         /* Load counties */
 918         gchar *text; gsize len;
 919         const gchar *file = PKGDATADIR G_DIR_SEPARATOR_S "fips.txt";
 920         if (!g_file_get_contents(file, &text, &len, NULL))
 921                 g_error("GritsPluginAlert: init - error loading fips polygons");
 922         fips_parse(text, &alert->counties, &alert->states);
 923         g_free(text);
 924 }
 925 static void grits_plugin_alert_dispose(GObject *gobject)
 926 {
 927         g_debug("GritsPluginAlert: dispose");
 928         GritsPluginAlert *alert = GRITS_PLUGIN_ALERT(gobject);
 929         /* Drop references */
 930         if (alert->viewer) {
 931                 GritsViewer *viewer = alert->viewer;
 932                 alert->viewer       = NULL;
 933                 g_signal_handler_disconnect(viewer, alert->refresh_id);
 934                 g_signal_handler_disconnect(viewer, alert->time_changed_id);
 935                 soup_session_abort(alert->http->soup);
 936                 g_thread_pool_free(alert->threads, TRUE, TRUE);
 937                 gtk_widget_destroy(alert->details);
 938                 while (gtk_events_pending())
 939                         gtk_main_iteration();
 940                 for (GList *cur = alert->msgs; cur; cur = cur->next) {
 941                         AlertMsg *msg = cur->data;
 942                         if (msg->county_based) grits_viewer_remove(viewer,
 943                                         GRITS_OBJECT(msg->county_based));
 944                         if (msg->storm_based) grits_viewer_remove(viewer,
 945                                         GRITS_OBJECT(msg->storm_based));
 946                 }
 947                 for (GList *cur = alert->states; cur; cur = cur->next)
 948                         grits_viewer_remove(viewer, cur->data);
 949                 g_object_unref(alert->prefs);
 950                 g_object_unref(viewer);
 951         }
 952         G_OBJECT_CLASS(grits_plugin_alert_parent_class)->dispose(gobject);
 953 }
 954 static gboolean _unref_county(gpointer key, gpointer val, gpointer data)
 955 {
 956         g_object_unref(val);
 957         return FALSE;
 958 }
 959 static void grits_plugin_alert_finalize(GObject *gobject)
 960 {
 961         g_debug("GritsPluginAlert: finalize");
 962         GritsPluginAlert *alert = GRITS_PLUGIN_ALERT(gobject);
 963         g_list_foreach(alert->msgs, (GFunc)msg_free, NULL);
 964         g_list_free(alert->msgs);
 965         g_list_free(alert->states);
 966         g_tree_foreach(alert->counties, (GTraverseFunc)_unref_county, NULL);
 967         g_tree_destroy(alert->counties);
 968         grits_http_free(alert->http);
 969         G_OBJECT_CLASS(grits_plugin_alert_parent_class)->finalize(gobject);
 970 }
 971 static void grits_plugin_alert_class_init(GritsPluginAlertClass *klass)
 972 {
 973         g_debug("GritsPluginAlert: class_init");
 974         GObjectClass *gobject_class = (GObjectClass*)klass;
 975         gobject_class->dispose  = grits_plugin_alert_dispose;
 976         gobject_class->finalize = grits_plugin_alert_finalize;
 977 }