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