Indicator for quick/bookmarks

3 files changed, 26 insertions, 0 deletions

diff --git a/src/dwb.c b/src/dwb.c
index 704cd1d8..2eb3e735 100644
--- a/src/dwb.c
+++ b/src/dwb.c
@@ -660,6 +660,19 @@ dwb_update_status_text(GList *gl, GtkAdjustment *a) {
       g_string_append_c(string, 'u');
     g_string_append(string, "</span>]");
   }
+  if (webkit_web_view_get_load_status(WEBVIEW(gl)) == WEBKIT_LOAD_FINISHED) {
+    const char *uri = webkit_web_view_get_uri(WEBVIEW(gl));
+    gboolean has_quickmark = g_list_find_custom(dwb.fc.quickmarks, uri, (GCompareFunc)util_quickmark_compare_uri) != NULL;
+    gboolean has_bookmark = g_list_find_custom(dwb.fc.bookmarks, uri, (GCompareFunc)util_navigation_compare_uri) != NULL;
+    if (has_quickmark || has_bookmark) {
+      g_string_append_c(string, '[');
+      if (has_quickmark) 
+        g_string_append_c(string, 'Q');
+      if (has_bookmark) 
+        g_string_append_c(string, 'B');
+      g_string_append_c(string, ']');
+    }
+  }
   if (v->status->progress != 0) {
     wchar_t *bar_blocks = PROGRESS_DEFAULT;
     wchar_t buffer[PBAR_LENGTH + 1] = { 0 };
diff --git a/src/util.c b/src/util.c
index 23974ddf..4fc99af5 100644
--- a/src/util.c
+++ b/src/util.c
@@ -213,6 +213,11 @@ int
 util_navigation_compare_first(Navigation *a, Navigation *b) {
   return (g_strcmp0(a->first, b->first));
 }/*}}}*/
+/* util_navigation_compare_uri {{{*/
+int
+util_navigation_compare_uri(Navigation *a, const char *uri) {
+  return (g_strcmp0(a->first, uri));
+}/*}}}*/
 /* util_navigation_sort_first {{{*/
 int
 util_navigation_compare_second(Navigation *a, Navigation *b) {
@@ -222,6 +227,10 @@ int
 util_quickmark_compare(Quickmark *a, Quickmark *b) {
   return g_strcmp0(a->key, b->key);
 }
+int 
+util_quickmark_compare_uri(Quickmark *a, const char *uri) {
+  return g_strcmp0(a->nav->first, uri);
+}
 /* util_keymap_sort_first(KeyMap *, KeyMap *) {{{*/
 int
 util_keymap_sort_first(KeyMap *a, KeyMap *b) {
diff --git a/src/util.h b/src/util.h
index 4baa2134..a9b8e8f7 100644
--- a/src/util.h
+++ b/src/util.h
@@ -41,6 +41,7 @@ Arg * util_char_to_arg(char *, DwbType );
 // sort 
 int util_navigation_compare_first(Navigation *, Navigation *);
 int util_navigation_compare_second(Navigation *, Navigation *);
+int util_navigation_compare_uri(Navigation *, const char *);
 
 int util_keymap_sort_first(KeyMap *, KeyMap *);
 int util_keymap_sort_second(KeyMap *, KeyMap *);
@@ -73,6 +74,7 @@ void dwb_quickmark_free(Quickmark *);
 Quickmark * dwb_quickmark_new(const char *, const char *, const char *);
 Quickmark * dwb_quickmark_new_from_line(const char *);
 int util_quickmark_compare(Quickmark *a, Quickmark *b);
+int util_quickmark_compare_uri(Quickmark *a, const char *);
 
 // useless
 char * dwb_return(const char *);
@@ -99,5 +101,7 @@ const char * util_str_chug(const char *str);
 Sanitize util_string_to_sanitize(const char *);
 
 char *util_create_json(int, ...);
+char *util_create_json(int, ...);
+
 
 #endif