usr.bin/ncal/ncal.c

   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 <libgen.h>
  37 #include <locale.h>
  38 #include <stdio.h>
  39 #include <stdlib.h>
  40 #include <string.h>
  41 #include <sysexits.h>
  42 #include <time.h>
  43 #include <unistd.h>
  44 #include <wchar.h>
  45 #include <wctype.h>
  46 #include <term.h>
  47 #undef lines                    /* term.h defines this */
  48
  49 /* Width of one month with backward compatibility and in regular mode*/
  50 #define MONTH_WIDTH_B_J 27
  51 #define MONTH_WIDTH_B 20
  52
  53 #define MONTH_WIDTH_R_J 24
  54 #define MONTH_WIDTH_R 18
  55
  56 #define MAX_WIDTH 64
  57
  58 typedef struct date date;
  59
  60 struct monthlines {
  61         wchar_t name[MAX_WIDTH + 1];
  62         char lines[7][MAX_WIDTH + 1];
  63         char weeks[MAX_WIDTH + 1];
  64         unsigned int extralen[7];
  65 };
  66
  67 struct weekdays {
  68         wchar_t names[7][4];
  69 };
  70
  71 /* The switches from Julian to Gregorian in some countries */
  72 static struct djswitch {
  73         const char *cc; /* Country code according to ISO 3166 */
  74         const char *nm; /* Name of country */
  75         date dt;        /* Last day of Julian calendar */
  76 } switches[] = {
  77         {"AL", "Albania",       {1912, 11, 30}},
  78         {"AT", "Austria",       {1583, 10,  5}},
  79         {"AU", "Australia",     {1752,  9,  2}},
  80         {"BE", "Belgium",       {1582, 12, 14}},
  81         {"BG", "Bulgaria",      {1916,  3, 18}},
  82         {"CA", "Canada",        {1752,  9,  2}},
  83         {"CH", "Switzerland",   {1655,  2, 28}},
  84         {"CN", "China",         {1911, 12, 18}},
  85         {"CZ", "Czech Republic",{1584,  1,  6}},
  86         {"DE", "Germany",       {1700,  2, 18}},
  87         {"DK", "Denmark",       {1700,  2, 18}},
  88         {"ES", "Spain",         {1582, 10,  4}},
  89         {"FI", "Finland",       {1753,  2, 17}},
  90         {"FR", "France",        {1582, 12,  9}},
  91         {"GB", "United Kingdom",{1752,  9,  2}},
  92         {"GR", "Greece",        {1924,  3,  9}},
  93         {"HU", "Hungary",       {1587, 10, 21}},
  94         {"IS", "Iceland",       {1700, 11, 16}},
  95         {"IT", "Italy",         {1582, 10,  4}},
  96         {"JP", "Japan",         {1918, 12, 18}},
  97         {"LI", "Lithuania",     {1918,  2,  1}},
  98         {"LN", "Latin",         {9999, 05, 31}},
  99         {"LU", "Luxembourg",    {1582, 12, 14}},
 100         {"LV", "Latvia",        {1918,  2,  1}},
 101         {"NL", "Netherlands",   {1582, 12, 14}},
 102         {"NO", "Norway",        {1700,  2, 18}},
 103         {"PL", "Poland",        {1582, 10,  4}},
 104         {"PT", "Portugal",      {1582, 10,  4}},
 105         {"RO", "Romania",       {1919,  3, 31}},
 106         {"RU", "Russia",        {1918,  1, 31}},
 107         {"SI", "Slovenia",      {1919,  3,  4}},
 108         {"SE", "Sweden",        {1753,  2, 17}},
 109         {"TR", "Turkey",        {1926, 12, 18}},
 110         {"US", "United States", {1752,  9,  2}},
 111         {"YU", "Yugoslavia",    {1919,  3,  4}}
 112 };
 113
 114 struct djswitch *dftswitch =
 115     switches + sizeof(switches) / sizeof(struct djswitch) - 2;
 116     /* default switch (should be "US") */
 117
 118 /* Table used to print day of month and week numbers */
 119 char daystr[] = "     1  2  3  4  5  6  7  8  9 10 11 12 13 14 15"
 120                 " 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31"
 121                 " 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47"
 122                 " 48 49 50 51 52 53";
 123
 124 /* Table used to print day of year and week numbers */
 125 char jdaystr[] = "       1   2   3   4   5   6   7   8   9"
 126                  "  10  11  12  13  14  15  16  17  18  19"
 127                  "  20  21  22  23  24  25  26  27  28  29"
 128                  "  30  31  32  33  34  35  36  37  38  39"
 129                  "  40  41  42  43  44  45  46  47  48  49"
 130                  "  50  51  52  53  54  55  56  57  58  59"
 131                  "  60  61  62  63  64  65  66  67  68  69"
 132                  "  70  71  72  73  74  75  76  77  78  79"
 133                  "  80  81  82  83  84  85  86  87  88  89"
 134                  "  90  91  92  93  94  95  96  97  98  99"
 135                  " 100 101 102 103 104 105 106 107 108 109"
 136                  " 110 111 112 113 114 115 116 117 118 119"
 137                  " 120 121 122 123 124 125 126 127 128 129"
 138                  " 130 131 132 133 134 135 136 137 138 139"
 139                  " 140 141 142 143 144 145 146 147 148 149"
 140                  " 150 151 152 153 154 155 156 157 158 159"
 141                  " 160 161 162 163 164 165 166 167 168 169"
 142                  " 170 171 172 173 174 175 176 177 178 179"
 143                  " 180 181 182 183 184 185 186 187 188 189"
 144                  " 190 191 192 193 194 195 196 197 198 199"
 145                  " 200 201 202 203 204 205 206 207 208 209"
 146                  " 210 211 212 213 214 215 216 217 218 219"
 147                  " 220 221 222 223 224 225 226 227 228 229"
 148                  " 230 231 232 233 234 235 236 237 238 239"
 149                  " 240 241 242 243 244 245 246 247 248 249"
 150                  " 250 251 252 253 254 255 256 257 258 259"
 151                  " 260 261 262 263 264 265 266 267 268 269"
 152                  " 270 271 272 273 274 275 276 277 278 279"
 153                  " 280 281 282 283 284 285 286 287 288 289"
 154                  " 290 291 292 293 294 295 296 297 298 299"
 155                  " 300 301 302 303 304 305 306 307 308 309"
 156                  " 310 311 312 313 314 315 316 317 318 319"
 157                  " 320 321 322 323 324 325 326 327 328 329"
 158                  " 330 331 332 333 334 335 336 337 338 339"
 159                  " 340 341 342 343 344 345 346 347 348 349"
 160                  " 350 351 352 353 354 355 356 357 358 359"
 161                  " 360 361 362 363 364 365 366";
 162
 163 int     flag_nohighlight;       /* user doesn't want a highlighted today */
 164 int     flag_weeks;             /* user wants number of week */
 165 int     nswitch;                /* user defined switch date */
 166 int     nswitchb;               /* switch date for backward compatibility */
 167 int     highlightdate;
 168
 169 char    *center(char *s, char *t, int w);
 170 wchar_t *wcenter(wchar_t *s, wchar_t *t, int w);
 171 int     firstday(int y, int m);
 172 void    highlight(char *dst, char *src, int len, int *extraletters);
 173 void    mkmonthr(int year, int month, int jd_flag, struct monthlines * monthl);
 174 void    mkmonthb(int year, int month, int jd_flag, struct monthlines * monthl);
 175 void    mkweekdays(struct weekdays * wds);
 176 void    monthranger(int year, int m, int jd_flag, int before, int after);
 177 void    monthrangeb(int year, int m, int jd_flag, int before, int after);
 178 int     parsemonth(const char *s, int *m, int *y);
 179 void    printcc(void);
 180 void    printeaster(int year, int julian, int orthodox);
 181 date    *sdater(int ndays, struct date * d);
 182 date    *sdateb(int ndays, struct date * d);
 183 int     sndaysr(struct date * d);
 184 int     sndaysb(struct date * d);
 185 static void     usage(void);
 186
 187 int
 188 main(int argc, char *argv[])
 189 {
 190         struct  djswitch *p, *q;        /* to search user defined switch date */
 191         date    never = {10000, 1, 1};  /* outside valid range of dates */
 192         date    ukswitch = {1752, 9, 2};/* switch date for Great Britain */
 193         date    dt;
 194         int     ch;                     /* holds the option character */
 195         int     m = 0;                  /* month */
 196         int     y = 0;                  /* year */
 197         int     flag_backward = 0;      /* user called cal--backward compat. */
 198         int     flag_wholeyear = 0;     /* user wants the whole year */
 199         int     flag_julian_cal = 0;    /* user wants Julian Calendar */
 200         int     flag_julian_day = 0;    /* user wants the Julian day numbers */
 201         int     flag_orthodox = 0;      /* user wants Orthodox easter */
 202         int     flag_easter = 0;        /* user wants easter date */
 203         int     flag_3months = 0;       /* user wants 3 month display (-3) */
 204         int     flag_after = 0;         /* user wants to see months after */
 205         int     flag_before = 0;        /* user wants to see months before */
 206         int     flag_specifiedmonth = 0;/* user wants to see this month (-m) */
 207         int     flag_givenmonth = 0;    /* user has specified month [n] */
 208         int     flag_givenyear = 0;     /* user has specified year [n] */
 209         char    *cp;                    /* character pointer */
 210         char    *flag_today = NULL;     /* debug: use date as being today */
 211         char    *flag_month = NULL;     /* requested month as string */
 212         char    *flag_highlightdate = NULL; /* debug: date to highlight */
 213         int     before, after;
 214         const char    *locale;          /* locale to get country code */
 215
 216         flag_nohighlight = 0;
 217         flag_weeks = 0;
 218
 219         /*
 220          * Use locale to determine the country code,
 221          * and use the country code to determine the default
 222          * switchdate and date format from the switches table.
 223          */
 224         if (setlocale(LC_ALL, "") == NULL)
 225                 warn("setlocale");
 226         locale = setlocale(LC_TIME, NULL);
 227         if (locale == NULL ||
 228             strcmp(locale, "C") == 0 ||
 229             strcmp(locale, "POSIX") == 0 ||
 230             strcmp(locale, "ASCII") == 0 ||
 231             strcmp(locale, "US-ASCII") == 0)
 232                 locale = "_US";
 233         q = switches + sizeof(switches) / sizeof(struct djswitch);
 234         for (p = switches; p != q; p++)
 235                 if ((cp = strstr(locale, p->cc)) != NULL && *(cp - 1) == '_')
 236                         break;
 237         if (p == q) {
 238                 nswitch = ndaysj(&dftswitch->dt);
 239         } else {
 240                 nswitch = ndaysj(&p->dt);
 241                 dftswitch = p;
 242         }
 243
 244
 245         /*
 246          * Get the filename portion of argv[0] and set flag_backward if
 247          * this program is called "cal".
 248          */
 249         if (strncmp(basename(argv[0]), "cal", strlen("cal")) == 0)
 250                 flag_backward = 1;
 251
 252         /* Set the switch date to United Kingdom if backwards compatible */
 253         if (flag_backward)
 254                 nswitchb = ndaysj(&ukswitch);
 255
 256         before = after = -1;
 257
 258         while ((ch = getopt(argc, argv, "3A:B:Cd:eH:hjJm:Nops:wy")) != -1)
 259                 switch (ch) {
 260                 case '3':
 261                         flag_3months = 1;
 262                         break;
 263                 case 'A':
 264                         if (flag_after > 0)
 265                                 errx(EX_USAGE, "Double -A specified");
 266                         flag_after = strtol(optarg, NULL, 10);
 267                         if (flag_after <= 0)
 268                                 errx(EX_USAGE,
 269                                     "Argument to -A must be positive");
 270                         break;
 271                 case 'B':
 272                         if (flag_before > 0)
 273                                 errx(EX_USAGE, "Double -A specified");
 274                         flag_before = strtol(optarg, NULL, 10);
 275                         if (flag_before <= 0)
 276                                 errx(EX_USAGE,
 277                                     "Argument to -B must be positive");
 278                         break;
 279                 case 'J':
 280                         if (flag_backward)
 281                                 usage();
 282                         nswitch = ndaysj(&never);
 283                         flag_julian_cal = 1;
 284                         break;
 285                 case 'C':
 286                         flag_backward = 1;
 287                         break;
 288                 case 'N':
 289                         flag_backward = 0;
 290                         break;
 291                 case 'd':
 292                         flag_today = optarg;
 293                         break;
 294                 case 'H':
 295                         flag_highlightdate = optarg;
 296                         break;
 297                 case 'h':
 298                         flag_nohighlight = 1;
 299                         break;
 300                 case 'e':
 301                         if (flag_backward)
 302                                 usage();
 303                         flag_easter = 1;
 304                         break;
 305                 case 'j':
 306                         flag_julian_day = 1;
 307                         break;
 308                 case 'm':
 309                         if (flag_specifiedmonth)
 310                                 errx(EX_USAGE, "Double -m specified");
 311                         flag_month = optarg;
 312                         flag_specifiedmonth = 1;
 313                         break;
 314                 case 'o':
 315                         if (flag_backward)
 316                                 usage();
 317                         flag_orthodox = 1;
 318                         flag_easter = 1;
 319                         break;
 320                 case 'p':
 321                         if (flag_backward)
 322                                 usage();
 323                         printcc();
 324                         return (0);
 325                         break;
 326                 case 's':
 327                         if (flag_backward)
 328                                 usage();
 329                         q = switches +
 330                             sizeof(switches) / sizeof(struct djswitch);
 331                         for (p = switches;
 332                              p != q && strcmp(p->cc, optarg) != 0; p++)
 333                                 ;
 334                         if (p == q)
 335                                 errx(EX_USAGE,
 336                                     "%s: invalid country code", optarg);
 337                         nswitch = ndaysj(&(p->dt));
 338                         break;
 339                 case 'w':
 340                         if (flag_backward)
 341                                 usage();
 342                         flag_weeks = 1;
 343                         break;
 344                 case 'y':
 345                         flag_wholeyear = 1;
 346                         break;
 347                 default:
 348                         usage();
 349                 }
 350
 351         argc -= optind;
 352         argv += optind;
 353
 354         switch (argc) {
 355         case 2:
 356                 if (flag_easter)
 357                         usage();
 358                 flag_month = *argv++;
 359                 flag_givenmonth = 1;
 360                 m = strtol(flag_month, NULL, 10);
 361                 /* FALLTHROUGH */
 362         case 1:
 363                 y = atoi(*argv);
 364                 if (y < 1 || y > 9999)
 365                         errx(EX_USAGE, "year `%s' not in range 1..9999", *argv);
 366                 argv++;
 367                 flag_givenyear = 1;
 368                 break;
 369         case 0:
 370                 if (flag_today != NULL) {
 371                         y = strtol(flag_today, NULL, 10);
 372                         m = strtol(flag_today + 5, NULL, 10);
 373                 } else {
 374                         time_t t;
 375                         struct tm *tm;
 376
 377                         t = time(NULL);
 378                         tm = localtime(&t);
 379                         y = tm->tm_year + 1900;
 380                         m = tm->tm_mon + 1;
 381                 }
 382                 break;
 383         default:
 384                 usage();
 385         }
 386
 387         if (flag_month != NULL) {
 388                 if (parsemonth(flag_month, &m, &y)) {
 389                         errx(EX_USAGE,
 390                             "%s is neither a month number (1..12) nor a name",
 391                             flag_month);
 392                 }
 393         }
 394
 395         /*
 396          * What is not supported:
 397          * -3 with -A or -B
 398          *      -3 displays 3 months, -A and -B change that behaviour.
 399          * -3 with -y
 400          *      -3 displays 3 months, -y says display a whole year.
 401          * -3 with a given year but no given month or without -m
 402          *      -3 displays 3 months, no month specified doesn't make clear
 403          *      which three months.
 404          * -m with a given month
 405          *      conflicting arguments, both specify the same field.
 406          * -y with -m
 407          *      -y displays the whole year, -m displays a single month.
 408          * -y with a given month
 409          *      -y displays the whole year, the given month displays a single
 410          *      month.
 411          * -y with -A or -B
 412          *      -y displays the whole year, -A and -B display extra months.
 413          */
 414
 415         /* -3 together with -A or -B. */
 416         if (flag_3months && (flag_after || flag_before))
 417                 errx(EX_USAGE, "-3 together with -A and -B is not supported.");
 418         /* -3 together with -y. */
 419         if (flag_3months && flag_wholeyear)
 420                 errx(EX_USAGE, "-3 together with -y is not supported.");
 421         /* -3 together with givenyear but no givenmonth. */
 422         if (flag_3months && flag_givenyear &&
 423             !(flag_givenmonth || flag_specifiedmonth))
 424                 errx(EX_USAGE,
 425                     "-3 together with a given year but no given month is "
 426                     "not supported.");
 427         /* -m together with xx xxxx. */
 428         if (flag_specifiedmonth && flag_givenmonth)
 429                 errx(EX_USAGE,
 430                     "-m together with a given month is not supported.");
 431         /* -y together with -m. */
 432         if (flag_wholeyear && flag_specifiedmonth)
 433                 errx(EX_USAGE, "-y together with -m is not supported.");
 434         /* -y together with xx xxxx. */
 435         if (flag_wholeyear && flag_givenmonth)
 436                 errx(EX_USAGE, "-y together a given month is not supported.");
 437         /* -y together with -A or -B. */
 438         if (flag_wholeyear && (flag_before > 0 || flag_after > 0))
 439                 errx(EX_USAGE, "-y together a -A or -B is not supported.");
 440         /* The rest should be fine. */
 441
 442         /* Select the period to display, in order of increasing priority .*/
 443         if (flag_wholeyear ||
 444             (flag_givenyear && !(flag_givenmonth || flag_specifiedmonth))) {
 445                 m = 1;
 446                 before = 0;
 447                 after = 11;
 448         }
 449         if (flag_givenyear && flag_givenmonth) {
 450                 before = 0;
 451                 after = 0;
 452         }
 453         if (flag_specifiedmonth) {
 454                 before = 0;
 455                 after = 0;
 456         }
 457         if (flag_before) {
 458                 before = flag_before;
 459         }
 460         if (flag_after) {
 461                 after = flag_after;
 462         }
 463         if (flag_3months) {
 464                 before = 1;
 465                 after = 1;
 466         }
 467         if (after == -1)
 468                 after = 0;
 469         if (before == -1)
 470                 before = 0;
 471
 472         /* Highlight a specified day or today .*/
 473         if (flag_highlightdate != NULL) {
 474                 dt.y = strtol(flag_highlightdate, NULL, 10);
 475                 dt.m = strtol(flag_highlightdate + 5, NULL, 10);
 476                 dt.d = strtol(flag_highlightdate + 8, NULL, 10);
 477         } else {
 478                 time_t t;
 479                 struct tm *tm1;
 480
 481                 t = time(NULL);
 482                 tm1 = localtime(&t);
 483                 dt.y = tm1->tm_year + 1900;
 484                 dt.m = tm1->tm_mon + 1;
 485                 dt.d = tm1->tm_mday;
 486         }
 487         highlightdate = sndaysb(&dt);
 488
 489         /* And now we finally start to calculate and output calendars. */
 490         if (flag_easter)
 491                 printeaster(y, flag_julian_cal, flag_orthodox);
 492         else
 493                 if (flag_backward)
 494                         monthrangeb(y, m, flag_julian_day, before, after);
 495                 else
 496                         monthranger(y, m, flag_julian_day, before, after);
 497         return (0);
 498 }
 499
 500 static void
 501 usage(void)
 502 {
 503
 504         fputs(
 505 "Usage: cal [general options] [-hjy] [[month] year]\n"
 506 "       cal [general options] [-hj] [-m month] [year]\n"
 507 "       ncal [general options] [-hJjpwy] [-s country_code] [[month] year]\n"
 508 "       ncal [general options] [-hJeo] [year]\n"
 509 "General options: [-NC3] [-A months] [-B months]\n"
 510 "For debug the highlighting: [-H yyyy-mm-dd] [-d yyyy-mm]\n",
 511             stderr);
 512         exit(EX_USAGE);
 513 }
 514
 515 /* Print the assumed switches for all countries. */
 516 void
 517 printcc(void)
 518 {
 519         struct djswitch *p;
 520         int n;  /* number of lines to print */
 521         int m;  /* offset from left to right table entry on the same line */
 522
 523 #define FSTR "%c%s %-15s%4d-%02d-%02d"
 524 #define DFLT(p) ((p) == dftswitch ? '*' : ' ')
 525 #define FSTRARG(p) DFLT(p), (p)->cc, (p)->nm, (p)->dt.y, (p)->dt.m, (p)->dt.d
 526
 527         n = sizeof(switches) / sizeof(struct djswitch);
 528         m = (n + 1) / 2;
 529         n /= 2;
 530         for (p = switches; p != switches + n; p++)
 531                 printf(FSTR"     "FSTR"\n", FSTRARG(p), FSTRARG(p+m));
 532         if (m != n)
 533                 printf(FSTR"\n", FSTRARG(p));
 534 }
 535
 536 /* Print the date of easter sunday. */
 537 void
 538 printeaster(int y, int julian, int orthodox)
 539 {
 540         date    dt;
 541         struct tm tm;
 542         char    buf[MAX_WIDTH];
 543         static int d_first = -1;
 544
 545         if (d_first < 0)
 546                 d_first = (*nl_langinfo(D_MD_ORDER) == 'd');
 547         /* force orthodox easter for years before 1583 */
 548         if (y < 1583)
 549                 orthodox = 1;
 550
 551         if (orthodox)
 552                 if (julian)
 553                         easteroj(y, &dt);
 554                 else
 555                         easterog(y, &dt);
 556         else
 557                 easterg(y, &dt);
 558
 559         memset(&tm, 0, sizeof(tm));
 560         tm.tm_year = dt.y - 1900;
 561         tm.tm_mon  = dt.m - 1;
 562         tm.tm_mday = dt.d;
 563         strftime(buf, sizeof(buf), d_first ? "%e %B %Y" : "%B %e %Y",  &tm);
 564         printf("%s\n", buf);
 565 }
 566
 567 #define MW(mw, me)              ((mw) + me)
 568 #define DECREASEMONTH(m, y)             \
 569                 if (--m == 0) {         \
 570                         m = 12;         \
 571                         y--;            \
 572                 }
 573 #define INCREASEMONTH(m, y)             \
 574                 if (++(m) == 13) {      \
 575                         (m) = 1;        \
 576                         (y)++;          \
 577                 }
 578 #define M2Y(m)  ((m) / 12)
 579 #define M2M(m)  (1 + (m) % 12)
 580
 581 /* Print all months for the period in the range [ before .. y-m .. after ]. */
 582 void
 583 monthrangeb(int y, int m, int jd_flag, int before, int after)
 584 {
 585         struct monthlines year[12];
 586         struct weekdays wds;
 587         char    s[MAX_WIDTH], t[MAX_WIDTH];
 588         wchar_t ws[MAX_WIDTH], ws1[MAX_WIDTH];
 589         const char      *wdss;
 590         int     i, j;
 591         int     mpl;
 592         int     mw;
 593         int     m1, m2;
 594         int     printyearheader;
 595         int     prevyear = -1;
 596
 597         mpl = jd_flag ? 2 : 3;
 598         mw = jd_flag ? MONTH_WIDTH_B_J : MONTH_WIDTH_B;
 599         wdss = (mpl == 2) ? " " : "";
 600
 601         while (before != 0) {
 602                 DECREASEMONTH(m, y);
 603                 before--;
 604                 after++;
 605         }
 606         m1 = y * 12 + m - 1;
 607         m2 = m1 + after;
 608
 609         mkweekdays(&wds);
 610
 611         /*
 612          * The year header is printed when there are more than 'mpl' months
 613          * and if the first month is a multitude of 'mpl'.
 614          * If not, it will print the year behind every month.
 615          */
 616         printyearheader = (after >= mpl - 1) && (M2M(m1) - 1) % mpl == 0;
 617
 618         m = m1;
 619         while (m <= m2) {
 620                 int count = 0;
 621                 for (i = 0; i != mpl && m + i <= m2; i++) {
 622                         mkmonthb(M2Y(m + i), M2M(m + i) - 1, jd_flag, year + i);
 623                         count++;
 624                 }
 625
 626                 /* Empty line between two rows of months */
 627                 if (m != m1)
 628                         printf("\n");
 629
 630                 /* Year at the top. */
 631                 if (printyearheader && M2Y(m) != prevyear) {
 632                         sprintf(s, "%d", M2Y(m));
 633                         printf("%s\n", center(t, s, mpl * mw));
 634                         prevyear = M2Y(m);
 635                 }
 636
 637                 /* Month names. */
 638                 for (i = 0; i < count; i++)
 639                         if (printyearheader)
 640                                 wprintf(L"%-*ls  ",
 641                                     mw, wcenter(ws, year[i].name, mw));
 642                         else {
 643                                 swprintf(ws, sizeof(ws), L"%-ls %d",
 644                                     year[i].name, M2Y(m + i));
 645                                 wprintf(L"%-*ls  ", mw, wcenter(ws1, ws, mw));
 646                         }
 647                 printf("\n");
 648
 649                 /* Day of the week names. */
 650                 for (i = 0; i < count; i++) {
 651                         wprintf(L"%s%ls%s%ls%s%ls%s%ls%s%ls%s%ls%s%ls ",
 652                                 wdss, wds.names[6], wdss, wds.names[0],
 653                                 wdss, wds.names[1], wdss, wds.names[2],
 654                                 wdss, wds.names[3], wdss, wds.names[4],
 655                                 wdss, wds.names[5]);
 656                 }
 657                 printf("\n");
 658
 659                 /* And the days of the month. */
 660                 for (i = 0; i != 6; i++) {
 661                         for (j = 0; j < count; j++)
 662                                 printf("%-*s  ",
 663                                     MW(mw, year[j].extralen[i]),
 664                                         year[j].lines[i]+1);
 665                         printf("\n");
 666                 }
 667
 668                 m += mpl;
 669         }
 670 }
 671
 672 void
 673 monthranger(int y, int m, int jd_flag, int before, int after)
 674 {
 675         struct monthlines year[12];
 676         struct weekdays wds;
 677         char    s[MAX_WIDTH], t[MAX_WIDTH];
 678         int     i, j;
 679         int     mpl;
 680         int     mw;
 681         int     m1, m2;
 682         int     prevyear = -1;
 683         int     printyearheader;
 684
 685         mpl = jd_flag ? 3 : 4;
 686         mw = jd_flag ? MONTH_WIDTH_R_J : MONTH_WIDTH_R;
 687
 688         while (before != 0) {
 689                 DECREASEMONTH(m, y);
 690                 before--;
 691                 after++;
 692         }
 693         m1 = y * 12 + m - 1;
 694         m2 = m1 + after;
 695
 696         mkweekdays(&wds);
 697
 698         /*
 699          * The year header is printed when there are more than 'mpl' months
 700          * and if the first month is a multitude of 'mpl'.
 701          * If not, it will print the year behind every month.
 702          */
 703         printyearheader = (after >= mpl - 1) && (M2M(m1) - 1) % mpl == 0;
 704
 705         m = m1;
 706         while (m <= m2) {
 707                 int count = 0;
 708                 for (i = 0; i != mpl && m + i <= m2; i++) {
 709                         mkmonthr(M2Y(m + i), M2M(m + i) - 1, jd_flag, year + i);
 710                         count++;
 711                 }
 712
 713                 /* Empty line between two rows of months. */
 714                 if (m != m1)
 715                         printf("\n");
 716
 717                 /* Year at the top. */
 718                 if (printyearheader && M2Y(m) != prevyear) {
 719                         sprintf(s, "%d", M2Y(m));
 720                         printf("%s\n", center(t, s, mpl * mw));
 721                         prevyear = M2Y(m);
 722                 }
 723
 724                 /* Month names. */
 725                 wprintf(L"    ");
 726                 for (i = 0; i < count; i++)
 727                         if (printyearheader)
 728                                 wprintf(L"%-*ls", mw, year[i].name);
 729                         else
 730                                 wprintf(L"%-ls %-*d", year[i].name,
 731                                     mw - wcslen(year[i].name) - 1, M2Y(m + i));
 732                 printf("\n");
 733
 734                 /* And the days of the month. */
 735                 for (i = 0; i != 7; i++) {
 736                         /* Week day */
 737                         wprintf(L"%.2ls", wds.names[i]);
 738
 739                         /* Full months */
 740                         for (j = 0; j < count; j++)
 741                                 printf("%-*s",
 742                                     MW(mw, year[j].extralen[i]),
 743                                         year[j].lines[i]);
 744                         printf("\n");
 745                 }
 746
 747                 /* Week numbers. */
 748                 if (flag_weeks) {
 749                         printf("  ");
 750                         for (i = 0; i < count; i++)
 751                                 printf("%-*s", mw, year[i].weeks);
 752                         printf("\n");
 753                 }
 754
 755                 m += mpl;
 756         }
 757         return;
 758 }
 759
 760 void
 761 mkmonthr(int y, int m, int jd_flag, struct monthlines *mlines)
 762 {
 763
 764         struct tm tm;           /* for strftime printing local names of
 765                                  * months */
 766         date    dt;             /* handy date */
 767         int     dw;             /* width of numbers */
 768         int     first;          /* first day of month */
 769         int     firstm;         /* first day of first week of month */
 770         int     i, j, k, l;     /* just indices */
 771         int     last;           /* the first day of next month */
 772         int     jan1 = 0;       /* the first day of this year */
 773         char   *ds;             /* pointer to day strings (daystr or
 774                                  * jdaystr) */
 775
 776         /* Set name of month. */
 777         memset(&tm, 0, sizeof(tm));
 778         tm.tm_mon = m;
 779         wcsftime(mlines->name, sizeof(mlines->name) / sizeof(mlines->name[0]),
 780                  L"%OB", &tm);
 781         mlines->name[0] = towupper(mlines->name[0]);
 782
 783         /*
 784          * Set first and last to the day number of the first day of this
 785          * month and the first day of next month respectively. Set jan1 to
 786          * the day number of the first day of this year.
 787          */
 788         first = firstday(y, m + 1);
 789         if (m == 11)
 790                 last = firstday(y + 1, 1);
 791         else
 792                 last = firstday(y, m + 2);
 793
 794         if (jd_flag)
 795                 jan1 = firstday(y, 1);
 796
 797         /*
 798          * Set firstm to the day number of monday of the first week of
 799          * this month. (This might be in the last month)
 800          */
 801         firstm = first - weekday(first);
 802
 803         /* Set ds (daystring) and dw (daywidth) according to the jd_flag. */
 804         if (jd_flag) {
 805                 ds = jdaystr;
 806                 dw = 4;
 807         } else {
 808                 ds = daystr;
 809                 dw = 3;
 810         }
 811
 812         /*
 813          * Fill the lines with day of month or day of year (julian day)
 814          * line index: i, each line is one weekday. column index: j, each
 815          * column is one day number. print column index: k.
 816          */
 817         for (i = 0; i != 7; i++) {
 818                 l = 0;
 819                 for (j = firstm + i, k = 0; j < last; j += 7, k += dw) {
 820                         if (j >= first) {
 821                                 if (jd_flag)
 822                                         dt.d = j - jan1 + 1;
 823                                 else
 824                                         sdater(j, &dt);
 825                                 if (j == highlightdate && !flag_nohighlight
 826                                  && isatty(STDOUT_FILENO))
 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 }