Merge from vendor branch NCURSES:
[dragonfly.git] / usr.bin / calendar / day.c
1 /*
2  * Copyright (c) 1989, 1993, 1994
3  *      The Regents of the University of California.  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  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *      This product includes software developed by the University of
16  *      California, Berkeley and its contributors.
17  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  *
33  * $FreeBSD: src/usr.bin/calendar/day.c,v 1.13.2.5 2003/04/06 20:04:57 dwmalone Exp $
34  * $DragonFly: src/usr.bin/calendar/day.c,v 1.4 2005/05/07 22:05:41 corecode Exp $
35  */
36
37 #include <sys/types.h>
38 #include <sys/uio.h>
39 #include <ctype.h>
40 #include <err.h>
41 #include <locale.h>
42 #include <stdio.h>
43 #include <stdlib.h>
44 #include <string.h>
45 #include <time.h>
46
47 #include "pathnames.h"
48 #include "calendar.h"
49
50 extern struct iovec header[];
51
52 struct tm *tp;
53 int *cumdays, offset, yrdays;
54 char dayname[10];
55
56
57 /* 1-based month, 0-based days, cumulative */
58 int daytab[][14] = {
59         { 0, -1, 30, 58, 89, 119, 150, 180, 211, 242, 272, 303, 333, 364 },
60         { 0, -1, 30, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334, 365 },
61 };
62
63 static char const *days[] = {
64         "sun", "mon", "tue", "wed", "thu", "fri", "sat", NULL,
65 };
66
67 static const char *months[] = {
68         "jan", "feb", "mar", "apr", "may", "jun",
69         "jul", "aug", "sep", "oct", "nov", "dec", NULL,
70 };
71
72 static struct fixs fndays[8];         /* full national days names */
73 static struct fixs ndays[8];          /* short national days names */
74
75 static struct fixs fnmonths[13];      /* full national months names */
76 static struct fixs nmonths[13];       /* short national month names */
77
78
79 void setnnames(void)
80 {
81         char buf[80];
82         int i, l;
83         struct tm tm;
84
85         for (i = 0; i < 7; i++) {
86                 tm.tm_wday = i;
87                 strftime(buf, sizeof(buf), "%a", &tm);
88                 for (l = strlen(buf);
89                      l > 0 && isspace((unsigned char)buf[l - 1]);
90                      l--)
91                         ;
92                 buf[l] = '\0';
93                 if (ndays[i].name != NULL)
94                         free(ndays[i].name);
95                 if ((ndays[i].name = strdup(buf)) == NULL)
96                         errx(1, "cannot allocate memory");
97                 ndays[i].len = strlen(buf);
98
99                 strftime(buf, sizeof(buf), "%A", &tm);
100                 for (l = strlen(buf);
101                      l > 0 && isspace((unsigned char)buf[l - 1]);
102                      l--)
103                         ;
104                 buf[l] = '\0';
105                 if (fndays[i].name != NULL)
106                         free(fndays[i].name);
107                 if ((fndays[i].name = strdup(buf)) == NULL)
108                         errx(1, "cannot allocate memory");
109                 fndays[i].len = strlen(buf);
110         }
111
112         for (i = 0; i < 12; i++) {
113                 tm.tm_mon = i;
114                 strftime(buf, sizeof(buf), "%b", &tm);
115                 for (l = strlen(buf);
116                      l > 0 && isspace((unsigned char)buf[l - 1]);
117                      l--)
118                         ;
119                 buf[l] = '\0';
120                 if (nmonths[i].name != NULL)
121                         free(nmonths[i].name);
122                 if ((nmonths[i].name = strdup(buf)) == NULL)
123                         errx(1, "cannot allocate memory");
124                 nmonths[i].len = strlen(buf);
125
126                 strftime(buf, sizeof(buf), "%B", &tm);
127                 for (l = strlen(buf);
128                      l > 0 && isspace((unsigned char)buf[l - 1]);
129                      l--)
130                         ;
131                 buf[l] = '\0';
132                 if (fnmonths[i].name != NULL)
133                         free(fnmonths[i].name);
134                 if ((fnmonths[i].name = strdup(buf)) == NULL)
135                         errx(1, "cannot allocate memory");
136                 fnmonths[i].len = strlen(buf);
137         }
138 }
139
140 void
141 settime(time_t now)
142 {
143         char *oldl, *lbufp;
144
145         tp = localtime(&now);
146         if ( isleap(tp->tm_year + 1900) ) {
147                 yrdays = 366;
148                 cumdays = daytab[1];
149         } else {
150                 yrdays = 365;
151                 cumdays = daytab[0];
152         }
153         /* Friday displays Monday's events */
154         offset = tp->tm_wday == Friday ? 3 : 1;
155         header[5].iov_base = dayname;
156
157         oldl = NULL;
158         lbufp = setlocale(LC_TIME, NULL);
159         if (lbufp != NULL && (oldl = strdup(lbufp)) == NULL)
160                 errx(1, "cannot allocate memory");
161         (void) setlocale(LC_TIME, "C");
162         header[5].iov_len = strftime(dayname, sizeof(dayname), "%A", tp);
163         (void) setlocale(LC_TIME, (oldl != NULL ? oldl : ""));
164         if (oldl != NULL)
165                 free(oldl);
166
167         setnnames();
168 }
169
170 /* convert Day[/Month][/Year] into unix time (since 1970)
171  * Day: two digits, Month: two digits, Year: digits
172  */
173 time_t Mktime (char *dp)
174 {
175     time_t t;
176     int d, m, y;
177     struct tm tm;
178
179     (void)time(&t);
180     tp = localtime(&t);
181
182     tm.tm_sec = 0;
183     tm.tm_min = 0;
184     tm.tm_hour = 0;
185     tm.tm_wday = 0;
186     tm.tm_mday = tp->tm_mday;
187     tm.tm_mon = tp->tm_mon;
188     tm.tm_year = tp->tm_year;
189
190     switch (sscanf(dp, "%d.%d.%d", &d, &m, &y)) {
191     case 3:
192         if (y > 1900)
193             y -= 1900;
194         tm.tm_year = y;
195         /* FALLTHROUGH */
196     case 2:
197         tm.tm_mon = m - 1;
198         /* FALLTHROUGH */
199     case 1:
200         tm.tm_mday = d;
201     }
202
203 #ifdef DEBUG
204     fprintf(stderr, "Mktime: %d %d %s\n", (int)mktime(&tm), (int)t,
205            asctime(&tm));
206 #endif
207     return(mktime(&tm));
208 }
209
210 /*
211  * Possible date formats include any combination of:
212  *      3-charmonth                     (January, Jan, Jan)
213  *      3-charweekday                   (Friday, Monday, mon.)
214  *      numeric month or day            (1, 2, 04)
215  *
216  * Any character may separate them, or they may not be separated.  Any line,
217  * following a line that is matched, that starts with "whitespace", is shown
218  * along with the matched line.
219  */
220 int
221 isnow(char *endp, int *monthp, int *dayp, int *varp)
222 {
223         int day, flags, month = 0, v1, v2;
224
225         /*
226          * CONVENTION
227          *
228          * Month:     1-12
229          * Monthname: Jan .. Dec
230          * Day:       1-31
231          * Weekday:   Mon-Sun
232          *
233          */
234
235         flags = 0;
236
237         /* read first field */
238         /* didn't recognize anything, skip it */
239         if (!(v1 = getfield(endp, &endp, &flags)))
240                 return (0);
241
242         /* Easter or Easter depending days */
243         if (flags & F_EASTER)
244             day = v1 - 1; /* days since January 1 [0-365] */
245
246          /*
247           * 1. {Weekday,Day} XYZ ...
248           *
249           *    where Day is > 12
250           */
251         else if (flags & F_ISDAY || v1 > 12) {
252
253                 /* found a day; day: 1-31 or weekday: 1-7 */
254                 day = v1;
255
256                 /* {Day,Weekday} {Month,Monthname} ... */
257                 /* if no recognizable month, assume just a day alone
258                  * in other words, find month or use current month */
259                 if (!(month = getfield(endp, &endp, &flags)))
260                         month = tp->tm_mon + 1;
261         }
262
263         /* 2. {Monthname} XYZ ... */
264         else if (flags & F_ISMONTH) {
265                 month = v1;
266
267                 /* Monthname {day,weekday} */
268                 /* if no recognizable day, assume the first day in month */
269                 if (!(day = getfield(endp, &endp, &flags)))
270                         day = 1;
271         }
272
273         /* Hm ... */
274         else {
275                 v2 = getfield(endp, &endp, &flags);
276
277                 /*
278                  * {Day} {Monthname} ...
279                  * where Day <= 12
280                  */
281                 if (flags & F_ISMONTH) {
282                         day = v1;
283                         month = v2;
284                         *varp = 0;
285                 }
286
287                 /* {Month} {Weekday,Day} ...  */
288                 else {
289                         /* F_ISDAY set, v2 > 12, or no way to tell */
290                         month = v1;
291                         /* if no recognizable day, assume the first */
292                         day = v2 ? v2 : 1;
293                         *varp = 0;
294                 }
295         }
296
297         /* convert Weekday into *next*  Day,
298          * e.g.: 'Sunday' -> 22
299          *       'SundayLast' -> ??
300          */
301         if (flags & F_ISDAY) {
302 #ifdef DEBUG
303             fprintf(stderr, "\nday: %d %s month %d\n", day, endp, month);
304 #endif
305
306             *varp = 1;
307             /* variable weekday, SundayLast, MondayFirst ... */
308             if (day < 0 || day >= 10) {
309
310                 /* negative offset; last, -4 .. -1 */
311                 if (day < 0) {
312                     v1 = day/10 - 1;          /* offset -4 ... -1 */
313                     day = 10 + (day % 10);    /* day 1 ... 7 */
314
315                     /* day, eg '22nd' */
316                     v2 = tp->tm_mday + (((day - 1) - tp->tm_wday + 7) % 7);
317
318                     /* (month length - day) / 7 + 1 */
319                     if (cumdays[month+1] - cumdays[month] >= v2
320                         && ((int)((cumdays[month+1] -
321                                cumdays[month] - v2) / 7) + 1) == -v1)
322                         /* bingo ! */
323                         day = v2;
324
325                     /* set to yesterday */
326                     else {
327                         day = tp->tm_mday - 1;
328                         if (day == 0)
329                             return (0);
330                     }
331                 }
332
333                 /* first, second ... +1 ... +5 */
334                 else {
335                     v1 = day/10;        /* offset: +1 (first Sunday) ... */
336                     day = day % 10;
337
338                     /* day, eg '22th' */
339                     v2 = tp->tm_mday + (((day - 1) - tp->tm_wday + 7) % 7);
340
341                     /* Hurrah! matched */
342                     if ( ((v2 - 1 + 7) / 7) == v1 )
343                         day = v2;
344
345                     /* set to yesterday */
346                     else {
347                         day = tp->tm_mday - 1;
348                         if (day == 0)
349                             return (0);
350                     }
351                 }
352             }
353
354             /* wired */
355             else {
356                 day = tp->tm_mday + (((day - 1) - tp->tm_wday + 7) % 7);
357                 *varp = 1;
358             }
359         }
360
361         if (!(flags & F_EASTER)) {
362             *monthp = month;
363             *dayp = day;
364             day = cumdays[month] + day;
365         }
366         else {
367             for (v1 = 0; day > cumdays[v1]; v1++)
368                 ;
369             *monthp = v1 - 1;
370             *dayp = day - cumdays[v1 - 1];
371             *varp = 1;
372         }
373
374 #ifdef DEBUG
375         fprintf(stderr, "day2: day %d(%d-%d) yday %d\n", *dayp, day, cumdays[month], tp->tm_yday);
376 #endif
377         /* if today or today + offset days */
378         if (day >= tp->tm_yday - f_dayBefore &&
379             day <= tp->tm_yday + offset + f_dayAfter)
380                 return (1);
381
382         /* if number of days left in this year + days to event in next year */
383         if (yrdays - tp->tm_yday + day <= offset + f_dayAfter ||
384             /* a year backward, eg. 6 Jan and 10 days before -> 27. Dec */
385             tp->tm_yday + day - f_dayBefore < 0
386             )
387                 return (1);
388         return (0);
389 }
390
391
392 int
393 getmonth(char *s)
394 {
395         const char **p;
396         struct fixs *n;
397
398         for (n = fnmonths; n->name; ++n)
399                 if (!strncasecmp(s, n->name, n->len))
400                         return ((n - fnmonths) + 1);
401         for (n = nmonths; n->name; ++n)
402                 if (!strncasecmp(s, n->name, n->len))
403                         return ((n - nmonths) + 1);
404         for (p = months; *p; ++p)
405                 if (!strncasecmp(s, *p, 3))
406                         return ((p - months) + 1);
407         return (0);
408 }
409
410
411 int
412 getday(char *s)
413 {
414         const char **p;
415         struct fixs *n;
416
417         for (n = fndays; n->name; ++n)
418                 if (!strncasecmp(s, n->name, n->len))
419                         return ((n - fndays) + 1);
420         for (n = ndays; n->name; ++n)
421                 if (!strncasecmp(s, n->name, n->len))
422                         return ((n - ndays) + 1);
423         for (p = days; *p; ++p)
424                 if (!strncasecmp(s, *p, 3))
425                         return ((p - days) + 1);
426         return (0);
427 }
428
429 /* return offset for variable weekdays
430  * -1 -> last weekday in month
431  * +1 -> first weekday in month
432  * ... etc ...
433  */
434 int
435 getdayvar(char *s)
436 {
437         int offs;
438
439
440         offs = strlen(s);
441
442
443         /* Sun+1 or Wednesday-2
444          *    ^              ^   */
445
446         /* fprintf(stderr, "x: %s %s %d\n", s, s + offs - 2, offs); */
447         switch(*(s + offs - 2)) {
448         case '-':
449             return(-(atoi(s + offs - 1)));
450             break;
451         case '+':
452             return(atoi(s + offs - 1));
453             break;
454         }
455
456
457         /*
458          * some aliases: last, first, second, third, fourth
459          */
460
461         /* last */
462         if      (offs > 4 && !strcasecmp(s + offs - 4, "last"))
463             return(-1);
464         else if (offs > 5 && !strcasecmp(s + offs - 5, "first"))
465             return(+1);
466         else if (offs > 6 && !strcasecmp(s + offs - 6, "second"))
467             return(+2);
468         else if (offs > 5 && !strcasecmp(s + offs - 5, "third"))
469             return(+3);
470         else if (offs > 6 && !strcasecmp(s + offs - 6, "fourth"))
471             return(+4);
472
473
474         /* no offset detected */
475         return(0);
476 }