xref: /freebsd/usr.bin/column/column.c (revision acd3428b7d3e94cef0e1881c868cb4b131d4ff41)
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 
34 #ifndef lint
35 static const char copyright[] =
36 "@(#) Copyright (c) 1989, 1993, 1994\n\
37 	The Regents of the University of California.  All rights reserved.\n";
38 #endif
39 
40 #if 0
41 #ifndef lint
42 static char sccsid[] = "@(#)column.c	8.4 (Berkeley) 5/4/95";
43 #endif
44 #endif
45 
46 #include <sys/cdefs.h>
47 __FBSDID("$FreeBSD$");
48 
49 #include <sys/types.h>
50 #include <sys/ioctl.h>
51 #include <sys/param.h>
52 
53 #include <err.h>
54 #include <limits.h>
55 #include <locale.h>
56 #include <stdio.h>
57 #include <stdlib.h>
58 #include <string.h>
59 #include <unistd.h>
60 #include <wchar.h>
61 #include <wctype.h>
62 
63 #define	TAB	8
64 
65 void  c_columnate(void);
66 void  input(FILE *);
67 void  maketbl(void);
68 void  print(void);
69 void  r_columnate(void);
70 void  usage(void);
71 int   width(const wchar_t *);
72 
73 int termwidth = 80;		/* default terminal width */
74 
75 int entries;			/* number of records */
76 int eval;			/* exit value */
77 int maxlength;			/* longest record */
78 wchar_t **list;			/* array of pointers to records */
79 const wchar_t *separator = L"\t ";	/* field separator for table option */
80 
81 int
82 main(int argc, char **argv)
83 {
84 	struct winsize win;
85 	FILE *fp;
86 	int ch, tflag, xflag;
87 	char *p;
88 	const char *src;
89 	wchar_t *newsep;
90 	size_t seplen;
91 
92 	setlocale(LC_ALL, "");
93 
94 	if (ioctl(1, TIOCGWINSZ, &win) == -1 || !win.ws_col) {
95 		if ((p = getenv("COLUMNS")))
96 			termwidth = atoi(p);
97 	} else
98 		termwidth = win.ws_col;
99 
100 	tflag = xflag = 0;
101 	while ((ch = getopt(argc, argv, "c:s:tx")) != -1)
102 		switch(ch) {
103 		case 'c':
104 			termwidth = atoi(optarg);
105 			break;
106 		case 's':
107 			src = optarg;
108 			seplen = mbsrtowcs(NULL, &src, 0, NULL);
109 			if (seplen == (size_t)-1)
110 				err(1, "bad separator");
111 			newsep = malloc((seplen + 1) * sizeof(wchar_t));
112 			if (newsep == NULL)
113 				err(1, NULL);
114 			mbsrtowcs(newsep, &src, seplen + 1, NULL);
115 			separator = newsep;
116 			break;
117 		case 't':
118 			tflag = 1;
119 			break;
120 		case 'x':
121 			xflag = 1;
122 			break;
123 		case '?':
124 		default:
125 			usage();
126 		}
127 	argc -= optind;
128 	argv += optind;
129 
130 	if (!*argv)
131 		input(stdin);
132 	else for (; *argv; ++argv)
133 		if ((fp = fopen(*argv, "r"))) {
134 			input(fp);
135 			(void)fclose(fp);
136 		} else {
137 			warn("%s", *argv);
138 			eval = 1;
139 		}
140 
141 	if (!entries)
142 		exit(eval);
143 
144 	maxlength = roundup(maxlength + TAB, TAB);
145 	if (tflag)
146 		maketbl();
147 	else if (maxlength >= termwidth)
148 		print();
149 	else if (xflag)
150 		c_columnate();
151 	else
152 		r_columnate();
153 	exit(eval);
154 }
155 
156 void
157 c_columnate(void)
158 {
159 	int chcnt, col, cnt, endcol, numcols;
160 	wchar_t **lp;
161 
162 	numcols = termwidth / maxlength;
163 	endcol = maxlength;
164 	for (chcnt = col = 0, lp = list;; ++lp) {
165 		wprintf(L"%ls", *lp);
166 		chcnt += width(*lp);
167 		if (!--entries)
168 			break;
169 		if (++col == numcols) {
170 			chcnt = col = 0;
171 			endcol = maxlength;
172 			putwchar('\n');
173 		} else {
174 			while ((cnt = roundup(chcnt + TAB, TAB)) <= endcol) {
175 				(void)putwchar('\t');
176 				chcnt = cnt;
177 			}
178 			endcol += maxlength;
179 		}
180 	}
181 	if (chcnt)
182 		putwchar('\n');
183 }
184 
185 void
186 r_columnate(void)
187 {
188 	int base, chcnt, cnt, col, endcol, numcols, numrows, row;
189 
190 	numcols = termwidth / maxlength;
191 	numrows = entries / numcols;
192 	if (entries % numcols)
193 		++numrows;
194 
195 	for (row = 0; row < numrows; ++row) {
196 		endcol = maxlength;
197 		for (base = row, chcnt = col = 0; col < numcols; ++col) {
198 			wprintf(L"%ls", list[base]);
199 			chcnt += width(list[base]);
200 			if ((base += numrows) >= entries)
201 				break;
202 			while ((cnt = roundup(chcnt + TAB, TAB)) <= endcol) {
203 				(void)putwchar('\t');
204 				chcnt = cnt;
205 			}
206 			endcol += maxlength;
207 		}
208 		putwchar('\n');
209 	}
210 }
211 
212 void
213 print(void)
214 {
215 	int cnt;
216 	wchar_t **lp;
217 
218 	for (cnt = entries, lp = list; cnt--; ++lp)
219 		(void)wprintf(L"%ls\n", *lp);
220 }
221 
222 typedef struct _tbl {
223 	wchar_t **list;
224 	int cols, *len;
225 } TBL;
226 #define	DEFCOLS	25
227 
228 void
229 maketbl(void)
230 {
231 	TBL *t;
232 	int coloff, cnt;
233 	wchar_t *p, **lp;
234 	int *lens, maxcols;
235 	TBL *tbl;
236 	wchar_t **cols;
237 	wchar_t *last;
238 
239 	if ((t = tbl = calloc(entries, sizeof(TBL))) == NULL)
240 		err(1, (char *)NULL);
241 	if ((cols = calloc((maxcols = DEFCOLS), sizeof(*cols))) == NULL)
242 		err(1, (char *)NULL);
243 	if ((lens = calloc(maxcols, sizeof(int))) == NULL)
244 		err(1, (char *)NULL);
245 	for (cnt = 0, lp = list; cnt < entries; ++cnt, ++lp, ++t) {
246 		for (coloff = 0, p = *lp;
247 		    (cols[coloff] = wcstok(p, separator, &last));
248 		    p = NULL)
249 			if (++coloff == maxcols) {
250 				if (!(cols = realloc(cols, ((u_int)maxcols +
251 				    DEFCOLS) * sizeof(char *))) ||
252 				    !(lens = realloc(lens,
253 				    ((u_int)maxcols + DEFCOLS) * sizeof(int))))
254 					err(1, NULL);
255 				memset((char *)lens + maxcols * sizeof(int),
256 				    0, DEFCOLS * sizeof(int));
257 				maxcols += DEFCOLS;
258 			}
259 		if ((t->list = calloc(coloff, sizeof(*t->list))) == NULL)
260 			err(1, (char *)NULL);
261 		if ((t->len = calloc(coloff, sizeof(int))) == NULL)
262 			err(1, (char *)NULL);
263 		for (t->cols = coloff; --coloff >= 0;) {
264 			t->list[coloff] = cols[coloff];
265 			t->len[coloff] = width(cols[coloff]);
266 			if (t->len[coloff] > lens[coloff])
267 				lens[coloff] = t->len[coloff];
268 		}
269 	}
270 	for (cnt = 0, t = tbl; cnt < entries; ++cnt, ++t) {
271 		for (coloff = 0; coloff < t->cols  - 1; ++coloff)
272 			(void)wprintf(L"%ls%*ls", t->list[coloff],
273 			    lens[coloff] - t->len[coloff] + 2, L" ");
274 		(void)wprintf(L"%ls\n", t->list[coloff]);
275 	}
276 }
277 
278 #define	DEFNUM		1000
279 #define	MAXLINELEN	(LINE_MAX + 1)
280 
281 void
282 input(FILE *fp)
283 {
284 	static int maxentry;
285 	int len;
286 	wchar_t *p, buf[MAXLINELEN];
287 
288 	if (!list)
289 		if ((list = calloc((maxentry = DEFNUM), sizeof(*list))) ==
290 		    NULL)
291 			err(1, (char *)NULL);
292 	while (fgetws(buf, MAXLINELEN, fp)) {
293 		for (p = buf; *p && iswspace(*p); ++p);
294 		if (!*p)
295 			continue;
296 		if (!(p = wcschr(p, L'\n'))) {
297 			warnx("line too long");
298 			eval = 1;
299 			continue;
300 		}
301 		*p = L'\0';
302 		len = width(buf);
303 		if (maxlength < len)
304 			maxlength = len;
305 		if (entries == maxentry) {
306 			maxentry += DEFNUM;
307 			if (!(list = realloc(list,
308 			    (u_int)maxentry * sizeof(*list))))
309 				err(1, NULL);
310 		}
311 		list[entries] = malloc((wcslen(buf) + 1) * sizeof(wchar_t));
312 		if (list[entries] == NULL)
313 			err(1, NULL);
314 		wcscpy(list[entries], buf);
315 		entries++;
316 	}
317 }
318 
319 /* Like wcswidth(), but ignores non-printing characters. */
320 int
321 width(const wchar_t *wcs)
322 {
323 	int w, cw;
324 
325 	for (w = 0; *wcs != L'\0'; wcs++)
326 		if ((cw = wcwidth(*wcs)) > 0)
327 			w += cw;
328 	return (w);
329 }
330 
331 void
332 usage(void)
333 {
334 
335 	(void)fprintf(stderr,
336 	    "usage: column [-tx] [-c columns] [-s sep] [file ...]\n");
337 	exit(1);
338 }
339