githubmirror/libuser

lib/prompt.c
170 строк5 KB
Miloslav Trmač
Update FSF address
06 фев 2012, 19:12
06 фев 2012, 19:1281ae3fa
/* Copyright (C) 2000-2002 Red Hat, Inc.
 *
 * This is free software; you can redistribute it and/or modify it under 
 * the terms of the GNU Library General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program 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
 * General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public 
 * License along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.
 */

#include <config.h>
#include <libintl.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <termios.h>
#include <unistd.h>
#include "user.h"
#include "user_private.h"

/**
 * SECTION:prompt
 * @short_description: Sample prompter functions for use with the libuser
 * library.
 * @include: libuser/prompt.h
 *
 * prompt.h declares two predefined prompter functions which applications can
 * use instead of providing their own.  These should suffice for most
 * command-line applications.  Authors of graphical applications are encouraged
 * to supply graphical implementations.
 */

/**
 * lu_prompt_console:
 * @prompts: An array of #lu_prompt structures.
 * @count: The number of elements in the @prompts array.
 * @callback_data: Ignored.
 * @error: The location to store error information in the event of an error.
 *
 * Prompts the user using a text console.
 *
 * Returns: A #gboolean indicating success or failure.
 */
gboolean
lu_prompt_console(struct lu_prompt *prompts, int count, gpointer callback_data,
		  struct lu_error **error)
{
	int i, is_tty;

	(void)callback_data;
	LU_ERROR_CHECK(error);

	if (count > 0) {
		g_assert(prompts != NULL);
	}

	is_tty = isatty(fileno(stdin));
	for (i = 0; i < count; i++) {
		char buf[LINE_MAX], *p;
		struct termios otermios, ntermios;

		if (prompts[i].prompt) {
			g_print("%s",
				prompts[i].domain ?
			       	dgettext(prompts[i].domain, prompts[i].prompt) :
				prompts[i].prompt);
		}
		if (prompts[i].visible && prompts[i].default_value) {
			g_print(" [%s]: ", prompts[i].default_value);
		} else {
			g_print(": ");
		}

		prompts[i].value = NULL;
		prompts[i].free_value = NULL;

		if (prompts[i].visible == FALSE && is_tty) {
			if (tcgetattr(fileno(stdin), &otermios) == -1) {
				lu_error_new(error, lu_error_terminal,
					     _("error reading terminal attributes"));
				return FALSE;
			}
			ntermios = otermios;
			ntermios.c_lflag &= ~ECHO;
			if (tcsetattr(fileno(stdin), TCSADRAIN, &ntermios) == -1) {
				lu_error_new(error, lu_error_terminal,
					     _("error setting terminal attributes"));
				return FALSE;
			}
		}
		if (fgets(buf, sizeof(buf), stdin) == NULL) {
			lu_error_new(error, lu_error_terminal,
				     _("error reading from terminal"));
			return FALSE;
		}
		if (prompts[i].visible == FALSE && is_tty) {
			if (tcsetattr(fileno(stdin), TCSADRAIN, &otermios) == -1) {
				lu_error_new(error, lu_error_terminal,
					     _("error setting terminal attributes"));
				return FALSE;
			}
			g_print("\n");
		}

		p = strchr(buf, '\r');
		if (p != NULL)
			*p = '\0';
		p = strchr(buf, '\n');
		if (p != NULL)
			*p = '\0';

		prompts[i].value = (strlen(buf) > 0) ?
			g_strdup(buf) :
			(prompts[i].default_value ?
			 g_strdup(prompts[i].default_value) :
			 g_strdup(""));
		prompts[i].free_value = (void *) g_free;
	}
	return TRUE;
}

/**
 * lu_prompt_console_quiet:
 * @prompts: An array of #lu_prompt structures.
 * @count: The number of elements in the @prompts array.
 * @callback_data: Ignored.
 * @error: The location to store error information in the event of an error.
 *
 * Prompts the user using a text console.  Unlike lu_prompt_console(), this
 * function will not prompt users for a question for which the calling
 * application or module supplies a default, and will simply return the
 * default.
 *
 * Returns: A #gboolean indicating success or failure.
 */
gboolean
lu_prompt_console_quiet(struct lu_prompt * prompts, int count,
			gpointer callback_data, struct lu_error ** error)
{
	int i;
	gboolean ret = TRUE;

	LU_ERROR_CHECK(error);

	if (count > 0) {
		g_return_val_if_fail(prompts != NULL, FALSE);
	}

	for (i = 0; (i < count) && ret; i++) {
		if (prompts[i].default_value) {
			prompts[i].value =
			    g_strdup(prompts[i].default_value);
			prompts[i].free_value = (void *) g_free;
		} else {
			ret = ret &&
			      lu_prompt_console(&prompts[i], 1, callback_data,
					        error);
		}
	}

	return ret;
}