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