gio tests: add a markup reader "test"

This is actually a utility program that dumps out a document by way of
GMarkupReader.

3 files changed, 176 insertions, 0 deletions

diff --git a/gio/tests/.gitignore b/gio/tests/.gitignore
index 302d6943d..e257ec5ab 100644
--- a/gio/tests/.gitignore
+++ b/gio/tests/.gitignore
@@ -92,6 +92,7 @@ icons
 inet-address
 io-stream
 live-g-file
+markup-read
 memory-input-stream
 memory-output-stream
 mimeapps
diff --git a/gio/tests/Makefile.am b/gio/tests/Makefile.am
index 4bb71a87c..2c888432f 100644
--- a/gio/tests/Makefile.am
+++ b/gio/tests/Makefile.am
@@ -95,6 +95,7 @@ uninstalled_test_extra_programs = \
 	gdbus-example-watch-proxy		\
 	gsubprocess-testprog			\
 	httpd					\
+	markup-read				\
 	proxy					\
 	resolver				\
 	send-data				\
diff --git a/gio/tests/markup-read.c b/gio/tests/markup-read.c
new file mode 100644
index 000000000..4ca07c60e
--- /dev/null
+++ b/gio/tests/markup-read.c
@@ -0,0 +1,174 @@
+/*
+ * Copyright © 2014 Canonical Limited
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2 of the licence, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the
+ * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ *
+ * Author: Ryan Lortie <desrt@desrt.ca>
+ */
+
+#include <gio/gio.h>
+#include <gio/gunixinputstream.h>
+
+#include <stdlib.h>
+
+static void try_to_read (GMarkupReader *reader);
+
+static gboolean sync_mode;
+
+static void
+print_token (GMarkupReader *reader)
+{
+  if (g_markup_reader_is_eof (reader))
+    {
+      g_print ("eof\n");
+      exit (0);
+    }
+
+  else if (g_markup_reader_is_start_element (reader, NULL))
+    g_print ("start %s\n", g_markup_reader_get_element_name (reader));
+
+  else if (g_markup_reader_is_end_element (reader))
+    g_print ("end %s\n", g_markup_reader_get_element_name (reader));
+
+  else if (g_markup_reader_is_text (reader))
+    g_print ("text\n");
+}
+
+static void
+advance_complete (GObject      *source,
+                  GAsyncResult *result,
+                  gpointer      user_data)
+{
+  GMarkupReader *reader = G_MARKUP_READER (source);
+  GError *error = NULL;
+
+  if (!g_markup_reader_advance_finish (reader, result, &error))
+    {
+      g_printerr ("\nerror advancing to next token: %s\n", error->message);
+      g_error_free (error);
+      exit (1);
+    }
+
+  print_token (reader);
+
+  try_to_read (reader);
+}
+
+static void
+try_to_read (GMarkupReader *reader)
+{
+  GError *error = NULL;
+
+  while (g_markup_reader_advance_nonblocking (reader, NULL, &error))
+    print_token (reader);
+
+  if (g_error_matches (error, G_IO_ERROR, G_IO_ERROR_WOULD_BLOCK))
+    {
+      g_printerr ("\nwould block -- doing async read\n");
+      g_error_free (error);
+
+      g_markup_reader_advance_async (reader, NULL, advance_complete, NULL);
+    }
+  else
+    {
+      g_printerr ("\nerror advancing to next token: %s\n", error->message);
+      g_error_free (error);
+      exit (1);
+    }
+}
+
+static void
+read_sync (GMarkupReader *reader)
+{
+  GError *error = NULL;
+
+  g_printerr ("stream cannot poll -- doing sync reads\n");
+
+  while (g_markup_reader_advance (reader, NULL, &error))
+    print_token (reader);
+
+  g_printerr ("\nerror advancing to next token: %s\n", error->message);
+  g_error_free (error);
+  exit (1);
+}
+
+static void
+got_stream (GInputStream *stream)
+{
+  GMarkupReader *reader;
+
+  g_printerr ("got stream\n");
+
+  reader = g_markup_reader_new (stream, 0);
+
+  if (sync_mode)
+    read_sync (reader);
+  else
+    try_to_read (reader);
+}
+
+static void
+read_async_complete (GObject      *source,
+                     GAsyncResult *result,
+                     gpointer      user_data)
+{
+  GFileInputStream *stream;
+  GError *error = NULL;
+
+  stream = g_file_read_finish (G_FILE (source), result, &error);
+
+  if (stream == NULL)
+    {
+      g_printerr ("failed to open stream: %s\n", error->message);
+      g_error_free (error);
+      exit (1);
+    }
+
+  got_stream (G_INPUT_STREAM (stream));
+
+  g_object_unref (stream);
+}
+
+int
+main (int     argc,
+      gchar **argv)
+{
+  if (argc != 2)
+    {
+      g_printerr ("requires one argument: a file or uri\n");
+      return 1;
+    }
+
+  if (g_str_equal (argv[1], "-"))
+    {
+      GInputStream *stream;
+
+      stream = g_unix_input_stream_new (0, FALSE);
+      got_stream (stream);
+      g_object_unref (stream);
+    }
+  else
+    {
+      GFile *file;
+
+      file = g_file_new_for_commandline_arg (argv[1]);
+      g_file_read_async (file, G_PRIORITY_DEFAULT, NULL, read_async_complete, NULL);
+      g_object_unref (file);
+    }
+
+  while (TRUE)
+    g_main_context_iteration (NULL, TRUE);
+}