xref: /freebsd-src/contrib/bsddialog/lib/timebox.c (revision 8c4f402881b3a926f1bafdf275b015c6d76a31b2)

/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2021 Alfonso Sabato Siciliano
 *
 * 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/param.h>

#ifdef PORTNCURSES
#include <ncurses/ncurses.h>
#else
#include <ncurses.h>
#endif

#include <ctype.h>
#include <string.h>

#include "bsddialog.h"
#include "lib_util.h"
#include "bsddialog_theme.h"

#define MINWDATE 25 /* 23 wins + 2 VBORDERS */
#define MINWTIME 16 /*14 wins + 2 VBORDERS */
#define MINHEIGHT 8 /* 2 for text */

/* "Time": timebox - datebox */

extern struct bsddialog_theme t;

static int
datetime_autosize(struct bsddialog_conf *conf, int rows, int cols, int *h,
    int *w, int minw, char *text, struct buttons bs)
{
	int maxword, maxline, nlines, line;

	if (get_text_properties(conf, text, &maxword, &maxline, &nlines) != 0)
		return BSDDIALOG_ERROR;

	if (cols == BSDDIALOG_AUTOSIZE) {
		*w = VBORDERS;
		/* buttons size */
		*w += bs.nbuttons * bs.sizebutton;
		*w += bs.nbuttons > 0 ? (bs.nbuttons-1) * t.button.space : 0;
		/* text size */
		line = maxline + VBORDERS + t.text.hmargin * 2;
		line = MAX(line, (int) (maxword + VBORDERS + t.text.hmargin * 2));
		*w = MAX(*w, line);
		/* date windows */
		*w = MAX(*w, minw);
		/* conf.auto_minwidth */
		*w = MAX(*w, (int)conf->auto_minwidth);
		/* avoid terminal overflow */
		*w = MIN(*w, widget_max_width(conf) -1);
	}

	if (rows == BSDDIALOG_AUTOSIZE) {
		*h = MINHEIGHT;
		if (maxword > 0)
			*h += MAX(nlines, (int)(*w / GET_ASPECT_RATIO(conf)));
		/* conf.auto_minheight */
		*h = MAX(*h, (int)conf->auto_minheight);
		/* avoid terminal overflow */
		*h = MIN(*h, widget_max_height(conf) -1);
	}

	return 0;
}

static int
datetime_checksize(int rows, int cols, char *text, int minw, struct buttons bs)
{
	int mincols;

	mincols = VBORDERS;
	mincols += bs.nbuttons * bs.sizebutton;
	mincols += bs.nbuttons > 0 ? (bs.nbuttons-1) * t.button.space : 0;
	mincols = MAX(minw, mincols);

	if (cols < mincols)
		RETURN_ERROR("Few cols for this timebox/datebox");

	if (rows < MINHEIGHT + (strlen(text) > 0 ? 1 : 0))
		RETURN_ERROR("Few rows for this timebox/datebox");

	return 0;
}

int bsddialog_timebox(struct bsddialog_conf *conf, char* text, int rows, int cols,
    unsigned int *hh, unsigned int *mm, unsigned int *ss)
{
	WINDOW *widget, *textpad, *shadow;
	int i, input, output, y, x, h, w, sel, htextpad;
	struct buttons bs;
	bool loop;
	struct myclockstruct {
		unsigned int max;
		unsigned int value;
		WINDOW *win;
	};

	if (hh == NULL || mm == NULL || ss == NULL)
		RETURN_ERROR("hh / mm / ss cannot be NULL");

	struct myclockstruct c[3] = {
		{23, *hh, NULL},
		{59, *mm, NULL},
		{59, *ss, NULL}
	};

	for (i = 0 ; i < 3; i++) {
		if (c[i].value > c[i].max)
			c[i].value = c[i].max;
	}

	get_buttons(conf, &bs, BUTTONLABEL(ok_label), BUTTONLABEL(extra_label),
	    BUTTONLABEL(cancel_label), BUTTONLABEL(help_label));

	if (set_widget_size(conf, rows, cols, &h, &w) != 0)
		return BSDDIALOG_ERROR;
	if (datetime_autosize(conf, rows, cols, &h, &w, MINWTIME, text, bs) != 0)
		return BSDDIALOG_ERROR;
	if (datetime_checksize(h, w, text, MINWTIME, bs) != 0)
		return BSDDIALOG_ERROR;
	if (set_widget_position(conf, &y, &x, h, w) != 0)
		return BSDDIALOG_ERROR;

	if (new_widget_withtextpad(conf, &shadow, &widget, y, x, h, w, RAISED,
	    &textpad, &htextpad, text, true) != 0)
		return BSDDIALOG_ERROR;

	draw_buttons(widget, h-2, w, bs, true);

	wrefresh(widget);

	prefresh(textpad, 0, 0, y+1, x+2, y+h-7, x+w-2);

	c[0].win = new_boxed_window(conf, y + h - 6, x + w/2 - 7, 3, 4, LOWERED);
	mvwaddch(widget, h - 5, w/2 - 3, ':');
	c[1].win = new_boxed_window(conf, y + h - 6, x + w/2 - 2, 3, 4, LOWERED);
	mvwaddch(widget, h - 5, w/2 + 2, ':');
	c[2].win = new_boxed_window(conf, y + h - 6, x + w/2 + 3, 3, 4, LOWERED);

	wrefresh(widget);

	sel = 0;
	curs_set(2);
	loop = true;
	while(loop) {
		for (i=0; i<3; i++) {
			mvwprintw(c[i].win, 1, 1, "%2d", c[i].value);
			wrefresh(c[i].win);
		}
		wmove(c[sel].win, 1, 2);
		wrefresh(c[sel].win);

		input = getch();
		switch(input) {
		case KEY_ENTER:
		case 10: /* Enter */
			output = bs.value[bs.curr];
			if (output == BSDDIALOG_OK) {
				*hh = c[0].value;
				*mm = c[1].value;
				*ss = c[2].value;
			}
			loop = false;
			break;
		case 27: /* Esc */
			output = BSDDIALOG_ESC;
			loop = false;
			break;
		case '\t': /* TAB */
			sel = (sel + 1) % 3;
			break;
		case KEY_LEFT:
			if (bs.curr > 0) {
				bs.curr--;
				draw_buttons(widget, h-2, w, bs, true);
				wrefresh(widget);
			}
			break;
		case KEY_RIGHT:
			if (bs.curr < (int) bs.nbuttons - 1) {
				bs.curr++;
				draw_buttons(widget, h-2, w, bs, true);
				wrefresh(widget);
			}
			break;
		case KEY_UP:
			c[sel].value = c[sel].value < c[sel].max ? c[sel].value + 1 : 0;
			break;
		case KEY_DOWN:
			c[sel].value = c[sel].value > 0 ? c[sel].value - 1 : c[sel].max;
			break;
		case KEY_F(1):
			if (conf->f1_file == NULL && conf->f1_message == NULL)
				break;
			curs_set(0);
			if (f1help(conf) != 0)
				return BSDDIALOG_ERROR;
			curs_set(2);
			/* No break! the terminal size can change */
		case KEY_RESIZE:
			hide_widget(y, x, h, w,conf->shadow);

			/*
			 * Unnecessary, but, when the columns decrease the
			 * following "refresh" seem not work
			 */
			refresh();

			if (set_widget_size(conf, rows, cols, &h, &w) != 0)
				return BSDDIALOG_ERROR;
			if (datetime_autosize(conf, rows, cols, &h, &w, MINWTIME, text, bs) != 0)
				return BSDDIALOG_ERROR;
			if (datetime_checksize(h, w, text, MINWTIME, bs) != 0)
				return BSDDIALOG_ERROR;
			if (set_widget_position(conf, &y, &x, h, w) != 0)
				return BSDDIALOG_ERROR;

			wclear(shadow);
			mvwin(shadow, y + t.shadow.h, x + t.shadow.w);
			wresize(shadow, h, w);

			wclear(widget);
			mvwin(widget, y, x);
			wresize(widget, h, w);

			htextpad = 1;
			wclear(textpad);
			wresize(textpad, 1, w - HBORDERS - t.text.hmargin * 2);

			if(update_widget_withtextpad(conf, shadow, widget, h, w,
			    RAISED, textpad, &htextpad, text, true) != 0)
				return BSDDIALOG_ERROR;

			mvwaddch(widget, h - 5, w/2 - 3, ':');
			mvwaddch(widget, h - 5, w/2 + 2, ':');

			draw_buttons(widget, h-2, w, bs, true);

			wrefresh(widget);

			prefresh(textpad, 0, 0, y+1, x+2, y+h-7, x+w-2);

			wclear(c[0].win);
			mvwin(c[0].win, y + h - 6, x + w/2 - 7);
			draw_borders(conf, c[0].win, 3, 4, LOWERED);
			wrefresh(c[0].win);

			wclear(c[1].win);
			mvwin(c[1].win, y + h - 6, x + w/2 - 2);
			draw_borders(conf, c[1].win, 3, 4, LOWERED);
			wrefresh(c[1].win);

			wclear(c[2].win);
			mvwin(c[2].win, y + h - 6, x + w/2 + 3);
			draw_borders(conf, c[2].win, 3, 4, LOWERED);
			wrefresh(c[2].win);

			/* Important to avoid grey lines expanding screen */
			refresh();
			break;
		default:
			for (i = 0; i < (int) bs.nbuttons; i++)
				if (tolower(input) == tolower((bs.label[i])[0])) {
					output = bs.value[i];
					loop = false;
			}
		}
	}

	curs_set(0);

	for (i=0; i<3; i++)
		delwin(c[i].win);
	end_widget_withtextpad(conf, widget, h, w, textpad, shadow);

	return output;
}

int
bsddialog_datebox(struct bsddialog_conf *conf, char* text, int rows, int cols,
    unsigned int *yy, unsigned int *mm, unsigned int *dd)
{
	WINDOW *widget, *textpad, *shadow;
	int i, input, output, y, x, h, w, sel, htextpad;
	struct buttons bs;
	bool loop;
	struct calendar {
		int max;
		int value;
		WINDOW *win;
		unsigned int x;
	};
	struct month {
		char *name;
		unsigned int days;
	};

	if (yy == NULL || mm == NULL || dd == NULL)
		RETURN_ERROR("yy / mm / dd cannot be NULL");

	struct calendar c[3] = {
		{9999, *yy, NULL, 4 },
		{12,   *mm, NULL, 9 },
		{31,   *dd, NULL, 2 }
	};

	struct month m[12] = {
		{ "January", 31 }, { "February", 28 }, { "March",     31 },
		{ "April",   30 }, { "May",      31 }, { "June",      30 },
		{ "July",    31 }, { "August",   31 }, { "September", 30 },
		{ "October", 31 }, { "November", 30 }, { "December",  31 }
	};

#define ISLEAF(year) ((year % 4 == 0 && year % 100 != 0) || year % 400 == 0)

	for (i = 0 ; i < 3; i++) {
		if (c[i].value > c[i].max)
			c[i].value = c[i].max;
		if (c[i].value < 1)
			c[i].value = 1;
	}
	c[2].max = m[c[1].value -1].days;
	if (c[1].value == 2 && ISLEAF(c[0].value))
		c[2].max = 29;
	if (c[2].value > c[2].max)
		c[2].value = c[2].max;

	get_buttons(conf, &bs, BUTTONLABEL(ok_label), BUTTONLABEL(extra_label),
	    BUTTONLABEL(cancel_label), BUTTONLABEL(help_label));

	if (set_widget_size(conf, rows, cols, &h, &w) != 0)
		return BSDDIALOG_ERROR;
	if (datetime_autosize(conf, rows, cols, &h, &w, MINWDATE, text, bs) != 0)
		return BSDDIALOG_ERROR;
	if (datetime_checksize(h, w, text, MINWDATE, bs) != 0)
		return BSDDIALOG_ERROR;
	if (set_widget_position(conf, &y, &x, h, w) != 0)
		return BSDDIALOG_ERROR;

	if (new_widget_withtextpad(conf, &shadow, &widget, y, x, h, w, RAISED,
	    &textpad, &htextpad, text, true) != 0)
		return BSDDIALOG_ERROR;

	draw_buttons(widget, h-2, w, bs, true);

	wrefresh(widget);

	prefresh(textpad, 0, 0, y+1, x+2, y+h-7, x+w-2);

	c[0].win = new_boxed_window(conf, y + h - 6, x + w/2 - 11, 3, 6, LOWERED);
	mvwaddch(widget, h - 5, w/2 - 5, '/');
	c[1].win = new_boxed_window(conf, y + h - 6, x + w/2 - 4, 3, 11, LOWERED);
	mvwaddch(widget, h - 5, w/2 + 7, '/');
	c[2].win = new_boxed_window(conf, y + h - 6, x + w/2 + 8, 3, 4, LOWERED);

	wrefresh(widget);

	sel = 2;
	curs_set(2);
	loop = true;
	while(loop) {
		mvwprintw(c[0].win, 1, 1, "%4d", c[0].value);
		mvwprintw(c[1].win, 1, 1, "%9s", m[c[1].value-1].name);
		mvwprintw(c[2].win, 1, 1, "%2d", c[2].value);
		for (i=0; i<3; i++) {
			wrefresh(c[i].win);
		}
		wmove(c[sel].win, 1, c[sel].x);
		wrefresh(c[sel].win);

		input = getch();
		switch(input) {
		case KEY_ENTER:
		case 10: /* Enter */
			output = bs.value[bs.curr];
			if (output == BSDDIALOG_OK) {
				*yy = c[0].value;
				*mm = c[1].value;
				*dd = c[2].value;
			}
			loop = false;
			break;
		case 27: /* Esc */
			output = BSDDIALOG_ESC;
			loop = false;
			break;
		case '\t': /* TAB */
			sel = (sel + 1) % 3;
			break;
		case KEY_LEFT:
			if (bs.curr > 0) {
				bs.curr--;
				draw_buttons(widget, h-2, w, bs, true);
				wrefresh(widget);
			}
			break;
		case KEY_RIGHT:
			if (bs.curr < (int) bs.nbuttons - 1) {
				bs.curr++;
				draw_buttons(widget, h-2, w, bs, true);
				wrefresh(widget);
			}
			break;
		case KEY_UP:
			c[sel].value = c[sel].value > 1 ? c[sel].value - 1 : c[sel].max ;
			/* if mount change */
			c[2].max = m[c[1].value -1].days;
			/* if year change */
			if (c[1].value == 2 && ISLEAF(c[0].value))
				c[2].max = 29;
			/* set new day */
			if (c[2].value > c[2].max)
				c[2].value = c[2].max;
			break;
		case KEY_DOWN:
			c[sel].value = c[sel].value < c[sel].max ? c[sel].value + 1 : 1;
			/* if mount change */
			c[2].max = m[c[1].value -1].days;
			/* if year change */
			if (c[1].value == 2 && ISLEAF(c[0].value))
				c[2].max = 29;
			/* set new day */
			if (c[2].value > c[2].max)
				c[2].value = c[2].max;
			break;
		case KEY_F(1):
			if (conf->f1_file == NULL && conf->f1_message == NULL)
				break;
			curs_set(0);
			if (f1help(conf) != 0)
				return BSDDIALOG_ERROR;
			curs_set(2);
			/* No break! the terminal size can change */
		case KEY_RESIZE:
			hide_widget(y, x, h, w,conf->shadow);

			/*
			 * Unnecessary, but, when the columns decrease the
			 * following "refresh" seem not work
			 */
			refresh();

			if (set_widget_size(conf, rows, cols, &h, &w) != 0)
				return BSDDIALOG_ERROR;
			if (datetime_autosize(conf, rows, cols, &h, &w, MINWDATE, text, bs) != 0)
				return BSDDIALOG_ERROR;
			if (datetime_checksize(h, w, text, MINWDATE, bs) != 0)
				return BSDDIALOG_ERROR;
			if (set_widget_position(conf, &y, &x, h, w) != 0)
				return BSDDIALOG_ERROR;

			wclear(shadow);
			mvwin(shadow, y + t.shadow.h, x + t.shadow.w);
			wresize(shadow, h, w);

			wclear(widget);
			mvwin(widget, y, x);
			wresize(widget, h, w);

			htextpad = 1;
			wclear(textpad);
			wresize(textpad, 1, w - HBORDERS - t.text.hmargin * 2);

			if(update_widget_withtextpad(conf, shadow, widget, h, w,
			    RAISED, textpad, &htextpad, text, true) != 0)
				return BSDDIALOG_ERROR;

			mvwaddch(widget, h - 5, w/2 - 5, '/');
			mvwaddch(widget, h - 5, w/2 + 7, '/');

			draw_buttons(widget, h-2, w, bs, true);

			wrefresh(widget);

			prefresh(textpad, 0, 0, y+1, x+2, y+h-7, x+w-2);

			wclear(c[0].win);
			mvwin(c[0].win, y + h - 6, x + w/2 - 11);
			draw_borders(conf, c[0].win, 3, 6, LOWERED);
			wrefresh(c[0].win);

			wclear(c[1].win);
			mvwin(c[1].win, y + h - 6, x + w/2 - 4);
			draw_borders(conf, c[1].win, 3, 11, LOWERED);
			wrefresh(c[1].win);

			wclear(c[2].win);
			mvwin(c[2].win, y + h - 6, x + w/2 + 8);
			draw_borders(conf, c[2].win, 3, 4, LOWERED);
			wrefresh(c[2].win);

			/* Important to avoid grey lines expanding screen */
			refresh();
			break;
		default:
			for (i = 0; i < (int) bs.nbuttons; i++)
				if (tolower(input) == tolower((bs.label[i])[0])) {
					output = bs.value[i];
					loop = false;
			}
		}
	}

	curs_set(0);

	for (i=0; i<3; i++)
		delwin(c[i].win);
	end_widget_withtextpad(conf, widget, h, w, textpad, shadow);

	return output;
}