ncal.c revision 300473
1/*-
2 * Copyright (c) 1997 Wolfgang Helbig
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 */
26
27#include <sys/cdefs.h>
28__FBSDID("$FreeBSD: stable/10/usr.bin/ncal/ncal.c 300473 2016-05-23 05:45:52Z truckman $");
29
30#include <calendar.h>
31#include <ctype.h>
32#include <err.h>
33#include <langinfo.h>
34#include <libgen.h>
35#include <locale.h>
36#include <stdio.h>
37#include <stdlib.h>
38#include <string.h>
39#include <sysexits.h>
40#include <time.h>
41#include <unistd.h>
42#include <wchar.h>
43#include <wctype.h>
44#include <term.h>
45#undef lines			/* term.h defines this */
46
47/* Width of one month with backward compatibility and in regular mode*/
48#define MONTH_WIDTH_B_J 27
49#define MONTH_WIDTH_B 20
50
51#define MONTH_WIDTH_R_J 24
52#define MONTH_WIDTH_R 18
53
54#define MAX_WIDTH 64
55
56typedef struct date date;
57
58struct monthlines {
59	wchar_t name[MAX_WIDTH + 1];
60	char lines[7][MAX_WIDTH + 1];
61	char weeks[MAX_WIDTH + 1];
62	unsigned int extralen[7];
63};
64
65struct weekdays {
66	wchar_t names[7][4];
67};
68
69/* The switches from Julian to Gregorian in some countries */
70static struct djswitch {
71	const char *cc;	/* Country code according to ISO 3166 */
72	const char *nm;	/* Name of country */
73	date dt;	/* Last day of Julian calendar */
74} switches[] = {
75	{"AL", "Albania",       {1912, 11, 30}},
76	{"AT", "Austria",       {1583, 10,  5}},
77	{"AU", "Australia",     {1752,  9,  2}},
78	{"BE", "Belgium",       {1582, 12, 14}},
79	{"BG", "Bulgaria",      {1916,  3, 18}},
80	{"CA", "Canada",        {1752,  9,  2}},
81	{"CH", "Switzerland",   {1655,  2, 28}},
82	{"CN", "China",         {1911, 12, 18}},
83	{"CZ", "Czech Republic",{1584,  1,  6}},
84	{"DE", "Germany",       {1700,  2, 18}},
85	{"DK", "Denmark",       {1700,  2, 18}},
86	{"ES", "Spain",         {1582, 10,  4}},
87	{"FI", "Finland",       {1753,  2, 17}},
88	{"FR", "France",        {1582, 12,  9}},
89	{"GB", "United Kingdom",{1752,  9,  2}},
90	{"GR", "Greece",        {1924,  3,  9}},
91	{"HU", "Hungary",       {1587, 10, 21}},
92	{"IS", "Iceland",       {1700, 11, 16}},
93	{"IT", "Italy",         {1582, 10,  4}},
94	{"JP", "Japan",         {1918, 12, 18}},
95	{"LI", "Lithuania",     {1918,  2,  1}},
96	{"LN", "Latin",         {9999, 05, 31}},
97	{"LU", "Luxembourg",    {1582, 12, 14}},
98	{"LV", "Latvia",        {1918,  2,  1}},
99	{"NL", "Netherlands",   {1582, 12, 14}},
100	{"NO", "Norway",        {1700,  2, 18}},
101	{"PL", "Poland",        {1582, 10,  4}},
102	{"PT", "Portugal",      {1582, 10,  4}},
103	{"RO", "Romania",       {1919,  3, 31}},
104	{"RU", "Russia",        {1918,  1, 31}},
105	{"SI", "Slovenia",      {1919,  3,  4}},
106	{"SE", "Sweden",        {1753,  2, 17}},
107	{"TR", "Turkey",        {1926, 12, 18}},
108	{"US", "United States", {1752,  9,  2}},
109	{"YU", "Yugoslavia",    {1919,  3,  4}}
110};
111
112static struct djswitch *dftswitch =
113    switches + sizeof(switches) / sizeof(struct djswitch) - 2;
114    /* default switch (should be "US") */
115
116/* Table used to print day of month and week numbers */
117static char daystr[] = "     1  2  3  4  5  6  7  8  9 10 11 12 13 14 15"
118		       " 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31"
119		       " 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47"
120		       " 48 49 50 51 52 53";
121
122/* Table used to print day of year and week numbers */
123static char jdaystr[] = "       1   2   3   4   5   6   7   8   9"
124			"  10  11  12  13  14  15  16  17  18  19"
125			"  20  21  22  23  24  25  26  27  28  29"
126			"  30  31  32  33  34  35  36  37  38  39"
127			"  40  41  42  43  44  45  46  47  48  49"
128			"  50  51  52  53  54  55  56  57  58  59"
129			"  60  61  62  63  64  65  66  67  68  69"
130			"  70  71  72  73  74  75  76  77  78  79"
131			"  80  81  82  83  84  85  86  87  88  89"
132			"  90  91  92  93  94  95  96  97  98  99"
133			" 100 101 102 103 104 105 106 107 108 109"
134			" 110 111 112 113 114 115 116 117 118 119"
135			" 120 121 122 123 124 125 126 127 128 129"
136			" 130 131 132 133 134 135 136 137 138 139"
137			" 140 141 142 143 144 145 146 147 148 149"
138			" 150 151 152 153 154 155 156 157 158 159"
139			" 160 161 162 163 164 165 166 167 168 169"
140			" 170 171 172 173 174 175 176 177 178 179"
141			" 180 181 182 183 184 185 186 187 188 189"
142			" 190 191 192 193 194 195 196 197 198 199"
143			" 200 201 202 203 204 205 206 207 208 209"
144			" 210 211 212 213 214 215 216 217 218 219"
145			" 220 221 222 223 224 225 226 227 228 229"
146			" 230 231 232 233 234 235 236 237 238 239"
147			" 240 241 242 243 244 245 246 247 248 249"
148			" 250 251 252 253 254 255 256 257 258 259"
149			" 260 261 262 263 264 265 266 267 268 269"
150			" 270 271 272 273 274 275 276 277 278 279"
151			" 280 281 282 283 284 285 286 287 288 289"
152			" 290 291 292 293 294 295 296 297 298 299"
153			" 300 301 302 303 304 305 306 307 308 309"
154			" 310 311 312 313 314 315 316 317 318 319"
155			" 320 321 322 323 324 325 326 327 328 329"
156			" 330 331 332 333 334 335 336 337 338 339"
157			" 340 341 342 343 344 345 346 347 348 349"
158			" 350 351 352 353 354 355 356 357 358 359"
159			" 360 361 362 363 364 365 366";
160
161static int flag_nohighlight;	/* user doesn't want a highlighted today */
162static int flag_weeks;		/* user wants number of week */
163static int nswitch;		/* user defined switch date */
164static int nswitchb;		/* switch date for backward compatibility */
165static int highlightdate;
166
167static char	*center(char *s, char *t, int w);
168static wchar_t *wcenter(wchar_t *s, wchar_t *t, int w);
169static int	firstday(int y, int m);
170static void	highlight(char *dst, char *src, int len, int *extraletters);
171static void	mkmonthr(int year, int month, int jd_flag,
172    struct monthlines * monthl);
173static void	mkmonthb(int year, int month, int jd_flag,
174    struct monthlines * monthl);
175static void	mkweekdays(struct weekdays * wds);
176static void	monthranger(int year, int m, int jd_flag,
177    int before, int after);
178static void	monthrangeb(int year, int m, int jd_flag,
179    int before, int after);
180static int	parsemonth(const char *s, int *m, int *y);
181static void	printcc(void);
182static void	printeaster(int year, int julian, int orthodox);
183static date	*sdater(int ndays, struct date * d);
184static date	*sdateb(int ndays, struct date * d);
185static int	sndaysr(struct date * d);
186static int	sndaysb(struct date * d);
187static void	usage(void);
188
189int
190main(int argc, char *argv[])
191{
192	struct  djswitch *p, *q;	/* to search user defined switch date */
193	date	never = {10000, 1, 1};	/* outside valid range of dates */
194	date	ukswitch = {1752, 9, 2};/* switch date for Great Britain */
195	date	dt;
196	int     ch;			/* holds the option character */
197	int     m = 0;			/* month */
198	int	y = 0;			/* year */
199	int     flag_backward = 0;	/* user called cal--backward compat. */
200	int     flag_wholeyear = 0;	/* user wants the whole year */
201	int	flag_julian_cal = 0;	/* user wants Julian Calendar */
202	int     flag_julian_day = 0;	/* user wants the Julian day numbers */
203	int	flag_orthodox = 0;	/* user wants Orthodox easter */
204	int	flag_easter = 0;	/* user wants easter date */
205	int	flag_3months = 0;	/* user wants 3 month display (-3) */
206	int	flag_after = 0;		/* user wants to see months after */
207	int	flag_before = 0;	/* user wants to see months before */
208	int	flag_specifiedmonth = 0;/* user wants to see this month (-m) */
209	int	flag_givenmonth = 0;	/* user has specified month [n] */
210	int	flag_givenyear = 0;	/* user has specified year [n] */
211	char	*cp;			/* character pointer */
212	char	*flag_today = NULL;	/* debug: use date as being today */
213	char	*flag_month = NULL;	/* requested month as string */
214	char	*flag_highlightdate = NULL; /* debug: date to highlight */
215	int	before, after;
216	const char    *locale;		/* locale to get country code */
217
218	flag_nohighlight = 0;
219	flag_weeks = 0;
220
221	/*
222	 * Use locale to determine the country code,
223	 * and use the country code to determine the default
224	 * switchdate and date format from the switches table.
225	 */
226	if (setlocale(LC_ALL, "") == NULL)
227		warn("setlocale");
228	locale = setlocale(LC_TIME, NULL);
229	if (locale == NULL ||
230	    strcmp(locale, "C") == 0 ||
231	    strcmp(locale, "POSIX") == 0 ||
232	    strcmp(locale, "ASCII") == 0 ||
233	    strcmp(locale, "US-ASCII") == 0)
234		locale = "_US";
235	q = switches + sizeof(switches) / sizeof(struct djswitch);
236	for (p = switches; p != q; p++)
237		if ((cp = strstr(locale, p->cc)) != NULL && *(cp - 1) == '_')
238			break;
239	if (p == q) {
240		nswitch = ndaysj(&dftswitch->dt);
241	} else {
242		nswitch = ndaysj(&p->dt);
243		dftswitch = p;
244	}
245
246
247	/*
248	 * Get the filename portion of argv[0] and set flag_backward if
249	 * this program is called "cal".
250	 */
251	if (strncmp(basename(argv[0]), "cal", strlen("cal")) == 0)
252		flag_backward = 1;
253
254	/* Set the switch date to United Kingdom if backwards compatible */
255	if (flag_backward)
256		nswitchb = ndaysj(&ukswitch);
257
258	before = after = -1;
259
260	while ((ch = getopt(argc, argv, "3A:B:Cd:eH:hjJm:Nops:wy")) != -1)
261		switch (ch) {
262		case '3':
263			flag_3months = 1;
264			break;
265		case 'A':
266			if (flag_after > 0)
267				errx(EX_USAGE, "Double -A specified");
268			flag_after = strtol(optarg, NULL, 10);
269			if (flag_after <= 0)
270				errx(EX_USAGE,
271				    "Argument to -A must be positive");
272			break;
273		case 'B':
274			if (flag_before > 0)
275				errx(EX_USAGE, "Double -A specified");
276			flag_before = strtol(optarg, NULL, 10);
277			if (flag_before <= 0)
278				errx(EX_USAGE,
279				    "Argument to -B must be positive");
280			break;
281		case 'J':
282			if (flag_backward)
283				usage();
284			nswitch = ndaysj(&never);
285			flag_julian_cal = 1;
286			break;
287		case 'C':
288			flag_backward = 1;
289			break;
290		case 'N':
291			flag_backward = 0;
292			break;
293		case 'd':
294			flag_today = optarg;
295			break;
296		case 'H':
297			flag_highlightdate = optarg;
298			break;
299		case 'h':
300			flag_nohighlight = 1;
301			break;
302		case 'e':
303			if (flag_backward)
304				usage();
305			flag_easter = 1;
306			break;
307		case 'j':
308			flag_julian_day = 1;
309			break;
310		case 'm':
311			if (flag_specifiedmonth)
312				errx(EX_USAGE, "Double -m specified");
313			flag_month = optarg;
314			flag_specifiedmonth = 1;
315			break;
316		case 'o':
317			if (flag_backward)
318				usage();
319			flag_orthodox = 1;
320			flag_easter = 1;
321			break;
322		case 'p':
323			if (flag_backward)
324				usage();
325			printcc();
326			return (0);
327			break;
328		case 's':
329			if (flag_backward)
330				usage();
331			q = switches +
332			    sizeof(switches) / sizeof(struct djswitch);
333			for (p = switches;
334			     p != q && strcmp(p->cc, optarg) != 0; p++)
335				;
336			if (p == q)
337				errx(EX_USAGE,
338				    "%s: invalid country code", optarg);
339			nswitch = ndaysj(&(p->dt));
340			break;
341		case 'w':
342			if (flag_backward)
343				usage();
344			flag_weeks = 1;
345			break;
346		case 'y':
347			flag_wholeyear = 1;
348			break;
349		default:
350			usage();
351		}
352
353	argc -= optind;
354	argv += optind;
355
356	switch (argc) {
357	case 2:
358		if (flag_easter)
359			usage();
360		flag_month = *argv++;
361		flag_givenmonth = 1;
362		m = strtol(flag_month, NULL, 10);
363		/* FALLTHROUGH */
364	case 1:
365		y = atoi(*argv);
366		if (y < 1 || y > 9999)
367			errx(EX_USAGE, "year `%s' not in range 1..9999", *argv);
368		argv++;
369		flag_givenyear = 1;
370		break;
371	case 0:
372		if (flag_today != NULL) {
373			y = strtol(flag_today, NULL, 10);
374			m = strtol(flag_today + 5, NULL, 10);
375		} else {
376			time_t t;
377			struct tm *tm;
378
379			t = time(NULL);
380			tm = localtime(&t);
381			y = tm->tm_year + 1900;
382			m = tm->tm_mon + 1;
383		}
384		break;
385	default:
386		usage();
387	}
388
389	if (flag_month != NULL) {
390		if (parsemonth(flag_month, &m, &y)) {
391			errx(EX_USAGE,
392			    "%s is neither a month number (1..12) nor a name",
393			    flag_month);
394		}
395	}
396
397	/*
398	 * What is not supported:
399	 * -3 with -A or -B
400	 *	-3 displays 3 months, -A and -B change that behaviour.
401	 * -3 with -y
402	 *	-3 displays 3 months, -y says display a whole year.
403	 * -3 with a given year but no given month or without -m
404	 *	-3 displays 3 months, no month specified doesn't make clear
405	 *      which three months.
406	 * -m with a given month
407	 *	conflicting arguments, both specify the same field.
408	 * -y with -m
409	 *	-y displays the whole year, -m displays a single month.
410	 * -y with a given month
411	 *	-y displays the whole year, the given month displays a single
412	 *	month.
413	 * -y with -A or -B
414	 *	-y displays the whole year, -A and -B display extra months.
415	 */
416
417	/* -3 together with -A or -B. */
418	if (flag_3months && (flag_after || flag_before))
419		errx(EX_USAGE, "-3 together with -A and -B is not supported.");
420	/* -3 together with -y. */
421	if (flag_3months && flag_wholeyear)
422		errx(EX_USAGE, "-3 together with -y is not supported.");
423	/* -3 together with givenyear but no givenmonth. */
424	if (flag_3months && flag_givenyear &&
425	    !(flag_givenmonth || flag_specifiedmonth))
426		errx(EX_USAGE,
427		    "-3 together with a given year but no given month is "
428		    "not supported.");
429	/* -m together with xx xxxx. */
430	if (flag_specifiedmonth && flag_givenmonth)
431		errx(EX_USAGE,
432		    "-m together with a given month is not supported.");
433	/* -y together with -m. */
434	if (flag_wholeyear && flag_specifiedmonth)
435		errx(EX_USAGE, "-y together with -m is not supported.");
436	/* -y together with xx xxxx. */
437	if (flag_wholeyear && flag_givenmonth)
438		errx(EX_USAGE, "-y together a given month is not supported.");
439	/* -y together with -A or -B. */
440	if (flag_wholeyear && (flag_before > 0 || flag_after > 0))
441		errx(EX_USAGE, "-y together a -A or -B is not supported.");
442	/* The rest should be fine. */
443
444	/* Select the period to display, in order of increasing priority .*/
445	if (flag_wholeyear ||
446	    (flag_givenyear && !(flag_givenmonth || flag_specifiedmonth))) {
447		m = 1;
448		before = 0;
449		after = 11;
450	}
451	if (flag_givenyear && flag_givenmonth) {
452		before = 0;
453		after = 0;
454	}
455	if (flag_specifiedmonth) {
456		before = 0;
457		after = 0;
458	}
459	if (flag_before) {
460		before = flag_before;
461	}
462	if (flag_after) {
463		after = flag_after;
464	}
465	if (flag_3months) {
466		before = 1;
467		after = 1;
468	}
469	if (after == -1)
470		after = 0;
471	if (before == -1)
472		before = 0;
473
474	/* Highlight a specified day or today .*/
475	if (flag_highlightdate != NULL) {
476		dt.y = strtol(flag_highlightdate, NULL, 10);
477		dt.m = strtol(flag_highlightdate + 5, NULL, 10);
478		dt.d = strtol(flag_highlightdate + 8, NULL, 10);
479	} else {
480		time_t t;
481		struct tm *tm1;
482
483		t = time(NULL);
484		tm1 = localtime(&t);
485		dt.y = tm1->tm_year + 1900;
486		dt.m = tm1->tm_mon + 1;
487		dt.d = tm1->tm_mday;
488	}
489	highlightdate = sndaysb(&dt);
490
491	/* And now we finally start to calculate and output calendars. */
492	if (flag_easter)
493		printeaster(y, flag_julian_cal, flag_orthodox);
494	else
495		if (flag_backward)
496			monthrangeb(y, m, flag_julian_day, before, after);
497		else
498			monthranger(y, m, flag_julian_day, before, after);
499	return (0);
500}
501
502static void
503usage(void)
504{
505
506	fputs(
507"Usage: cal [general options] [-hjy] [[month] year]\n"
508"       cal [general options] [-hj] [-m month] [year]\n"
509"       ncal [general options] [-hJjpwy] [-s country_code] [[month] year]\n"
510"       ncal [general options] [-hJeo] [year]\n"
511"General options: [-NC3] [-A months] [-B months]\n"
512"For debug the highlighting: [-H yyyy-mm-dd] [-d yyyy-mm]\n",
513	    stderr);
514	exit(EX_USAGE);
515}
516
517/* Print the assumed switches for all countries. */
518static void
519printcc(void)
520{
521	struct djswitch *p;
522	int n;	/* number of lines to print */
523	int m;	/* offset from left to right table entry on the same line */
524
525#define FSTR "%c%s %-15s%4d-%02d-%02d"
526#define DFLT(p) ((p) == dftswitch ? '*' : ' ')
527#define FSTRARG(p) DFLT(p), (p)->cc, (p)->nm, (p)->dt.y, (p)->dt.m, (p)->dt.d
528
529	n = sizeof(switches) / sizeof(struct djswitch);
530	m = (n + 1) / 2;
531	n /= 2;
532	for (p = switches; p != switches + n; p++)
533		printf(FSTR"     "FSTR"\n", FSTRARG(p), FSTRARG(p+m));
534	if (m != n)
535		printf(FSTR"\n", FSTRARG(p));
536}
537
538/* Print the date of easter sunday. */
539static void
540printeaster(int y, int julian, int orthodox)
541{
542	date    dt;
543	struct tm tm;
544	char    buf[MAX_WIDTH];
545	static int d_first = -1;
546
547	if (d_first < 0)
548		d_first = (*nl_langinfo(D_MD_ORDER) == 'd');
549	/* force orthodox easter for years before 1583 */
550	if (y < 1583)
551		orthodox = 1;
552
553	if (orthodox)
554		if (julian)
555			easteroj(y, &dt);
556		else
557			easterog(y, &dt);
558	else
559		easterg(y, &dt);
560
561	memset(&tm, 0, sizeof(tm));
562	tm.tm_year = dt.y - 1900;
563	tm.tm_mon  = dt.m - 1;
564	tm.tm_mday = dt.d;
565	strftime(buf, sizeof(buf), d_first ? "%e %B %Y" : "%B %e %Y",  &tm);
566	printf("%s\n", buf);
567}
568
569#define MW(mw, me)		((mw) + me)
570#define	DECREASEMONTH(m, y) 		\
571		if (--m == 0) {		\
572			m = 12;		\
573			y--;		\
574		}
575#define	INCREASEMONTH(m, y)		\
576		if (++(m) == 13) {	\
577			(m) = 1;	\
578			(y)++;		\
579		}
580#define	M2Y(m)	((m) / 12)
581#define	M2M(m)	(1 + (m) % 12)
582
583/* Print all months for the period in the range [ before .. y-m .. after ]. */
584static void
585monthrangeb(int y, int m, int jd_flag, int before, int after)
586{
587	struct monthlines year[12];
588	struct weekdays wds;
589	char	s[MAX_WIDTH], t[MAX_WIDTH];
590	wchar_t	ws[MAX_WIDTH], ws1[MAX_WIDTH];
591	const char	*wdss;
592	int     i, j;
593	int     mpl;
594	int     mw;
595	int	m1, m2;
596	int	printyearheader;
597	int	prevyear = -1;
598
599	mpl = jd_flag ? 2 : 3;
600	mw = jd_flag ? MONTH_WIDTH_B_J : MONTH_WIDTH_B;
601	wdss = (mpl == 2) ? " " : "";
602
603	while (before != 0) {
604		DECREASEMONTH(m, y);
605		before--;
606		after++;
607	}
608	m1 = y * 12 + m - 1;
609	m2 = m1 + after;
610
611	mkweekdays(&wds);
612
613	/*
614	 * The year header is printed when there are more than 'mpl' months
615	 * and if the first month is a multitude of 'mpl'.
616	 * If not, it will print the year behind every month.
617	 */
618	printyearheader = (after >= mpl - 1) && (M2M(m1) - 1) % mpl == 0;
619
620	m = m1;
621	while (m <= m2) {
622		int count = 0;
623		for (i = 0; i != mpl && m + i <= m2; i++) {
624			mkmonthb(M2Y(m + i), M2M(m + i) - 1, jd_flag, year + i);
625			count++;
626		}
627
628		/* Empty line between two rows of months */
629		if (m != m1)
630			printf("\n");
631
632		/* Year at the top. */
633		if (printyearheader && M2Y(m) != prevyear) {
634			sprintf(s, "%d", M2Y(m));
635			printf("%s\n", center(t, s, mpl * mw));
636			prevyear = M2Y(m);
637		}
638
639		/* Month names. */
640		for (i = 0; i < count; i++)
641			if (printyearheader)
642				wprintf(L"%-*ls  ",
643				    mw, wcenter(ws, year[i].name, mw));
644			else {
645				swprintf(ws, sizeof(ws)/sizeof(ws[0]),
646				    L"%-ls %d", year[i].name, M2Y(m + i));
647				wprintf(L"%-*ls  ", mw, wcenter(ws1, ws, mw));
648			}
649		printf("\n");
650
651		/* Day of the week names. */
652		for (i = 0; i < count; i++) {
653			wprintf(L"%s%ls%s%ls%s%ls%s%ls%s%ls%s%ls%s%ls ",
654				wdss, wds.names[6], wdss, wds.names[0],
655				wdss, wds.names[1], wdss, wds.names[2],
656				wdss, wds.names[3], wdss, wds.names[4],
657				wdss, wds.names[5]);
658		}
659		printf("\n");
660
661		/* And the days of the month. */
662		for (i = 0; i != 6; i++) {
663			for (j = 0; j < count; j++)
664				printf("%-*s  ",
665				    MW(mw, year[j].extralen[i]),
666					year[j].lines[i]+1);
667			printf("\n");
668		}
669
670		m += mpl;
671	}
672}
673
674static void
675monthranger(int y, int m, int jd_flag, int before, int after)
676{
677	struct monthlines year[12];
678	struct weekdays wds;
679	char    s[MAX_WIDTH], t[MAX_WIDTH];
680	int     i, j;
681	int     mpl;
682	int     mw;
683	int	m1, m2;
684	int	prevyear = -1;
685	int	printyearheader;
686
687	mpl = jd_flag ? 3 : 4;
688	mw = jd_flag ? MONTH_WIDTH_R_J : MONTH_WIDTH_R;
689
690	while (before != 0) {
691		DECREASEMONTH(m, y);
692		before--;
693		after++;
694	}
695	m1 = y * 12 + m - 1;
696	m2 = m1 + after;
697
698	mkweekdays(&wds);
699
700	/*
701	 * The year header is printed when there are more than 'mpl' months
702	 * and if the first month is a multitude of 'mpl'.
703	 * If not, it will print the year behind every month.
704	 */
705	printyearheader = (after >= mpl - 1) && (M2M(m1) - 1) % mpl == 0;
706
707	m = m1;
708	while (m <= m2) {
709		int count = 0;
710		for (i = 0; i != mpl && m + i <= m2; i++) {
711			mkmonthr(M2Y(m + i), M2M(m + i) - 1, jd_flag, year + i);
712			count++;
713		}
714
715		/* Empty line between two rows of months. */
716		if (m != m1)
717			printf("\n");
718
719		/* Year at the top. */
720		if (printyearheader && M2Y(m) != prevyear) {
721			sprintf(s, "%d", M2Y(m));
722			printf("%s\n", center(t, s, mpl * mw));
723			prevyear = M2Y(m);
724		}
725
726		/* Month names. */
727		wprintf(L"    ");
728		for (i = 0; i < count; i++)
729			if (printyearheader)
730				wprintf(L"%-*ls", mw, year[i].name);
731			else
732				wprintf(L"%-ls %-*d", year[i].name,
733				    mw - wcslen(year[i].name) - 1, M2Y(m + i));
734		printf("\n");
735
736		/* And the days of the month. */
737		for (i = 0; i != 7; i++) {
738			/* Week day */
739			wprintf(L"%.2ls", wds.names[i]);
740
741			/* Full months */
742			for (j = 0; j < count; j++)
743				printf("%-*s",
744				    MW(mw, year[j].extralen[i]),
745					year[j].lines[i]);
746			printf("\n");
747		}
748
749		/* Week numbers. */
750		if (flag_weeks) {
751			printf("  ");
752			for (i = 0; i < count; i++)
753				printf("%-*s", mw, year[i].weeks);
754			printf("\n");
755		}
756
757		m += mpl;
758	}
759	return;
760}
761
762static void
763mkmonthr(int y, int m, int jd_flag, struct monthlines *mlines)
764{
765
766	struct tm tm;		/* for strftime printing local names of
767				 * months */
768	date    dt;		/* handy date */
769	int     dw;		/* width of numbers */
770	int     first;		/* first day of month */
771	int     firstm;		/* first day of first week of month */
772	int     i, j, k, l;	/* just indices */
773	int     last;		/* the first day of next month */
774	int     jan1 = 0;	/* the first day of this year */
775	char   *ds;		/* pointer to day strings (daystr or
776				 * jdaystr) */
777
778	/* Set name of month. */
779	memset(&tm, 0, sizeof(tm));
780	tm.tm_mon = m;
781	wcsftime(mlines->name, sizeof(mlines->name) / sizeof(mlines->name[0]),
782		 L"%OB", &tm);
783	mlines->name[0] = towupper(mlines->name[0]);
784
785	/*
786	 * Set first and last to the day number of the first day of this
787	 * month and the first day of next month respectively. Set jan1 to
788	 * the day number of the first day of this year.
789	 */
790	first = firstday(y, m + 1);
791	if (m == 11)
792		last = firstday(y + 1, 1);
793	else
794		last = firstday(y, m + 2);
795
796	if (jd_flag)
797		jan1 = firstday(y, 1);
798
799	/*
800	 * Set firstm to the day number of monday of the first week of
801	 * this month. (This might be in the last month)
802	 */
803	firstm = first - weekday(first);
804
805	/* Set ds (daystring) and dw (daywidth) according to the jd_flag. */
806	if (jd_flag) {
807		ds = jdaystr;
808		dw = 4;
809	} else {
810		ds = daystr;
811		dw = 3;
812	}
813
814	/*
815	 * Fill the lines with day of month or day of year (julian day)
816	 * line index: i, each line is one weekday. column index: j, each
817	 * column is one day number. print column index: k.
818	 */
819	for (i = 0; i != 7; i++) {
820		l = 0;
821		for (j = firstm + i, k = 0; j < last; j += 7, k += dw) {
822			if (j >= first) {
823				if (jd_flag)
824					dt.d = j - jan1 + 1;
825				else
826					sdater(j, &dt);
827				if (j == highlightdate && !flag_nohighlight
828				 && isatty(STDOUT_FILENO))
829					highlight(mlines->lines[i] + k,
830					    ds + dt.d * dw, dw, &l);
831				else
832					memcpy(mlines->lines[i] + k + l,
833					       ds + dt.d * dw, dw);
834			} else
835				memcpy(mlines->lines[i] + k + l, "    ", dw);
836		}
837		mlines->lines[i][k + l] = '\0';
838		mlines->extralen[i] = l;
839	}
840
841	/* fill the weeknumbers. */
842	if (flag_weeks) {
843		for (j = firstm, k = 0; j < last;  k += dw, j += 7)
844			if (j <= nswitch)
845				memset(mlines->weeks + k, ' ', dw);
846			else
847				memcpy(mlines->weeks + k,
848				    ds + week(j, &i)*dw, dw);
849		mlines->weeks[k] = '\0';
850	}
851}
852
853static void
854mkmonthb(int y, int m, int jd_flag, struct monthlines *mlines)
855{
856
857	struct tm tm;		/* for strftime printing local names of
858				 * months */
859	date    dt;		/* handy date */
860	int     dw;		/* width of numbers */
861	int     first;		/* first day of month */
862	int     firsts;		/* sunday of first week of month */
863	int     i, j, k, l;	/* just indices */
864	int     jan1 = 0;	/* the first day of this year */
865	int     last;		/* the first day of next month */
866	char   *ds;		/* pointer to day strings (daystr or
867				 * jdaystr) */
868
869	/* Set ds (daystring) and dw (daywidth) according to the jd_flag */
870	if (jd_flag) {
871		ds = jdaystr;
872		dw = 4;
873	} else {
874		ds = daystr;
875		dw = 3;
876	}
877
878	/* Set name of month centered. */
879	memset(&tm, 0, sizeof(tm));
880	tm.tm_mon = m;
881	wcsftime(mlines->name, sizeof(mlines->name) / sizeof(mlines->name[0]),
882		 L"%OB", &tm);
883	mlines->name[0] = towupper(mlines->name[0]);
884
885	/*
886	 * Set first and last to the day number of the first day of this
887	 * month and the first day of next month respectively. Set jan1 to
888	 * the day number of Jan 1st of this year.
889	 */
890	dt.y = y;
891	dt.m = m + 1;
892	dt.d = 1;
893	first = sndaysb(&dt);
894	if (m == 11) {
895		dt.y = y + 1;
896		dt.m = 1;
897		dt.d = 1;
898	} else {
899		dt.y = y;
900		dt.m = m + 2;
901		dt.d = 1;
902	}
903	last = sndaysb(&dt);
904
905	if (jd_flag) {
906		dt.y = y;
907		dt.m = 1;
908		dt.d = 1;
909		jan1 = sndaysb(&dt);
910	}
911
912	/*
913	 * Set firsts to the day number of sunday of the first week of
914	 * this month. (This might be in the last month)
915	 */
916	firsts = first - (weekday(first)+1) % 7;
917
918	/*
919	 * Fill the lines with day of month or day of year (Julian day)
920	 * line index: i, each line is one week. column index: j, each
921	 * column is one day number. print column index: k.
922	 */
923	for (i = 0; i != 6; i++) {
924		l = 0;
925		for (j = firsts + 7 * i, k = 0; j < last && k != dw * 7;
926		    j++, k += dw) {
927			if (j >= first) {
928				if (jd_flag)
929					dt.d = j - jan1 + 1;
930				else
931					sdateb(j, &dt);
932				if (j == highlightdate && !flag_nohighlight)
933					highlight(mlines->lines[i] + k,
934					    ds + dt.d * dw, dw, &l);
935				else
936					memcpy(mlines->lines[i] + k + l,
937					       ds + dt.d * dw, dw);
938			} else
939				memcpy(mlines->lines[i] + k + l, "    ", dw);
940		}
941		if (k == 0)
942			mlines->lines[i][1] = '\0';
943		else
944			mlines->lines[i][k + l] = '\0';
945		mlines->extralen[i] = l;
946	}
947}
948
949/* Put the local names of weekdays into the wds. */
950static void
951mkweekdays(struct weekdays *wds)
952{
953	int i, len, width = 0;
954	struct tm tm;
955	wchar_t buf[20];
956
957	memset(&tm, 0, sizeof(tm));
958
959	for (i = 0; i != 7; i++) {
960		tm.tm_wday = (i+1) % 7;
961		wcsftime(buf, sizeof(buf)/sizeof(buf[0]), L"%a", &tm);
962		for (len = 2; len > 0; --len) {
963			if ((width = wcswidth(buf, len)) <= 2)
964				break;
965		}
966		wmemset(wds->names[i], L'\0', 4);
967		if (width == 1)
968			wds->names[i][0] = L' ';
969		wcsncat(wds->names[i], buf, len);
970		wcsncat(wds->names[i], L" ", 1);
971	}
972}
973
974/*
975 * Compute the day number of the first existing date after the first day in
976 * month. (the first day in month and even the month might not exist!)
977 */
978static int
979firstday(int y, int m)
980{
981	date dt;
982	int nd;
983
984	dt.y = y;
985	dt.m = m;
986	dt.d = 1;
987	nd = sndaysr(&dt);
988	for (;;) {
989		sdater(nd, &dt);
990		if ((dt.m >= m && dt.y == y) || dt.y > y)
991			return (nd);
992		else
993			nd++;
994	}
995	/* NEVER REACHED */
996}
997
998/*
999 * Compute the number of days from date, obey the local switch from
1000 * Julian to Gregorian if specified by the user.
1001 */
1002static int
1003sndaysr(struct date *d)
1004{
1005
1006	if (nswitch != 0)
1007		if (nswitch < ndaysj(d))
1008			return (ndaysg(d));
1009		else
1010			return (ndaysj(d));
1011	else
1012		return ndaysg(d);
1013}
1014
1015/*
1016 * Compute the number of days from date, obey the switch from
1017 * Julian to Gregorian as used by UK and her colonies.
1018 */
1019static int
1020sndaysb(struct date *d)
1021{
1022
1023	if (nswitchb < ndaysj(d))
1024		return (ndaysg(d));
1025	else
1026		return (ndaysj(d));
1027}
1028
1029/* Inverse of sndays. */
1030static struct date *
1031sdater(int nd, struct date *d)
1032{
1033
1034	if (nswitch < nd)
1035		return (gdate(nd, d));
1036	else
1037		return (jdate(nd, d));
1038}
1039
1040/* Inverse of sndaysb. */
1041static struct date *
1042sdateb(int nd, struct date *d)
1043{
1044
1045	if (nswitchb < nd)
1046		return (gdate(nd, d));
1047	else
1048		return (jdate(nd, d));
1049}
1050
1051/* Center string t in string s of length w by putting enough leading blanks. */
1052static char *
1053center(char *s, char *t, int w)
1054{
1055	char blanks[MAX_WIDTH];
1056
1057	memset(blanks, ' ', sizeof(blanks));
1058	sprintf(s, "%.*s%s", (int)(w - strlen(t)) / 2, blanks, t);
1059	return (s);
1060}
1061
1062/* Center string t in string s of length w by putting enough leading blanks. */
1063static wchar_t *
1064wcenter(wchar_t *s, wchar_t *t, int w)
1065{
1066	char blanks[MAX_WIDTH];
1067
1068	memset(blanks, ' ', sizeof(blanks));
1069	swprintf(s, MAX_WIDTH, L"%.*s%ls", (int)(w - wcslen(t)) / 2, blanks, t);
1070	return (s);
1071}
1072
1073static int
1074parsemonth(const char *s, int *m, int *y)
1075{
1076	int nm, ny;
1077	char *cp;
1078	struct tm tm;
1079
1080	nm = (int)strtol(s, &cp, 10);
1081	if (cp != s) {
1082		ny = *y;
1083		if (*cp == '\0') {
1084			;	/* no special action */
1085		} else if (*cp == 'f' || *cp == 'F') {
1086			if (nm <= *m)
1087				ny++;
1088		} else if (*cp == 'p' || *cp == 'P') {
1089			if (nm >= *m)
1090				ny--;
1091		} else
1092			return (1);
1093		if (nm < 1 || nm > 12)
1094			return 1;
1095		*m = nm;
1096		*y = ny;
1097		return (0);
1098	}
1099	if (strptime(s, "%B", &tm) != NULL || strptime(s, "%b", &tm) != NULL) {
1100		*m = tm.tm_mon + 1;
1101		return (0);
1102	}
1103	return (1);
1104}
1105
1106static void
1107highlight(char *dst, char *src, int len, int *extralen)
1108{
1109	static int first = 1;
1110	static const char *term_so, *term_se;
1111
1112	if (first) {
1113		char tbuf[1024], cbuf[512], *b;
1114
1115		term_se = term_so = NULL;
1116
1117		/* On how to highlight on this type of terminal (if any). */
1118		if (isatty(STDOUT_FILENO) && tgetent(tbuf, NULL) == 1) {
1119			b = cbuf;
1120			term_so = tgetstr("so", &b);
1121			term_se = tgetstr("se", &b);
1122		}
1123
1124		first = 0;
1125	}
1126
1127	/*
1128	 * This check is not necessary, should have been handled before calling
1129	 * this function.
1130	 */
1131	if (flag_nohighlight) {
1132		memcpy(dst, src, len);
1133		return;
1134	}
1135
1136	/*
1137	 * If it is a real terminal, use the data from the termcap database.
1138	 */
1139	if (term_so != NULL && term_se != NULL) {
1140		/* separator. */
1141		dst[0] = ' ';
1142		dst++;
1143		/* highlight on. */
1144		memcpy(dst, term_so, strlen(term_so));
1145		dst += strlen(term_so);
1146		/* the actual text. (minus leading space) */
1147		len--;
1148		src++;
1149		memcpy(dst, src, len);
1150		dst += len;
1151		/* highlight off. */
1152		memcpy(dst, term_se, strlen(term_se));
1153		*extralen = strlen(term_so) + strlen(term_se);
1154		return;
1155	}
1156
1157	/*
1158	 * Otherwise, print a _, backspace and the letter.
1159	 */
1160	*extralen = 0;
1161	/* skip leading space. */
1162	src++;
1163	len--;
1164	/* separator. */
1165	dst[0] = ' ';
1166	dst++;
1167	while (len > 0) {
1168		/* _ and backspace. */
1169		memcpy(dst, "_\010", 2);
1170		dst += 2;
1171		*extralen += 2;
1172		/* the character. */
1173		*dst++ = *src++;
1174		len--;
1175	}
1176	return;
1177}
1178