/*	$NetBSD: gettext.c,v 1.3 2015/07/12 11:40:52 christos Exp $	*/

/*-
 * Copyright (c) 2015 William Orr <will@worrbase.com>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */
#include <sys/cdefs.h>
__RCSID("$NetBSD: gettext.c,v 1.3 2015/07/12 11:40:52 christos Exp $");

#include <err.h>
#include <errno.h>
#include <getopt.h>
#include <libintl.h>
#include <locale.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <util.h>

static struct option longopts[] = {
	{ "help",       no_argument,        NULL,   'h' },
	{ "domain",	required_argument,  NULL,   'd' },
	{ NULL,         0,                  NULL,   '\0' },
};

static __dead void
usage(int exit_status)
{

	fprintf(stderr, "Usage: %s [-ehn] [[<textdomain>] <msgid>]\n",
	    getprogname());
	fprintf(stderr, "Usage: %s [-ehn] -d <textdomain> <msgid>\n",
	    getprogname());
	fprintf(stderr, "Usage: %s -s [<msgid>]...\n", getprogname());
	exit(exit_status);
}

static bool
expand(char *str)
{
	char *fp, *sp, ch, pl;
	bool nflag = false;

	for (fp = str, sp = str; *fp != 0;) {
		if (*fp == '\\') {
			switch (*++fp) {
			case 'a':
				*sp++ = '\a';
				fp++;
				break;
			case 'b':
				*sp++ = '\b';
				fp++;
				break;
			case 'c':
				nflag = true;
				fp++;
				break;
			case 'f':
				*sp++ = '\f';
				fp++;
				break;
			case 'n':
				*sp++ = '\n';
				fp++;
				break;
			case 'r':
				*sp++ = '\r';
				fp++;
				break;
			case 't':
				*sp++ = '\t';
				fp++;
				break;
			case 'v':
				*sp++ = '\v';
				fp++;
				break;
			case '\\':
				*sp++ = '\\';
				fp++;
				break;
			case '0':
			case '1':
			case '2':
			case '3':
			case '4':
			case '5':
			case '6':
			case '7':
				ch = *fp++ - '0';
				pl = 0;
				while (*fp >= '0' && *fp <= '7' && pl < 2) {
					ch *= 8;
					ch += *fp++ - '0';
					pl++;
				}

				*sp++ = ch;
				break;
			default:
				*sp++ = '\\';
				break;
			}
			continue;
		}
		*sp++ = *fp++;
	}

	*sp = '\0';
	return nflag;
}

int
main(int argc, char **argv)
{
	char *msgdomain = NULL;
	char *msgdomaindir = NULL;
	char *translation = NULL;
	char *s;
	bool eflag = false;
	bool sflag = false;
	bool nflag = false;
	int ch;

	setlocale(LC_ALL, "");
	setprogname(argv[0]);

	while ((ch = getopt_long(argc, argv, "d:eEhnsV", longopts, NULL)) != -1)
	{
		switch (ch) {
		case 'd':
			msgdomain = estrdup(optarg);
			break;
		case 'E':
			/* GNU gettext compat */
			break;
		case 'e':
			eflag = true;
			break;
		case 'V':
		case 'h':
			free(msgdomain);
			usage(EXIT_SUCCESS);
			/* NOTREACHED */
		case 'n':
			nflag = true;
			break;
		case 's':
			sflag = true;
			break;
		default:
			free(msgdomain);
			usage(EXIT_FAILURE);
			/* NOTREACHED */
		}
	}
	argc -= optind;
	argv += optind;

	if (argc == 0) {
		free(msgdomain);
		warnx("missing msgid");
		usage(EXIT_FAILURE);
	}

	/* msgdomain can be passed as optional arg iff -s is not passed */
	if (!sflag) {
		if (argc == 2) {
			free(msgdomain);
			msgdomain = estrdup(argv[0]);

			argc -= 1;
			argv += 1;
		} else if (argc > 2) {
			warnx("too many arguments");
			usage(EXIT_FAILURE);
		}
	}

	/* msgdomain can be passed as env var */
	if (msgdomain == NULL) {
		if ((s = getenv("TEXTDOMAIN")) != NULL)
			msgdomain = estrdup(s);
	}

	if (msgdomain != NULL) {
		if ((s = getenv("TEXTDOMAINDIR")) != NULL)
			msgdomaindir = estrdup(s);
		if (msgdomaindir)
			bindtextdomain(msgdomain, msgdomaindir);
	}

	do {
		if (eflag)
			nflag |= expand(*argv);

		translation = dgettext(msgdomain, argv[0]);
		printf("%s", translation);

		argc--;
		argv++;
		if (argc)
			printf(" ");
	} while (sflag && argc != 0);

	if (sflag && !nflag)
		printf("\n");

	free(msgdomain);
	free(msgdomaindir);

	return EXIT_SUCCESS;
}