Rename tools, make them installable

1 files changed, 313 insertions, 0 deletions

diff --git a/src/tdv-query-tool.c b/src/tdv-query-tool.c
new file mode 100644
index 0000000..6cfdc66
--- /dev/null
+++ b/src/tdv-query-tool.c
@@ -0,0 +1,313 @@
+/*
+ * A tool to query multiple dictionaries for the specified word
+ *
+ * Intended for use in IRC bots and similar silly things---words go in,
+ * one per each line, and entries come out, one dictionary at a time,
+ * finalised with an empty line.  Newlines are escaped with `\n',
+ * backslashes with `\\'.
+ *
+ * So far only the `m', `g`, and `x` fields are supported, as in tdv.
+ *
+ * Copyright (c) 2013 - 2021, Přemysl Eric Janouch <p@janouch.name>
+ *
+ * Permission to use, copy, modify, and/or distribute this software for any
+ * purpose with or without fee is hereby granted.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
+ * SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
+ * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
+ * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ *
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <errno.h>
+
+#include <glib.h>
+#include <gio/gio.h>
+#include <pango/pango.h>
+
+#include "stardict.h"
+#include "stardict-private.h"
+#include "generator.h"
+#include "utils.h"
+
+
+// --- Output formatting -------------------------------------------------------
+
+/// Transform Pango attributes to in-line formatting sequences (non-reentrant)
+typedef const gchar *(*FormatterFunc) (PangoAttrIterator *);
+
+static const gchar *
+pango_attrs_ignore (G_GNUC_UNUSED PangoAttrIterator *iterator)
+{
+	return "";
+}
+
+static const gchar *
+pango_attrs_to_irc (PangoAttrIterator *iterator)
+{
+	static gchar buf[5];
+	gchar *p = buf;
+	*p++ = 0x0f;
+
+	if (!iterator)
+		goto reset_formatting;
+
+	PangoAttrInt *attr = NULL;
+	if ((attr = (PangoAttrInt *) pango_attr_iterator_get (iterator,
+			PANGO_ATTR_WEIGHT)) && attr->value >= PANGO_WEIGHT_BOLD)
+		*p++ = 0x02;
+	if ((attr = (PangoAttrInt *) pango_attr_iterator_get (iterator,
+			PANGO_ATTR_UNDERLINE)) && attr->value == PANGO_UNDERLINE_SINGLE)
+		*p++ = 0x1f;
+	if ((attr = (PangoAttrInt *) pango_attr_iterator_get (iterator,
+			PANGO_ATTR_STYLE)) && attr->value == PANGO_STYLE_ITALIC)
+		*p++ = 0x1d;
+
+reset_formatting:
+	*p++ = 0;
+	return buf;
+}
+
+static const gchar *
+pango_attrs_to_ansi (PangoAttrIterator *iterator)
+{
+	static gchar buf[16];
+	g_strlcpy (buf, "\x1b[0", sizeof buf);
+	if (!iterator)
+		goto reset_formatting;
+
+	PangoAttrInt *attr = NULL;
+	if ((attr = (PangoAttrInt *) pango_attr_iterator_get (iterator,
+			PANGO_ATTR_WEIGHT)) && attr->value >= PANGO_WEIGHT_BOLD)
+		g_strlcat (buf, ";1", sizeof buf);
+	if ((attr = (PangoAttrInt *) pango_attr_iterator_get (iterator,
+			PANGO_ATTR_UNDERLINE)) && attr->value == PANGO_UNDERLINE_SINGLE)
+		g_strlcat (buf, ";4", sizeof buf);
+	if ((attr = (PangoAttrInt *) pango_attr_iterator_get (iterator,
+			PANGO_ATTR_STYLE)) && attr->value == PANGO_STYLE_ITALIC)
+		g_strlcat (buf, ";3", sizeof buf);
+
+reset_formatting:
+	g_strlcat (buf, "m", sizeof buf);
+	return buf;
+}
+
+static gchar *
+pango_to_output_text (const gchar *markup, FormatterFunc formatter)
+{
+	// This function skips leading whitespace, but it's the canonical one
+	gchar *text = NULL;
+	PangoAttrList *attrs = NULL;
+	if (!pango_parse_markup (markup, -1, 0, &attrs, &text, NULL, NULL))
+		return g_strdup_printf ("<%s>", ("error in entry"));
+
+	PangoAttrIterator *iterator = pango_attr_list_get_iterator (attrs);
+	GString *result = g_string_new ("");
+	do
+	{
+		gint start = 0, end = 0;
+		pango_attr_iterator_range (iterator, &start, &end);
+		if (end == G_MAXINT)
+			end = strlen (text);
+
+		g_string_append (result, formatter (iterator));
+		g_string_append_len (result, text + start, end - start);
+	}
+	while (pango_attr_iterator_next (iterator));
+	g_string_append (result, formatter (NULL));
+
+	g_free (text);
+	pango_attr_iterator_destroy (iterator);
+	pango_attr_list_unref (attrs);
+	return g_string_free (result, FALSE);
+}
+
+static gchar *
+field_to_output_text (const StardictEntryField *field, FormatterFunc formatter)
+{
+	const gchar *definition = field->data;
+	if (field->type == STARDICT_FIELD_MEANING)
+		return g_strdup (definition);
+	if (field->type == STARDICT_FIELD_PANGO)
+		return pango_to_output_text (definition, formatter);
+	if (field->type == STARDICT_FIELD_XDXF)
+	{
+		gchar *markup = xdxf_to_pango_markup_with_reduced_effort (definition);
+		gchar *result = pango_to_output_text (markup, formatter);
+		g_free (markup);
+		return result;
+	}
+	return NULL;
+}
+
+// --- Main --------------------------------------------------------------------
+
+static guint
+count_equal_chars (const gchar *a, const gchar *b)
+{
+	guint count = 0;
+	while (*a && *b)
+		if (*a++ == *b++)
+			count++;
+	return count;
+}
+
+static void
+do_dictionary (StardictDict *dict, const gchar *word, FormatterFunc formatter)
+{
+	gboolean found;
+	StardictIterator *iter = stardict_dict_search (dict, word, &found);
+	if (!found)
+		goto out;
+
+	// Default Stardict ordering is ASCII case-insensitive,
+	// which may be further exacerbated by our own collation feature.
+	// Try to find a better matching entry:
+
+	gint64 best_offset = stardict_iterator_get_offset (iter);
+	guint best_score = count_equal_chars
+		(stardict_iterator_get_word (iter), word);
+
+	while (TRUE)
+	{
+		stardict_iterator_next (iter);
+		if (!stardict_iterator_is_valid (iter))
+			break;
+
+		const gchar *iter_word = stardict_iterator_get_word (iter);
+		if (g_ascii_strcasecmp (iter_word, word))
+			break;
+
+		guint score = count_equal_chars (iter_word, word);
+		if (score > best_score)
+		{
+			best_offset = stardict_iterator_get_offset (iter);
+			best_score = score;
+		}
+	}
+
+	stardict_iterator_set_offset (iter, best_offset, FALSE);
+
+	StardictEntry *entry = stardict_iterator_get_entry (iter);
+	StardictInfo *info = stardict_dict_get_info (dict);
+	const GList *list = stardict_entry_get_fields (entry);
+	for (; list; list = list->next)
+	{
+		StardictEntryField *field = list->data;
+		gchar *definitions = field_to_output_text (field, formatter);
+		if (!definitions)
+			continue;
+
+		printf ("%s\t", info->book_name);
+		for (const gchar *p = definitions; *p; p++)
+		{
+			if (*p == '\\')
+				printf ("\\\\");
+			else if (*p == '\n')
+				printf ("\\n");
+			else
+				putchar (*p);
+		}
+		putchar ('\n');
+		g_free (definitions);
+	}
+	g_object_unref (entry);
+out:
+	g_object_unref (iter);
+}
+
+static FormatterFunc
+parse_options (int *argc, char ***argv)
+{
+	GError *error = NULL;
+	GOptionContext *ctx = g_option_context_new
+		("DICTIONARY.ifo... - query multiple dictionaries");
+
+	gboolean format_with_ansi = FALSE;
+	gboolean format_with_irc = FALSE;
+	GOptionEntry entries[] =
+	{
+		{ "ansi", 'a', 0, G_OPTION_ARG_NONE, &format_with_ansi,
+		  "Format with ANSI sequences", NULL },
+		{ "irc", 'i', 0, G_OPTION_ARG_NONE, &format_with_irc,
+		  "Format with IRC codes", NULL },
+		{ }
+	};
+
+	g_option_context_add_main_entries (ctx, entries, NULL);
+	if (!g_option_context_parse (ctx, argc, argv, &error))
+	{
+		g_printerr ("Error: option parsing failed: %s\n", error->message);
+		exit (EXIT_FAILURE);
+	}
+	if (*argc < 2)
+	{
+		g_printerr ("%s\n", g_option_context_get_help (ctx, TRUE, NULL));
+		exit (EXIT_FAILURE);
+	}
+	g_option_context_free (ctx);
+
+	if (format_with_ansi)
+		return pango_attrs_to_ansi;
+	if (format_with_irc)
+		return pango_attrs_to_irc;
+
+	return pango_attrs_ignore;
+}
+
+int
+main (int argc, char *argv[])
+{
+G_GNUC_BEGIN_IGNORE_DEPRECATIONS
+	if (glib_check_version (2, 36, 0))
+		g_type_init ();
+G_GNUC_END_IGNORE_DEPRECATIONS
+
+	FormatterFunc formatter = parse_options (&argc, &argv);
+
+	guint n_dicts = argc - 1;
+	StardictDict **dicts = g_alloca (sizeof *dicts * n_dicts);
+
+	guint i;
+	for (i = 1; i <= n_dicts; i++)
+	{
+		GError *error = NULL;
+		dicts[i - 1] = stardict_dict_new (argv[i], &error);
+		if (error)
+		{
+			g_printerr ("Error: opening dictionary `%s' failed: %s\n",
+				argv[i], error->message);
+			exit (EXIT_FAILURE);
+		}
+	}
+
+	gint c;
+	do
+	{
+		GString *s = g_string_new (NULL);
+		while ((c = getchar ()) != EOF && c != '\n')
+			if (c != '\r')
+				g_string_append_c (s, c);
+
+		if (s->len)
+			for (i = 0; i < n_dicts; i++)
+				do_dictionary (dicts[i], s->str, formatter);
+
+		printf ("\n");
+		fflush (NULL);
+		g_string_free (s, TRUE);
+	}
+	while (c != EOF);
+
+	for (i = 0; i < n_dicts; i++)
+		g_object_unref (dicts[i]);
+
+	return 0;
+}