1 files changed, 120 insertions, 0 deletions
diff --git a/src/libwebvi/mainmenu.c b/src/libwebvi/mainmenu.c
new file mode 100644
index 0000000..df08dce
--- /dev/null
+++ b/src/libwebvi/mainmenu.c
@@ -0,0 +1,120 @@
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
+#include <libxml/tree.h>
+#include <libxml/parser.h>
+#include "mainmenu.h"
+#include "menubuilder.h"
+
+static GPtrArray *load_websites(const char *path);
+static gint title_cmp(gconstpointer a, gconstpointer b);
+static gchar *get_site_title(gchar *sitemenu, gsize sitemenu_len);
+
+char *build_mainmenu(const char *path) {
+  g_log(LIBWEBVI_LOG_DOMAIN, G_LOG_LEVEL_DEBUG, "building main menu %s", path);
+
+  MenuBuilder *menu_builder = menu_builder_create();
+  menu_builder_set_title(menu_builder, "Select video source");
+
+  GPtrArray *websites = load_websites(path);
+  menu_builder_append_link_list(menu_builder, websites);
+  char *menu = menu_builder_to_string(menu_builder);
+
+  g_ptr_array_free(websites, TRUE);
+  menu_builder_delete(menu_builder);
+  return menu;
+}
+
+/*
+ * Load known websites from the given directory.
+ *
+ * Traverses each subdirectory looking for file called menu.xml. If
+ * found, reads the file to find site title. Returns an array of
+ * titles and wvt references. The caller must call g_ptr_array_free()
+ * on the returned array (the content of the array will be freed
+ * automatically).
+ */
+GPtrArray *load_websites(const char *path) {
+  GPtrArray *websites = g_ptr_array_new_with_free_func(g_free_link);
+
+  GDir *dir = g_dir_open(path, 0, NULL);
+  if (!dir)
+    return websites;
+
+  const gchar *dirname;
+  while ((dirname = g_dir_read_name(dir))) {
+    gchar *menudir = g_strconcat(path, "/", dirname, NULL);
+    if (g_file_test(menudir, G_FILE_TEST_IS_DIR)) {
+      gchar *menufile = g_strconcat(menudir, "/menu.xml", NULL);
+
+      g_log(LIBWEBVI_LOG_DOMAIN, G_LOG_LEVEL_DEBUG, "processing website menu %s", menufile);
+
+      gchar *sitemenu = NULL;
+      gsize sitemenu_len;
+      if (g_file_get_contents(menufile, &sitemenu, &sitemenu_len, NULL)) {
+        gchar *title = get_site_title(sitemenu, sitemenu_len);
+        if (!title) {
+          title = g_strdup(dirname);
+        }
+
+        gchar *href = g_strconcat("wvt://", menufile, NULL);
+        Link *menuitem = link_create(href, title, LINK_ACTION_PARSE);
+        g_ptr_array_add(websites, menuitem);
+        g_free(href);
+        g_free(title);
+        g_free(sitemenu);
+      }
+
+      g_free(menufile);
+    }
+
+    g_free(menudir);
+  }
+  
+  g_dir_close(dir);
+
+  g_ptr_array_sort(websites, title_cmp);
+
+  return websites;
+}
+
+gint title_cmp(gconstpointer a, gconstpointer b) {
+  // a and b are pointers to Link pointers!
+  Link *link1 = *(Link **)a;
+  Link *link2 = *(Link **)b;
+  
+  return g_ascii_strcasecmp(link_get_title(link1),
+                            link_get_title(link2));
+}
+
+/*
+ * Parse the contents of website menu.xml and return site's title.
+ */
+gchar *get_site_title(gchar *menuxml, gsize menuxml_len) {
+  gchar *title = NULL;
+  xmlDocPtr doc = xmlReadMemory(menuxml, menuxml_len, "", NULL,
+                                XML_PARSE_NOWARNING | XML_PARSE_NONET);
+  if (!doc)
+    return NULL;
+
+  xmlNode *root = xmlDocGetRootElement(doc);
+  xmlNode *node = root->children;
+  while (node) {
+    if (node->type == XML_ELEMENT_NODE &&
+        xmlStrEqual(node->name, BAD_CAST "title")) {
+      xmlChar *xmltitle = xmlNodeGetContent(node);
+      if (xmltitle) {
+        title = g_strdup((gchar *)xmltitle);
+        xmlFree(xmltitle);
+
+        break;
+      }
+    }
+
+    node = node->next;
+  }
+
+  xmlFreeDoc(doc);
+
+  return title;
+}