From 965c28d14da39210b6f427e5afb07d6f5538920f Mon Sep 17 00:00:00 2001
From: Robert Manea <gotmor@gmail.com>
Date: Tue, 26 May 2009 14:40:01 +0200
Subject: [PATCH] added html_mode_timeout and an alrm handler that returns to
 normal command mode after the timeout

---
 uzbl.c |   55 ++++++++++++++++++++++++++++++++++++++++++++++++-------
 uzbl.h |   13 ++++++++++---
 2 files changed, 58 insertions(+), 10 deletions(-)

diff --git a/uzbl.c b/uzbl.c
index 6fdbf2e..9b07593 100644
--- a/uzbl.c
+++ b/uzbl.c
@@ -41,6 +41,7 @@
 #include <sys/types.h>
 #include <sys/un.h>
 #include <sys/utsname.h>
+#include <sys/time.h>
 #include <webkit/webkit.h>
 #include <stdio.h>
 #include <string.h>
@@ -99,6 +100,7 @@ const struct {
     { "inject_html",         PTR(uzbl.behave.inject_html,         STR, cmd_inject_html)},
     { "base_url",            PTR(uzbl.behave.base_url,            STR, NULL)},
     { "html_endmarker",      PTR(uzbl.behave.html_endmarker,      STR, NULL)},
+    { "html_mode_timeout",   PTR(uzbl.behave.html_timeout,        INT, NULL)},
     { "status_message",      PTR(uzbl.gui.sbar.msg,               STR, update_title)},
     { "show_status",         PTR(uzbl.behave.show_status,         INT, cmd_set_status)},
     { "status_top",          PTR(uzbl.behave.status_top,          INT, move_statusbar)},
@@ -289,6 +291,19 @@ clean_up(void) {
     g_hash_table_destroy(uzbl.behave.commands);
 }
 
+/* used for html_mode_timeout 
+ * be sure to extend this function to use
+ * more timers if needed in other places
+*/
+static void
+set_timeout(int seconds) {
+    struct itimerval t;
+    memset(&t, 0, sizeof t);
+
+    t.it_value.tv_sec =  seconds;
+    t.it_value.tv_usec = 0;
+    setitimer(ITIMER_REAL, &t, NULL);
+}
 
 /* --- SIGNAL HANDLER --- */
 
@@ -305,6 +320,15 @@ catch_sigint(int s) {
     exit(EXIT_SUCCESS);
 }
 
+static void
+catch_alrm(int s) {
+    (void) s;
+
+    set_var_value("mode", "0");
+    render_html();
+}
+
+
 /* --- CALLBACKS --- */
 
 static gboolean
@@ -1408,6 +1432,18 @@ runcmd(WebKitWebView* page, GArray *argv) {
     parse_cmd_line(argv_idx(argv, 0));
 }
 
+static void
+render_html() {
+    Behaviour *b = &uzbl.behave;
+
+    if(b->html_buffer->str) {
+        webkit_web_view_load_html_string (uzbl.gui.web_view,
+                b->html_buffer->str, b->base_url);
+        g_string_free(b->html_buffer, TRUE);
+        b->html_buffer = g_string_new("");
+    }
+}
+
 enum {M_CMD, M_HTML};
 static void
 parse_cmd_line(const char *ctl_line) {
@@ -1415,17 +1451,18 @@ parse_cmd_line(const char *ctl_line) {
     Behaviour *b = &uzbl.behave;
 
     if(b->mode == M_HTML) {
+
         if(!strncmp(b->html_endmarker, ctl_line, strlen(b->html_endmarker))) {
+            set_timeout(0);
             set_var_value("mode", "0");
-            if(b->html_buffer->str)
-                webkit_web_view_load_html_string (uzbl.gui.web_view,
-                        b->html_buffer->str, b->base_url);
-            g_string_free(b->html_buffer, TRUE);
-            b->html_buffer = g_string_new("");
+            render_html();
             return;
         }
-        else
+        else {
+            /* set an alarm to kill us after the timeout */
+            set_timeout(b->html_timeout);
             g_string_append(b->html_buffer, ctl_line);
+        }
     }
     else {
         /* SET command */
@@ -2200,6 +2237,9 @@ main (int argc, char* argv[]) {
         fprintf(stderr, "uzbl: error hooking SIGTERM\n");
     if(setup_signal(SIGINT, catch_sigint) == SIG_ERR)
         fprintf(stderr, "uzbl: error hooking SIGINT\n");
+    if(setup_signal(SIGALRM, catch_alrm) == SIG_ERR)
+        fprintf(stderr, "uzbl: error hooking SIGALARM\n");
+
 
     if(uname(&uzbl.state.unameinfo) == -1)
         g_printerr("Can't retrieve unameinfo.  Your useragent might appear wrong.\n");
@@ -2208,9 +2248,10 @@ main (int argc, char* argv[]) {
     uzbl.gui.sbar.progress_u = g_strdup("·");
     uzbl.gui.sbar.progress_w = 10;
 
-    /* HTML mode*/
+    /* HTML mode defaults*/
     uzbl.behave.html_buffer = g_string_new("");
     uzbl.behave.html_endmarker = g_strdup(".");
+    uzbl.behave.html_timeout = 60;
     uzbl.behave.base_url = g_strdup("http://invalid");
 
     setup_regex();
diff --git a/uzbl.h b/uzbl.h
index 14346fc..fd81614 100644
--- a/uzbl.h
+++ b/uzbl.h
@@ -172,6 +172,7 @@ typedef struct {
     gchar*   base_url;
     gchar*   html_endmarker;
     GString* html_buffer;
+    guint    html_timeout;  
 
     /* command list: name -> Command  */
     GHashTable* commands;
@@ -243,6 +244,9 @@ static sigfunc *
 setup_signal(int signe, sigfunc *shandler);
 
 static gboolean
+set_var_value(gchar *name, gchar *val);
+
+static gboolean
 new_window_cb (WebKitWebView *web_view, WebKitWebFrame *frame, WebKitNetworkRequest *request, WebKitWebNavigationAction *navigation_action, WebKitWebPolicyDecision *policy_decision, gpointer user_data);
 
 WebKitWebView*
@@ -409,6 +413,12 @@ save_cookies (SoupMessage *msg,
 static void
 set_var(WebKitWebView *page, GArray *argv);
 
+static void
+render_html();
+
+static void
+set_timeout(int seconds);
+
 
 /* Command callbacks */
 static void
@@ -495,7 +505,4 @@ cmd_inject_html();
 static void 
 cmd_caret_browsing();
 
-static void 
-cmd_set_mode();
-
 /* vi: set et ts=4: */
-- 
1.7.9.5