aboutsummaryrefslogtreecommitdiff
path: root/toys/cal.c
blob: 1c018e2f35a63d5d70c94e449dc0a04d4138332c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
/* vi: set sw=4 ts=4:
 *
 * cal.c - show calendar.
 *
 * Copyright 2011 Rob Landley <rob@landley.net>
 *
 * See http://opengroup.org/onlinepubs/9699919799/utilities/cal.html

USE_CAL(NEWTOY(cal, ">2", TOYFLAG_USR|TOYFLAG_BIN))

config CAL
	bool "cal"
	default y
	help
	  usage: cal [[month] year]
	  Print a calendar.

	  With one argument, prints all months of the specified year.
	  With two arguments, prints calendar for month and year.
*/

#include "toys.h"

// Write calendar into buffer: each line is 20 chars wide, end indicated
// by empty string.

static char *calstrings(char *buf, struct tm *tm)
{
	char temp[21];
	int wday, mday, start, len, line;

	// header
	len = strftime(temp, 21, "%B %Y", tm);
	len += (20-len)/2;
	buf += sprintf(buf, "%*s%*s ", len, temp, 20-len, "");
	buf++;
	buf += sprintf(buf, "Su Mo Tu We Th Fr Sa ");
	buf++;

	// What day of the week does this month start on?
	if (tm->tm_mday>1)
		start = (36+tm->tm_wday-tm->tm_mday)%7;
	else start = tm->tm_wday;

	// What day does this month end on?  Alas, libc doesn't tell us...
	len = 31;
	if (tm->tm_mon == 1) {
		int year = tm->tm_year;
		len = 28;
		if (!(year & 3) && !((year&100) && !(year&400))) len++;
	} else if ((tm->tm_mon+(tm->tm_mon>6 ? 1 : 0)) & 1) len = 30;

	for (mday=line=0;line<6;line++) {
		for (wday=0; wday<7; wday++) {
			char *pat = "   ";
			if (!mday ? wday==start : mday<len) {
				pat = "%2d ";
				mday++;
			}
			buf += sprintf(buf, pat, mday);
		}
		buf++;
	}

	return buf;
}

void xcheckrange(long val, long low, long high)
{
	char *err = "%ld %s than %ld";

	if (val < low) error_exit(err, val, "less", low);
	if (val > high) error_exit(err, val, "greater", high);
}

// Worst case scenario toybuf usage: sizeof(struct tm) plus 21 bytes/line
// plus 8 lines/month plus 12 months, comes to a bit over 2k of our 4k buffer.

void cal_main(void)
{
	struct tm *tm;
	char *buf = toybuf;

	if (toys.optc) {
		// Conveniently starts zeroed
		tm = (struct tm *)toybuf;
		buf += sizeof(struct tm);

		// Last argument is year, one before that (if any) is month.
		xcheckrange(tm->tm_year = atol(toys.optargs[--toys.optc]),1,9999);
		tm->tm_year -= 1900;
		tm->tm_mday = 1;
		tm->tm_hour = 12;  // noon to avoid timezone weirdness
		if (toys.optc) {
			xcheckrange(tm->tm_mon = atol(toys.optargs[--toys.optc]),1,12);
			tm->tm_mon--;

		// Print 12 months of the year

		} else {
			char *bufs[12];
			int i, j, k;

			for (i=0; i<12; i++) {
				tm->tm_mon=i;
				mktime(tm);
				buf = calstrings(bufs[i]=buf, tm);
			}

			// 4 rows, 6 lines each, 3 columns
			for (i=0; i<4; i++) {
				for (j=0; j<8; j++) {
					for(k=0; k<3; k++) {
						char **b = bufs+(k+i*3);
						*b += printf("%s ", *b);
					}
					puts("");
				}
			}
			return;
		}

		// What day of the week does that start on?
		mktime(tm);

	} else {
		time_t now;
		time(&now);
		tm = localtime(&now);
	}

	calstrings(buf, tm);
	while (*buf) buf += printf("%s\n", buf);
}