ba70a6b50edc6ed287b9c07637bce0df5c7a90b6
[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.5 2006/09/16 18:38:00 pavalos 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; 
54
55 static char dayname[10];
56 static int offset, yrdays;
57
58 /* 1-based month, 0-based days, cumulative */
59 static int daytab[][14] = {
60         { 0, -1, 30, 58, 89, 119, 150, 180, 211, 242, 272, 303, 333, 364 },
61         { 0, -1, 30, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334, 365 },
62 };
63
64 static char const *days[] = {
65         "sun", "mon", "tue", "wed", "thu", "fri", "sat", NULL,
66 };
67
68 static const char *months[] = {
69         "jan", "feb", "mar", "apr", "may", "jun",
70         "jul", "aug", "sep", "oct", "nov", "dec", NULL,
71 };
72
73 static struct fixs fndays[8];         /* full national days names */
74 static struct fixs ndays[8];          /* short national days names */
75
76 static struct fixs fnmonths[13];      /* full national months names */
77 static struct fixs nmonths[13];       /* short national month names */
78
79 void
80 setnnames(void)
81 {
82         char buf[80];
83         int i, l;
84         struct tm tm;
85
86         for (i = 0; i < 7; i++) {
87                 tm.tm_wday = i;
88                 strftime(buf, sizeof(buf), "%a", &tm);
89                 for (l = strlen(buf);
90                      l > 0 && isspace((unsigned char)buf[l - 1]);
91                      l--)
92                         ;
93                 buf[l] = '\0';
94                 if (ndays[i].name != NULL)
95                         free(ndays[i].name);
96                 if ((ndays[i].name = strdup(buf)) == NULL)
97                         errx(EXIT_FAILURE, "cannot allocate memory");
98                 ndays[i].len = strlen(buf);
99
100                 strftime(buf, sizeof(buf), "%A", &tm);
101                 for (l = strlen(buf);
102                      l > 0 && isspace((unsigned char)buf[l - 1]);
103                      l--)
104                         ;
105                 buf[l] = '\0';
106                 if (fndays[i].name != NULL)
107                         free(fndays[i].name);
108                 if ((fndays[i].name = strdup(buf)) == NULL)
109                         errx(EXIT_FAILURE, "cannot allocate memory");
110                 fndays[i].len = strlen(buf);
111         }
112
113         for (i = 0; i < 12; i++) {
114                 tm.tm_mon = i;
115                 strftime(buf, sizeof(buf), "%b", &tm);
116                 for (l = strlen(buf);
117                      l > 0 && isspace((unsigned char)buf[l - 1]);
118                      l--)
119                         ;
120                 buf[l] = '\0';
121                 if (nmonths[i].name != NULL)
122                         free(nmonths[i].name);
123                 if ((nmonths[i].name = strdup(buf)) == NULL)
124                         errx(EXIT_FAILURE, "cannot allocate memory");
125                 nmonths[i].len = strlen(buf);
126
127                 strftime(buf, sizeof(buf), "%B", &tm);
128                 for (l = strlen(buf);
129                      l > 0 && isspace((unsigned char)buf[l - 1]);
130                      l--)
131                         ;
132                 buf[l] = '\0';
133                 if (fnmonths[i].name != NULL)
134                         free(fnmonths[i].name);
135                 if ((fnmonths[i].name = strdup(buf)) == NULL)
136                         errx(EXIT_FAILURE, "cannot allocate memory");
137                 fnmonths[i].len = strlen(buf);
138         }
139 }
140
141 void
142 settime(time_t now)
143 {
144         char *oldl, *lbufp;
145
146         tp = localtime(&now);
147         if (isleap(tp->tm_year + 1900)) {
148                 yrdays = 366;
149                 cumdays = daytab[1];
150         } else {
151                 yrdays = 365;
152                 cumdays = daytab[0];
153         }
154         /* Friday displays Monday's events */
155         offset = tp->tm_wday == Friday ? 3 : 1;
156         header[5].iov_base = dayname;
157
158         oldl = NULL;
159         lbufp = setlocale(LC_TIME, NULL);
160         if (lbufp != NULL && (oldl = strdup(lbufp)) == NULL)
161                 errx(EXIT_FAILURE, "cannot allocate memory");
162         setlocale(LC_TIME, "C");
163         header[5].iov_len = strftime(dayname, sizeof(dayname), "%A", tp);
164         setlocale(LC_TIME, (oldl != NULL ? oldl : ""));
165         if (oldl != NULL)
166                 free(oldl);
167
168         setnnames();
169 }
170
171 /* convert Day[/Month][/Year] into unix time (since 1970)
172  * Day: two digits, Month: two digits, Year: digits
173  */
174 time_t
175 Mktime(char *dp)
176 {
177     time_t t;
178     int d, m, y;
179     struct tm tm;
180
181     time(&t);
182     tp = localtime(&t);
183
184     tm.tm_sec = 0;
185     tm.tm_min = 0;
186     tm.tm_hour = 0;
187     tm.tm_wday = 0;
188     tm.tm_mday = tp->tm_mday;
189     tm.tm_mon = tp->tm_mon;
190     tm.tm_year = tp->tm_year;
191
192     switch (sscanf(dp, "%d.%d.%d", &d, &m, &y)) {
193     case 3:
194         if (y > 1900)
195             y -= 1900;
196         tm.tm_year = y;
197         /* FALLTHROUGH */
198     case 2:
199         tm.tm_mon = m - 1;
200         /* FALLTHROUGH */
201     case 1:
202         tm.tm_mday = d;
203     }
204
205 #ifdef DEBUG
206     fprintf(stderr, "Mktime: %d %d %s\n", (int)mktime(&tm), (int)t,
207            asctime(&tm));
208 #endif
209     return(mktime(&tm));
210 }
211
212 /*
213  * Possible date formats include any combination of:
214  *      3-charmonth                     (January, Jan, Jan)
215  *      3-charweekday                   (Friday, Monday, mon.)
216  *      numeric month or day            (1, 2, 04)
217  *
218  * Any character may separate them, or they may not be separated.  Any line,
219  * following a line that is matched, that starts with "whitespace", is shown
220  * along with the matched line.
221  */
222 int
223 isnow(char *endp, int *monthp, int *dayp, int *varp)
224 {
225         int day, flags, month = 0, v1, v2;
226
227         /*
228          * CONVENTION
229          *
230          * Month:     1-12
231          * Monthname: Jan .. Dec
232          * Day:       1-31
233          * Weekday:   Mon-Sun
234          *
235          */
236
237         flags = 0;
238
239         /* read first field */
240         /* didn't recognize anything, skip it */
241         if (!(v1 = getfield(endp, &endp, &flags)))
242                 return (0);
243
244         /* Easter or Easter depending days */
245         if (flags & F_EASTER)
246             day = v1 - 1; /* days since January 1 [0-365] */
247
248          /*
249           * 1. {Weekday,Day} XYZ ...
250           *
251           *    where Day is > 12
252           */
253         else if (flags & F_ISDAY || v1 > 12) {
254
255                 /* found a day; day: 1-31 or weekday: 1-7 */
256                 day = v1;
257
258                 /* {Day,Weekday} {Month,Monthname} ... */
259                 /* if no recognizable month, assume just a day alone
260                  * in other words, find month or use current month */
261                 if (!(month = getfield(endp, &endp, &flags)))
262                         month = tp->tm_mon + 1;
263         }
264
265         /* 2. {Monthname} XYZ ... */
266         else if (flags & F_ISMONTH) {
267                 month = v1;
268
269                 /* Monthname {day,weekday} */
270                 /* if no recognizable day, assume the first day in month */
271                 if (!(day = getfield(endp, &endp, &flags)))
272                         day = 1;
273         }
274
275         /* Hm ... */
276         else {
277                 v2 = getfield(endp, &endp, &flags);
278
279                 /*
280                  * {Day} {Monthname} ...
281                  * where Day <= 12
282                  */
283                 if (flags & F_ISMONTH) {
284                         day = v1;
285                         month = v2;
286                         *varp = 0;
287                 }
288
289                 /* {Month} {Weekday,Day} ...  */
290                 else {
291                         /* F_ISDAY set, v2 > 12, or no way to tell */
292                         month = v1;
293                         /* if no recognizable day, assume the first */
294                         day = v2 ? v2 : 1;
295                         *varp = 0;
296                 }
297         }
298
299         /* convert Weekday into *next*  Day,
300          * e.g.: 'Sunday' -> 22
301          *       'SundayLast' -> ??
302          */
303         if (flags & F_ISDAY) {
304 #ifdef DEBUG
305             fprintf(stderr, "\nday: %d %s month %d\n", day, endp, month);
306 #endif
307
308             *varp = 1;
309             /* variable weekday, SundayLast, MondayFirst ... */
310             if (day < 0 || day >= 10) {
311
312                 /* negative offset; last, -4 .. -1 */
313                 if (day < 0) {
314                     v1 = day/10 - 1;          /* offset -4 ... -1 */
315                     day = 10 + (day % 10);    /* day 1 ... 7 */
316
317                     /* day, eg '22nd' */
318                     v2 = tp->tm_mday + (((day - 1) - tp->tm_wday + 7) % 7);
319
320                     /* (month length - day) / 7 + 1 */
321                     if (cumdays[month+1] - cumdays[month] >= v2
322                         && ((int)((cumdays[month+1] -
323                                cumdays[month] - v2) / 7) + 1) == -v1)
324                         /* bingo ! */
325                         day = v2;
326
327                     /* set to yesterday */
328                     else {
329                         day = tp->tm_mday - 1;
330                         if (day == 0)
331                             return (0);
332                     }
333                 }
334
335                 /* first, second ... +1 ... +5 */
336                 else {
337                     v1 = day/10;        /* offset: +1 (first Sunday) ... */
338                     day = day % 10;
339
340                     /* day, eg '22th' */
341                     v2 = tp->tm_mday + (((day - 1) - tp->tm_wday + 7) % 7);
342
343                     /* Hurrah! matched */
344                     if (((v2 - 1 + 7) / 7) == v1)
345                         day = v2;
346
347                     /* set to yesterday */
348                     else {
349                         day = tp->tm_mday - 1;
350                         if (day == 0)
351                             return (0);
352                     }
353                 }
354             }
355
356             /* wired */
357             else {
358                 day = tp->tm_mday + (((day - 1) - tp->tm_wday + 7) % 7);
359                 *varp = 1;
360             }
361         }
362
363         if (!(flags & F_EASTER)) {
364             *monthp = month;
365             *dayp = day;
366             day = cumdays[month] + day;
367         }
368         else {
369             for (v1 = 0; day > cumdays[v1]; v1++)
370                 ;
371             *monthp = v1 - 1;
372             *dayp = day - cumdays[v1 - 1];
373             *varp = 1;
374         }
375
376 #ifdef DEBUG
377         fprintf(stderr, "day2: day %d(%d-%d) yday %d\n", *dayp, day, cumdays[month], tp->tm_yday);
378 #endif
379         /* if today or today + offset days */
380         if (day >= tp->tm_yday - f_dayBefore &&
381             day <= tp->tm_yday + offset + f_dayAfter)
382                 return(1);
383
384         /* if number of days left in this year + days to event in next year */
385         if (yrdays - tp->tm_yday + day <= offset + f_dayAfter ||
386             /* a year backward, eg. 6 Jan and 10 days before -> 27. Dec */
387             tp->tm_yday + day - f_dayBefore < 0
388             )
389                 return(1);
390         return(0);
391 }
392
393
394 int
395 getmonth(char *s)
396 {
397         const char **p;
398         struct fixs *n;
399
400         for (n = fnmonths; n->name; ++n)
401                 if (!strncasecmp(s, n->name, n->len))
402                         return ((n - fnmonths) + 1);
403         for (n = nmonths; n->name; ++n)
404                 if (!strncasecmp(s, n->name, n->len))
405                         return ((n - nmonths) + 1);
406         for (p = months; *p; ++p)
407                 if (!strncasecmp(s, *p, 3))
408                         return ((p - months) + 1);
409         return(0);
410 }
411
412
413 int
414 getday(char *s)
415 {
416         const char **p;
417         struct fixs *n;
418
419         for (n = fndays; n->name; ++n)
420                 if (!strncasecmp(s, n->name, n->len))
421                         return((n - fndays) + 1);
422         for (n = ndays; n->name; ++n)
423                 if (!strncasecmp(s, n->name, n->len))
424                         return((n - ndays) + 1);
425         for (p = days; *p; ++p)
426                 if (!strncasecmp(s, *p, 3))
427                         return((p - days) + 1);
428         return(0);
429 }
430
431 /* return offset for variable weekdays
432  * -1 -> last weekday in month
433  * +1 -> first weekday in month
434  * ... etc ...
435  */
436 int
437 getdayvar(char *s)
438 {
439         int offs;
440
441
442         offs = strlen(s);
443
444
445         /* Sun+1 or Wednesday-2
446          *    ^              ^   */
447
448         /* fprintf(stderr, "x: %s %s %d\n", s, s + offs - 2, offs); */
449         switch(*(s + offs - 2)) {
450         case '-':
451             return(-(atoi(s + offs - 1)));
452             break;
453         case '+':
454             return(atoi(s + offs - 1));
455             break;
456         }
457
458
459         /*
460          * some aliases: last, first, second, third, fourth
461          */
462
463         /* last */
464         if      (offs > 4 && !strcasecmp(s + offs - 4, "last"))
465             return(-1);
466         else if (offs > 5 && !strcasecmp(s + offs - 5, "first"))
467             return(+1);
468         else if (offs > 6 && !strcasecmp(s + offs - 6, "second"))
469             return(+2);
470         else if (offs > 5 && !strcasecmp(s + offs - 5, "third"))
471             return(+3);
472         else if (offs > 6 && !strcasecmp(s + offs - 6, "fourth"))
473             return(+4);
474
475
476         /* no offset detected */
477         return(0);
478 }