sn.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3
   4 #include <gtk/gtk.h>
   5 #include <gdk/gdkx.h>
   6 #include <webkit/webkit.h>
   7
   8
   9 #define DOWNLOAD_DIR "/tmp/tmp"
  10
  11
  12 static void sn_destroy_client(GtkWidget *, gpointer);
  13 static gboolean sn_do_download(WebKitWebView *, WebKitDownload *, gpointer);
  14 static gboolean sn_download_request(WebKitWebView *, WebKitWebFrame *,
  15                                     WebKitNetworkRequest *, gchar *,
  16                                     WebKitWebPolicyDecision *, gpointer);
  17 static void sn_new_client(const gchar *uri);
  18 static gboolean sn_new_client_request(WebKitWebView *, WebKitWebFrame *,
  19                                       WebKitNetworkRequest *,
  20                                       WebKitWebNavigationAction *,
  21                                       WebKitWebPolicyDecision *, gpointer);
  22 static void sn_title_changed(GObject *, GParamSpec *, gpointer);
  23
  24
  25 static Window embed = 0;
  26 static int clients = 0;
  27 static double global_zoom = 1.0;
  28
  29
  30 struct Client
  31 {
  32         GtkWidget *win;
  33         GtkWidget *scroll;
  34         GtkWidget *web_view;
  35 };
  36
  37
  38 void
  39 sn_destroy_client(GtkWidget *obj, gpointer data)
  40 {
  41         struct Client *c = (struct Client *)data;
  42
  43         (void)obj;
  44
  45         webkit_web_view_stop_loading(WEBKIT_WEB_VIEW(c->web_view));
  46         gtk_widget_destroy(c->web_view);
  47         gtk_widget_destroy(c->scroll);
  48         gtk_widget_destroy(c->win);
  49         free(c);
  50
  51         clients--;
  52         if (clients == 0)
  53                 gtk_main_quit();
  54 }
  55
  56 gboolean
  57 sn_do_download(WebKitWebView *web_view, WebKitDownload *download, gpointer data)
  58 {
  59         const gchar *uri;
  60         char id[16] = "";
  61         int ret;
  62
  63         (void)web_view;
  64         (void)data;
  65
  66         uri = webkit_download_get_uri(download);
  67         if (fork() == 0)
  68         {
  69                 chdir(DOWNLOAD_DIR);
  70                 if (embed == 0)
  71                         ret = execlp("xterm", "xterm", "-hold", "-e", "wget", uri, NULL);
  72                 else
  73                 {
  74                         if (snprintf(id, 16, "%ld", embed) >= 16)
  75                         {
  76                                 fprintf(stderr, "sn: id for xterm embed truncated!\n");
  77                                 exit(EXIT_FAILURE);
  78                         }
  79                         ret = execlp("xterm", "xterm", "-hold", "-into", id, "-e", "wget",
  80                                      uri, NULL);
  81                 }
  82
  83                 if (ret == -1)
  84                 {
  85                         fprintf(stderr, "sn: exec'ing xterm for download");
  86                         perror(" failed");
  87                         exit(EXIT_FAILURE);
  88                 }
  89         }
  90
  91         return FALSE;
  92 }
  93
  94 gboolean
  95 sn_download_request(WebKitWebView *web_view, WebKitWebFrame *frame,
  96                     WebKitNetworkRequest *request, gchar *mime_type,
  97                     WebKitWebPolicyDecision *policy_decision,
  98                     gpointer data)
  99 {
 100         (void)frame;
 101         (void)request;
 102         (void)data;
 103
 104         if (!webkit_web_view_can_show_mime_type(web_view, mime_type))
 105         {
 106                 webkit_web_policy_decision_download(policy_decision);
 107                 return TRUE;
 108         }
 109         return FALSE;
 110 }
 111
 112 void
 113 sn_new_client(const gchar *uri)
 114 {
 115         struct Client *c = malloc(sizeof(struct Client));
 116         if (!c)
 117         {
 118                 fprintf(stderr, "sn: fatal: malloc failed\n");
 119                 exit(EXIT_FAILURE);
 120         }
 121
 122         if (embed == 0)
 123         {
 124                 c->win = gtk_window_new(GTK_WINDOW_TOPLEVEL);
 125         }
 126         else
 127         {
 128                 c->win = gtk_plug_new(embed);
 129         }
 130
 131         /* When using Gtk2, sn only shows a white area when run in suckless'
 132          * tabbed. It appears we need to set a default window size for this
 133          * to work. This is not needed when using Gtk3. */
 134         gtk_window_set_default_size(GTK_WINDOW(c->win), 1024, 768);
 135
 136         g_signal_connect(G_OBJECT(c->win), "destroy", G_CALLBACK(sn_destroy_client),
 137                          c);
 138         gtk_window_set_title(GTK_WINDOW(c->win), "sn");
 139
 140         c->web_view = webkit_web_view_new();
 141         webkit_web_view_set_full_content_zoom(WEBKIT_WEB_VIEW(c->web_view), TRUE);
 142         webkit_web_view_set_zoom_level(WEBKIT_WEB_VIEW(c->web_view), global_zoom);
 143         g_signal_connect(G_OBJECT(c->web_view), "notify::title",
 144                          G_CALLBACK(sn_title_changed), c->win);
 145         g_signal_connect(G_OBJECT(c->web_view),
 146                          "new-window-policy-decision-requested",
 147                          G_CALLBACK(sn_new_client_request), NULL);
 148         g_signal_connect(G_OBJECT(c->web_view), "mime-type-policy-decision-requested",
 149                          G_CALLBACK(sn_download_request), NULL);
 150         g_signal_connect(G_OBJECT(c->web_view), "download-requested",
 151                          G_CALLBACK(sn_do_download), NULL);
 152
 153         c->scroll = gtk_scrolled_window_new(NULL, NULL);
 154
 155         gtk_container_add(GTK_CONTAINER(c->scroll), c->web_view);
 156         gtk_container_add(GTK_CONTAINER(c->win), c->scroll);
 157
 158         gtk_widget_show_all(c->win);
 159
 160         webkit_web_view_load_uri(WEBKIT_WEB_VIEW(c->web_view), uri);
 161
 162         clients++;
 163 }
 164
 165 gboolean
 166 sn_new_client_request(WebKitWebView *web_view, WebKitWebFrame *frame,
 167                       WebKitNetworkRequest *request,
 168                       WebKitWebNavigationAction *navigation_action,
 169                       WebKitWebPolicyDecision *policy_decision,
 170                       gpointer user_data)
 171 {
 172         (void)web_view;
 173         (void)frame;
 174         (void)navigation_action;
 175         (void)user_data;
 176
 177         webkit_web_policy_decision_ignore(policy_decision);
 178         sn_new_client(webkit_network_request_get_uri(request));
 179
 180         return TRUE;
 181 }
 182
 183 void
 184 sn_title_changed(GObject *obj, GParamSpec *pspec, gpointer data)
 185 {
 186         const gchar *t;
 187         WebKitWebView *view = WEBKIT_WEB_VIEW(obj);
 188         GtkWindow *win = GTK_WINDOW(data);
 189
 190         (void)pspec;
 191
 192         t = webkit_web_view_get_title(view);
 193         gtk_window_set_title(win, (t == NULL ? "sn" : t));
 194 }
 195
 196 int
 197 main(int argc, char **argv)
 198 {
 199         int opt;
 200
 201         gtk_init(&argc, &argv);
 202
 203         while ((opt = getopt(argc, argv, "z:e:")) != -1)
 204         {
 205                 switch (opt)
 206                 {
 207                         case 'z':
 208                                 global_zoom = atof(optarg);
 209                                 break;
 210                         case 'e':
 211                                 embed = atol(optarg);
 212                                 break;
 213                 }
 214         }
 215
 216         if (optind >= argc)
 217         {
 218                 fprintf(stderr, "Usage: sn [OPTIONS] <URI>\n");
 219                 exit(EXIT_FAILURE);
 220         }
 221
 222         sn_new_client(argv[optind]);
 223         gtk_main();
 224         exit(EXIT_SUCCESS);
 225 }